Updating app.py with chat_with_tools

app.py

@@ -13,6 +13,7 @@ from langgraph.prebuilt import ToolNode, tools_condition
 from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
 from typing import TypedDict, Annotated
 from langgraph.graph.message import add_messages
+from langchain_ollama.chat_models import ChatOllama
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -20,30 +21,35 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+class BasicAgent:
+    
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-        # initializing the DuckDuckGo tool
-        self.search_tool = DuckDuckGoSearchRun()
-        self.tools = [self.search_tool]
+        # Initialize the DuckDuckGo search tool
+        search_tool = DuckDuckGoSearchRun()
+        tools = [search_tool]
+
+        chat = ChatOllama(model="qwen2.5-coder:7b", temperature=0, base_url="http://localhost:11434")
+        chat_with_tools = chat.bind_tools(tools)
 
-        # building LangGraph agent
         class AgentState(TypedDict):
             messages: Annotated[list[AnyMessage], add_messages]
 
         def assistant(state: AgentState):
-            user_message = state["messages"][-1]
-            if isinstance(user_message, HumanMessage):
-                result = self.search_tool.invoke(user_message.content)
-                return {"messages": [AIMessage(content=result)]}
-            return {"messages": [AIMessage(content="I can only answer human questions.")]}
+            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()
 
-    
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         messages = [HumanMessage(content=question)]