Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -201,12 +201,3 @@ def build_graph(provider: str = "google"):
|
|
201 |
# Compile graph
|
202 |
return builder.compile()
|
203 |
|
204 |
-
if __name__ == "__main__":
|
205 |
-
question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
|
206 |
-
# Build the graph
|
207 |
-
graph = build_graph(provider="groq")
|
208 |
-
# Run the graph
|
209 |
-
messages = [HumanMessage(content=question)]
|
210 |
-
messages = graph.invoke({"messages": messages})
|
211 |
-
for m in messages["messages"]:
|
212 |
-
m.pretty_print()
|
|
|
201 |
# Compile graph
|
202 |
return builder.compile()
|
203 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|