changes to agent.py
Browse files
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="
|
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)
|