Update app.py

app.py

@@ -8,16 +8,53 @@ import pandas as pd
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+from agents import tool_list
+
+
+# Generate the AgentState and Agent graph
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
+
+def get_assistant(chat_with_tools):
+    def assistant(state: AgentState):
+        return {
+            "messages": [chat_with_tools.invoke(state["messages"])],
+        }
+    return assistant
+
 class BasicAgent:
     def __init__(self):
+        llm = HuggingFaceEndpoint(repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",)
+        chat = ChatHuggingFace(llm=llm, verbose=True)
+
+        # define tools
+        tools = tool_list()
+        chat_with_tools = chat.bind_tools(tools)
+
+        ## The graph
+        graph_builder = StateGraph(AgentState)
+
+        chat_assistant_tool = get_assistant(chat_with_tools)
+
+        graph_builder.add_node("assistant", chat_assistant_tool)
+        graph_builder.add_node("tools", ToolNode(tools))
+
+        # edges
+        graph_builder.add_edge(START, "assistant")
+        graph_builder.add_conditional_edges("assistant", tools_condition)
+        graph_builder.add_edge("tools", "assistant")
+
+        self.okmich_agent = graph_builder.compile()
+
         print("BasicAgent initialized.")
+
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        messages = [HumanMessage(content=question)]
         fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        _response = self.okmich_agent.invoke({"messages": messages})
+        return _response['messages'][-1].content if _response else fixed_answer
+
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """