letta-client 0.1.323__py3-none-any.whl → 1.0.0a1__py3-none-any.whl

letta_client/templates/agents/__init__.py

@@ -2,72 +2,3 @@
 
 # isort: skip_file
 
-from .types import (
-    AgentsCreateRequestInitialMessageSequenceItem,
-    AgentsCreateRequestInitialMessageSequenceItemRole,
-    AgentsCreateResponse,
-    AgentsCreateResponseGroup,
-    AgentsCreateResponseGroupBaseTemplateId,
-    AgentsCreateResponseGroupBaseTemplateIdItem,
-    AgentsCreateResponseGroupDeploymentId,
-    AgentsCreateResponseGroupDeploymentIdItem,
-    AgentsCreateResponseGroupHidden,
-    AgentsCreateResponseGroupHiddenItem,
-    AgentsCreateResponseGroupLastProcessedMessageId,
-    AgentsCreateResponseGroupLastProcessedMessageIdItem,
-    AgentsCreateResponseGroupManagerAgentId,
-    AgentsCreateResponseGroupManagerAgentIdItem,
-    AgentsCreateResponseGroupManagerType,
-    AgentsCreateResponseGroupMaxMessageBufferLength,
-    AgentsCreateResponseGroupMaxMessageBufferLengthItem,
-    AgentsCreateResponseGroupMaxTurns,
-    AgentsCreateResponseGroupMaxTurnsItem,
-    AgentsCreateResponseGroupMinMessageBufferLength,
-    AgentsCreateResponseGroupMinMessageBufferLengthItem,
-    AgentsCreateResponseGroupProjectId,
-    AgentsCreateResponseGroupProjectIdItem,
-    AgentsCreateResponseGroupSharedBlockIds,
-    AgentsCreateResponseGroupSleeptimeAgentFrequency,
-    AgentsCreateResponseGroupSleeptimeAgentFrequencyItem,
-    AgentsCreateResponseGroupTemplateId,
-    AgentsCreateResponseGroupTemplateIdItem,
-    AgentsCreateResponseGroupTerminationToken,
-    AgentsCreateResponseGroupTerminationTokenItem,
-    AgentsCreateResponseGroupTurnsCounter,
-    AgentsCreateResponseGroupTurnsCounterItem,
-)
-
-__all__ = [
-    "AgentsCreateRequestInitialMessageSequenceItem",
-    "AgentsCreateRequestInitialMessageSequenceItemRole",
-    "AgentsCreateResponse",
-    "AgentsCreateResponseGroup",
-    "AgentsCreateResponseGroupBaseTemplateId",
-    "AgentsCreateResponseGroupBaseTemplateIdItem",
-    "AgentsCreateResponseGroupDeploymentId",
-    "AgentsCreateResponseGroupDeploymentIdItem",
-    "AgentsCreateResponseGroupHidden",
-    "AgentsCreateResponseGroupHiddenItem",
-    "AgentsCreateResponseGroupLastProcessedMessageId",
-    "AgentsCreateResponseGroupLastProcessedMessageIdItem",
-    "AgentsCreateResponseGroupManagerAgentId",
-    "AgentsCreateResponseGroupManagerAgentIdItem",
-    "AgentsCreateResponseGroupManagerType",
-    "AgentsCreateResponseGroupMaxMessageBufferLength",
-    "AgentsCreateResponseGroupMaxMessageBufferLengthItem",
-    "AgentsCreateResponseGroupMaxTurns",
-    "AgentsCreateResponseGroupMaxTurnsItem",
-    "AgentsCreateResponseGroupMinMessageBufferLength",
-    "AgentsCreateResponseGroupMinMessageBufferLengthItem",
-    "AgentsCreateResponseGroupProjectId",
-    "AgentsCreateResponseGroupProjectIdItem",
-    "AgentsCreateResponseGroupSharedBlockIds",
-    "AgentsCreateResponseGroupSleeptimeAgentFrequency",
-    "AgentsCreateResponseGroupSleeptimeAgentFrequencyItem",
-    "AgentsCreateResponseGroupTemplateId",
-    "AgentsCreateResponseGroupTemplateIdItem",
-    "AgentsCreateResponseGroupTerminationToken",
-    "AgentsCreateResponseGroupTerminationTokenItem",
-    "AgentsCreateResponseGroupTurnsCounter",
-    "AgentsCreateResponseGroupTurnsCounterItem",
-]

letta_client/templates/agents/client.py

@@ -5,11 +5,6 @@ import typing
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from .raw_client import AsyncRawAgentsClient, RawAgentsClient
-from .types.agents_create_request_initial_message_sequence_item import AgentsCreateRequestInitialMessageSequenceItem
-from .types.agents_create_response import AgentsCreateResponse
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
 
 
 class AgentsClient:
@@ -28,56 +23,21 @@ class AgentsClient:
         return self._raw_client
 
     def create(
-        self,
-        project: str,
-        template_version: str,
-        *,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        agent_name: typing.Optional[str] = OMIT,
-        initial_message_sequence: typing.Optional[
-            typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]
-        ] = OMIT,
-        memory_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        tool_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        identity_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> AgentsCreateResponse:
+        self, project: str, template_version: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
-        Creates an Agent or multiple Agents from a template
-
         Parameters
         ----------
         project : str
-            The project slug
 
         template_version : str
-            The template version, formatted as {template-name}:{version-number} or {template-name}:latest
-
-        tags : typing.Optional[typing.Sequence[str]]
-            The tags to assign to the agent
-
-        agent_name : typing.Optional[str]
-            The name of the agent, optional otherwise a random one will be assigned
-
-        initial_message_sequence : typing.Optional[typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]]
-            Set an initial sequence of messages, if not provided, the agent will start with the default message sequence, if an empty array is provided, the agent will start with no messages
-
-        memory_variables : typing.Optional[typing.Dict[str, str]]
-            The memory variables to assign to the agent
-
-        tool_variables : typing.Optional[typing.Dict[str, str]]
-            The tool variables to assign to the agent
-
-        identity_ids : typing.Optional[typing.Sequence[str]]
-            The identity ids to assign to the agent
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AgentsCreateResponse
-            201
+        None
 
         Examples
         --------
@@ -92,17 +52,7 @@ class AgentsClient:
             template_version="template_version",
         )
         """
-        _response = self._raw_client.create(
-            project,
-            template_version,
-            tags=tags,
-            agent_name=agent_name,
-            initial_message_sequence=initial_message_sequence,
-            memory_variables=memory_variables,
-            tool_variables=tool_variables,
-            identity_ids=identity_ids,
-            request_options=request_options,
-        )
+        _response = self._raw_client.create(project, template_version, request_options=request_options)
         return _response.data
 
 
@@ -122,56 +72,21 @@ class AsyncAgentsClient:
         return self._raw_client
 
     async def create(
-        self,
-        project: str,
-        template_version: str,
-        *,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        agent_name: typing.Optional[str] = OMIT,
-        initial_message_sequence: typing.Optional[
-            typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]
-        ] = OMIT,
-        memory_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        tool_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        identity_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> AgentsCreateResponse:
+        self, project: str, template_version: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
-        Creates an Agent or multiple Agents from a template
-
         Parameters
         ----------
         project : str
-            The project slug
 
         template_version : str
-            The template version, formatted as {template-name}:{version-number} or {template-name}:latest
-
-        tags : typing.Optional[typing.Sequence[str]]
-            The tags to assign to the agent
-
-        agent_name : typing.Optional[str]
-            The name of the agent, optional otherwise a random one will be assigned
-
-        initial_message_sequence : typing.Optional[typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]]
-            Set an initial sequence of messages, if not provided, the agent will start with the default message sequence, if an empty array is provided, the agent will start with no messages
-
-        memory_variables : typing.Optional[typing.Dict[str, str]]
-            The memory variables to assign to the agent
-
-        tool_variables : typing.Optional[typing.Dict[str, str]]
-            The tool variables to assign to the agent
-
-        identity_ids : typing.Optional[typing.Sequence[str]]
-            The identity ids to assign to the agent
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AgentsCreateResponse
-            201
+        None
 
         Examples
         --------
@@ -194,15 +109,5 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.create(
-            project,
-            template_version,
-            tags=tags,
-            agent_name=agent_name,
-            initial_message_sequence=initial_message_sequence,
-            memory_variables=memory_variables,
-            tool_variables=tool_variables,
-            identity_ids=identity_ids,
-            request_options=request_options,
-        )
+        _response = await self._raw_client.create(project, template_version, request_options=request_options)
         return _response.data

letta_client/templates/agents/raw_client.py

@@ -8,15 +8,6 @@ from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.http_response import AsyncHttpResponse, HttpResponse
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.request_options import RequestOptions
-from ...core.serialization import convert_and_respect_annotation_metadata
-from ...core.unchecked_base_model import construct_type
-from ...errors.payment_required_error import PaymentRequiredError
-from ...types.payment_required_error_body import PaymentRequiredErrorBody
-from .types.agents_create_request_initial_message_sequence_item import AgentsCreateRequestInitialMessageSequenceItem
-from .types.agents_create_response import AgentsCreateResponse
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
 
 
 class RawAgentsClient:
@@ -24,99 +15,30 @@ class RawAgentsClient:
         self._client_wrapper = client_wrapper
 
     def create(
-        self,
-        project: str,
-        template_version: str,
-        *,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        agent_name: typing.Optional[str] = OMIT,
-        initial_message_sequence: typing.Optional[
-            typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]
-        ] = OMIT,
-        memory_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        tool_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        identity_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[AgentsCreateResponse]:
+        self, project: str, template_version: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[None]:
         """
-        Creates an Agent or multiple Agents from a template
-
         Parameters
         ----------
         project : str
-            The project slug
 
         template_version : str
-            The template version, formatted as {template-name}:{version-number} or {template-name}:latest
-
-        tags : typing.Optional[typing.Sequence[str]]
-            The tags to assign to the agent
-
-        agent_name : typing.Optional[str]
-            The name of the agent, optional otherwise a random one will be assigned
-
-        initial_message_sequence : typing.Optional[typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]]
-            Set an initial sequence of messages, if not provided, the agent will start with the default message sequence, if an empty array is provided, the agent will start with no messages
-
-        memory_variables : typing.Optional[typing.Dict[str, str]]
-            The memory variables to assign to the agent
-
-        tool_variables : typing.Optional[typing.Dict[str, str]]
-            The tool variables to assign to the agent
-
-        identity_ids : typing.Optional[typing.Sequence[str]]
-            The identity ids to assign to the agent
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        HttpResponse[AgentsCreateResponse]
-            201
+        HttpResponse[None]
         """
         _response = self._client_wrapper.httpx_client.request(
             f"v1/templates/{jsonable_encoder(project)}/{jsonable_encoder(template_version)}/agents",
             method="POST",
-            json={
-                "tags": tags,
-                "agent_name": agent_name,
-                "initial_message_sequence": convert_and_respect_annotation_metadata(
-                    object_=initial_message_sequence,
-                    annotation=typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem],
-                    direction="write",
-                ),
-                "memory_variables": memory_variables,
-                "tool_variables": tool_variables,
-                "identity_ids": identity_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    AgentsCreateResponse,
-                    construct_type(
-                        type_=AgentsCreateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return HttpResponse(response=_response, data=_data)
-            if _response.status_code == 402:
-                raise PaymentRequiredError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        PaymentRequiredErrorBody,
-                        construct_type(
-                            type_=PaymentRequiredErrorBody,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
+                return HttpResponse(response=_response, data=None)
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
@@ -128,99 +50,30 @@ class AsyncRawAgentsClient:
         self._client_wrapper = client_wrapper
 
     async def create(
-        self,
-        project: str,
-        template_version: str,
-        *,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        agent_name: typing.Optional[str] = OMIT,
-        initial_message_sequence: typing.Optional[
-            typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]
-        ] = OMIT,
-        memory_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        tool_variables: typing.Optional[typing.Dict[str, str]] = OMIT,
-        identity_ids: typing.Optional[typing.Sequence[str]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[AgentsCreateResponse]:
+        self, project: str, template_version: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[None]:
         """
-        Creates an Agent or multiple Agents from a template
-
         Parameters
         ----------
         project : str
-            The project slug
 
         template_version : str
-            The template version, formatted as {template-name}:{version-number} or {template-name}:latest
-
-        tags : typing.Optional[typing.Sequence[str]]
-            The tags to assign to the agent
-
-        agent_name : typing.Optional[str]
-            The name of the agent, optional otherwise a random one will be assigned
-
-        initial_message_sequence : typing.Optional[typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem]]
-            Set an initial sequence of messages, if not provided, the agent will start with the default message sequence, if an empty array is provided, the agent will start with no messages
-
-        memory_variables : typing.Optional[typing.Dict[str, str]]
-            The memory variables to assign to the agent
-
-        tool_variables : typing.Optional[typing.Dict[str, str]]
-            The tool variables to assign to the agent
-
-        identity_ids : typing.Optional[typing.Sequence[str]]
-            The identity ids to assign to the agent
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[AgentsCreateResponse]
-            201
+        AsyncHttpResponse[None]
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"v1/templates/{jsonable_encoder(project)}/{jsonable_encoder(template_version)}/agents",
             method="POST",
-            json={
-                "tags": tags,
-                "agent_name": agent_name,
-                "initial_message_sequence": convert_and_respect_annotation_metadata(
-                    object_=initial_message_sequence,
-                    annotation=typing.Sequence[AgentsCreateRequestInitialMessageSequenceItem],
-                    direction="write",
-                ),
-                "memory_variables": memory_variables,
-                "tool_variables": tool_variables,
-                "identity_ids": identity_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    AgentsCreateResponse,
-                    construct_type(
-                        type_=AgentsCreateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return AsyncHttpResponse(response=_response, data=_data)
-            if _response.status_code == 402:
-                raise PaymentRequiredError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        PaymentRequiredErrorBody,
-                        construct_type(
-                            type_=PaymentRequiredErrorBody,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
+                return AsyncHttpResponse(response=_response, data=None)
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)