letta-client 0.1.256__tar.gz → 0.1.258__tar.gz

{letta_client-0.1.256 → letta_client-0.1.258}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-client
-Version: 0.1.256
+Version: 0.1.258
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{letta_client-0.1.256 → letta_client-0.1.258}/pyproject.toml

@@ -3,7 +3,7 @@ name = "letta-client"
 
 [tool.poetry]
 name = "letta-client"
-version = "0.1.256"
+version = "0.1.258"
 description = ""
 readme = "README.md"
 authors = []

{letta_client-0.1.256 → letta_client-0.1.258}/src/letta_client/__init__.py

@@ -234,6 +234,7 @@ from .types import (
     StdioServerConfig,
     Step,
     StepFeedback,
+    StepMetrics,
     StepStatus,
     StopReasonType,
     StreamableHttpServerConfig,
@@ -641,6 +642,7 @@ __all__ = [
     "StdioServerConfig",
     "Step",
     "StepFeedback",
+    "StepMetrics",
     "StepStatus",
     "StepsListRequestFeedback",
     "StopReasonType",

{letta_client-0.1.256 → letta_client-0.1.258}/src/letta_client/core/client_wrapper.py

@@ -24,10 +24,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "letta-client/0.1.256",
+            "User-Agent": "letta-client/0.1.258",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.256",
+            "X-Fern-SDK-Version": "0.1.258",
             **(self.get_custom_headers() or {}),
         }
         if self._project is not None:

{letta_client-0.1.256 → letta_client-0.1.258}/src/letta_client/steps/client.py

@@ -5,6 +5,7 @@ import typing
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.step import Step
+from ..types.step_metrics import StepMetrics
 from .feedback.client import AsyncFeedbackClient, FeedbackClient
 from .raw_client import AsyncRawStepsClient, RawStepsClient
 from .types.steps_list_request_feedback import StepsListRequestFeedback
@@ -156,6 +157,39 @@ class StepsClient:
         _response = self._raw_client.retrieve(step_id, request_options=request_options)
         return _response.data
 
+    def retrieve_step_metrics(
+        self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> StepMetrics:
+        """
+        Get step metrics by step ID.
+
+        Parameters
+        ----------
+        step_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        StepMetrics
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+        client.steps.retrieve_step_metrics(
+            step_id="step_id",
+        )
+        """
+        _response = self._raw_client.retrieve_step_metrics(step_id, request_options=request_options)
+        return _response.data
+
 
 class AsyncStepsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -318,3 +352,44 @@ class AsyncStepsClient:
         """
         _response = await self._raw_client.retrieve(step_id, request_options=request_options)
         return _response.data
+
+    async def retrieve_step_metrics(
+        self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> StepMetrics:
+        """
+        Get step metrics by step ID.
+
+        Parameters
+        ----------
+        step_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        StepMetrics
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.steps.retrieve_step_metrics(
+                step_id="step_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.retrieve_step_metrics(step_id, request_options=request_options)
+        return _response.data

{letta_client-0.1.256 → letta_client-0.1.258}/src/letta_client/steps/raw_client.py

@@ -12,6 +12,7 @@ from ..core.unchecked_base_model import construct_type
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.http_validation_error import HttpValidationError
 from ..types.step import Step
+from ..types.step_metrics import StepMetrics
 from .types.steps_list_request_feedback import StepsListRequestFeedback
 
 
@@ -183,6 +184,55 @@ class RawStepsClient:
             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)
 
+    def retrieve_step_metrics(
+        self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[StepMetrics]:
+        """
+        Get step metrics by step ID.
+
+        Parameters
+        ----------
+        step_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[StepMetrics]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/steps/{jsonable_encoder(step_id)}/metrics",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    StepMetrics,
+                    construct_type(
+                        type_=StepMetrics,  # 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 AsyncRawStepsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -353,3 +403,52 @@ class AsyncRawStepsClient:
         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)
+
+    async def retrieve_step_metrics(
+        self, step_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[StepMetrics]:
+        """
+        Get step metrics by step ID.
+
+        Parameters
+        ----------
+        step_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[StepMetrics]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/steps/{jsonable_encoder(step_id)}/metrics",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    StepMetrics,
+                    construct_type(
+                        type_=StepMetrics,  # 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-0.1.256 → letta_client-0.1.258}/src/letta_client/types/__init__.py

@@ -245,6 +245,7 @@ from .sse_server_config import SseServerConfig
 from .stdio_server_config import StdioServerConfig
 from .step import Step
 from .step_feedback import StepFeedback
+from .step_metrics import StepMetrics
 from .step_status import StepStatus
 from .stop_reason_type import StopReasonType
 from .streamable_http_server_config import StreamableHttpServerConfig
@@ -530,6 +531,7 @@ __all__ = [
     "StdioServerConfig",
     "Step",
     "StepFeedback",
+    "StepMetrics",
     "StepStatus",
     "StopReasonType",
     "StreamableHttpServerConfig",

{letta_client-0.1.256 → letta_client-0.1.258}/src/letta_client/types/embedding_config_embedding_endpoint_type.py

@@ -7,7 +7,6 @@ EmbeddingConfigEmbeddingEndpointType = typing.Union[
         "openai",
         "anthropic",
         "bedrock",
-        "cohere",
         "google_ai",
         "google_vertex",
         "azure",

{letta_client-0.1.256 → letta_client-0.1.258}/src/letta_client/types/llm_config_model_endpoint_type.py

@@ -6,7 +6,6 @@ LlmConfigModelEndpointType = typing.Union[
     typing.Literal[
         "openai",
         "anthropic",
-        "cohere",
         "google_ai",
         "google_vertex",
         "azure",

{letta_client-0.1.256 → letta_client-0.1.258}/src/letta_client/types/provider_type.py

@@ -20,7 +20,6 @@ ProviderType = typing.Union[
         "azure",
         "vllm",
         "bedrock",
-        "cohere",
     ],
     typing.Any,
 ]

letta_client-0.1.258/src/letta_client/types/step_metrics.py

@@ -0,0 +1,68 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.unchecked_base_model import UncheckedBaseModel
+
+
+class StepMetrics(UncheckedBaseModel):
+    id: str = pydantic.Field()
+    """
+    The id of the step this metric belongs to (matches steps.id).
+    """
+
+    provider_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The unique identifier of the provider.
+    """
+
+    job_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The unique identifier of the job.
+    """
+
+    agent_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The unique identifier of the agent.
+    """
+
+    llm_request_ns: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Time spent on LLM requests in nanoseconds.
+    """
+
+    tool_execution_ns: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Time spent on tool execution in nanoseconds.
+    """
+
+    step_ns: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Total time for the step in nanoseconds.
+    """
+
+    base_template_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The base template ID that the step belongs to (cloud only).
+    """
+
+    template_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The template ID that the step belongs to (cloud only).
+    """
+
+    project_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The project that the step belongs to (cloud only).
+    """
+
+    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