Commit 2b
Browse files
app.py
CHANGED
|
@@ -24,9 +24,10 @@ class BasicAgent:
|
|
| 24 |
api_key=os.environ.get("api_key"),
|
| 25 |
model_id=os.environ.get("model")
|
| 26 |
)
|
|
|
|
| 27 |
agent = CodeAgent(
|
| 28 |
model=model,
|
| 29 |
-
tools=[DuckDuckGoSearchTool(),
|
| 30 |
additional_authorized_imports=["pandas"],
|
| 31 |
max_steps=20,
|
| 32 |
)
|
|
@@ -34,7 +35,6 @@ class BasicAgent:
|
|
| 34 |
|
| 35 |
def __call__(self, question: str) -> str:
|
| 36 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
| 37 |
-
fixed_answer = "This is a default answer."
|
| 38 |
answer = agent.run(question)
|
| 39 |
print(f"Agent returning answer: {answer}")
|
| 40 |
return answer
|
|
|
|
| 24 |
api_key=os.environ.get("api_key"),
|
| 25 |
model_id=os.environ.get("model")
|
| 26 |
)
|
| 27 |
+
|
| 28 |
agent = CodeAgent(
|
| 29 |
model=model,
|
| 30 |
+
tools=[DuckDuckGoSearchTool(),VisitWebpageTool(),FinalAnswerTool()],
|
| 31 |
additional_authorized_imports=["pandas"],
|
| 32 |
max_steps=20,
|
| 33 |
)
|
|
|
|
| 35 |
|
| 36 |
def __call__(self, question: str) -> str:
|
| 37 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
|
|
|
| 38 |
answer = agent.run(question)
|
| 39 |
print(f"Agent returning answer: {answer}")
|
| 40 |
return answer
|