limited web search results and added speech to txt tool
Browse files
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)
|