Update app.py
Browse files
app.py
CHANGED
|
@@ -42,6 +42,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
| 42 |
|
| 43 |
# --- Basic Agent Definition ---
|
| 44 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
class BasicAgent:
|
| 46 |
def __init__(self):
|
| 47 |
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
|
|
@@ -55,7 +60,7 @@ class BasicAgent:
|
|
| 55 |
prompts = yaml.safe_load(stream)
|
| 56 |
|
| 57 |
#prompt_templates.final_answer.pre_messages = prompts['final_answer']['pre_messages']
|
| 58 |
-
|
| 59 |
self.agent = CodeAgent(
|
| 60 |
model=model,
|
| 61 |
tools=[FinalAnswerTool(), DuckDuckGoSearchTool(), VisitWebpageTool()], ## add your tools here (don't remove final answer),
|
|
@@ -65,13 +70,15 @@ class BasicAgent:
|
|
| 65 |
#step_callbacks=[delay_execution_10]
|
| 66 |
)
|
| 67 |
|
| 68 |
-
|
|
|
|
|
|
|
| 69 |
|
| 70 |
print("BasicAgent initialized.")
|
| 71 |
|
| 72 |
def __call__(self, question: str) -> str:
|
| 73 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
| 74 |
-
answer = self.agent.run(question)
|
| 75 |
print(f"Agent returning answer: {answer}")
|
| 76 |
return answer
|
| 77 |
|
|
|
|
| 42 |
|
| 43 |
# --- Basic Agent Definition ---
|
| 44 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
| 45 |
+
|
| 46 |
+
# RUNTIME PARAMS
|
| 47 |
+
set_inference_on = False
|
| 48 |
+
|
| 49 |
+
# Agent stuff starts here
|
| 50 |
class BasicAgent:
|
| 51 |
def __init__(self):
|
| 52 |
# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
|
|
|
|
| 60 |
prompts = yaml.safe_load(stream)
|
| 61 |
|
| 62 |
#prompt_templates.final_answer.pre_messages = prompts['final_answer']['pre_messages']
|
| 63 |
+
|
| 64 |
self.agent = CodeAgent(
|
| 65 |
model=model,
|
| 66 |
tools=[FinalAnswerTool(), DuckDuckGoSearchTool(), VisitWebpageTool()], ## add your tools here (don't remove final answer),
|
|
|
|
| 70 |
#step_callbacks=[delay_execution_10]
|
| 71 |
)
|
| 72 |
|
| 73 |
+
print('*$%')
|
| 74 |
+
|
| 75 |
+
print(self.agent.prompt_templates['system_prompt'])
|
| 76 |
|
| 77 |
print("BasicAgent initialized.")
|
| 78 |
|
| 79 |
def __call__(self, question: str) -> str:
|
| 80 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
| 81 |
+
answer = self.agent.run(question) if set_inference_on else "DEFAULT: INFERENCE IS SET TO INACTIVE."
|
| 82 |
print(f"Agent returning answer: {answer}")
|
| 83 |
return answer
|
| 84 |
|