韩宇
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
|