phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-15 06:00:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Incorporated suggested code modification

Browse source 
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
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
docs/source/getting_started/demo_script.py
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):
log.print()
# Only call `AgentEventLogger().log(response)` for streaming responses.
if use_stream:
for log in AgentEventLogger().log(response):
log.print()
else:
print(response)