韩宇
commited on
Commit
·
0b0cf33
1
Parent(s):
9a91383
init req
Browse files- agent/conclude/webpage_conclude.py +1 -1
- webpage.py +1 -1
agent/conclude/webpage_conclude.py
CHANGED
@@ -76,6 +76,6 @@ class WebpageConclude(BaseLLMBackend, BaseWorker):
|
|
76 |
agent_id=self.workflow_instance_id,
|
77 |
msg=f'Answer: {chat_complete_res["choices"][0]["message"]["content"]}',
|
78 |
)
|
79 |
-
self.callback.send_answer(agent_id=self.workflow_instance_id, msg=f"Token usage: {self.token_usage}")
|
80 |
self.stm(self.workflow_instance_id).clear()
|
81 |
return {"last_output": last_output}
|
|
|
76 |
agent_id=self.workflow_instance_id,
|
77 |
msg=f'Answer: {chat_complete_res["choices"][0]["message"]["content"]}',
|
78 |
)
|
79 |
+
# self.callback.send_answer(agent_id=self.workflow_instance_id, msg=f"Token usage: {self.token_usage}")
|
80 |
self.stm(self.workflow_instance_id).clear()
|
81 |
return {"last_output": last_output}
|
webpage.py
CHANGED
@@ -106,7 +106,7 @@ class WebpageClient:
|
|
106 |
def load_local_video() -> dict:
|
107 |
result = {}
|
108 |
for root, _, files in os.walk(video_root_path):
|
109 |
-
for file in filter(lambda x: x.split('.')[-1] in (
|
110 |
'mp4', 'avi', 'mov', 'wmv', 'flv', 'mkv', 'webm', 'm4v'), files):
|
111 |
file_obs_path = os.path.join(root, file)
|
112 |
result[Path(file_obs_path).name] = file_obs_path
|
|
|
106 |
def load_local_video() -> dict:
|
107 |
result = {}
|
108 |
for root, _, files in os.walk(video_root_path):
|
109 |
+
for file in filter(lambda x: x.split('.')[-1].lower() in (
|
110 |
'mp4', 'avi', 'mov', 'wmv', 'flv', 'mkv', 'webm', 'm4v'), files):
|
111 |
file_obs_path = os.path.join(root, file)
|
112 |
result[Path(file_obs_path).name] = file_obs_path
|