abtsousa commited on
Commit
f439125
·
1 Parent(s): a40ea82

Changed test question to one in the set.

Browse files
Files changed (1) hide show
  1. 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 = "When was Andrej Karpathy born?"
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"]: