Pretty print output
This commit is contained in:
parent
f43293819f
commit
86bd32a9e6
@ -1,3 +1,4 @@
|
|||||||
|
import json # Import the json module
|
||||||
from qwen_agent.agents import Assistant
|
from qwen_agent.agents import Assistant
|
||||||
|
|
||||||
# Define LLM
|
# Define LLM
|
||||||
@ -40,7 +41,15 @@ tools = [
|
|||||||
bot = Assistant(llm=llm_cfg, function_list=tools)
|
bot = Assistant(llm=llm_cfg, function_list=tools)
|
||||||
|
|
||||||
# Streaming generation
|
# Streaming generation
|
||||||
messages = [{'role': 'user', 'content': 'https://qwenlm.github.io/blog/ Introduce the latest developments of Qwen'}]
|
messages = [{'role': 'user', 'content': 'https://qwenlm.github.io/blog/ Sumarize the latest developments of Qwen'}]
|
||||||
|
# Initialize responses variable before the loop in case the loop doesn't run
|
||||||
|
final_responses = None
|
||||||
for responses in bot.run(messages=messages):
|
for responses in bot.run(messages=messages):
|
||||||
pass
|
# The loop assigns the latest response to final_responses
|
||||||
print(responses)
|
final_responses = responses
|
||||||
|
|
||||||
|
# Pretty-print the final response object
|
||||||
|
if final_responses:
|
||||||
|
print(json.dumps(final_responses, indent=2)) # Use indent=2 (or 4) for pretty printing
|
||||||
|
else:
|
||||||
|
print("No response received from the agent.")
|
Loading…
x
Reference in New Issue
Block a user