make sure the new code is being run.
Browse files- app.py +6 -0
- basic_agent.py +7 -3
app.py
CHANGED
@@ -31,7 +31,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
|
|
31 |
|
32 |
# 1. Instantiate Agent ( modify this part to create your agent)
|
33 |
try:
|
|
|
34 |
agent = basic_agent.BasicAgent()
|
|
|
35 |
except Exception as e:
|
36 |
print(f"Error instantiating agent: {e}")
|
37 |
return f"Error initializing agent: {e}", None
|
@@ -71,7 +73,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
|
|
71 |
print(f"Skipping item with missing task_id or question: {item}")
|
72 |
continue
|
73 |
try:
|
|
|
74 |
submitted_answer = agent(question_text)
|
|
|
75 |
answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
|
76 |
results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
|
77 |
except Exception as e:
|
@@ -183,5 +187,7 @@ if __name__ == "__main__":
|
|
183 |
|
184 |
print("-"*(60 + len(" App Starting ")) + "\n")
|
185 |
|
|
|
|
|
186 |
print("Launching Gradio Interface for Basic Agent Evaluation...")
|
187 |
demo.launch(debug=True, share=False)
|
|
|
31 |
|
32 |
# 1. Instantiate Agent ( modify this part to create your agent)
|
33 |
try:
|
34 |
+
print("This is new code creating the new basic_agent")
|
35 |
agent = basic_agent.BasicAgent()
|
36 |
+
print("This is new code creating the new basic_agent is now done")
|
37 |
except Exception as e:
|
38 |
print(f"Error instantiating agent: {e}")
|
39 |
return f"Error initializing agent: {e}", None
|
|
|
73 |
print(f"Skipping item with missing task_id or question: {item}")
|
74 |
continue
|
75 |
try:
|
76 |
+
print(f"calling agent")
|
77 |
submitted_answer = agent(question_text)
|
78 |
+
print(f"called agent: {submitted_answer=}")
|
79 |
answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
|
80 |
results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
|
81 |
except Exception as e:
|
|
|
187 |
|
188 |
print("-"*(60 + len(" App Starting ")) + "\n")
|
189 |
|
190 |
+
print("*"*10, "This is new code!", "*"*10)
|
191 |
+
|
192 |
print("Launching Gradio Interface for Basic Agent Evaluation...")
|
193 |
demo.launch(debug=True, share=False)
|
basic_agent.py
CHANGED
@@ -17,7 +17,8 @@ Take the time to plan the steps to reach the solution. Show the steps and then e
|
|
17 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
18 |
class BasicAgent:
|
19 |
def __init__(self, model_id=None):
|
20 |
-
print("BasicAgent
|
|
|
21 |
LOG.warning("logging BasicAgent initialized.")
|
22 |
|
23 |
if model_id:
|
@@ -33,6 +34,7 @@ class BasicAgent:
|
|
33 |
# max_new_tokens=32000,
|
34 |
# )
|
35 |
|
|
|
36 |
self.model = smolagents.HfApiModel(
|
37 |
max_tokens=32000,
|
38 |
temperature=0.3,
|
@@ -41,6 +43,7 @@ class BasicAgent:
|
|
41 |
)
|
42 |
self.tools = [smolagents.DuckDuckGoSearchTool(), smolagents.VisitWebpageTool(), smolagents.FinalAnswerTool()]
|
43 |
|
|
|
44 |
self.search_agent = smolagents.CodeAgent(
|
45 |
name="search_agent",
|
46 |
description="Search the web",
|
@@ -52,6 +55,7 @@ class BasicAgent:
|
|
52 |
additional_authorized_imports=["duckduckgo_search"],
|
53 |
)
|
54 |
|
|
|
55 |
self.manager_agent = smolagents.CodeAgent(
|
56 |
name="manager_agent",
|
57 |
description="Manger of other agents",
|
@@ -65,8 +69,8 @@ class BasicAgent:
|
|
65 |
)
|
66 |
|
67 |
def __call__(self, question: str) -> str:
|
68 |
-
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
69 |
prompt = f"{SYSTEM_PROMPT}\n\n{question}"
|
70 |
answer = self.manager_agent.run(prompt)
|
71 |
-
|
72 |
return answer
|
|
|
17 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
18 |
class BasicAgent:
|
19 |
def __init__(self, model_id=None):
|
20 |
+
print("BasicAgent initializing.")
|
21 |
+
# Logs appear to be swallowed.
|
22 |
LOG.warning("logging BasicAgent initialized.")
|
23 |
|
24 |
if model_id:
|
|
|
34 |
# max_new_tokens=32000,
|
35 |
# )
|
36 |
|
37 |
+
print("BasicAgent making model.")
|
38 |
self.model = smolagents.HfApiModel(
|
39 |
max_tokens=32000,
|
40 |
temperature=0.3,
|
|
|
43 |
)
|
44 |
self.tools = [smolagents.DuckDuckGoSearchTool(), smolagents.VisitWebpageTool(), smolagents.FinalAnswerTool()]
|
45 |
|
46 |
+
print("BasicAgent making search tool.")
|
47 |
self.search_agent = smolagents.CodeAgent(
|
48 |
name="search_agent",
|
49 |
description="Search the web",
|
|
|
55 |
additional_authorized_imports=["duckduckgo_search"],
|
56 |
)
|
57 |
|
58 |
+
print("BasicAgent making manager.")
|
59 |
self.manager_agent = smolagents.CodeAgent(
|
60 |
name="manager_agent",
|
61 |
description="Manger of other agents",
|
|
|
69 |
)
|
70 |
|
71 |
def __call__(self, question: str) -> str:
|
72 |
+
print(f"NEW Agent received question (first 50 chars): {question[:50]}...")
|
73 |
prompt = f"{SYSTEM_PROMPT}\n\n{question}"
|
74 |
answer = self.manager_agent.run(prompt)
|
75 |
+
print(f"NEW {answer=}")
|
76 |
return answer
|