Incorporated suggested code modification

Signed-off-by: Dean Wampler <dean.wampler@ibm.com>
This commit is contained in:
Dean Wampler 2025-08-07 11:57:46 -04:00
parent 688f42ba3a
commit 3ed082e69a
No known key found for this signature in database

View file

@ -52,14 +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,
)
# NOTE: `AgentEventLogger().log(response)` only works with streaming responses.
# So, if you change the value for `stream` to `False` in the `create_turn` call,
# comment out the following two lines!
for log in AgentEventLogger().log(response):
# Only call `AgentEventLogger().log(response)` for streaming responses.
if use_stream:
for log in AgentEventLogger().log(response):
log.print()
else:
print(response)