Update my_agent.py

my_agent.py

@@ -1,18 +1,41 @@
+from typing import BinaryIO
+
 import yaml
-from smolagents import OpenAIServerModel, VisitWebpageTool
-from smolagents import CodeAgent, FinalAnswerTool, WikipediaSearchTool, Tool
+from smolagents import OpenAIServerModel, VisitWebpageTool, PythonInterpreterTool, SpeechToTextTool
+from smolagents import CodeAgent, FinalAnswerTool, WikipediaSearchTool, Tool, tool
 from langchain_community.tools import DuckDuckGoSearchRun
 
 import os
 
 # Load environment variables from .env file
-# with open('.env') as f:
-#     for line in f:
-#         if line.strip() and not line.startswith('#'):
-#             key, value = line.strip().split('=', 1)
-#             os.environ[key] = value
+with open('.env') as f:
+    for line in f:
+        if line.strip() and not line.startswith('#'):
+            key, value = line.strip().split('=', 1)
+            os.environ[key] = value
+
+@tool
+def get_file_from_documents(file_name: str) -> BinaryIO:
+    """
+    opens and returns a file as a BinaryIO object from the 'documents' directory provided the file name.
+
+    Args:
+        file_name (str): The name of the file to read.
 
+    Returns:
+        BinaryIO: A file-like object containing the content of the file.
 
+    Raises:
+        FileNotFoundError: If the file does not exist in the 'documents' directory.
+    """
+    documents_dir = os.path.join(os.getcwd(), "documents")
+    file_path = os.path.join(documents_dir, file_name)
+
+    if not os.path.exists(file_path):
+        raise FileNotFoundError(f"The file '{file_name}' does not exist in the 'documents' directory.")
+
+    file = open(file_path, "rb")
+    return file
 
 model = OpenAIServerModel(
     model_id="CLAUDE_3_SONNET_35",  # Replace with the desired model ID
@@ -23,13 +46,13 @@ model = OpenAIServerModel(
     client_kwargs={
         "default_headers": {"Subscription-Key": os.getenv("SUBSCRIPTION_KEY")},  # Additional headers,
     },
-    # web_search_options={
-    #     "enabled": True,
-    #     "dynamic_threshold": 0.5
-    # }
+    web_search_options={
+        "enabled": True,
+        "dynamic_threshold": 0.5
+    }
 )
 
-# Load system prompt from prompt.yaml file
+#Load system prompt from prompt.yaml file
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
@@ -38,18 +61,20 @@ final_answer = FinalAnswerTool()
 ddg_tool = Tool.from_langchain(DuckDuckGoSearchRun())
 
 agent = CodeAgent(
-    tools=[final_answer, VisitWebpageTool(), WikipediaSearchTool(), ddg_tool],
+    tools=[final_answer, VisitWebpageTool(), WikipediaSearchTool(), ddg_tool, get_file_from_documents, PythonInterpreterTool(), SpeechToTextTool()],
     add_base_tools=True,
-    max_steps=10,
+    max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
     description=None,
     prompt_templates=prompt_templates,
-    additional_authorized_imports=['pandas', 'numpy', 'openpyxl', 'xlrd', 'os'],
+    additional_authorized_imports=['pandas', 'numpy', 'openpyxl', 'xlrd', 'os', ],
     model=model)
 
+#agent.run("""What is the final numeric output from the attached Python code? (File Name: f918266a-b3e0-4914-865d-4faa564f1aef.py) """)
+
 
 
 #