Spaces:
Runtime error
Runtime error
abtsousa
commited on
Commit
·
f439125
1
Parent(s):
a40ea82
Changed test question to one in the set.
Browse files- agent/agent.py +2 -1
agent/agent.py
CHANGED
|
@@ -31,13 +31,14 @@ def get_agent():
|
|
| 31 |
if __name__ == "__main__":
|
| 32 |
import os
|
| 33 |
|
| 34 |
-
question = "
|
| 35 |
messages = [HumanMessage(content=question)]
|
| 36 |
|
| 37 |
try:
|
| 38 |
graph = get_agent()
|
| 39 |
from agent.config import create_agent_config
|
| 40 |
config = create_agent_config("OracleBot")
|
|
|
|
| 41 |
result = graph.invoke({"messages": messages}, config)
|
| 42 |
print("\n=== Agent Response ===")
|
| 43 |
for m in result["messages"]:
|
|
|
|
| 31 |
if __name__ == "__main__":
|
| 32 |
import os
|
| 33 |
|
| 34 |
+
question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
|
| 35 |
messages = [HumanMessage(content=question)]
|
| 36 |
|
| 37 |
try:
|
| 38 |
graph = get_agent()
|
| 39 |
from agent.config import create_agent_config
|
| 40 |
config = create_agent_config("OracleBot")
|
| 41 |
+
from app import start_phoenix
|
| 42 |
result = graph.invoke({"messages": messages}, config)
|
| 43 |
print("\n=== Agent Response ===")
|
| 44 |
for m in result["messages"]:
|