Add Responses API resource and client.responses (with tests)

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: rushitat

src/gradient/_client.py

@@ -42,6 +42,7 @@
         retrieve,
         databases,
         inference,
+        responses,
         gpu_droplets,
         knowledge_bases,
     )
@@ -58,6 +59,10 @@
     from .resources.models.models import ModelsResource, AsyncModelsResource
     from .resources.databases.databases import DatabasesResource, AsyncDatabasesResource
     from .resources.inference.inference import InferenceResource, AsyncInferenceResource
+    from .resources.responses.responses import (
+        ResponsesResource,
+        AsyncResponsesResource,
+    )
     from .resources.knowledge_bases.knowledge_bases import (
         KnowledgeBasesResource,
         AsyncKnowledgeBasesResource,
@@ -195,6 +200,12 @@ def chat(self) -> ChatResource:
 
         return ChatResource(self)
 
+    @cached_property
+    def responses(self) -> ResponsesResource:
+        from .resources.responses import ResponsesResource
+
+        return ResponsesResource(self)
+
     @cached_property
     def images(self) -> ImagesResource:
         from .resources.images import ImagesResource
@@ -353,14 +364,10 @@ def copy(
         Create a new client instance re-using the same options given to the current client with optional overriding.
         """
         if default_headers is not None and set_default_headers is not None:
-            raise ValueError(
-                "The `default_headers` and `set_default_headers` arguments are mutually exclusive"
-            )
+            raise ValueError("The `default_headers` and `set_default_headers` arguments are mutually exclusive")
 
         if default_query is not None and set_default_query is not None:
-            raise ValueError(
-                "The `default_query` and `set_default_query` arguments are mutually exclusive"
-            )
+            raise ValueError("The `default_query` and `set_default_query` arguments are mutually exclusive")
 
         headers = self._custom_headers
         if default_headers is not None:
@@ -411,14 +418,10 @@ def _make_status_error(
             return _exceptions.BadRequestError(err_msg, response=response, body=body)
 
         if response.status_code == 401:
-            return _exceptions.AuthenticationError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.AuthenticationError(err_msg, response=response, body=body)
 
         if response.status_code == 403:
-            return _exceptions.PermissionDeniedError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.PermissionDeniedError(err_msg, response=response, body=body)
 
         if response.status_code == 404:
             return _exceptions.NotFoundError(err_msg, response=response, body=body)
@@ -427,17 +430,13 @@ def _make_status_error(
             return _exceptions.ConflictError(err_msg, response=response, body=body)
 
         if response.status_code == 422:
-            return _exceptions.UnprocessableEntityError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.UnprocessableEntityError(err_msg, response=response, body=body)
 
         if response.status_code == 429:
             return _exceptions.RateLimitError(err_msg, response=response, body=body)
 
         if response.status_code >= 500:
-            return _exceptions.InternalServerError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.InternalServerError(err_msg, response=response, body=body)
         return APIStatusError(err_msg, response=response, body=body)
 
 
@@ -561,6 +560,12 @@ def chat(self) -> AsyncChatResource:
 
         return AsyncChatResource(self)
 
+    @cached_property
+    def responses(self) -> AsyncResponsesResource:
+        from .resources.responses import AsyncResponsesResource
+
+        return AsyncResponsesResource(self)
+
     @cached_property
     def images(self) -> AsyncImagesResource:
         from .resources.images import AsyncImagesResource
@@ -719,14 +724,10 @@ def copy(
         Create a new client instance re-using the same options given to the current client with optional overriding.
         """
         if default_headers is not None and set_default_headers is not None:
-            raise ValueError(
-                "The `default_headers` and `set_default_headers` arguments are mutually exclusive"
-            )
+            raise ValueError("The `default_headers` and `set_default_headers` arguments are mutually exclusive")
 
         if default_query is not None and set_default_query is not None:
-            raise ValueError(
-                "The `default_query` and `set_default_query` arguments are mutually exclusive"
-            )
+            raise ValueError("The `default_query` and `set_default_query` arguments are mutually exclusive")
 
         headers = self._custom_headers
         if default_headers is not None:
@@ -777,14 +778,10 @@ def _make_status_error(
             return _exceptions.BadRequestError(err_msg, response=response, body=body)
 
         if response.status_code == 401:
-            return _exceptions.AuthenticationError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.AuthenticationError(err_msg, response=response, body=body)
 
         if response.status_code == 403:
-            return _exceptions.PermissionDeniedError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.PermissionDeniedError(err_msg, response=response, body=body)
 
         if response.status_code == 404:
             return _exceptions.NotFoundError(err_msg, response=response, body=body)
@@ -793,17 +790,13 @@ def _make_status_error(
             return _exceptions.ConflictError(err_msg, response=response, body=body)
 
         if response.status_code == 422:
-            return _exceptions.UnprocessableEntityError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.UnprocessableEntityError(err_msg, response=response, body=body)
 
         if response.status_code == 429:
             return _exceptions.RateLimitError(err_msg, response=response, body=body)
 
         if response.status_code >= 500:
-            return _exceptions.InternalServerError(
-                err_msg, response=response, body=body
-            )
+            return _exceptions.InternalServerError(err_msg, response=response, body=body)
         return APIStatusError(err_msg, response=response, body=body)
 
 
@@ -825,6 +818,12 @@ def chat(self) -> chat.ChatResourceWithRawResponse:
 
         return ChatResourceWithRawResponse(self._client.chat)
 
+    @cached_property
+    def responses(self) -> responses.ResponsesResourceWithRawResponse:
+        from .resources.responses import ResponsesResourceWithRawResponse
+
+        return ResponsesResourceWithRawResponse(self._client.responses)
+
     @cached_property
     def images(self) -> images.ImagesResourceWithRawResponse:
         from .resources.images import ImagesResourceWithRawResponse
@@ -898,6 +897,12 @@ def chat(self) -> chat.AsyncChatResourceWithRawResponse:
 
         return AsyncChatResourceWithRawResponse(self._client.chat)
 
+    @cached_property
+    def responses(self) -> responses.AsyncResponsesResourceWithRawResponse:
+        from .resources.responses import AsyncResponsesResourceWithRawResponse
+
+        return AsyncResponsesResourceWithRawResponse(self._client.responses)
+
     @cached_property
     def images(self) -> images.AsyncImagesResourceWithRawResponse:
         from .resources.images import AsyncImagesResourceWithRawResponse
@@ -975,6 +980,12 @@ def chat(self) -> chat.ChatResourceWithStreamingResponse:
 
         return ChatResourceWithStreamingResponse(self._client.chat)
 
+    @cached_property
+    def responses(self) -> responses.ResponsesResourceWithStreamingResponse:
+        from .resources.responses import ResponsesResourceWithStreamingResponse
+
+        return ResponsesResourceWithStreamingResponse(self._client.responses)
+
     @cached_property
     def images(self) -> images.ImagesResourceWithStreamingResponse:
         from .resources.images import ImagesResourceWithStreamingResponse
@@ -1052,6 +1063,12 @@ def chat(self) -> chat.AsyncChatResourceWithStreamingResponse:
 
         return AsyncChatResourceWithStreamingResponse(self._client.chat)
 
+    @cached_property
+    def responses(self) -> responses.AsyncResponsesResourceWithStreamingResponse:
+        from .resources.responses import AsyncResponsesResourceWithStreamingResponse
+
+        return AsyncResponsesResourceWithStreamingResponse(self._client.responses)
+
     @cached_property
     def images(self) -> images.AsyncImagesResourceWithStreamingResponse:
         from .resources.images import AsyncImagesResourceWithStreamingResponse
@@ -1082,9 +1099,7 @@ def knowledge_bases(
             AsyncKnowledgeBasesResourceWithStreamingResponse,
         )
 
-        return AsyncKnowledgeBasesResourceWithStreamingResponse(
-            self._client.knowledge_bases
-        )
+        return AsyncKnowledgeBasesResourceWithStreamingResponse(self._client.knowledge_bases)
 
     @cached_property
     def models(self) -> models.AsyncModelsResourceWithStreamingResponse:

src/gradient/resources/__init__.py

@@ -72,6 +72,14 @@
     InferenceResourceWithStreamingResponse,
     AsyncInferenceResourceWithStreamingResponse,
 )
+from .responses import (
+    ResponsesResource,
+    AsyncResponsesResource,
+    ResponsesResourceWithRawResponse,
+    AsyncResponsesResourceWithRawResponse,
+    ResponsesResourceWithStreamingResponse,
+    AsyncResponsesResourceWithStreamingResponse,
+)
 from .gpu_droplets import (
     GPUDropletsResource,
     AsyncGPUDropletsResource,
@@ -150,6 +158,12 @@
     "AsyncNfsResourceWithRawResponse",
     "NfsResourceWithStreamingResponse",
     "AsyncNfsResourceWithStreamingResponse",
+    "ResponsesResource",
+    "AsyncResponsesResource",
+    "ResponsesResourceWithRawResponse",
+    "AsyncResponsesResourceWithRawResponse",
+    "ResponsesResourceWithStreamingResponse",
+    "AsyncResponsesResourceWithStreamingResponse",
     "RetrieveResource",
     "AsyncRetrieveResource",
     "RetrieveResourceWithRawResponse",

src/gradient/resources/responses/__init__.py

@@ -0,0 +1,21 @@
+# Responses API. See docs/RESPONSES_API_PR_BREAKDOWN.md.
+
+from __future__ import annotations
+
+from .responses import (
+    ResponsesResource,
+    AsyncResponsesResource,
+    ResponsesResourceWithRawResponse,
+    AsyncResponsesResourceWithRawResponse,
+    ResponsesResourceWithStreamingResponse,
+    AsyncResponsesResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "AsyncResponsesResource",
+    "AsyncResponsesResourceWithRawResponse",
+    "AsyncResponsesResourceWithStreamingResponse",
+    "ResponsesResource",
+    "ResponsesResourceWithRawResponse",
+    "ResponsesResourceWithStreamingResponse",
+]