CakeCN/hsa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

logger time

Browse Source
This commit is contained in:
CakeCN
2025-12-11 15:02:33 +08:00
parent 1bf13aacf6
commit ae91b62332
1 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
Html/apps/extension_ase/ase_client/client.py
View File

@@ -433,17 +433,20 @@ class HSAEngineClient(ASEngineClient):
super().__init__(server_url, namespace, id) super().__init__(server_url, namespace, id)
self.chatmanager = chatmanager self.chatmanager = chatmanager
def loadmarkdown(self, fmd, fmdp, fsp): def loadmarkdown(self, fmd, fmdp, fsp):
now = time.time()
# 并行发送三个消息，获取事件对象和ack_id # 并行发送三个消息，获取事件对象和ack_id
event1, ack_id1 = self.send_text('markdown-in', fmd, self.id, wait_for_ack=True, return_event=True) event1, ack_id1 = self.send_text('markdown-in', fmd, self.id, wait_for_ack=True, return_event=True)
event2, ack_id2 = self.send_text('markdown-prompt-in', fmdp, self.id, wait_for_ack=True, return_event=True) event2, ack_id2 = self.send_text('markdown-prompt-in', fmdp, self.id, wait_for_ack=True, return_event=True)
event3, ack_id3 = self.send_text('score-prompt-in', fsp, self.id, wait_for_ack=True, return_event=True) event3, ack_id3 = self.send_text('score-prompt-in', fsp, self.id, wait_for_ack=True, return_event=True)
self.chatmanager.app.logger.info(f"Load markdown time cost: {time.time() - now}")
now = time.time()
# 等待所有事件完成 # 等待所有事件完成
timeout = 5 # 超时时间 timeout = 5 # 超时时间
success1 = event1.wait(timeout=timeout) success1 = event1.wait(timeout=timeout)
success2 = event2.wait(timeout=timeout) success2 = event2.wait(timeout=timeout)
success3 = event3.wait(timeout=timeout) success3 = event3.wait(timeout=timeout)
self.chatmanager.app.logger.info(f"Wait markdown time cost: {time.time() - now}")
now = time.time()
# 清理ACK事件 # 清理ACK事件
if ack_id1 in self.ack_events: if ack_id1 in self.ack_events:
del self.ack_events[ack_id1] del self.ack_events[ack_id1]
@@ -451,13 +454,4 @@ class HSAEngineClient(ASEngineClient):
del self.ack_events[ack_id2] del self.ack_events[ack_id2]
if ack_id3 in self.ack_events: if ack_id3 in self.ack_events:
del self.ack_events[ack_id3] del self.ack_events[ack_id3]
# 检查是否所有ACK都成功收到
if not success1:
print(f"ACK timeout for message: {ack_id1}")
if not success2:
print(f"ACK timeout for message: {ack_id2}")
if not success3:
print(f"ACK timeout for message: {ack_id3}")
return success1 and success2 and success3 return success1 and success2 and success3