SergeyO7 commited on
Commit
8706eb6
·
verified ·
1 Parent(s): de0487a

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +6 -5
agent.py CHANGED
@@ -179,7 +179,8 @@ class SpeechToTextTool(Tool):
179
  def forward(self, audio_path: str) -> str:
180
  if not os.path.exists(audio_path):
181
  return f"Error: File not found at {audio_path}"
182
- result = self.model.transcribe(audio_path)
 
183
  return result.get("text", "")
184
 
185
  class ExcelReaderTool(Tool):
@@ -286,10 +287,10 @@ class MagAgent:
286
  DownloadTaskAttachmentTool(rate_limiter=rate_limiter),
287
  # DuckDuckGoSearchTool(),
288
  WikipediaSearchTool(),
289
- SpeechToTextTool,
290
- ExcelReaderTool,
291
- VisitWebpageTool,
292
- PythonCodeReaderTool,
293
  GoogleSearchTool,
294
  # ImageAnalysisTool,
295
  ],
 
179
  def forward(self, audio_path: str) -> str:
180
  if not os.path.exists(audio_path):
181
  return f"Error: File not found at {audio_path}"
182
+ result = self.model.
183
+ transcribe(audio_path)
184
  return result.get("text", "")
185
 
186
  class ExcelReaderTool(Tool):
 
287
  DownloadTaskAttachmentTool(rate_limiter=rate_limiter),
288
  # DuckDuckGoSearchTool(),
289
  WikipediaSearchTool(),
290
+ SpeechToTextTool(),
291
+ ExcelReaderTool(),
292
+ VisitWebpageTool(),
293
+ PythonCodeReaderTool(),
294
  GoogleSearchTool,
295
  # ImageAnalysisTool,
296
  ],