Update app.py

app.py

@@ -4,20 +4,85 @@ import requests
 import inspect
 import pandas as pd
 
+from typing import List, TypedDict, Annotated, Optional
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
+from langgraph.graph.message import add_messages
+from langgraph.graph import START, StateGraph
+from langgraph.prebuilt import ToolNode, tools_condition
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
+    
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
+
+        self.MODEL_NAME = "qwen3.6-plus"
+        # export DASHSCOPE_API_KEY="sk-**"
+        self.api_key = os.getenv("DASHSCOPE_API_KEY")
+        if not self.api_key:
+            raise RuntimeError("NO DASHSCOPE_API_KEY")
+        
+        self.model = ChatOpenAI(
+            model=self.MODEL_NAME,
+            api_key=self.api_key,
+            base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
+            temperature=0,
+        )
+        self.tools = []
+        self.chat_with_tools = self.model.bind_tools(self.tools)
+        # The graph
+        builder = StateGraph(AgentState)
+        
+        # Define nodes: these do the work
+        builder.add_node("assistant", self.assistant)
+        builder.add_node("tools", ToolNode(self.tools))
+        
+        # Define edges: these determine how the control flow moves
+        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_edge("tools", "assistant")
+        self.react_graph = builder.compile()
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        print(f"Agent received question: {question}")
+
+        try:
+
+            answer = self.answer_question(question)
+
+            return str(answer).strip()
+
+        except Exception as e:
+            print(f"Agent error: {e}")
+            return ""
+    def assistant(self, state: AgentState):
+        sys_msg = SystemMessage(content=f"""
+            You are a precise question-answering agent. 
+            Answer the question directly and concisely. 
+            Return only the final answer, without explanation, unless explanation is necessary.
+        """)
+        return {
+            "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],
+        }
+        
+    def answer_question(self, question: str) -> str:
+        # TODO: 
+        messages = [HumanMessage(content=question)]
+        messages = self.react_graph.invoke({"messages" : messages})
+        return messages["messages"][-1].content
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """