Update app.py

app.py

@@ -29,6 +29,7 @@ class BasicAgent:
         
         chat = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
 
+        print("check1")
         # llm = HuggingFaceEndpoint(
         #     repo_id="mistralai/Mistral-7B-Instruct-v0.2",
         #     huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN"),
@@ -43,6 +44,7 @@ class BasicAgent:
         # Bind LLM with tools
         chat_with_tools = chat.bind_tools(tools)
 
+        print("check2")
         # Assistant function (process one step)
         def assistant(state: AgentState):
             return {
@@ -51,11 +53,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)