letta-client 0.1.233__py3-none-any.whl → 0.1.234__py3-none-any.whl

letta_client/telemetry/client.py

@@ -1,21 +1,27 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.client_wrapper import SyncClientWrapper
 import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.provider_trace import ProviderTrace
-from ..core.jsonable_encoder import jsonable_encoder
-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
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper
+from .raw_client import AsyncRawTelemetryClient, RawTelemetryClient
 
 
 class TelemetryClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawTelemetryClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawTelemetryClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawTelemetryClient
+        """
+        return self._raw_client
 
     def retrieve_provider_trace(
         self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -45,39 +51,24 @@ class TelemetryClient:
             step_id="step_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/telemetry/{jsonable_encoder(step_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProviderTrace,
-                    construct_type(
-                        type_=ProviderTrace,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.retrieve_provider_trace(step_id, request_options=request_options)
+        return _response.data
 
 
 class AsyncTelemetryClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawTelemetryClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawTelemetryClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawTelemetryClient
+        """
+        return self._raw_client
 
     async def retrieve_provider_trace(
         self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -115,31 +106,5 @@ class AsyncTelemetryClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/telemetry/{jsonable_encoder(step_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProviderTrace,
-                    construct_type(
-                        type_=ProviderTrace,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        construct_type(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = await self._raw_client.retrieve_provider_trace(step_id, request_options=request_options)
+        return _response.data

letta_client/telemetry/raw_client.py

@@ -0,0 +1,118 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+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.unchecked_base_model import construct_type
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.http_validation_error import HttpValidationError
+from ..types.provider_trace import ProviderTrace
+
+
+class RawTelemetryClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def retrieve_provider_trace(
+        self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[ProviderTrace]:
+        """
+        Parameters
+        ----------
+        step_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[ProviderTrace]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/telemetry/{jsonable_encoder(step_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ProviderTrace,
+                    construct_type(
+                        type_=ProviderTrace,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        construct_type(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawTelemetryClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def retrieve_provider_trace(
+        self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[ProviderTrace]:
+        """
+        Parameters
+        ----------
+        step_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[ProviderTrace]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/telemetry/{jsonable_encoder(step_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ProviderTrace,
+                    construct_type(
+                        type_=ProviderTrace,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        construct_type(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

letta_client/templates/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .types import TemplatesListResponse, TemplatesListResponseTemplatesItem
 from . import agents
 from .agents import (

letta_client/templates/agents/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .types import (
     AgentsCreateRequestInitialMessageSequenceItem,
     AgentsCreateRequestInitialMessageSequenceItemRole,

letta_client/templates/agents/client.py

@@ -1,18 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ...core.client_wrapper import SyncClientWrapper
-from .types.agents_create_request_initial_message_sequence_item import AgentsCreateRequestInitialMessageSequenceItem
+
+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
-from ...core.jsonable_encoder import jsonable_encoder
-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 json.decoder import JSONDecodeError
-from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -20,7 +14,18 @@ OMIT = typing.cast(typing.Any, ...)
 
 class AgentsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawAgentsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawAgentsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawAgentsClient
+        """
+        return self._raw_client
 
     def create(
         self,
@@ -87,55 +92,34 @@ class AgentsClient:
             template_version="template_version",
         )
         """
-        _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",
-            },
+        _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,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentsCreateResponse,
-                    construct_type(
-                        type_=AgentsCreateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 402:
-                raise PaymentRequiredError(
-                    typing.cast(
-                        PaymentRequiredErrorBody,
-                        construct_type(
-                            type_=PaymentRequiredErrorBody,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
 
 class AsyncAgentsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawAgentsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawAgentsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawAgentsClient
+        """
+        return self._raw_client
 
     async def create(
         self,
@@ -210,47 +194,15 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _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",
-            },
+        _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,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentsCreateResponse,
-                    construct_type(
-                        type_=AgentsCreateResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 402:
-                raise PaymentRequiredError(
-                    typing.cast(
-                        PaymentRequiredErrorBody,
-                        construct_type(
-                            type_=PaymentRequiredErrorBody,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data

letta_client/templates/agents/raw_client.py

@@ -0,0 +1,227 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+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:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        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]:
+        """
+        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
+        """
+        _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(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawAgentsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        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]:
+        """
+        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
+        """
+        _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(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

letta_client/templates/agents/types/__init__.py

@@ -1,5 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .agents_create_request_initial_message_sequence_item import AgentsCreateRequestInitialMessageSequenceItem
 from .agents_create_request_initial_message_sequence_item_role import AgentsCreateRequestInitialMessageSequenceItemRole
 from .agents_create_response import AgentsCreateResponse