0f3dy commited on
Commit
fef9d97
·
verified ·
1 Parent(s): 83e4c3c

Upload 8 files

Browse files
Files changed (8) hide show
  1. README.md +15 -0
  2. agent.py +190 -0
  3. app.py +203 -0
  4. gitattributes +35 -0
  5. gitignore +15 -0
  6. metadata.jsonl +0 -0
  7. requirements.txt +18 -0
  8. system_prompt.txt +5 -0
README.md ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: Template Final Assignment
3
+ emoji: 🕵🏻‍♂️
4
+ colorFrom: indigo
5
+ colorTo: indigo
6
+ sdk: gradio
7
+ sdk_version: 5.25.2
8
+ app_file: app.py
9
+ pinned: false
10
+ hf_oauth: true
11
+ # optional, default duration is 8 hours/480 minutes. Max duration is 30 days/43200 minutes.
12
+ hf_oauth_expiration_minutes: 480
13
+ ---
14
+
15
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
agent.py ADDED
@@ -0,0 +1,190 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from dotenv import load_dotenv
3
+ from langchain.tools import tool
4
+ from langchain_core.messages import SystemMessage, HumanMessage
5
+ from langchain_core.tools import tool
6
+ from langchain.tools.retriever import create_retriever_tool
7
+ from langchain_community.tools.tavily_search import TavilySearchResults
8
+ from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
9
+ from langchain_community.vectorstores import SupabaseVectorStore
10
+ from langchain_groq import ChatGroq
11
+ from langgraph.prebuilt import tools_condition, ToolNode
12
+ from langgraph.graph import START, StateGraph, MessagesState
13
+ from langchain_huggingface import HuggingFaceEmbeddings
14
+ from langchain_huggingface import ChatHuggingFace
15
+ from supabase.client import Client, create_client
16
+
17
+ load_dotenv()
18
+
19
+ @tool
20
+ def multiply(a: int, b: int) -> int:
21
+ """Multiply two numbers.
22
+
23
+ Args:
24
+ a: first int
25
+ b: second int
26
+ """
27
+ return a * b
28
+
29
+ @tool
30
+ def add(a: int, b: int) -> int:
31
+ """Add two numbers.
32
+
33
+ Args:
34
+ a: first int
35
+ b: second int
36
+ """
37
+ return a + b
38
+
39
+ @tool
40
+ def subtract(a: int, b: int) -> int:
41
+ """Subtract two numbers.
42
+
43
+ Args:
44
+ a: first int
45
+ b: second int
46
+ """
47
+ return a - b
48
+
49
+ @tool
50
+ def divide(a: int, b: int) -> int:
51
+ """Divide two numbers.
52
+
53
+ Args:
54
+ a: first int
55
+ b: second int
56
+ """
57
+ if b == 0:
58
+ raise ValueError("Cannot divide by zero.")
59
+ return a / b
60
+
61
+ @tool
62
+ def modulus(a: int, b: int) -> int:
63
+ """Get the modulus of two numbers.
64
+
65
+ Args:
66
+ a: first int
67
+ b: second int
68
+ """
69
+ return a % b
70
+
71
+ @tool
72
+ def wiki_search(query: str) -> str:
73
+ """Search Wikipedia for a query and return maximum 2 results.
74
+
75
+ Args:
76
+ query: The search query."""
77
+ search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
78
+ formatted_search_docs = "\n\n---\n\n".join(
79
+ [
80
+ f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
81
+ for doc in search_docs
82
+ ])
83
+ return {"wiki_results": formatted_search_docs}
84
+
85
+ @tool
86
+ def web_search(query: str) -> str:
87
+ """Search Tavily for a query and return maximum 3 results.
88
+
89
+ Args:
90
+ query: The search query."""
91
+ search_docs = TavilySearchResults(max_results=3).invoke(query=query)
92
+ formatted_search_docs = "\n\n---\n\n".join(
93
+ [
94
+ f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
95
+ for doc in search_docs
96
+ ])
97
+ return {"web_results": formatted_search_docs}
98
+
99
+ @tool
100
+ def arvix_search(query: str) -> str:
101
+ """Search Arxiv for a query and return maximum 3 result.
102
+
103
+ Args:
104
+ query: The search query."""
105
+ search_docs = ArxivLoader(query=query, load_max_docs=3).load()
106
+ formatted_search_docs = "\n\n---\n\n".join(
107
+ [
108
+ f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
109
+ for doc in search_docs
110
+ ])
111
+ return {"arvix_results": formatted_search_docs}
112
+
113
+ # load the system prompt from the file
114
+ with open("system_prompt.txt", "r", encoding="utf-8") as f:
115
+ system_prompt = f.read()
116
+
117
+ # System message
118
+ sys_msg = SystemMessage(content=system_prompt)
119
+
120
+ # build a retriever
121
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") # dim=768
122
+ supabase: Client = create_client(
123
+ os.environ.get("SUPABASE_URL"),
124
+ os.environ.get("SUPABASE_SERVICE_KEY"))
125
+ vector_store = SupabaseVectorStore(
126
+ client=supabase,
127
+ embedding= embeddings,
128
+ table_name="documents",
129
+ query_name="get_docs",
130
+ )
131
+ create_retriever_tool = create_retriever_tool(
132
+ retriever=vector_store.as_retriever(),
133
+ name="Question Search",
134
+ description="A tool to retrieve similar questions from a vector store.",
135
+ )
136
+
137
+
138
+ tools = [
139
+ multiply,
140
+ add,
141
+ subtract,
142
+ divide,
143
+ modulus,
144
+ wiki_search,
145
+ web_search,
146
+ arvix_search,
147
+ ]
148
+
149
+ # Build the state graph
150
+ def build_graph():
151
+ llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
152
+ llm_with_tools = llm.bind_tools(tools)
153
+
154
+ def assistant_node(state: MessagesState):
155
+ """Assistant node"""
156
+ return {"messages": [llm_with_tools.invoke(state["messages"])]}
157
+
158
+ def retriever_node(state: MessagesState):
159
+ """Retriever node"""
160
+ similar_question = vector_store.similarity_search(state["messages"][0].content)
161
+ if not similar_question:
162
+ return {"messages": [HumanMessage(content="No similar questions found in the database.")]}
163
+ example_msg = HumanMessage(
164
+ content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
165
+ )
166
+ return {"messages": [sys_msg] + state["messages"] + [example_msg]}
167
+
168
+ build_graph = StateGraph(MessagesState)
169
+ build_graph.add_node("retreiver", retriever_node)
170
+ build_graph.add_node("assistant", assistant_node)
171
+ build_graph.add_node("tools", ToolNode(tools=tools))
172
+ build_graph.add_edge(START, "retreiver")
173
+ build_graph.add_edge("retreiver", "assistant")
174
+ build_graph.add_conditional_edges(
175
+ "assistant",
176
+ tools_condition
177
+ )
178
+ build_graph.add_edge("tools", "assistant")
179
+ return build_graph.compile()
180
+
181
+ # test
182
+ if __name__ == "__main__":
183
+ question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
184
+ # Build the graph
185
+ graph = build_graph()
186
+ # Run the graph
187
+ messages = [HumanMessage(content=question)]
188
+ messages = graph.invoke({"messages": messages})
189
+ for m in messages["messages"]:
190
+ m.pretty_print()
app.py ADDED
@@ -0,0 +1,203 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+ import requests
4
+ import inspect
5
+ import pandas as pd
6
+ from langchain_core.messages import HumanMessage
7
+ from agent import build_graph
8
+
9
+ # (Keep Constants as is)
10
+ # --- Constants ---
11
+ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
12
+
13
+ # --- Basic Agent Definition ---
14
+ # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
15
+ class BasicAgent:
16
+ """A langgraph agent."""
17
+ def __init__(self):
18
+ print("BasicAgent initialized.")
19
+ self.graph = build_graph()
20
+
21
+ def __call__(self, question: str) -> str:
22
+ print(f"Agent received question (first 50 chars): {question[:50]}...")
23
+ # Wrap the question in a HumanMessage from langchain_core
24
+ messages = [HumanMessage(content=question)]
25
+ messages = self.graph.invoke({"messages": messages})
26
+ answer = messages['messages'][-1].content
27
+ return answer[14:]
28
+
29
+ def run_and_submit_all( profile: gr.OAuthProfile | None):
30
+ """
31
+ Fetches all questions, runs the BasicAgent on them, submits all answers,
32
+ and displays the results.
33
+ """
34
+ # --- Determine HF Space Runtime URL and Repo URL ---
35
+ space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
36
+
37
+ if profile:
38
+ username= f"{profile.username}"
39
+ print(f"User logged in: {username}")
40
+ else:
41
+ print("User not logged in.")
42
+ return "Please Login to Hugging Face with the button.", None
43
+
44
+ api_url = DEFAULT_API_URL
45
+ questions_url = f"{api_url}/questions"
46
+ submit_url = f"{api_url}/submit"
47
+
48
+ # 1. Instantiate Agent ( modify this part to create your agent)
49
+ try:
50
+ agent = BasicAgent()
51
+ except Exception as e:
52
+ print(f"Error instantiating agent: {e}")
53
+ return f"Error initializing agent: {e}", None
54
+ # In the case of an app running as a hugging Face space, this link points toward your codebase ( usefull for others so please keep it public)
55
+ agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
56
+ print(agent_code)
57
+
58
+ # 2. Fetch Questions
59
+ print(f"Fetching questions from: {questions_url}")
60
+ try:
61
+ response = requests.get(questions_url, timeout=15)
62
+ response.raise_for_status()
63
+ questions_data = response.json()
64
+ if not questions_data:
65
+ print("Fetched questions list is empty.")
66
+ return "Fetched questions list is empty or invalid format.", None
67
+ print(f"Fetched {len(questions_data)} questions.")
68
+ except requests.exceptions.RequestException as e:
69
+ print(f"Error fetching questions: {e}")
70
+ return f"Error fetching questions: {e}", None
71
+ except requests.exceptions.JSONDecodeError as e:
72
+ print(f"Error decoding JSON response from questions endpoint: {e}")
73
+ print(f"Response text: {response.text[:500]}")
74
+ return f"Error decoding server response for questions: {e}", None
75
+ except Exception as e:
76
+ print(f"An unexpected error occurred fetching questions: {e}")
77
+ return f"An unexpected error occurred fetching questions: {e}", None
78
+
79
+ # 3. Run your Agent
80
+ results_log = []
81
+ answers_payload = []
82
+ print(f"Running agent on {len(questions_data)} questions...")
83
+ for item in questions_data:
84
+ task_id = item.get("task_id")
85
+ question_text = item.get("question")
86
+ if not task_id or question_text is None:
87
+ print(f"Skipping item with missing task_id or question: {item}")
88
+ continue
89
+ try:
90
+ submitted_answer = agent(question_text)
91
+ answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
92
+ results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
93
+ except Exception as e:
94
+ print(f"Error running agent on task {task_id}: {e}")
95
+ results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
96
+
97
+ if not answers_payload:
98
+ print("Agent did not produce any answers to submit.")
99
+ return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
100
+
101
+ # 4. Prepare Submission
102
+ submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
103
+ status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
104
+ print(status_update)
105
+
106
+ # 5. Submit
107
+ print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
108
+ try:
109
+ response = requests.post(submit_url, json=submission_data, timeout=60)
110
+ response.raise_for_status()
111
+ result_data = response.json()
112
+ final_status = (
113
+ f"Submission Successful!\n"
114
+ f"User: {result_data.get('username')}\n"
115
+ f"Overall Score: {result_data.get('score', 'N/A')}% "
116
+ f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
117
+ f"Message: {result_data.get('message', 'No message received.')}"
118
+ )
119
+ print("Submission successful.")
120
+ results_df = pd.DataFrame(results_log)
121
+ return final_status, results_df
122
+ except requests.exceptions.HTTPError as e:
123
+ error_detail = f"Server responded with status {e.response.status_code}."
124
+ try:
125
+ error_json = e.response.json()
126
+ error_detail += f" Detail: {error_json.get('detail', e.response.text)}"
127
+ except requests.exceptions.JSONDecodeError:
128
+ error_detail += f" Response: {e.response.text[:500]}"
129
+ status_message = f"Submission Failed: {error_detail}"
130
+ print(status_message)
131
+ results_df = pd.DataFrame(results_log)
132
+ return status_message, results_df
133
+ except requests.exceptions.Timeout:
134
+ status_message = "Submission Failed: The request timed out."
135
+ print(status_message)
136
+ results_df = pd.DataFrame(results_log)
137
+ return status_message, results_df
138
+ except requests.exceptions.RequestException as e:
139
+ status_message = f"Submission Failed: Network error - {e}"
140
+ print(status_message)
141
+ results_df = pd.DataFrame(results_log)
142
+ return status_message, results_df
143
+ except Exception as e:
144
+ status_message = f"An unexpected error occurred during submission: {e}"
145
+ print(status_message)
146
+ results_df = pd.DataFrame(results_log)
147
+ return status_message, results_df
148
+
149
+
150
+ # --- Build Gradio Interface using Blocks ---
151
+ with gr.Blocks() as demo:
152
+ gr.Markdown("# Basic Agent Evaluation Runner")
153
+ gr.Markdown(
154
+ """
155
+ **Instructions:**
156
+
157
+ 1. Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
158
+ 2. Log in to your Hugging Face account using the button below. This uses your HF username for submission.
159
+ 3. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
160
+
161
+ ---
162
+ **Disclaimers:**
163
+ Once clicking on the "submit button, it can take quite some time ( this is the time for the agent to go through all the questions).
164
+ This space provides a basic setup and is intentionally sub-optimal to encourage you to develop your own, more robust solution. For instance for the delay process of the submit button, a solution could be to cache the answers and submit in a seperate action or even to answer the questions in async.
165
+ """
166
+ )
167
+
168
+ gr.LoginButton()
169
+
170
+ run_button = gr.Button("Run Evaluation & Submit All Answers")
171
+
172
+ status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
173
+ # Removed max_rows=10 from DataFrame constructor
174
+ results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
175
+
176
+ run_button.click(
177
+ fn=run_and_submit_all,
178
+ outputs=[status_output, results_table]
179
+ )
180
+
181
+ if __name__ == "__main__":
182
+ print("\n" + "-"*30 + " App Starting " + "-"*30)
183
+ # Check for SPACE_HOST and SPACE_ID at startup for information
184
+ space_host_startup = os.getenv("SPACE_HOST")
185
+ space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup
186
+
187
+ if space_host_startup:
188
+ print(f"✅ SPACE_HOST found: {space_host_startup}")
189
+ print(f" Runtime URL should be: https://{space_host_startup}.hf.space")
190
+ else:
191
+ print("ℹ️ SPACE_HOST environment variable not found (running locally?).")
192
+
193
+ if space_id_startup: # Print repo URLs if SPACE_ID is found
194
+ print(f"✅ SPACE_ID found: {space_id_startup}")
195
+ print(f" Repo URL: https://huggingface.co/spaces/{space_id_startup}")
196
+ print(f" Repo Tree URL: https://huggingface.co/spaces/{space_id_startup}/tree/main")
197
+ else:
198
+ print("ℹ️ SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
199
+
200
+ print("-"*(60 + len(" App Starting ")) + "\n")
201
+
202
+ print("Launching Gradio Interface for Basic Agent Evaluation...")
203
+ demo.launch(debug=True, share=False)
gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
gitignore ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Python gitignore
2
+ __pycache__/
3
+ *.py[cod]
4
+ *.class
5
+ *.so
6
+ .env
7
+ .venv
8
+ env/
9
+ venv/
10
+ ENV/
11
+ .idea/
12
+ .vscode/
13
+ *.log
14
+ .DS_Store
15
+ .env
metadata.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
requirements.txt ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ gradio
2
+ requests
3
+ langchain
4
+ langchain-community
5
+ langchain-core
6
+ langchain-google-genai
7
+ langchain-huggingface
8
+ langchain-groq
9
+ langchain-tavily
10
+ langchain-chroma
11
+ langgraph
12
+ huggingface_hub
13
+ supabase
14
+ arxiv
15
+ pymupdf
16
+ wikipedia
17
+ pgvector
18
+ python-dotenv
system_prompt.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ You are a helpful assistant tasked with answering questions using a set of tools.
2
+ Now, I will ask you a question. Report your thoughts, and finish your answer with the following template:
3
+ FINAL ANSWER: [YOUR FINAL ANSWER].
4
+ 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.
5
+ Your answer should only start with "FINAL ANSWER: ", then follows with the answer.