quick and dirty test

app.py

@@ -1,15 +1,103 @@
 import os
+import re
 import gradio as gr
 import requests
 import inspect
 import pandas as pd
 
+from typing import TypedDict, Annotated
+from langgraph.graph.message import add_messages
+from langchain_core.messages import AnyMessage, HumanMessage, AIMessage, SystemMessage
+from langgraph.prebuilt import ToolNode
+from langgraph.graph import START, StateGraph
+from langgraph.prebuilt import tools_condition
+from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_community.tools import DuckDuckGoSearchRun
+
 # (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 ------
+
+# Generate the AgentState and Agent graph
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
+
+# --- Agent ---
+class GAIAAgent:
+    # --- LLM ---
+    llm = HuggingFaceEndpoint(
+        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        huggingfacehub_api_token=os.getenv('HF_TOKEN'),
+    )
+    chat = ChatHuggingFace(llm=llm, verbose=True)
+
+    # --- Tools ---
+    search_tool = DuckDuckGoSearchRun()
+    tools = [search_tool]
+
+    # --- System Prompt ---
+    system = """
+    You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your 
+    answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should 
+    be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If 
+    you are asked for a number, don't use comma to write your number neither use units such as $ or 
+    percent sign unless specified otherwise. If you are asked for a string, don't use articles, 
+    neither abbreviations (e.g. for cities), and write the digits in plain text unless specified 
+    otherwise. If you are asked for a comma separated list, apply the above rules depending of 
+    whether the element to be put in the list is a number or a string.
+    """.replace('\n', '')
+
+    def __init__(self):
+        self.build_graph()
+        print("GAIAAgent initialized.")
+
+    def build_graph(self):
+        self.chat_with_tools = self.chat.bind_tools(self.tools)
+
+        builder = StateGraph(AgentState)
+
+        def assistant(state: AgentState):
+            return {
+                "messages": [self.chat_with_tools.invoke(state["messages"])],
+            }
+
+        # Define nodes: these do the work
+        builder.add_node("assistant", 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.graph = builder.compile()
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+
+        answer = self.graph.invoke(
+            {"messages": [
+                SystemMessage(content=self.system),
+                HumanMessage(content=question)
+            ]})
+        
+        # extract answer
+        match = re.search(r"(?<=FINAL ANSWER:\s).*", answer)
+        if match:
+            final_answer = match.group(0)
+        else:
+            final_answer = "I don't know"
+
+        print(f"Agent returning answer: {final_answer}")
+        return final_answer
+
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")