GoldsWolf commited on
Commit
9a880aa
·
1 Parent(s): 3f0e951

changes to agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -390,7 +390,7 @@ def build_graph(provider: str = "groq"):
390
  return builder.compile()
391
 
392
  if __name__ == "__main__":
393
- app = build_graph(provider="groq")
394
  inputs = {"messages": [HumanMessage(content="I was reading this book last year, and it had this really cool description where it referred to something as looking like a manta ray. I\u2019m trying to figure out when I read that part, but I can\u2019t find what book it\u2019s from. This file I attached has a list of the books I read last year, with the date I started and finished reading each one. I\u2019d like for you to tell me the month in which I likely read the passage about the manta ray.")]}
395
  result = app.invoke(inputs)
396
  print(result)
 
390
  return builder.compile()
391
 
392
  if __name__ == "__main__":
393
+ app = build_graph(provider="huggingface")
394
  inputs = {"messages": [HumanMessage(content="I was reading this book last year, and it had this really cool description where it referred to something as looking like a manta ray. I\u2019m trying to figure out when I read that part, but I can\u2019t find what book it\u2019s from. This file I attached has a list of the books I read last year, with the date I started and finished reading each one. I\u2019d like for you to tell me the month in which I likely read the passage about the manta ray.")]}
395
  result = app.invoke(inputs)
396
  print(result)