mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
27 lines
956 B
Python
27 lines
956 B
Python
def condition_to_stop_receiving(response):
|
|
"""Determines when to stop listening to the server"""
|
|
if response.get("type") in ["agent_response_end", "agent_response_error", "command_response", "server_error"]:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def print_server_response(response):
|
|
"""Turn response json into a nice print"""
|
|
if response["type"] == "agent_response_start":
|
|
print("[agent.step start]")
|
|
elif response["type"] == "agent_response_end":
|
|
print("[agent.step end]")
|
|
elif response["type"] == "agent_response":
|
|
msg = response["message"]
|
|
if response["message_type"] == "internal_monologue":
|
|
print(f"[inner thoughts] {msg}")
|
|
elif response["message_type"] == "assistant_message":
|
|
print(f"{msg}")
|
|
elif response["message_type"] == "function_message":
|
|
pass
|
|
else:
|
|
print(response)
|
|
else:
|
|
print(response)
|