Update app.py

app.py

@@ -27,9 +27,8 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         
-        chat = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
+        chat = ChatOpenAI(model="gpt-4o", temperature=0)
 
-        print("check1")
         # llm = HuggingFaceEndpoint(
         #     repo_id="mistralai/Mistral-7B-Instruct-v0.2",
         #     huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
@@ -39,13 +38,11 @@ class BasicAgent:
 
         # Set up tools
         search_tool = DuckDuckGoSearchRun()
-        print("check1.1")
         tools = [search_tool]
-        print("check1.2")
-        # Bind LLM with tools
-        chat_with_tools = chat.bind_tools(tools)
+        
+        # # Bind LLM with tools
+        # chat_with_tools = chat.bind_tools(tools)
 
-        print("check2")
         # Assistant function (process one step)
         def assistant(state: AgentState):
             return {
@@ -54,11 +51,11 @@ class BasicAgent:
 
         # Create a StateGraph
         builder = StateGraph(AgentState)
-        print("check3")
+        
         # Add nodes
         builder.add_node("assistant", assistant)
         builder.add_node("tools", ToolNode(tools))
-        print("check4")
+       
         # Define edges
         builder.add_edge(START, "assistant")
         builder.add_conditional_edges("assistant", tools_condition)