Updating app.py

app.py

@@ -31,22 +31,12 @@ SERPER_API_KEY = os.environ.get("SERPER_API_KEY")
 HUGGINGFACEHUB_API_TOKEN=os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
-        self.system_prompt = (
-            "You are a general AI assistant. I will ask you a question. "
-            "Report your thoughts, and finish your answer with the following template: "
-            "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. "
-            "If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
-            "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."
-        )
         llm = ChatOpenAI(
         model = 'gpt-4o-mini',
         temperature = 0,
         openai_api_key = OPENAI_KEY,
         max_tokens = 4096
         )
-        chat = ChatHuggingFace(llm=llm, verbose=True)
         search = GoogleSerperAPIWrapper()
         wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
         tools = [Tool(
@@ -59,30 +49,22 @@ class BasicAgent:
             func=wikipedia.run,
             description="useful for when you need to ask with Wikipedia",
         )]
-        chat_with_tools = chat.bind_tools(tools)
-
-        class AgentState(TypedDict):
-            messages: Annotated[list[AnyMessage], add_messages]
-
-        def assistant(state: AgentState):
-            return {
-                "messages": [chat_with_tools.invoke(state["messages"])],
-            }
-
-        builder = StateGraph(AgentState)
-        builder.add_node("assistant", assistant)
-        builder.add_node("tools", ToolNode(tools))
-        builder.add_edge(START, "assistant")
-        builder.add_conditional_edges("assistant", tools_condition)
-        builder.add_edge("tools", "assistant")
-        self.alfred = builder.compile()
-  
+        
+        self.agent = initialize_agent(
+        tools=tools,
+        llm=llm,
+        agent=AgentType.OPENAI_FUNCTIONS,
+        verbose=True,
+        agent_kwargs={
+        "system_message": SystemMessage(content="You are a general AI assistant. I will ask you a question. "
+            "Report your thoughts, and finish your answer with the following template: "
+            "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. "
+            "If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
+            "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." }
+        )
 
     def __call__(self, question: str) -> str:
-        messages = [
-            SystemMessage(content=self.system_prompt),
-            HumanMessage(content=question)
-        ]
         def extract_model_answer(full_response: str) -> str:
             marker = "FINAL ANSWER:"
             idx = full_response.find(marker)