letta-client 0.1.4__py3-none-any.whl → 0.1.9__py3-none-any.whl

letta/agents/messages/types/letta_streaming_response.py

@@ -1,20 +1,143 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
+from ....core.unchecked_base_model import UncheckedBaseModel
 import typing
-from ....types.system_message_output import SystemMessageOutput
-from ....types.user_message_output import UserMessageOutput
-from ....types.reasoning_message import ReasoningMessage
-from ....types.tool_call_message import ToolCallMessage
-from ....types.tool_return_message import ToolReturnMessage
-from ....types.assistant_message_output import AssistantMessageOutput
-from ....types.letta_usage_statistics import LettaUsageStatistics
-
-LettaStreamingResponse = typing.Union[
-    SystemMessageOutput,
-    UserMessageOutput,
-    ReasoningMessage,
-    ToolCallMessage,
-    ToolReturnMessage,
-    AssistantMessageOutput,
-    LettaUsageStatistics,
+import datetime as dt
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ....types.tool_call_message_tool_call import ToolCallMessageToolCall
+from ....types.tool_return_message_status import ToolReturnMessageStatus
+import typing_extensions
+from ....core.unchecked_base_model import UnionMetadata
+
+
+class LettaStreamingResponse_SystemMessage(UncheckedBaseModel):
+    message_type: typing.Literal["system_message"] = "system_message"
+    id: str
+    date: dt.datetime
+    message: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class LettaStreamingResponse_UserMessage(UncheckedBaseModel):
+    message_type: typing.Literal["user_message"] = "user_message"
+    id: str
+    date: dt.datetime
+    message: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class LettaStreamingResponse_ReasoningMessage(UncheckedBaseModel):
+    message_type: typing.Literal["reasoning_message"] = "reasoning_message"
+    id: str
+    date: dt.datetime
+    reasoning: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class LettaStreamingResponse_ToolCallMessage(UncheckedBaseModel):
+    message_type: typing.Literal["tool_call_message"] = "tool_call_message"
+    id: str
+    date: dt.datetime
+    tool_call: ToolCallMessageToolCall
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class LettaStreamingResponse_ToolReturnMessage(UncheckedBaseModel):
+    message_type: typing.Literal["tool_return_message"] = "tool_return_message"
+    id: str
+    date: dt.datetime
+    tool_return: str
+    status: ToolReturnMessageStatus
+    tool_call_id: str
+    stdout: typing.Optional[typing.List[str]] = None
+    stderr: typing.Optional[typing.List[str]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class LettaStreamingResponse_AssistantMessage(UncheckedBaseModel):
+    message_type: typing.Literal["assistant_message"] = "assistant_message"
+    id: str
+    date: dt.datetime
+    assistant_message: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class LettaStreamingResponse_UsageStatistics(UncheckedBaseModel):
+    message_type: typing.Literal["usage_statistics"] = "usage_statistics"
+    completion_tokens: typing.Optional[int] = None
+    prompt_tokens: typing.Optional[int] = None
+    total_tokens: typing.Optional[int] = None
+    step_count: typing.Optional[int] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+LettaStreamingResponse = typing_extensions.Annotated[
+    typing.Union[
+        LettaStreamingResponse_SystemMessage,
+        LettaStreamingResponse_UserMessage,
+        LettaStreamingResponse_ReasoningMessage,
+        LettaStreamingResponse_ToolCallMessage,
+        LettaStreamingResponse_ToolReturnMessage,
+        LettaStreamingResponse_AssistantMessage,
+        LettaStreamingResponse_UsageStatistics,
+    ],
+    UnionMetadata(discriminant="message_type"),
 ]

letta/agents/messages/types/messages_list_response_item.py

@@ -1,13 +1,125 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
+from ....core.unchecked_base_model import UncheckedBaseModel
 import typing
-from ....types.system_message_output import SystemMessageOutput
-from ....types.user_message_output import UserMessageOutput
-from ....types.reasoning_message import ReasoningMessage
-from ....types.tool_call_message import ToolCallMessage
-from ....types.tool_return_message import ToolReturnMessage
-from ....types.assistant_message_output import AssistantMessageOutput
-
-MessagesListResponseItem = typing.Union[
-    SystemMessageOutput, UserMessageOutput, ReasoningMessage, ToolCallMessage, ToolReturnMessage, AssistantMessageOutput
+import datetime as dt
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ....types.tool_call_message_tool_call import ToolCallMessageToolCall
+from ....types.tool_return_message_status import ToolReturnMessageStatus
+import typing_extensions
+from ....core.unchecked_base_model import UnionMetadata
+
+
+class MessagesListResponseItem_SystemMessage(UncheckedBaseModel):
+    message_type: typing.Literal["system_message"] = "system_message"
+    id: str
+    date: dt.datetime
+    message: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class MessagesListResponseItem_UserMessage(UncheckedBaseModel):
+    message_type: typing.Literal["user_message"] = "user_message"
+    id: str
+    date: dt.datetime
+    message: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class MessagesListResponseItem_ReasoningMessage(UncheckedBaseModel):
+    message_type: typing.Literal["reasoning_message"] = "reasoning_message"
+    id: str
+    date: dt.datetime
+    reasoning: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class MessagesListResponseItem_ToolCallMessage(UncheckedBaseModel):
+    message_type: typing.Literal["tool_call_message"] = "tool_call_message"
+    id: str
+    date: dt.datetime
+    tool_call: ToolCallMessageToolCall
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class MessagesListResponseItem_ToolReturnMessage(UncheckedBaseModel):
+    message_type: typing.Literal["tool_return_message"] = "tool_return_message"
+    id: str
+    date: dt.datetime
+    tool_return: str
+    status: ToolReturnMessageStatus
+    tool_call_id: str
+    stdout: typing.Optional[typing.List[str]] = None
+    stderr: typing.Optional[typing.List[str]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class MessagesListResponseItem_AssistantMessage(UncheckedBaseModel):
+    message_type: typing.Literal["assistant_message"] = "assistant_message"
+    id: str
+    date: dt.datetime
+    assistant_message: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+MessagesListResponseItem = typing_extensions.Annotated[
+    typing.Union[
+        MessagesListResponseItem_SystemMessage,
+        MessagesListResponseItem_UserMessage,
+        MessagesListResponseItem_ReasoningMessage,
+        MessagesListResponseItem_ToolCallMessage,
+        MessagesListResponseItem_ToolReturnMessage,
+        MessagesListResponseItem_AssistantMessage,
+    ],
+    UnionMetadata(discriminant="message_type"),
 ]

letta/agents/recall_memory/client.py

@@ -5,7 +5,7 @@ import typing
 from ...core.request_options import RequestOptions
 from ...types.recall_memory_summary import RecallMemorySummary
 from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import parse_obj_as
+from ...core.unchecked_base_model import construct_type
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
@@ -55,7 +55,7 @@ class RecallMemoryClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     RecallMemorySummary,
-                    parse_obj_as(
+                    construct_type(
                         type_=RecallMemorySummary,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -64,7 +64,7 @@ class RecallMemoryClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -126,7 +126,7 @@ class AsyncRecallMemoryClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     RecallMemorySummary,
-                    parse_obj_as(
+                    construct_type(
                         type_=RecallMemorySummary,  # type: ignore
                         object_=_response.json(),
                     ),
@@ -135,7 +135,7 @@ class AsyncRecallMemoryClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),

letta/agents/sources/client.py

@@ -5,7 +5,7 @@ import typing
 from ...core.request_options import RequestOptions
 from ...types.source import Source
 from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import parse_obj_as
+from ...core.unchecked_base_model import construct_type
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
@@ -53,7 +53,7 @@ class SourcesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     typing.List[Source],
-                    parse_obj_as(
+                    construct_type(
                         type_=typing.List[Source],  # type: ignore
                         object_=_response.json(),
                     ),
@@ -62,7 +62,7 @@ class SourcesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),
@@ -124,7 +124,7 @@ class AsyncSourcesClient:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
                     typing.List[Source],
-                    parse_obj_as(
+                    construct_type(
                         type_=typing.List[Source],  # type: ignore
                         object_=_response.json(),
                     ),
@@ -133,7 +133,7 @@ class AsyncSourcesClient:
                 raise UnprocessableEntityError(
                     typing.cast(
                         HttpValidationError,
-                        parse_obj_as(
+                        construct_type(
                             type_=HttpValidationError,  # type: ignore
                             object_=_response.json(),
                         ),

letta/agents/templates/__init__.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .types import TemplatesMigrateResponse
+
+__all__ = ["TemplatesMigrateResponse"]