add test version of the agent

agent.py

@@ -0,0 +1,58 @@
+import os
+from typing import TypedDict, Annotated
+from dotenv import load_dotenv
+from langgraph.graph.message import add_messages
+from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
+from langgraph.prebuilt import ToolNode, tools_condition
+from langgraph.graph import START, StateGraph, MessagesState
+from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from tools import wiki_search, tavily_search, arxiv_search, add, subtract, multiply, divide, power, sqrt, modulus
+
+
+load_dotenv()
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+TOOLS = [
+    wiki_search,
+    tavily_search,
+    arxiv_search,
+    add,
+    subtract,
+    multiply,
+    divide,
+    power,
+    sqrt,
+    modulus
+]
+
+def build_agent():
+    # Define llm from Hugging Face
+    llm = HuggingFaceEndpoint(
+        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        huggingfacehub_api_token=HF_TOKEN
+    )
+
+    # Define chat interface and the tools
+    chat = ChatHuggingFace(llm=llm, verbose=True)
+    chat_w_tools = chat.bind_tools(TOOLS)
+
+    # Node
+    def assistant(state: MessagesState):
+        """Assistant node"""
+        return {"messages": [chat_w_tools.invoke(state["messages"])]}
+
+
+    builder = StateGraph(MessagesState)
+
+    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")
+
+    # Compile graph
+    return builder.compile()

app.py

@@ -3,6 +3,9 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from agent import build_agent
+from config import SYSTEM_PROMPT, SPACE_ID
+from langchain_core.messages import SystemMessage, HumanMessage
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -12,12 +15,19 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
+        self.agent = build_agent()
         print("BasicAgent initialized.")
     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
+        messages = [
+            SystemMessage(content=SYSTEM_PROMPT),
+            HumanMessage(content=question)
+        ]
+        messages = self.graph.invoke({"messages": messages})
+
+        answer = messages['messages'][-1].content
+        final_answer = answer.split("FINAL ANSWER: ")[-1].strip()
+
+        return final_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -25,7 +35,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
-    space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
+    # space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
+    space_id = SPACE_ID
 
     if profile:
         username= f"{profile.username}"

config.py

@@ -1,7 +1,13 @@
 import os
+import yaml
 from dotenv import load_dotenv
 
 
 load_dotenv()
 
-TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
+TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
+SPACE_ID = os.getenv("SPACE_ID")
+
+with open("system_prompt.yaml", "r") as f:
+    SYSTEM_PROMPT = yaml.safe_load(f)
+    SYSTEM_PROMPT = SYSTEM_PROMPT["system_prompt"]

requirements.txt

@@ -1,2 +1,14 @@
 gradio
-requests
+requests
+pandas
+langchain
+python-dotenv
+langgraph
+langchain_openai
+langchain_community
+langchain_tavily
+wikipedia
+pymupdf
+arxiv
+flake8
+langchain_huggingface

system_prompt.yaml

@@ -0,0 +1,11 @@
+system_prompt: |
+  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