韩宇 commited on
Commit
0b0cf33
·
1 Parent(s): 9a91383
Files changed (2) hide show
  1. agent/conclude/webpage_conclude.py +1 -1
  2. 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