Files
letta-server/letta/server/ws_api/protocol.py
Shubham Naik 5a743d1dc4 Add 'apps/core/' from commit 'ea2a7395f4023f5b9fab03e6273db3b64a1181d5'
git-subtree-dir: apps/core
git-subtree-mainline: a8963e11e7a5a0059acbc849ce768e1eee80df61
git-subtree-split: ea2a7395f4023f5b9fab03e6273db3b64a1181d5
2024-12-22 20:31:22 -08:00

101 lines
1.8 KiB
Python

from letta.utils import json_dumps
# Server -> client
def server_error(msg):
"""General server error"""
return json_dumps(
{
"type": "server_error",
"message": msg,
}
)
def server_command_response(status):
return json_dumps(
{
"type": "command_response",
"status": status,
}
)
def server_agent_response_error(msg):
return json_dumps(
{
"type": "agent_response_error",
"message": msg,
}
)
def server_agent_response_start():
return json_dumps(
{
"type": "agent_response_start",
}
)
def server_agent_response_end():
return json_dumps(
{
"type": "agent_response_end",
}
)
def server_agent_internal_monologue(msg):
return json_dumps(
{
"type": "agent_response",
"message_type": "internal_monologue",
"message": msg,
}
)
def server_agent_assistant_message(msg):
return json_dumps(
{
"type": "agent_response",
"message_type": "assistant_message",
"message": msg,
}
)
def server_agent_function_message(msg):
return json_dumps(
{
"type": "agent_response",
"message_type": "function_message",
"message": msg,
}
)
# Client -> server
def client_user_message(msg, agent_id=None):
return json_dumps(
{
"type": "user_message",
"message": msg,
"agent_id": agent_id,
}
)
def client_command_create(config):
return json_dumps(
{
"type": "command",
"command": "create_agent",
"config": config,
}
)