Update agent and tools

agent.py

@@ -17,7 +17,7 @@ load_dotenv()
 def build_agent_graph():
     """Build the graph"""
     # Load environment variables from .env file
-    llm = ChatOpenAI(model="gpt-4o")
+    llm = ChatOpenAI(model="gpt-4o-mini")
     
     # Bind tools to LLM
     llm_with_tools = llm.bind_tools(level1_tools)
@@ -64,14 +64,17 @@ class MyGAIAAgent:
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # Wrap the question in a HumanMessage from langchain_core
-        '''
+        
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
+        
         '''
         user_input = {"messages": [("user", question)]}
-        answer = self.graph.invoke(user_input)["messages"][-1].content
-        return self._clean_answer(answer)
+        answer1 = self.graph.invoke(user_input)["messages"][-1].content
+        print (answer1)
+        '''
+        return answer
     
     def _clean_answer(self, answer: any) -> str:
         """
@@ -137,5 +140,6 @@ class MyGAIAAgent:
 if __name__ == "__main__":
     question1 = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)?"
     question2 = "Convert 10 miles to kilometers."       
+    question3 = "Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec. What does Teal'c say in response to the question ""Isn't that hot?"
     agent = MyGAIAAgent()
     print(agent(question1))

tools.py

@@ -6,7 +6,7 @@ import re
 
 from typing import Dict
 from pathlib import Path
-from markitdown import MarkItDown
+#from markitdown import MarkItDown
 from urllib.parse import urlparse
 
 from langchain_core.tools import tool