Update langgraph_agent.py

langgraph_agent.py

@@ -5,11 +5,12 @@ from langchain_core.messages import AnyMessage, HumanMessage
 from langchain_core.runnables import RunnableConfig
 from langgraph.prebuilt import create_react_agent
 from langgraph.prebuilt.chat_agent_executor import AgentState
-
+from smolagents import VisitWebpageTool, PythonInterpreterTool, SpeechToTextTool
 from langchain_community.tools import DuckDuckGoSearchRun
 from langchain_experimental.utilities import PythonREPL
 from langchain_community.tools import WikipediaQueryRun
 from langchain_community.utilities import WikipediaAPIWrapper
+from langchain_core.tools import tool
 
 load_dotenv()
 
@@ -28,6 +29,45 @@ duckduckgo = DuckDuckGoSearchRun()
 python_repl = PythonREPL()
 wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
 
+# Instantiate smolagents tools
+visit_webpage_tool = VisitWebpageTool()
+python_interpreter_tool = PythonInterpreterTool()
+speech_to_text_tool = SpeechToTextTool()
+
+@tool
+def visit_webpage(url: str) -> str:
+    """
+    Visit a webpage and return its content as a string.
+    Args:
+        url (str): The URL of the webpage to visit.
+    Returns:
+        str: The content of the webpage.
+    """
+    return visit_webpage_tool(url=url)
+
+@tool
+def python_interpreter(code: str) -> str:
+    """
+    Execute Python code and return the output as a string.
+    Args:
+        code (str): The Python code to execute.
+    Returns:
+        str: The output of the executed code.
+    """
+    return python_interpreter_tool(code=code)
+
+@tool
+def speech_to_text(audio_path: str) -> str:
+    """
+    Convert speech in an audio file to text.
+    Args:
+        audio_path (str): The path to the audio file.
+    Returns:
+        str: The transcribed text from the audio.
+    """
+    return speech_to_text_tool(audio_path=audio_path)
+
+
 def prompt(state: AgentState, config: RunnableConfig) -> list[AnyMessage]:
     file_path = config["configurable"].get("file_path", "")
     system_msg = (
@@ -44,6 +84,6 @@ def prompt(state: AgentState, config: RunnableConfig) -> list[AnyMessage]:
 
 react_agent = create_react_agent(
     model=chat,
-    tools=[duckduckgo, wikipedia],
+    tools=[duckduckgo, wikipedia, visit_webpage, python_interpreter, speech_to_text],
     prompt=prompt,
 )