8ae1e64987
Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
from typing import List, Union
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from letta.schemas.enums import MessageStreamStatus
|
|
from letta.schemas.letta_message import LettaMessage
|
|
from letta.schemas.message import Message
|
|
from letta.schemas.usage import LettaUsageStatistics
|
|
|
|
# TODO: consider moving into own file
|
|
|
|
|
|
class LettaResponse(BaseModel):
|
|
"""
|
|
Response object from an agent interaction, consisting of the new messages generated by the agent and usage statistics.
|
|
The type of the returned messages can be either `Message` or `LettaMessage`, depending on what was specified in the request.
|
|
|
|
Attributes:
|
|
messages (List[Union[Message, LettaMessage]]): The messages returned by the agent.
|
|
usage (LettaUsageStatistics): The usage statistics
|
|
"""
|
|
|
|
messages: Union[List[Message], List[LettaMessage]] = Field(..., description="The messages returned by the agent.")
|
|
usage: LettaUsageStatistics = Field(..., description="The usage statistics of the agent.")
|
|
|
|
|
|
# The streaming response is either [DONE], [DONE_STEP], [DONE], an error, or a LettaMessage
|
|
LettaStreamingResponse = Union[LettaMessage, MessageStreamStatus]
|