Update app.py
Browse files
app.py
CHANGED
|
@@ -126,7 +126,6 @@ def wikipedia_search(query: str) -> str:
|
|
| 126 |
# === Constants ===
|
| 127 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
| 128 |
|
| 129 |
-
# === Agent Definition ===
|
| 130 |
class BasicAgent:
|
| 131 |
def __init__(self):
|
| 132 |
model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
|
|
@@ -191,10 +190,9 @@ class BasicAgent:
|
|
| 191 |
)
|
| 192 |
print("✅ Agent initialized with all tools")
|
| 193 |
|
| 194 |
-
|
| 195 |
response = self.agent.run(question)
|
| 196 |
|
| 197 |
-
# Use fallback RAG answers if the output is inconclusive
|
| 198 |
if (
|
| 199 |
"Final Answer:" not in response
|
| 200 |
or "None" in response
|
|
@@ -207,7 +205,6 @@ class BasicAgent:
|
|
| 207 |
if key.lower() in question.lower():
|
| 208 |
return f"Tool response inconclusive. Based on known data: Final Answer: {fallback}"
|
| 209 |
|
| 210 |
-
# Default to last line as Final Answer if not present
|
| 211 |
if "Final Answer:" not in response:
|
| 212 |
last_line = response.strip().split("\n")[-1]
|
| 213 |
response += f"\nFinal Answer: {last_line}"
|
|
|
|
| 126 |
# === Constants ===
|
| 127 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
| 128 |
|
|
|
|
| 129 |
class BasicAgent:
|
| 130 |
def __init__(self):
|
| 131 |
model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
|
|
|
|
| 190 |
)
|
| 191 |
print("✅ Agent initialized with all tools")
|
| 192 |
|
| 193 |
+
def __call__(self, question: str) -> str:
|
| 194 |
response = self.agent.run(question)
|
| 195 |
|
|
|
|
| 196 |
if (
|
| 197 |
"Final Answer:" not in response
|
| 198 |
or "None" in response
|
|
|
|
| 205 |
if key.lower() in question.lower():
|
| 206 |
return f"Tool response inconclusive. Based on known data: Final Answer: {fallback}"
|
| 207 |
|
|
|
|
| 208 |
if "Final Answer:" not in response:
|
| 209 |
last_line = response.strip().split("\n")[-1]
|
| 210 |
response += f"\nFinal Answer: {last_line}"
|