Merge branch 'main' into chunk-ui

This commit is contained in:
Francisco Arceo 2025-08-07 12:13:57 -06:00 committed by GitHub
commit f658b25159
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -52,11 +52,16 @@ agent = Agent(
prompt = "How do you do great work?"
print("prompt>", prompt)
use_stream = True
response = agent.create_turn(
messages=[{"role": "user", "content": prompt}],
session_id=agent.create_session("rag_session"),
stream=True,
stream=use_stream,
)
for log in AgentEventLogger().log(response):
log.print()
# Only call `AgentEventLogger().log(response)` for streaming responses.
if use_stream:
for log in AgentEventLogger().log(response):
log.print()
else:
print(response)