Updating app.py

app.py

@@ -36,17 +36,24 @@ class BasicAgent:
             "If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. "
             "If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
         )
-        # Initialize the DuckDuckGo search tool
-        search_tool = DuckDuckGoSearchRun()
-        tools = [search_tool]
-
-        llm = HuggingFaceEndpoint(
-        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-        huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+        chat = ChatOpenAI(
+        model = 'gpt-4o-mini',
+        temperature = 0,
+        openai_api_key = OPENAI_KEY,
+        max_tokens = 4096
         )
-
-        chat = ChatHuggingFace(llm=llm, verbose=True)
-        tools = [search_tool]
+        search = GoogleSerperAPIWrapper()
+        wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
+        tools = [Tool(
+        name="Web_Search",
+        func=search.run,
+        description="useful for when you need to ask with search",
+        ),
+        Tool(
+            name="wikipedia_search",
+            func=wikipedia.run,
+            description="useful for when you need to ask with Wikipedia",
+        )]
         chat_with_tools = chat.bind_tools(tools)
 
         class AgentState(TypedDict):
@@ -64,6 +71,7 @@ class BasicAgent:
         builder.add_conditional_edges("assistant", tools_condition)
         builder.add_edge("tools", "assistant")
         self.alfred = builder.compile()
+  
 
     def __call__(self, question: str) -> str:
         messages = [