Qscar KIM commited on
Commit
19eab17
ยท
1 Parent(s): 81917a3

update codes

Browse files
Files changed (3) hide show
  1. agent.py +33 -0
  2. app.py +45 -134
  3. requirements.txt +7 -2
agent.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from smolagents import CodeAgent, HfApiClientModel, Tool
3
+ from langchain_community.tools import DuckDuckGoSearchRun
4
+
5
+ class WebSearchTool(Tool):
6
+ name = "web_search"
7
+ description = "์ธํ„ฐ๋„ท์—์„œ ์ตœ์‹  ์ •๋ณด๋‚˜ ์ง€์‹์„ ๊ฒ€์ƒ‰ํ•  ๋•Œ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค."
8
+ inputs = {"query": {"type": "string", "description": "๊ฒ€์ƒ‰ํ•  ๋‹จ์–ด๋‚˜ ๋ฌธ์žฅ"}}
9
+ output_type = "string"
10
+
11
+ def __init__(self):
12
+ super().__init__()
13
+ self.search = DuckDuckGoSearchRun()
14
+
15
+ def forward(self, query: str) -> str:
16
+ try:
17
+ return self.search.run(query)
18
+ except Exception as e:
19
+ return f"๊ฒ€์ƒ‰ ์‹คํŒจ: {str(e)}"
20
+
21
+ def create_gaia_agent():
22
+ model = HfApiClientModel(
23
+ model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
24
+ )
25
+
26
+ search_tool = WebSearchTool()
27
+ agent = CodeAgent(
28
+ tools=[search_tool],
29
+ model=model,
30
+ additional_authorized_imports=["pandas", "numpy", "json", "math", "re"]
31
+ )
32
+
33
+ return agent
app.py CHANGED
@@ -1,169 +1,101 @@
1
  import os
2
  import gradio as gr
3
  import requests
4
- import inspect
5
  import pandas as pd
 
 
6
 
7
- # (Keep Constants as is)
8
- # --- Constants ---
9
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
10
 
11
- # --- Basic Agent Definition ---
12
- # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
  class BasicAgent:
14
  def __init__(self):
15
- print("BasicAgent initialized.")
 
 
 
16
  def __call__(self, question: str) -> str:
17
- print(f"Agent received question (first 50 chars): {question[:50]}...")
18
- fixed_answer = "This is a default answer."
19
- print(f"Agent returning fixed answer: {fixed_answer}")
20
- return fixed_answer
21
-
22
- def run_and_submit_all( profile: gr.OAuthProfile | None):
23
- """
24
- Fetches all questions, runs the BasicAgent on them, submits all answers,
25
- and displays the results.
26
- """
27
- # --- Determine HF Space Runtime URL and Repo URL ---
28
- space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
 
 
 
29
 
30
  if profile:
31
- username= f"{profile.username}"
32
- print(f"User logged in: {username}")
33
  else:
34
- print("User not logged in.")
35
- return "Please Login to Hugging Face with the button.", None
36
 
37
- api_url = DEFAULT_API_URL
38
- questions_url = f"{api_url}/questions"
39
- submit_url = f"{api_url}/submit"
40
 
41
- # 1. Instantiate Agent ( modify this part to create your agent)
42
  try:
43
  agent = BasicAgent()
44
  except Exception as e:
45
- print(f"Error instantiating agent: {e}")
46
- return f"Error initializing agent: {e}", None
47
- # 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)
48
  agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
49
- print(agent_code)
50
 
51
- # 2. Fetch Questions
52
- print(f"Fetching questions from: {questions_url}")
53
  try:
54
  response = requests.get(questions_url, timeout=15)
55
  response.raise_for_status()
56
  questions_data = response.json()
57
- if not questions_data:
58
- print("Fetched questions list is empty.")
59
- return "Fetched questions list is empty or invalid format.", None
60
- print(f"Fetched {len(questions_data)} questions.")
61
- except requests.exceptions.RequestException as e:
62
- print(f"Error fetching questions: {e}")
63
- return f"Error fetching questions: {e}", None
64
- except requests.exceptions.JSONDecodeError as e:
65
- print(f"Error decoding JSON response from questions endpoint: {e}")
66
- print(f"Response text: {response.text[:500]}")
67
- return f"Error decoding server response for questions: {e}", None
68
  except Exception as e:
69
- print(f"An unexpected error occurred fetching questions: {e}")
70
- return f"An unexpected error occurred fetching questions: {e}", None
71
 
72
- # 3. Run your Agent
73
  results_log = []
74
  answers_payload = []
75
- print(f"Running agent on {len(questions_data)} questions...")
76
  for item in questions_data:
77
  task_id = item.get("task_id")
78
  question_text = item.get("question")
79
  if not task_id or question_text is None:
80
- print(f"Skipping item with missing task_id or question: {item}")
81
  continue
82
  try:
83
  submitted_answer = agent(question_text)
84
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
85
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
 
 
86
  except Exception as e:
87
- print(f"Error running agent on task {task_id}: {e}")
88
- results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
89
 
90
  if not answers_payload:
91
- print("Agent did not produce any answers to submit.")
92
- return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
93
 
94
- # 4. Prepare Submission
95
  submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
96
- status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
97
- print(status_update)
98
-
99
- # 5. Submit
100
- print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
101
  try:
102
- response = requests.post(submit_url, json=submission_data, timeout=60)
103
  response.raise_for_status()
104
  result_data = response.json()
105
  final_status = (
106
- f"Submission Successful!\n"
107
  f"User: {result_data.get('username')}\n"
108
- f"Overall Score: {result_data.get('score', 'N/A')}% "
109
- f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
110
- f"Message: {result_data.get('message', 'No message received.')}"
111
  )
112
- print("Submission successful.")
113
- results_df = pd.DataFrame(results_log)
114
- return final_status, results_df
115
- except requests.exceptions.HTTPError as e:
116
- error_detail = f"Server responded with status {e.response.status_code}."
117
- try:
118
- error_json = e.response.json()
119
- error_detail += f" Detail: {error_json.get('detail', e.response.text)}"
120
- except requests.exceptions.JSONDecodeError:
121
- error_detail += f" Response: {e.response.text[:500]}"
122
- status_message = f"Submission Failed: {error_detail}"
123
- print(status_message)
124
- results_df = pd.DataFrame(results_log)
125
- return status_message, results_df
126
- except requests.exceptions.Timeout:
127
- status_message = "Submission Failed: The request timed out."
128
- print(status_message)
129
- results_df = pd.DataFrame(results_log)
130
- return status_message, results_df
131
- except requests.exceptions.RequestException as e:
132
- status_message = f"Submission Failed: Network error - {e}"
133
- print(status_message)
134
- results_df = pd.DataFrame(results_log)
135
- return status_message, results_df
136
  except Exception as e:
137
- status_message = f"An unexpected error occurred during submission: {e}"
138
- print(status_message)
139
- results_df = pd.DataFrame(results_log)
140
- return status_message, results_df
141
-
142
 
143
- # --- Build Gradio Interface using Blocks ---
144
  with gr.Blocks() as demo:
145
- gr.Markdown("# Basic Agent Evaluation Runner")
146
- gr.Markdown(
147
- """
148
- **Instructions:**
149
-
150
- 1. Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
151
- 2. Log in to your Hugging Face account using the button below. This uses your HF username for submission.
152
- 3. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
153
-
154
- ---
155
- **Disclaimers:**
156
- 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).
157
- 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.
158
- """
159
- )
160
-
161
  gr.LoginButton()
162
-
163
  run_button = gr.Button("Run Evaluation & Submit All Answers")
164
-
165
  status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
166
- # Removed max_rows=10 from DataFrame constructor
167
  results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
168
 
169
  run_button.click(
@@ -172,25 +104,4 @@ with gr.Blocks() as demo:
172
  )
173
 
174
  if __name__ == "__main__":
175
- print("\n" + "-"*30 + " App Starting " + "-"*30)
176
- # Check for SPACE_HOST and SPACE_ID at startup for information
177
- space_host_startup = os.getenv("SPACE_HOST")
178
- space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup
179
-
180
- if space_host_startup:
181
- print(f"โœ… SPACE_HOST found: {space_host_startup}")
182
- print(f" Runtime URL should be: https://{space_host_startup}.hf.space")
183
- else:
184
- print("โ„น๏ธ SPACE_HOST environment variable not found (running locally?).")
185
-
186
- if space_id_startup: # Print repo URLs if SPACE_ID is found
187
- print(f"โœ… SPACE_ID found: {space_id_startup}")
188
- print(f" Repo URL: https://huggingface.co/spaces/{space_id_startup}")
189
- print(f" Repo Tree URL: https://huggingface.co/spaces/{space_id_startup}/tree/main")
190
- else:
191
- print("โ„น๏ธ SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
192
-
193
- print("-"*(60 + len(" App Starting ")) + "\n")
194
-
195
- print("Launching Gradio Interface for Basic Agent Evaluation...")
196
- demo.launch(debug=True, share=False)
 
1
  import os
2
  import gradio as gr
3
  import requests
 
4
  import pandas as pd
5
+ import time
6
+ from agent import create_gaia_agent
7
 
 
 
8
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
9
 
 
 
10
  class BasicAgent:
11
  def __init__(self):
12
+ print("GAIA ์—์ด์ „ํŠธ๋ฅผ ์ค€๋น„ํ•ฉ๋‹ˆ๋‹ค...")
13
+ self.real_agent = create_gaia_agent()
14
+ print("์—์ด์ „ํŠธ ์ค€๋น„ ์™„๋ฃŒ!")
15
+
16
  def __call__(self, question: str) -> str:
17
+ print(f"๋ฐ›์€ ์งˆ๋ฌธ: {question[:50]}...")
18
+ try:
19
+ prompt = (
20
+ f"You are a precise assistant. Solve this task: {question}\n\n"
21
+ f"Provide the final answer as briefly as possible at the very end. "
22
+ f"No explanations in the last line."
23
+ )
24
+ result = self.real_agent.run(prompt)
25
+ return str(result).strip()
26
+ except Exception as e:
27
+ print(f"์—๋Ÿฌ ๋ฐœ์ƒ: {e}")
28
+ return "unknown"
29
+
30
+ def run_and_submit_all(profile: gr.OAuthProfile | None):
31
+ space_id = os.getenv("SPACE_ID")
32
 
33
  if profile:
34
+ username = f"{profile.username}"
35
+ print(f"๋กœ๊ทธ์ธ ์œ ์ €: {username}")
36
  else:
37
+ return "์˜ค๋ฅธ์ชฝ ์œ„ Login ๋ฒ„ํŠผ์„ ๋ˆŒ๋Ÿฌ Hugging Face ๊ณ„์ •์œผ๋กœ ๋จผ์ € ๋กœ๊ทธ์ธํ•ด ์ฃผ์„ธ์š”!", None
 
38
 
39
+ questions_url = f"{DEFAULT_API_URL}/questions"
40
+ submit_url = f"{DEFAULT_API_URL}/submit"
 
41
 
 
42
  try:
43
  agent = BasicAgent()
44
  except Exception as e:
45
+ return f"์—์ด์ „ํŠธ ์ƒ์„ฑ ์‹คํŒจ: {e}", None
46
+
 
47
  agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
48
 
 
 
49
  try:
50
  response = requests.get(questions_url, timeout=15)
51
  response.raise_for_status()
52
  questions_data = response.json()
 
 
 
 
 
 
 
 
 
 
 
53
  except Exception as e:
54
+ return f"๋ฌธ์ œ๋ฅผ ๊ฐ€์ ธ์˜ค์ง€ ๋ชปํ–ˆ์Šต๋‹ˆ๋‹ค: {e}", None
 
55
 
 
56
  results_log = []
57
  answers_payload = []
58
+
59
  for item in questions_data:
60
  task_id = item.get("task_id")
61
  question_text = item.get("question")
62
  if not task_id or question_text is None:
 
63
  continue
64
  try:
65
  submitted_answer = agent(question_text)
66
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
67
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
68
+
69
+ time.sleep(2)
70
  except Exception as e:
71
+ results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"ERROR: {e}"})
 
72
 
73
  if not answers_payload:
74
+ return "์ œ์ถœํ•  ์ •๋‹ต์ด ๋น„์–ด์žˆ์Šต๋‹ˆ๋‹ค.", pd.DataFrame(results_log)
 
75
 
 
76
  submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
77
+
 
 
 
 
78
  try:
79
+ response = requests.post(submit_url, json=submission_data, timeout=90)
80
  response.raise_for_status()
81
  result_data = response.json()
82
  final_status = (
83
+ f" Submission Successful!\n"
84
  f"User: {result_data.get('username')}\n"
85
+ f"Overall Score: {result_data.get('score', 'N/A')}%\n"
86
+ f"Message: {result_data.get('message', '์ฑ„์  ์™„๋ฃŒ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.')}"
 
87
  )
88
+ return final_status, pd.DataFrame(results_log)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
89
  except Exception as e:
90
+ return f"์„œ๋ฒ„ ์ „์†ก ์‹คํŒจ: {e}", pd.DataFrame(results_log)
 
 
 
 
91
 
92
+ # --- Gradio UI ํŒŒํŠธ (๊ธฐ๋ณธ ๊ณจ์กฐ ์œ ์ง€) ---
93
  with gr.Blocks() as demo:
94
+ gr.Markdown("# ๐Ÿš€ Open-Source Model GAIA Agent Runner")
95
+ gr.Markdown("์™ธ๋ถ€ ์œ ๋ฃŒ Key ์—†์ด Hugging Face Inference API ๋ชจ๋ธ๋กœ ๋™์ž‘ํ•˜๋Š” ์ž์œจ ์—์ด์ „ํŠธ์ž…๋‹ˆ๋‹ค.")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
96
  gr.LoginButton()
 
97
  run_button = gr.Button("Run Evaluation & Submit All Answers")
 
98
  status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
 
99
  results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
100
 
101
  run_button.click(
 
104
  )
105
 
106
  if __name__ == "__main__":
107
+ demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -1,2 +1,7 @@
1
- gradio
2
- requests
 
 
 
 
 
 
1
+ smolagents>=1.0.0
2
+ duckduckgo-search
3
+ pandas
4
+ openpyxl
5
+ pillow
6
+ requests
7
+ gradio