change code

app.py

@@ -4,7 +4,13 @@ import requests
 import inspect
 import pandas as pd
 import uuid
-from smolagents import OpenAIServerModel, ToolCallingAgent, DuckDuckGoSearchTool
+
+from langchain.chat_models import init_chat_model
+from langgraph.prebuilt import create_react_agent
+from langchain_community.tools import WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
+
+from langchain_community.tools import DuckDuckGoSearchResults
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -15,16 +21,16 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        model = OpenAIServerModel(
-            model_id=os.getenv("MODEL_ID"),
-            api_key=os.getenv("API_KEY"),
-            api_base=os.getenv("API_BASE"),
-        )
-        self.agent = ToolCallingAgent(tools=[DuckDuckGoSearchTool()], model=model)
+        llm = init_chat_model(temperature=0, model=os.getenv("MODEL_ID"), api_key=os.getenv("API_KEY"), base_url=os.getenv("API_BASE"))
+
+        api_wrapper = WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=100)
+        wiki = WikipediaQueryRun(api_wrapper=api_wrapper)
+
+        self.agent = create_react_agent(llm, tools=[wiki, DuckDuckGoSearchResults()])
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = self.agent(question)
+        answer = self.agent.invoke(inputs = {"messages": [("user", question)]})
         print(f"Agent returning answer: {answer}")
         return answer
 

requirements.txt

@@ -1,3 +1,12 @@
 gradio
 requests
-smolagents[openai]
+# smolagents[openai]
+
+langgraph-cli[inmem]
+langchain
+openai
+duckduckgo-search
+langchain-community
+langchain-openai
+wikipedia
+