feat(api): manual updates

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 76
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradientai-e8b3cbc80e18e4f7f277010349f25e1319156704f359911dc464cc21a0d077a6.yml
 openapi_spec_hash: c773d792724f5647ae25a5ae4ccec208
-config_hash: 0bc3af28d4abd9be8bcc81f615bc832d
+config_hash: 651ae9b93d723d383facbf979fd97fee

README.md

@@ -31,16 +31,16 @@ client = GradientAI(
     api_key=os.environ.get("GRADIENTAI_API_KEY"),  # This is the default and can be omitted
 )
 
-completion = client.agents.chat.completions.create(
+completion = client.chat.completions.create(
     messages=[
         {
-            "content": "string",
-            "role": "system",
+            "role": "user",
+            "content": "What is the capital of France?",
         }
     ],
-    model="llama3-8b-instruct",
+    model="llama3.3-70b-instruct",
 )
-print(completion.id)
+print(completion.choices)
 ```
 
 While you can provide an `api_key` keyword argument,
@@ -63,16 +63,16 @@ client = AsyncGradientAI(
 
 
 async def main() -> None:
-    completion = await client.agents.chat.completions.create(
+    completion = await client.chat.completions.create(
         messages=[
             {
-                "content": "string",
-                "role": "system",
+                "role": "user",
+                "content": "What is the capital of France?",
             }
         ],
-        model="llama3-8b-instruct",
+        model="llama3.3-70b-instruct",
     )
-    print(completion.id)
+    print(completion.choices)
 
 
 asyncio.run(main())
@@ -105,16 +105,16 @@ async def main() -> None:
         api_key=os.environ.get("GRADIENTAI_API_KEY"),  # This is the default and can be omitted
         http_client=DefaultAioHttpClient(),
     ) as client:
-        completion = await client.agents.chat.completions.create(
+        completion = await client.chat.completions.create(
             messages=[
                 {
-                    "content": "string",
-                    "role": "system",
+                    "role": "user",
+                    "content": "What is the capital of France?",
                 }
             ],
-            model="llama3-8b-instruct",
+            model="llama3.3-70b-instruct",
         )
-        print(completion.id)
+        print(completion.choices)
 
 
 asyncio.run(main())
@@ -138,7 +138,7 @@ from gradientai import GradientAI
 
 client = GradientAI()
 
-completion = client.agents.chat.completions.create(
+completion = client.chat.completions.create(
     messages=[
         {
             "content": "string",

api.md

@@ -4,6 +4,20 @@
 from gradientai.types import APILinks, APIMeta, ChatCompletionTokenLogprob
 ```
 
+# Chat
+
+## Completions
+
+Types:
+
+```python
+from gradientai.types.chat import ChatCompletionChunk, CompletionCreateResponse
+```
+
+Methods:
+
+- <code title="post /chat/completions">client.chat.completions.<a href="./src/gradientai/resources/chat/completions.py">create</a>(\*\*<a href="src/gradientai/types/chat/completion_create_params.py">params</a>) -> <a href="./src/gradientai/types/chat/completion_create_response.py">CompletionCreateResponse</a></code>
+
 # Agents
 
 Types:
@@ -389,20 +403,6 @@ Methods:
 - <code title="get /v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources">client.knowledge_bases.indexing_jobs.<a href="./src/gradientai/resources/knowledge_bases/indexing_jobs.py">retrieve_data_sources</a>(indexing_job_uuid) -> <a href="./src/gradientai/types/knowledge_bases/indexing_job_retrieve_data_sources_response.py">IndexingJobRetrieveDataSourcesResponse</a></code>
 - <code title="put /v2/gen-ai/indexing_jobs/{uuid}/cancel">client.knowledge_bases.indexing_jobs.<a href="./src/gradientai/resources/knowledge_bases/indexing_jobs.py">update_cancel</a>(path_uuid, \*\*<a href="src/gradientai/types/knowledge_bases/indexing_job_update_cancel_params.py">params</a>) -> <a href="./src/gradientai/types/knowledge_bases/indexing_job_update_cancel_response.py">IndexingJobUpdateCancelResponse</a></code>
 
-# Chat
-
-## Completions
-
-Types:
-
-```python
-from gradientai.types.chat import ChatCompletionChunk, CompletionCreateResponse
-```
-
-Methods:
-
-- <code title="post /chat/completions">client.chat.completions.<a href="./src/gradientai/resources/chat/completions.py">create</a>(\*\*<a href="src/gradientai/types/chat/completion_create_params.py">params</a>) -> <a href="./src/gradientai/types/chat/completion_create_response.py">CompletionCreateResponse</a></code>
-
 # Inference
 
 ## APIKeys

src/gradientai/_client.py

@@ -119,6 +119,12 @@ def __init__(
 
         self._default_stream_cls = Stream
 
+    @cached_property
+    def chat(self) -> ChatResource:
+        from .resources.chat import ChatResource
+
+        return ChatResource(self)
+
     @cached_property
     def agents(self) -> AgentsResource:
         from .resources.agents import AgentsResource
@@ -143,12 +149,6 @@ def knowledge_bases(self) -> KnowledgeBasesResource:
 
         return KnowledgeBasesResource(self)
 
-    @cached_property
-    def chat(self) -> ChatResource:
-        from .resources.chat import ChatResource
-
-        return ChatResource(self)
-
     @cached_property
     def inference(self) -> InferenceResource:
         from .resources.inference import InferenceResource
@@ -359,6 +359,12 @@ def __init__(
 
         self._default_stream_cls = AsyncStream
 
+    @cached_property
+    def chat(self) -> AsyncChatResource:
+        from .resources.chat import AsyncChatResource
+
+        return AsyncChatResource(self)
+
     @cached_property
     def agents(self) -> AsyncAgentsResource:
         from .resources.agents import AsyncAgentsResource
@@ -383,12 +389,6 @@ def knowledge_bases(self) -> AsyncKnowledgeBasesResource:
 
         return AsyncKnowledgeBasesResource(self)
 
-    @cached_property
-    def chat(self) -> AsyncChatResource:
-        from .resources.chat import AsyncChatResource
-
-        return AsyncChatResource(self)
-
     @cached_property
     def inference(self) -> AsyncInferenceResource:
         from .resources.inference import AsyncInferenceResource
@@ -539,6 +539,12 @@ class GradientAIWithRawResponse:
     def __init__(self, client: GradientAI) -> None:
         self._client = client
 
+    @cached_property
+    def chat(self) -> chat.ChatResourceWithRawResponse:
+        from .resources.chat import ChatResourceWithRawResponse
+
+        return ChatResourceWithRawResponse(self._client.chat)
+
     @cached_property
     def agents(self) -> agents.AgentsResourceWithRawResponse:
         from .resources.agents import AgentsResourceWithRawResponse
@@ -563,12 +569,6 @@ def knowledge_bases(self) -> knowledge_bases.KnowledgeBasesResourceWithRawRespon
 
         return KnowledgeBasesResourceWithRawResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def chat(self) -> chat.ChatResourceWithRawResponse:
-        from .resources.chat import ChatResourceWithRawResponse
-
-        return ChatResourceWithRawResponse(self._client.chat)
-
     @cached_property
     def inference(self) -> inference.InferenceResourceWithRawResponse:
         from .resources.inference import InferenceResourceWithRawResponse
@@ -588,6 +588,12 @@ class AsyncGradientAIWithRawResponse:
     def __init__(self, client: AsyncGradientAI) -> None:
         self._client = client
 
+    @cached_property
+    def chat(self) -> chat.AsyncChatResourceWithRawResponse:
+        from .resources.chat import AsyncChatResourceWithRawResponse
+
+        return AsyncChatResourceWithRawResponse(self._client.chat)
+
     @cached_property
     def agents(self) -> agents.AsyncAgentsResourceWithRawResponse:
         from .resources.agents import AsyncAgentsResourceWithRawResponse
@@ -612,12 +618,6 @@ def knowledge_bases(self) -> knowledge_bases.AsyncKnowledgeBasesResourceWithRawR
 
         return AsyncKnowledgeBasesResourceWithRawResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def chat(self) -> chat.AsyncChatResourceWithRawResponse:
-        from .resources.chat import AsyncChatResourceWithRawResponse
-
-        return AsyncChatResourceWithRawResponse(self._client.chat)
-
     @cached_property
     def inference(self) -> inference.AsyncInferenceResourceWithRawResponse:
         from .resources.inference import AsyncInferenceResourceWithRawResponse
@@ -637,6 +637,12 @@ class GradientAIWithStreamedResponse:
     def __init__(self, client: GradientAI) -> None:
         self._client = client
 
+    @cached_property
+    def chat(self) -> chat.ChatResourceWithStreamingResponse:
+        from .resources.chat import ChatResourceWithStreamingResponse
+
+        return ChatResourceWithStreamingResponse(self._client.chat)
+
     @cached_property
     def agents(self) -> agents.AgentsResourceWithStreamingResponse:
         from .resources.agents import AgentsResourceWithStreamingResponse
@@ -661,12 +667,6 @@ def knowledge_bases(self) -> knowledge_bases.KnowledgeBasesResourceWithStreaming
 
         return KnowledgeBasesResourceWithStreamingResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def chat(self) -> chat.ChatResourceWithStreamingResponse:
-        from .resources.chat import ChatResourceWithStreamingResponse
-
-        return ChatResourceWithStreamingResponse(self._client.chat)
-
     @cached_property
     def inference(self) -> inference.InferenceResourceWithStreamingResponse:
         from .resources.inference import InferenceResourceWithStreamingResponse
@@ -686,6 +686,12 @@ class AsyncGradientAIWithStreamedResponse:
     def __init__(self, client: AsyncGradientAI) -> None:
         self._client = client
 
+    @cached_property
+    def chat(self) -> chat.AsyncChatResourceWithStreamingResponse:
+        from .resources.chat import AsyncChatResourceWithStreamingResponse
+
+        return AsyncChatResourceWithStreamingResponse(self._client.chat)
+
     @cached_property
     def agents(self) -> agents.AsyncAgentsResourceWithStreamingResponse:
         from .resources.agents import AsyncAgentsResourceWithStreamingResponse
@@ -710,12 +716,6 @@ def knowledge_bases(self) -> knowledge_bases.AsyncKnowledgeBasesResourceWithStre
 
         return AsyncKnowledgeBasesResourceWithStreamingResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def chat(self) -> chat.AsyncChatResourceWithStreamingResponse:
-        from .resources.chat import AsyncChatResourceWithStreamingResponse
-
-        return AsyncChatResourceWithStreamingResponse(self._client.chat)
-
     @cached_property
     def inference(self) -> inference.AsyncInferenceResourceWithStreamingResponse:
         from .resources.inference import AsyncInferenceResourceWithStreamingResponse

src/gradientai/resources/__init__.py

@@ -58,6 +58,12 @@
 )
 
 __all__ = [
+    "ChatResource",
+    "AsyncChatResource",
+    "ChatResourceWithRawResponse",
+    "AsyncChatResourceWithRawResponse",
+    "ChatResourceWithStreamingResponse",
+    "AsyncChatResourceWithStreamingResponse",
     "AgentsResource",
     "AsyncAgentsResource",
     "AgentsResourceWithRawResponse",
@@ -82,12 +88,6 @@
     "AsyncKnowledgeBasesResourceWithRawResponse",
     "KnowledgeBasesResourceWithStreamingResponse",
     "AsyncKnowledgeBasesResourceWithStreamingResponse",
-    "ChatResource",
-    "AsyncChatResource",
-    "ChatResourceWithRawResponse",
-    "AsyncChatResourceWithRawResponse",
-    "ChatResourceWithStreamingResponse",
-    "AsyncChatResourceWithStreamingResponse",
     "InferenceResource",
     "AsyncInferenceResource",
     "InferenceResourceWithRawResponse",