c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a6__py3-none-any.whl → 0.1.0a8__py3-none-any.whl

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python
-Version: 0.1.0a6
+Version: 0.1.0a8
 Summary: The official Python library for GradientAI
 Project-URL: Homepage, https://github.com/digitalocean/gradientai-python
 Project-URL: Repository, https://github.com/digitalocean/gradientai-python
@@ -65,7 +65,7 @@ client = GradientAI(
     api_key=os.environ.get("GRADIENTAI_API_KEY"),  # This is the default and can be omitted
 )
 
-completion = client.chat.completions.create(
+completion = client.agents.chat.completions.create(
     messages=[
         {
             "content": "string",
@@ -97,7 +97,7 @@ client = AsyncGradientAI(
 
 
 async def main() -> None:
-    completion = await client.chat.completions.create(
+    completion = await client.agents.chat.completions.create(
         messages=[
             {
                 "content": "string",
@@ -139,7 +139,7 @@ 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.chat.completions.create(
+        completion = await client.agents.chat.completions.create(
             messages=[
                 {
                     "content": "string",
@@ -172,7 +172,7 @@ from gradientai import GradientAI
 
 client = GradientAI()
 
-completion = client.chat.completions.create(
+completion = client.agents.chat.completions.create(
     messages=[
         {
             "content": "string",

{c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8.dist-info}/RECORD

@@ -1,6 +1,6 @@
 gradientai/__init__.py,sha256=yqy3yZnX0JVUS-H01MAvroChzqS67Qf072OrPfNckjI,2655
 gradientai/_base_client.py,sha256=TADFnPHK7WpsNuJUY76SxMFf2IVoOdz_tlloQEXkutk,66719
-gradientai/_client.py,sha256=fVSjsof_G2ocWJyWlghrJYbpHg_z8gbJDOISRrsAvUY,26908
+gradientai/_client.py,sha256=SwygmkQnjfqCEKgMA7rovhfWiEOk3BjyenunyiDS0F8,27425
 gradientai/_compat.py,sha256=VWemUKbj6DDkQ-O4baSpHVLJafotzeXmCQGJugfVTIw,6580
 gradientai/_constants.py,sha256=S14PFzyN9-I31wiV7SmIlL5Ga0MLHxdvegInGdXH7tM,462
 gradientai/_exceptions.py,sha256=o1GvaW36c7_LMj5WasVKUBOpae8tzETBJsfbVphb3Vk,3228
@@ -9,9 +9,9 @@ gradientai/_models.py,sha256=G1vczEodX0vUySeVKbF-mbzlaObNL1oVAYH4c65agRk,29131
 gradientai/_qs.py,sha256=AOkSz4rHtK4YI3ZU_kzea-zpwBUgEY8WniGmTPyEimc,4846
 gradientai/_resource.py,sha256=4NZbH2h8dQ-t-DQPida4VANJ_oZJNA7qxV84mwJT8oM,1124
 gradientai/_response.py,sha256=RhlDdupxTcKNyDDj045MZD3-a_lsEc3yjiOzxWg0cDc,28842
-gradientai/_streaming.py,sha256=AWqY4cmmmTplZperXnkkMkeQ11gmpqYbt6TIXByqCv8,10116
+gradientai/_streaming.py,sha256=3KH-GBmqhoS1KAOhecADOsbW9WuzhIi8wSdmrEj5PPA,11404
 gradientai/_types.py,sha256=22gBoIuoGJ1R6l5nPwquWCRzJodKhO-3e7k22-h37JQ,6201
-gradientai/_version.py,sha256=6DEgmFh2VPj98nJ2tVgNYDCU5i68Lf33js_QO9Ry45c,170
+gradientai/_version.py,sha256=vl2rc2vAkezojVTxFct2GJxzipzCGzHOxgS9Ld5ASmM,170
 gradientai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gradientai/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
 gradientai/_utils/_logs.py,sha256=1QuZcxzSKHcqNFbPoz_pvfLD5eDfOMBzGMyanAm_2gw,787
@@ -25,10 +25,10 @@ gradientai/_utils/_typing.py,sha256=D0DbbNu8GnYQTSICnTSHDGsYXj8TcAKyhejb0XcnjtY,
 gradientai/_utils/_utils.py,sha256=ts4CiiuNpFiGB6YMdkQRh2SZvYvsl7mAF-JWHCcLDf4,12312
 gradientai/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
 gradientai/resources/__init__.py,sha256=Sej63-wOANl_4F78LH6m2Zx2Qu_-wGPHirIiq_20gwk,3488
-gradientai/resources/models.py,sha256=QgmB03fBAdteKRmdTvwLnBcWYzvvaWnU6wtv8ZTE4vw,8597
+gradientai/resources/models.py,sha256=SKdBMKr6NP1ldD_xzbr-gUL8hdo33omUAz5UPAJ0KWQ,9434
 gradientai/resources/regions.py,sha256=c4KX_qFTu83Svil1CO8CP4XdkEIfNQD3zwU_E6w734A,7172
-gradientai/resources/agents/__init__.py,sha256=Rysev19rr3O3IVlHCvKUac0VwMV2xcByK5jpXPiFaVk,5363
-gradientai/resources/agents/agents.py,sha256=be3ucbfRa3gVya1iP8xzzUUMs77Y-UiUYSb8WuN3BFE,45161
+gradientai/resources/agents/__init__.py,sha256=2LkcBGQQJzBhZhVaeAy_IiqAPu28nV-d3_S-b__lTmk,5787
+gradientai/resources/agents/agents.py,sha256=8hp-HMwlAH9VO-TBbTQMUpkkhRDFYzpc47VN9URJY_0,46177
 gradientai/resources/agents/api_keys.py,sha256=1Uh77vhUD0WYUkvX0RD4VcLCOWFG6vL6Lz1EQRDx-t4,24632
 gradientai/resources/agents/evaluation_datasets.py,sha256=mhs9lTIIZHfUFgN93jBdJYQZ0iGHQwN_EQlTFk2IiDU,12594
 gradientai/resources/agents/evaluation_runs.py,sha256=LDaEWVWUjrPcLMqQ6TQldzoc1a1ydz2rzpAcP1KNpxI,19483
@@ -37,6 +37,9 @@ gradientai/resources/agents/functions.py,sha256=ZAOelsCnOl8lRuF5RpZDVGMN-E0yTG4V
 gradientai/resources/agents/knowledge_bases.py,sha256=rFIrqsT7nXFJ_Sy4OWpUc8QmBGKSN_PhYpFpWvYHTfg,15191
 gradientai/resources/agents/routes.py,sha256=pDoK5hUgluhz5awJ9FR_krF8DIbkRzLX__MTYYQa2bA,21957
 gradientai/resources/agents/versions.py,sha256=hmHDaawFkpqoUEYI4Vi5jckH3mamkPKupXO3cZccyKE,11546
+gradientai/resources/agents/chat/__init__.py,sha256=BVAfz9TM3DT5W9f_mt0P9YRxL_MsUxKCWAH6u1iogmA,1041
+gradientai/resources/agents/chat/chat.py,sha256=nt97777qa-xM71JQBKDFG_x7fUDJRvy6rF5LoBiMOKE,3698
+gradientai/resources/agents/chat/completions.py,sha256=jq62v8gN3hJ7POxBCHMcUJUi6Zj4IgvNGlr7D36W1M8,46188
 gradientai/resources/agents/evaluation_metrics/__init__.py,sha256=qUCsT_vI2TrZrUBPe8h-VMNBE4ytcoe0RXshDctV0g0,1198
 gradientai/resources/agents/evaluation_metrics/evaluation_metrics.py,sha256=BueqcWT0lqbElo-lgntkr1s7J0-qi2KqEZV2qzc7dcI,7089
 gradientai/resources/agents/evaluation_metrics/workspaces/__init__.py,sha256=Zf3wa7WSVOKyG1x9Fppny38_ewQCbdghxbptqGFVfOM,1054
@@ -60,7 +63,7 @@ gradientai/resources/model_providers/anthropic/keys.py,sha256=WyRqOn51jfb3Iuu8ml
 gradientai/resources/model_providers/openai/__init__.py,sha256=77LFIe5ko8imMJR7pkAn3E9r4rzw7j2fF9ZSpBaLZyo,976
 gradientai/resources/model_providers/openai/keys.py,sha256=HAwxE3WVrBLPZnn799iHL9LRzMWTCdldZ5-PNeWhQrs,26388
 gradientai/resources/model_providers/openai/openai.py,sha256=MfV9fB4EJVa6RYIfVS2QJW_BSkShLKCscDuxDV8wCBE,3567
-gradientai/types/__init__.py,sha256=95EVUFJ_xrIP7K7ewBh2MeDPRrmhwGedv0rQ2GkTkv4,2784
+gradientai/types/__init__.py,sha256=U5bsYUOgm6miO6k6D6oMl_dbFfzP5UsJT_L8QSyj7Bc,2918
 gradientai/types/agent_create_params.py,sha256=nvLpM8V64762cwAn7ggyvbSzQ0mgjwOopOUZ_bVL1-I,912
 gradientai/types/agent_create_response.py,sha256=frBMp76xLUzznVzWP70fb49drn3r3-2l4kX1xpQrNmg,332
 gradientai/types/agent_delete_response.py,sha256=_1ciIOuRDvbeuaBvbCOpLV7QF6TufMX5MXNsBZv6H4s,332
@@ -78,6 +81,7 @@ gradientai/types/api_agreement.py,sha256=hMeoGBF2HySVHDm4dFTyamq19BexwhezL6Y1e_R
 gradientai/types/api_anthropic_api_key_info.py,sha256=Ii_1zvwZselJjf9XwUGDFWq-fT7VlgC_l-cfXQfpc-s,489
 gradientai/types/api_deployment_visibility.py,sha256=WT2fj4lgBorEOB9EirE50Xn0-J56a5v1zXuNpKCrzf0,340
 gradientai/types/api_knowledge_base.py,sha256=LoZAFXubzOBaV3edIYCr8-N7cwwf4BEyzKudw1hQjZQ,839
+gradientai/types/api_model.py,sha256=cTBrApG2za1_mR6Cca9vL254XczcxZM0lVMfXEqhFnE,718
 gradientai/types/api_model_version.py,sha256=heTuejUUfP0GVFThpYEhwPqBc331hrMBGF_8Uq2qcME,313
 gradientai/types/api_openai_api_key_info.py,sha256=_vfDX_fAIwHB1BXqNk7dUeA7meSU5GVTLbdDblSK5rY,582
 gradientai/types/api_retrieval_method.py,sha256=hbD5272cHgAqDCwPv7tqIUlqNnDAOIjkwDUTciglitE,377
@@ -90,8 +94,8 @@ gradientai/types/knowledge_base_list_response.py,sha256=z8mUZAd328EEvs_3YnHZVcbT
 gradientai/types/knowledge_base_retrieve_response.py,sha256=zwsKaL1KEfgFbglBriT4XorDz6PUwmvgkIrnD0mzWpI,758
 gradientai/types/knowledge_base_update_params.py,sha256=8TRWSRY6c1tRTKPwBNkjwdw_1FbJ1Jmy7vX3covlj64,681
 gradientai/types/knowledge_base_update_response.py,sha256=hNxnfwd-eyRIAuy7KxcQfG-dEK3TZITMYmAv7VH7AyE,342
-gradientai/types/model.py,sha256=DMw8KwQx8B6S6sAI038D0xdzkmYdY5-r0oMhCUG4l6w,532
-gradientai/types/model_list_response.py,sha256=fUKK6I7am-Sx4cnC7nC3mHEhOKZXbkTRHLzxiApK3pA,329
+gradientai/types/model_list_params.py,sha256=RGl48iwa2txemByFNe1EypUyE3CjPgsyr_bOu2mKesY,1335
+gradientai/types/model_list_response.py,sha256=TK6-R0XAWZH2LPrgJMVsjaFO317n5TfdSeBXOAc5B1U,451
 gradientai/types/region_list_params.py,sha256=_oeMCCqhoSpB4rb3BvQw97w-Zo6eblgye3iuIoTjMHs,418
 gradientai/types/region_list_response.py,sha256=JyPfcqxMTafktlkuq5S2Vog77rMQ1RJ6_3cCa-gi0yA,515
 gradientai/types/agents/__init__.py,sha256=h0Uk85a9xEa8HmelfaLVLHmB82wRmIZxj4eVXO4GR28,5117
@@ -146,6 +150,10 @@ gradientai/types/agents/version_list_params.py,sha256=0_3DhUbFDRyjUnn2G4saFOvuot
 gradientai/types/agents/version_list_response.py,sha256=Y0Y8CSPUPVHKRA3zTRRfQ8gC2aFyBacSCaGz-D5k8nk,2725
 gradientai/types/agents/version_update_params.py,sha256=j9tOda5wXmSOHsmcxQONo2mM-hEtrYi5-19HfGU_XnI,379
 gradientai/types/agents/version_update_response.py,sha256=nspPIkxQskT82tcW0JyG7bBVlXq_KU6CZzodTd9jfkQ,709
+gradientai/types/agents/chat/__init__.py,sha256=c-PmEwuvWZQ4CRBTs9gzbKAq2sxL7V7JlVxddeoaGl0,381
+gradientai/types/agents/chat/chat_completion_chunk.py,sha256=1K-F0JdUmQ_4idDk3oUGUB_mhxAxCzjq0C8hMhGtDuY,3048
+gradientai/types/agents/chat/completion_create_params.py,sha256=ADEJ0N3MMsouT9AqBLE1-rho4FVVhlp9U9E-buqIAYs,7165
+gradientai/types/agents/chat/completion_create_response.py,sha256=9uKS3memEoV0_Xd1CZwI0jQGsQyfVkhXRPGlRO3rUIc,2415
 gradientai/types/agents/evaluation_metrics/__init__.py,sha256=XWH_utxMx-JwArRpr-rHQfmoxQRGK6GciKOllbkqg40,894
 gradientai/types/agents/evaluation_metrics/workspace_create_params.py,sha256=LrccBST52BUMlIidiba8K_7vU9HLZW8TTQ2E227UF8Y,343
 gradientai/types/agents/evaluation_metrics/workspace_create_response.py,sha256=jCVs2l3APGqA7gTetoepZS0Fkf7XJdK4e9pspgI3zpo,360
@@ -160,10 +168,9 @@ gradientai/types/agents/evaluation_metrics/workspaces/agent_list_params.py,sha25
 gradientai/types/agents/evaluation_metrics/workspaces/agent_list_response.py,sha256=W4O5v1LoWh2hQJTpUm5SfQCYcv6Q9Yz1Id5Pm1sPQNA,503
 gradientai/types/agents/evaluation_metrics/workspaces/agent_move_params.py,sha256=4INiLEvgT9UDqFbrGwp3nuWOzFhwv7sX_YCr1Um1RaQ,422
 gradientai/types/agents/evaluation_metrics/workspaces/agent_move_response.py,sha256=j2uoTmFwnLNU7kGI5LZdPpMUxWmUk-HdQIVVDOwSy4Y,350
-gradientai/types/chat/__init__.py,sha256=zdTzozOxKOf2jRrwHBHU2Rmtz4nOTwQkYQl2ZbLJZyg,403
-gradientai/types/chat/chat_completion_token_logprob.py,sha256=6-ipUFfsXMf5L7FDFi127NaVkDtmEooVgGBF6Ts965A,1769
+gradientai/types/chat/__init__.py,sha256=A5VCUPqJZydjjOqEXC01GXmcDkKM3bq6zuCu9lmi5Es,303
 gradientai/types/chat/completion_create_params.py,sha256=F4Dcrt5aYC_GEWUSN2OA3Zm5ImevZ7tFuTxBH7RyooM,6635
-gradientai/types/chat/completion_create_response.py,sha256=5nCsU9BeXjB5HmkBNemRZHyFwegvKouSBHjzxvb4EFM,2405
+gradientai/types/chat/completion_create_response.py,sha256=nNPWSXZYbyYLjT_ikVvDcjRw3f9eRGHFsUrLKtQHYGI,2413
 gradientai/types/inference/__init__.py,sha256=4Dt7-03NeP9ehdHLkLsZMiL_YLQwZsl92D0mMoDQ5g0,857
 gradientai/types/inference/api_key_create_params.py,sha256=MOy5Bdr1wNBqCvqzyZ0FLfFY2a97q6eXCzgCR1wcLAE,263
 gradientai/types/inference/api_key_create_response.py,sha256=HjTGC3rpPKqYPE-D6Z9jIE8fqrmo2hTTGBrt00cMwIU,335
@@ -221,10 +228,11 @@ gradientai/types/model_providers/openai/key_retrieve_agents_response.py,sha256=I
 gradientai/types/model_providers/openai/key_retrieve_response.py,sha256=96ZitCFjLlTzk8UhB2Q1ekmLYF2fer_9TGZ1kzhT2Ck,339
 gradientai/types/model_providers/openai/key_update_params.py,sha256=Y3soYvQEpYbnofgxQoLdeZFolk-NcUSnlUg8BQ2BliU,398
 gradientai/types/model_providers/openai/key_update_response.py,sha256=918Y7yxffOe0DmFci54fq0-zTO_1Lc6jruSok8WsBX4,335
-gradientai/types/shared/__init__.py,sha256=S9SLB1APW-dDg_jhLB7iJbyycEkZGof0AJi0ejhNGWI,172
+gradientai/types/shared/__init__.py,sha256=YA2_qLkZLySOac1HrqOfCTEz6GeipnjIJh1mK_QYwYU,272
 gradientai/types/shared/api_links.py,sha256=Iq5iQwOkRYuwLcuDLk54dUfrq0f2ZVEOXSpF744gYgA,403
 gradientai/types/shared/api_meta.py,sha256=-KyinzQqM5GSjD7E5xm7A4UALXAvLOyVNR1SYVOUFJM,297
-c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info/METADATA,sha256=2yYN_5FGbHKS8wNSw4eW7fNmY93uA2s0Z-Qo6ZmB1_I,15021
-c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info/licenses/LICENSE,sha256=AzxEF8mEks6hu5V_87CXF8gLdL875WeO8FmQtEZTFok,11341
-c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info/RECORD,,
+gradientai/types/shared/chat_completion_token_logprob.py,sha256=6-ipUFfsXMf5L7FDFi127NaVkDtmEooVgGBF6Ts965A,1769
+c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8.dist-info/METADATA,sha256=xwI-Z-rvG_Z_-L5Xu3x_NW2F7CauiUQmYG8wtunHk-Q,15049
+c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8.dist-info/licenses/LICENSE,sha256=AzxEF8mEks6hu5V_87CXF8gLdL875WeO8FmQtEZTFok,11341
+c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a8.dist-info/RECORD,,

gradientai/_client.py

@@ -57,12 +57,14 @@ class GradientAI(SyncAPIClient):
     # client options
     api_key: str | None
     inference_key: str | None
+    agent_domain: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -96,6 +98,8 @@ class GradientAI(SyncAPIClient):
             inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
         self.inference_key = inference_key
 
+        self.agent_domain = agent_domain
+
         if base_url is None:
             base_url = os.environ.get("GRADIENT_AI_BASE_URL")
         self._base_url_overridden = base_url is not None
@@ -113,6 +117,8 @@ class GradientAI(SyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = Stream
+
     @cached_property
     def agents(self) -> AgentsResource:
         from .resources.agents import AgentsResource
@@ -201,6 +207,7 @@ class GradientAI(SyncAPIClient):
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.Client | None = None,
@@ -236,6 +243,7 @@ class GradientAI(SyncAPIClient):
         client = self.__class__(
             api_key=api_key or self.api_key,
             inference_key=inference_key or self.inference_key,
+            agent_domain=agent_domain or self.agent_domain,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -289,12 +297,14 @@ class AsyncGradientAI(AsyncAPIClient):
     # client options
     api_key: str | None
     inference_key: str | None
+    agent_domain: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -328,6 +338,8 @@ class AsyncGradientAI(AsyncAPIClient):
             inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
         self.inference_key = inference_key
 
+        self.agent_domain = agent_domain
+
         if base_url is None:
             base_url = os.environ.get("GRADIENT_AI_BASE_URL")
         self._base_url_overridden = base_url is not None
@@ -345,6 +357,8 @@ class AsyncGradientAI(AsyncAPIClient):
             _strict_response_validation=_strict_response_validation,
         )
 
+        self._default_stream_cls = AsyncStream
+
     @cached_property
     def agents(self) -> AsyncAgentsResource:
         from .resources.agents import AsyncAgentsResource
@@ -433,6 +447,7 @@ class AsyncGradientAI(AsyncAPIClient):
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.AsyncClient | None = None,
@@ -468,6 +483,7 @@ class AsyncGradientAI(AsyncAPIClient):
         client = self.__class__(
             api_key=api_key or self.api_key,
             inference_key=inference_key or self.inference_key,
+            agent_domain=agent_domain or self.agent_domain,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,

gradientai/_streaming.py

@@ -9,7 +9,8 @@ from typing_extensions import Self, Protocol, TypeGuard, override, get_origin, r
 
 import httpx
 
-from ._utils import extract_type_var_from_base
+from ._utils import is_mapping, extract_type_var_from_base
+from ._exceptions import APIError
 
 if TYPE_CHECKING:
     from ._client import GradientAI, AsyncGradientAI
@@ -55,7 +56,25 @@ class Stream(Generic[_T]):
         iterator = self._iter_events()
 
         for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+            if sse.data.startswith("[DONE]"):
+                break
+
+            data = sse.json()
+            if is_mapping(data) and data.get("error"):
+                message = None
+                error = data.get("error")
+                if is_mapping(error):
+                    message = error.get("message")
+                if not message or not isinstance(message, str):
+                    message = "An error occurred during streaming"
+
+                raise APIError(
+                    message=message,
+                    request=self.response.request,
+                    body=data["error"],
+                )
+
+            yield process_data(data=data, cast_to=cast_to, response=response)
 
         # Ensure the entire stream is consumed
         for _sse in iterator:
@@ -119,7 +138,25 @@ class AsyncStream(Generic[_T]):
         iterator = self._iter_events()
 
         async for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+            if sse.data.startswith("[DONE]"):
+                break
+
+            data = sse.json()
+            if is_mapping(data) and data.get("error"):
+                message = None
+                error = data.get("error")
+                if is_mapping(error):
+                    message = error.get("message")
+                if not message or not isinstance(message, str):
+                    message = "An error occurred during streaming"
+
+                raise APIError(
+                    message=message,
+                    request=self.response.request,
+                    body=data["error"],
+                )
+
+            yield process_data(data=data, cast_to=cast_to, response=response)
 
         # Ensure the entire stream is consumed
         async for _sse in iterator:

gradientai/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "gradientai"
-__version__ = "0.1.0-alpha.6"  # x-release-please-version
+__version__ = "0.1.0-alpha.8"  # x-release-please-version

gradientai/resources/agents/__init__.py

@@ -1,5 +1,13 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from .chat import (
+    ChatResource,
+    AsyncChatResource,
+    ChatResourceWithRawResponse,
+    AsyncChatResourceWithRawResponse,
+    ChatResourceWithStreamingResponse,
+    AsyncChatResourceWithStreamingResponse,
+)
 from .agents import (
     AgentsResource,
     AsyncAgentsResource,
@@ -88,6 +96,12 @@ __all__ = [
     "AsyncAPIKeysResourceWithRawResponse",
     "APIKeysResourceWithStreamingResponse",
     "AsyncAPIKeysResourceWithStreamingResponse",
+    "ChatResource",
+    "AsyncChatResource",
+    "ChatResourceWithRawResponse",
+    "AsyncChatResourceWithRawResponse",
+    "ChatResourceWithStreamingResponse",
+    "AsyncChatResourceWithStreamingResponse",
     "EvaluationMetricsResource",
     "AsyncEvaluationMetricsResource",
     "EvaluationMetricsResourceWithRawResponse",

gradientai/resources/agents/agents.py

@@ -41,6 +41,14 @@ from .versions import (
     AsyncVersionsResourceWithStreamingResponse,
 )
 from ..._compat import cached_property
+from .chat.chat import (
+    ChatResource,
+    AsyncChatResource,
+    ChatResourceWithRawResponse,
+    AsyncChatResourceWithRawResponse,
+    ChatResourceWithStreamingResponse,
+    AsyncChatResourceWithStreamingResponse,
+)
 from .functions import (
     FunctionsResource,
     AsyncFunctionsResource,
@@ -114,6 +122,10 @@ class AgentsResource(SyncAPIResource):
     def api_keys(self) -> APIKeysResource:
         return APIKeysResource(self._client)
 
+    @cached_property
+    def chat(self) -> ChatResource:
+        return ChatResource(self._client)
+
     @cached_property
     def evaluation_metrics(self) -> EvaluationMetricsResource:
         return EvaluationMetricsResource(self._client)
@@ -498,6 +510,10 @@ class AsyncAgentsResource(AsyncAPIResource):
     def api_keys(self) -> AsyncAPIKeysResource:
         return AsyncAPIKeysResource(self._client)
 
+    @cached_property
+    def chat(self) -> AsyncChatResource:
+        return AsyncChatResource(self._client)
+
     @cached_property
     def evaluation_metrics(self) -> AsyncEvaluationMetricsResource:
         return AsyncEvaluationMetricsResource(self._client)
@@ -904,6 +920,10 @@ class AgentsResourceWithRawResponse:
     def api_keys(self) -> APIKeysResourceWithRawResponse:
         return APIKeysResourceWithRawResponse(self._agents.api_keys)
 
+    @cached_property
+    def chat(self) -> ChatResourceWithRawResponse:
+        return ChatResourceWithRawResponse(self._agents.chat)
+
     @cached_property
     def evaluation_metrics(self) -> EvaluationMetricsResourceWithRawResponse:
         return EvaluationMetricsResourceWithRawResponse(self._agents.evaluation_metrics)
@@ -964,6 +984,10 @@ class AsyncAgentsResourceWithRawResponse:
     def api_keys(self) -> AsyncAPIKeysResourceWithRawResponse:
         return AsyncAPIKeysResourceWithRawResponse(self._agents.api_keys)
 
+    @cached_property
+    def chat(self) -> AsyncChatResourceWithRawResponse:
+        return AsyncChatResourceWithRawResponse(self._agents.chat)
+
     @cached_property
     def evaluation_metrics(self) -> AsyncEvaluationMetricsResourceWithRawResponse:
         return AsyncEvaluationMetricsResourceWithRawResponse(self._agents.evaluation_metrics)
@@ -1024,6 +1048,10 @@ class AgentsResourceWithStreamingResponse:
     def api_keys(self) -> APIKeysResourceWithStreamingResponse:
         return APIKeysResourceWithStreamingResponse(self._agents.api_keys)
 
+    @cached_property
+    def chat(self) -> ChatResourceWithStreamingResponse:
+        return ChatResourceWithStreamingResponse(self._agents.chat)
+
     @cached_property
     def evaluation_metrics(self) -> EvaluationMetricsResourceWithStreamingResponse:
         return EvaluationMetricsResourceWithStreamingResponse(self._agents.evaluation_metrics)
@@ -1084,6 +1112,10 @@ class AsyncAgentsResourceWithStreamingResponse:
     def api_keys(self) -> AsyncAPIKeysResourceWithStreamingResponse:
         return AsyncAPIKeysResourceWithStreamingResponse(self._agents.api_keys)
 
+    @cached_property
+    def chat(self) -> AsyncChatResourceWithStreamingResponse:
+        return AsyncChatResourceWithStreamingResponse(self._agents.chat)
+
     @cached_property
     def evaluation_metrics(self) -> AsyncEvaluationMetricsResourceWithStreamingResponse:
         return AsyncEvaluationMetricsResourceWithStreamingResponse(self._agents.evaluation_metrics)

gradientai/resources/agents/chat/__init__.py

@@ -0,0 +1,33 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .chat import (
+    ChatResource,
+    AsyncChatResource,
+    ChatResourceWithRawResponse,
+    AsyncChatResourceWithRawResponse,
+    ChatResourceWithStreamingResponse,
+    AsyncChatResourceWithStreamingResponse,
+)
+from .completions import (
+    CompletionsResource,
+    AsyncCompletionsResource,
+    CompletionsResourceWithRawResponse,
+    AsyncCompletionsResourceWithRawResponse,
+    CompletionsResourceWithStreamingResponse,
+    AsyncCompletionsResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "CompletionsResource",
+    "AsyncCompletionsResource",
+    "CompletionsResourceWithRawResponse",
+    "AsyncCompletionsResourceWithRawResponse",
+    "CompletionsResourceWithStreamingResponse",
+    "AsyncCompletionsResourceWithStreamingResponse",
+    "ChatResource",
+    "AsyncChatResource",
+    "ChatResourceWithRawResponse",
+    "AsyncChatResourceWithRawResponse",
+    "ChatResourceWithStreamingResponse",
+    "AsyncChatResourceWithStreamingResponse",
+]

gradientai/resources/agents/chat/chat.py

@@ -0,0 +1,102 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from ...._compat import cached_property
+from .completions import (
+    CompletionsResource,
+    AsyncCompletionsResource,
+    CompletionsResourceWithRawResponse,
+    AsyncCompletionsResourceWithRawResponse,
+    CompletionsResourceWithStreamingResponse,
+    AsyncCompletionsResourceWithStreamingResponse,
+)
+from ...._resource import SyncAPIResource, AsyncAPIResource
+
+__all__ = ["ChatResource", "AsyncChatResource"]
+
+
+class ChatResource(SyncAPIResource):
+    @cached_property
+    def completions(self) -> CompletionsResource:
+        return CompletionsResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> ChatResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
+        """
+        return ChatResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> ChatResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
+        """
+        return ChatResourceWithStreamingResponse(self)
+
+
+class AsyncChatResource(AsyncAPIResource):
+    @cached_property
+    def completions(self) -> AsyncCompletionsResource:
+        return AsyncCompletionsResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> AsyncChatResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncChatResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncChatResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
+        """
+        return AsyncChatResourceWithStreamingResponse(self)
+
+
+class ChatResourceWithRawResponse:
+    def __init__(self, chat: ChatResource) -> None:
+        self._chat = chat
+
+    @cached_property
+    def completions(self) -> CompletionsResourceWithRawResponse:
+        return CompletionsResourceWithRawResponse(self._chat.completions)
+
+
+class AsyncChatResourceWithRawResponse:
+    def __init__(self, chat: AsyncChatResource) -> None:
+        self._chat = chat
+
+    @cached_property
+    def completions(self) -> AsyncCompletionsResourceWithRawResponse:
+        return AsyncCompletionsResourceWithRawResponse(self._chat.completions)
+
+
+class ChatResourceWithStreamingResponse:
+    def __init__(self, chat: ChatResource) -> None:
+        self._chat = chat
+
+    @cached_property
+    def completions(self) -> CompletionsResourceWithStreamingResponse:
+        return CompletionsResourceWithStreamingResponse(self._chat.completions)
+
+
+class AsyncChatResourceWithStreamingResponse:
+    def __init__(self, chat: AsyncChatResource) -> None:
+        self._chat = chat
+
+    @cached_property
+    def completions(self) -> AsyncCompletionsResourceWithStreamingResponse:
+        return AsyncCompletionsResourceWithStreamingResponse(self._chat.completions)