Update agent.py

agent.py

@@ -1,8 +1,8 @@
 from langchain_core.messages import HumanMessage, AIMessage
-from langchain_community.tools import DuckDuckGoSearchRun
 from langchain.prompts import PromptTemplate
 from langchain_openai import ChatOpenAI
 from langchain_core.runnables import RunnableSequence
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool
 import os
 
 class AdvancedAgent:
@@ -10,14 +10,21 @@ class AdvancedAgent:
         print("AdvancedAgent initialized.")
         # Initialize the language model (assumes OPENAI_API_KEY is set)
         try:
-            self.llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.7)
+            model = OpenAIServerModel(model_id="gpt-3.5-turbo", temperature=0.7)
+            search_tool = DuckDuckGoSearchTool()
+            wiki_search = WikipediaSearchTool()
+
+            self.agent = CodeAgent(
+                model = model,
+                tools=[
+                    search_tool,
+                    wiki_search
+                ]
+            )
         except Exception as e:
             print(f"Error initializing LLM: {e}")
             raise e
 
-        # Initialize web search tool
-        self.search_tool = DuckDuckGoSearchRun()
-
         # Define prompt template for answering questions
         self.prompt_template = PromptTemplate(
             input_variables=["question", "context"],