Update app.py

app.py

@@ -35,36 +35,36 @@ class BasicAgent:
         chat = ChatOpenAI(
             model="gpt-4",
             temperature=0.0,
-            openai_api_key=os.environ.get("OPENAI_API_KEY"),  # use OpenAI API key securely from environment variable
+            openai_api_key=os.environ.get("OPENAI_API_KEY"),  # Secure API Key
             verbose=True
         )
 
-        # chat = ChatHuggingFace(llm=llm, verbose=True)
+        # Set up tools
         search_tool = DuckDuckGoSearchRun()
         tools = [search_tool]
+
+        # Bind LLM with tools
         chat_with_tools = chat.bind_tools(tools)
-        
+
+        # Assistant function (process one step)
         def assistant(state: AgentState):
             return {
-                "messages": [chat_with_tools.invoke(state["messages"])],
+                "messages": [chat_with_tools.invoke(state["messages"])]
             }
-        
-        ## The graph
+
+        # Create a StateGraph
         builder = StateGraph(AgentState)
-        
-        # Define nodes: these do the work
+
+        # Add nodes
         builder.add_node("assistant", assistant)
         builder.add_node("tools", ToolNode(tools))
-        
-        # Define edges: these determine how the control flow moves
+
+        # Define edges
         builder.add_edge(START, "assistant")
-        builder.add_conditional_edges(
-            "assistant",
-            # If the latest message requires a tool, route to tools
-            # Otherwise, provide a direct response
-            tools_condition,
-        )
+        builder.add_conditional_edges("assistant", tools_condition)
         builder.add_edge("tools", "assistant")
+
+        # Compile the agent
         self.agent = builder.compile()
         
     def __call__(self, question: str) -> str: