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

letta_client/jobs/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

letta_client/jobs/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.job import Job
-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.jsonable_encoder import jsonable_encoder
-from ..core.client_wrapper import AsyncClientWrapper
+from .raw_client import AsyncRawJobsClient, RawJobsClient
 
 
 class JobsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawJobsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawJobsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawJobsClient
+        """
+        return self._raw_client
 
     def list(
         self,
@@ -66,41 +72,15 @@ class JobsClient:
         )
         client.jobs.list()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/jobs/",
-            method="GET",
-            params={
-                "source_id": source_id,
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "ascending": ascending,
-            },
+        _response = self._raw_client.list(
+            source_id=source_id,
+            before=before,
+            after=after,
+            limit=limit,
+            ascending=ascending,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Job],
-                    construct_type(
-                        type_=typing.List[Job],  # 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)
+        return _response.data
 
     def list_active(
         self,
@@ -150,41 +130,15 @@ class JobsClient:
         )
         client.jobs.list_active()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/jobs/active",
-            method="GET",
-            params={
-                "source_id": source_id,
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "ascending": ascending,
-            },
+        _response = self._raw_client.list_active(
+            source_id=source_id,
+            before=before,
+            after=after,
+            limit=limit,
+            ascending=ascending,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Job],
-                    construct_type(
-                        type_=typing.List[Job],  # 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)
+        return _response.data
 
     def retrieve(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
         """
@@ -214,34 +168,8 @@ class JobsClient:
             job_id="job_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/jobs/{jsonable_encoder(job_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Job,
-                    construct_type(
-                        type_=Job,  # 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(job_id, request_options=request_options)
+        return _response.data
 
     def delete(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
         """
@@ -271,34 +199,8 @@ class JobsClient:
             job_id="job_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/jobs/{jsonable_encoder(job_id)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Job,
-                    construct_type(
-                        type_=Job,  # 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.delete(job_id, request_options=request_options)
+        return _response.data
 
     def cancel_job(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
         """
@@ -331,39 +233,24 @@ class JobsClient:
             job_id="job_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/jobs/{jsonable_encoder(job_id)}/cancel",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Job,
-                    construct_type(
-                        type_=Job,  # 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.cancel_job(job_id, request_options=request_options)
+        return _response.data
 
 
 class AsyncJobsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawJobsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawJobsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawJobsClient
+        """
+        return self._raw_client
 
     async def list(
         self,
@@ -422,41 +309,15 @@ class AsyncJobsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/jobs/",
-            method="GET",
-            params={
-                "source_id": source_id,
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "ascending": ascending,
-            },
+        _response = await self._raw_client.list(
+            source_id=source_id,
+            before=before,
+            after=after,
+            limit=limit,
+            ascending=ascending,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Job],
-                    construct_type(
-                        type_=typing.List[Job],  # 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)
+        return _response.data
 
     async def list_active(
         self,
@@ -514,41 +375,15 @@ class AsyncJobsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/jobs/active",
-            method="GET",
-            params={
-                "source_id": source_id,
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "ascending": ascending,
-            },
+        _response = await self._raw_client.list_active(
+            source_id=source_id,
+            before=before,
+            after=after,
+            limit=limit,
+            ascending=ascending,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[Job],
-                    construct_type(
-                        type_=typing.List[Job],  # 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)
+        return _response.data
 
     async def retrieve(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
         """
@@ -586,34 +421,8 @@ class AsyncJobsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/jobs/{jsonable_encoder(job_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Job,
-                    construct_type(
-                        type_=Job,  # 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(job_id, request_options=request_options)
+        return _response.data
 
     async def delete(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
         """
@@ -651,34 +460,8 @@ class AsyncJobsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/jobs/{jsonable_encoder(job_id)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Job,
-                    construct_type(
-                        type_=Job,  # 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.delete(job_id, request_options=request_options)
+        return _response.data
 
     async def cancel_job(self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Job:
         """
@@ -719,31 +502,5 @@ class AsyncJobsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/jobs/{jsonable_encoder(job_id)}/cancel",
-            method="PATCH",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Job,
-                    construct_type(
-                        type_=Job,  # 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.cancel_job(job_id, request_options=request_options)
+        return _response.data