Nioi commited on
Commit
69c427f
·
1 Parent(s): 035edd7

limited web search results and added speech to txt tool

Browse files
Files changed (1) hide show
  1. agent.py +6 -4
agent.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- from smolagents import CodeAgent, tool, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, PythonInterpreterTool
3
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
4
  from tools.csv_reader import CsvReaderTool
5
  from tools.excel_reader import ExcelReaderTool
@@ -116,8 +116,8 @@ def arvix_search(query: str) -> str:
116
 
117
 
118
  def get_agent() -> CodeAgent:
119
- search_tool = DuckDuckGoSearchTool()
120
- web_page_tool = VisitWebpageTool()
121
  py_interpreter = PythonInterpreterTool()
122
 
123
  csv_reader = CsvReaderTool()
@@ -126,6 +126,8 @@ def get_agent() -> CodeAgent:
126
  img_txt_extractor = ImageTextExtractorTool()
127
  pdf_reader = PdfReaderTool()
128
 
 
 
129
  api_key = os.getenv('CODESTRAL_API_KEY')
130
 
131
  model = OpenAIServerModel(
@@ -133,4 +135,4 @@ def get_agent() -> CodeAgent:
133
  api_base="https://codestral.mistral.ai/v1/",
134
  api_key=api_key)
135
 
136
- return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool, web_page_tool, py_interpreter, wiki_search, arvix_search, csv_reader, excel_reader, file_downloader, img_txt_extractor, pdf_reader], model=model)
 
1
  import os
2
+ from smolagents import CodeAgent, tool, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, PythonInterpreterTool, SpeechToTextTool
3
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
4
  from tools.csv_reader import CsvReaderTool
5
  from tools.excel_reader import ExcelReaderTool
 
116
 
117
 
118
  def get_agent() -> CodeAgent:
119
+ search_tool = DuckDuckGoSearchTool(max_results=5)
120
+ web_page_tool = VisitWebpageTool(max_output_length=1_000_000)
121
  py_interpreter = PythonInterpreterTool()
122
 
123
  csv_reader = CsvReaderTool()
 
126
  img_txt_extractor = ImageTextExtractorTool()
127
  pdf_reader = PdfReaderTool()
128
 
129
+ speech_txt = SpeechToTextTool()
130
+
131
  api_key = os.getenv('CODESTRAL_API_KEY')
132
 
133
  model = OpenAIServerModel(
 
135
  api_base="https://codestral.mistral.ai/v1/",
136
  api_key=api_key)
137
 
138
+ return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool, web_page_tool, py_interpreter, wiki_search, arvix_search, csv_reader, excel_reader, file_downloader, img_txt_extractor, pdf_reader, speech_txt], model=model)