Skip to content

Commit 4212f62

Browse files
feat(api): update via SDK Studio
1 parent b16cceb commit 4212f62

28 files changed

+396
-518
lines changed

.stats.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 58
22
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradientai-e40feaac59c85aace6aa42d2749b20e0955dbbae58b06c3a650bc03adafcd7b5.yml
33
openapi_spec_hash: 825c1a4816938e9f594b7a8c06692667
4-
config_hash: 6082607b38b030ffbcb6f681788d1a88
4+
config_hash: ae932f39d93e617d3f513271503efbcf

api.md

Lines changed: 29 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,12 @@ Types:
66
from gradientai.types import (
77
APIAgent,
88
APIAgentAPIKeyInfo,
9+
APIAgentModel,
910
APIAnthropicAPIKeyInfo,
1011
APIDeploymentVisibility,
1112
APIOpenAIAPIKeyInfo,
1213
APIRetrievalMethod,
14+
APIWorkspace,
1315
AgentCreateResponse,
1416
AgentRetrieveResponse,
1517
AgentUpdateResponse,
@@ -174,12 +176,34 @@ Methods:
174176
Types:
175177

176178
```python
177-
from gradientai.types import RegionListResponse
179+
from gradientai.types import APIEvaluationMetric, RegionListResponse
178180
```
179181

180182
Methods:
181183

182-
- <code title="get /v2/gen-ai/regions">client.regions.<a href="./src/gradientai/resources/regions.py">list</a>(\*\*<a href="src/gradientai/types/region_list_params.py">params</a>) -> <a href="./src/gradientai/types/region_list_response.py">RegionListResponse</a></code>
184+
- <code title="get /v2/gen-ai/regions">client.regions.<a href="./src/gradientai/resources/regions/regions.py">list</a>(\*\*<a href="src/gradientai/types/region_list_params.py">params</a>) -> <a href="./src/gradientai/types/region_list_response.py">RegionListResponse</a></code>
185+
186+
## EvaluationRuns
187+
188+
### Results
189+
190+
Types:
191+
192+
```python
193+
from gradientai.types.regions.evaluation_runs import (
194+
APIEvaluationMetricResult,
195+
APIEvaluationRun,
196+
APIPrompt,
197+
)
198+
```
199+
200+
## EvaluationTestCases
201+
202+
Types:
203+
204+
```python
205+
from gradientai.types.regions import APIEvaluationTestCase, APIStarMetric
206+
```
183207

184208
# IndexingJobs
185209

@@ -237,6 +261,7 @@ from gradientai.types.knowledge_bases import (
237261
APIKnowledgeBaseDataSource,
238262
APISpacesDataSource,
239263
APIWebCrawlerDataSource,
264+
AwsDataSource,
240265
DataSourceCreateResponse,
241266
DataSourceListResponse,
242267
DataSourceDeleteResponse,
@@ -256,7 +281,7 @@ Methods:
256281
Types:
257282

258283
```python
259-
from gradientai.types.chat import CompletionCreateResponse
284+
from gradientai.types.chat import ChatCompletionTokenLogprob, CompletionCreateResponse
260285
```
261286

262287
Methods:
@@ -306,7 +331,7 @@ Methods:
306331
Types:
307332

308333
```python
309-
from gradientai.types import APIModel, ModelListResponse
334+
from gradientai.types import APIAgreement, APIModel, APIModelVersion, ModelListResponse
310335
```
311336

312337
Methods:

src/gradientai/_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,10 @@
3333
if TYPE_CHECKING:
3434
from .resources import chat, agents, models, regions, inference, providers, indexing_jobs, knowledge_bases
3535
from .resources.models import ModelsResource, AsyncModelsResource
36-
from .resources.regions import RegionsResource, AsyncRegionsResource
3736
from .resources.chat.chat import ChatResource, AsyncChatResource
3837
from .resources.agents.agents import AgentsResource, AsyncAgentsResource
3938
from .resources.indexing_jobs import IndexingJobsResource, AsyncIndexingJobsResource
39+
from .resources.regions.regions import RegionsResource, AsyncRegionsResource
4040
from .resources.inference.inference import InferenceResource, AsyncInferenceResource
4141
from .resources.providers.providers import ProvidersResource, AsyncProvidersResource
4242
from .resources.knowledge_bases.knowledge_bases import KnowledgeBasesResource, AsyncKnowledgeBasesResource

src/gradientai/resources/knowledge_bases/data_sources.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
data_source_list_params,
2020
data_source_create_params,
2121
)
22+
from ...types.knowledge_bases.aws_data_source_param import AwsDataSourceParam
2223
from ...types.knowledge_bases.data_source_list_response import DataSourceListResponse
2324
from ...types.knowledge_bases.data_source_create_response import DataSourceCreateResponse
2425
from ...types.knowledge_bases.data_source_delete_response import DataSourceDeleteResponse
@@ -52,7 +53,7 @@ def create(
5253
self,
5354
path_knowledge_base_uuid: str,
5455
*,
55-
aws_data_source: data_source_create_params.AwsDataSource | NotGiven = NOT_GIVEN,
56+
aws_data_source: AwsDataSourceParam | NotGiven = NOT_GIVEN,
5657
body_knowledge_base_uuid: str | NotGiven = NOT_GIVEN,
5758
spaces_data_source: APISpacesDataSourceParam | NotGiven = NOT_GIVEN,
5859
web_crawler_data_source: APIWebCrawlerDataSourceParam | NotGiven = NOT_GIVEN,
@@ -219,7 +220,7 @@ async def create(
219220
self,
220221
path_knowledge_base_uuid: str,
221222
*,
222-
aws_data_source: data_source_create_params.AwsDataSource | NotGiven = NOT_GIVEN,
223+
aws_data_source: AwsDataSourceParam | NotGiven = NOT_GIVEN,
223224
body_knowledge_base_uuid: str | NotGiven = NOT_GIVEN,
224225
spaces_data_source: APISpacesDataSourceParam | NotGiven = NOT_GIVEN,
225226
web_crawler_data_source: APIWebCrawlerDataSourceParam | NotGiven = NOT_GIVEN,
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2+
3+
from .regions import (
4+
RegionsResource,
5+
AsyncRegionsResource,
6+
RegionsResourceWithRawResponse,
7+
AsyncRegionsResourceWithRawResponse,
8+
RegionsResourceWithStreamingResponse,
9+
AsyncRegionsResourceWithStreamingResponse,
10+
)
11+
12+
__all__ = [
13+
"RegionsResource",
14+
"AsyncRegionsResource",
15+
"RegionsResourceWithRawResponse",
16+
"AsyncRegionsResourceWithRawResponse",
17+
"RegionsResourceWithStreamingResponse",
18+
"AsyncRegionsResourceWithStreamingResponse",
19+
]

src/gradientai/resources/regions.py renamed to src/gradientai/resources/regions/regions.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,19 +4,19 @@
44

55
import httpx
66

7-
from ..types import region_list_params
8-
from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
9-
from .._utils import maybe_transform, async_maybe_transform
10-
from .._compat import cached_property
11-
from .._resource import SyncAPIResource, AsyncAPIResource
12-
from .._response import (
7+
from ...types import region_list_params
8+
from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
9+
from ..._utils import maybe_transform, async_maybe_transform
10+
from ..._compat import cached_property
11+
from ..._resource import SyncAPIResource, AsyncAPIResource
12+
from ..._response import (
1313
to_raw_response_wrapper,
1414
to_streamed_response_wrapper,
1515
async_to_raw_response_wrapper,
1616
async_to_streamed_response_wrapper,
1717
)
18-
from .._base_client import make_request_options
19-
from ..types.region_list_response import RegionListResponse
18+
from ..._base_client import make_request_options
19+
from ...types.region_list_response import RegionListResponse
2020

2121
__all__ = ["RegionsResource", "AsyncRegionsResource"]
2222

src/gradientai/types/__init__.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,12 @@
44

55
from .api_agent import APIAgent as APIAgent
66
from .api_model import APIModel as APIModel
7+
from .api_agreement import APIAgreement as APIAgreement
8+
from .api_workspace import APIWorkspace as APIWorkspace
9+
from .api_agent_model import APIAgentModel as APIAgentModel
710
from .api_indexing_job import APIIndexingJob as APIIndexingJob
811
from .agent_list_params import AgentListParams as AgentListParams
12+
from .api_model_version import APIModelVersion as APIModelVersion
913
from .model_list_params import ModelListParams as ModelListParams
1014
from .api_knowledge_base import APIKnowledgeBase as APIKnowledgeBase
1115
from .region_list_params import RegionListParams as RegionListParams
@@ -18,6 +22,7 @@
1822
from .agent_create_response import AgentCreateResponse as AgentCreateResponse
1923
from .agent_delete_response import AgentDeleteResponse as AgentDeleteResponse
2024
from .agent_update_response import AgentUpdateResponse as AgentUpdateResponse
25+
from .api_evaluation_metric import APIEvaluationMetric as APIEvaluationMetric
2126
from .api_agent_api_key_info import APIAgentAPIKeyInfo as APIAgentAPIKeyInfo
2227
from .agent_retrieve_response import AgentRetrieveResponse as AgentRetrieveResponse
2328
from .api_openai_api_key_info import APIOpenAIAPIKeyInfo as APIOpenAIAPIKeyInfo

src/gradientai/types/agent_list_response.py

Lines changed: 3 additions & 136 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
from .._models import BaseModel
88
from .agents.api_meta import APIMeta
9+
from .api_agent_model import APIAgentModel
910
from .agents.api_links import APILinks
1011
from .api_knowledge_base import APIKnowledgeBase
1112
from .api_retrieval_method import APIRetrievalMethod
@@ -17,14 +18,8 @@
1718
"AgentChatbot",
1819
"AgentChatbotIdentifier",
1920
"AgentDeployment",
20-
"AgentModel",
21-
"AgentModelAgreement",
22-
"AgentModelVersion",
2321
"AgentTemplate",
2422
"AgentTemplateGuardrail",
25-
"AgentTemplateModel",
26-
"AgentTemplateModelAgreement",
27-
"AgentTemplateModelVersion",
2823
]
2924

3025

@@ -74,140 +69,12 @@ class AgentDeployment(BaseModel):
7469
visibility: Optional[APIDeploymentVisibility] = None
7570

7671

77-
class AgentModelAgreement(BaseModel):
78-
description: Optional[str] = None
79-
80-
name: Optional[str] = None
81-
82-
url: Optional[str] = None
83-
84-
uuid: Optional[str] = None
85-
86-
87-
class AgentModelVersion(BaseModel):
88-
major: Optional[int] = None
89-
90-
minor: Optional[int] = None
91-
92-
patch: Optional[int] = None
93-
94-
95-
class AgentModel(BaseModel):
96-
agreement: Optional[AgentModelAgreement] = None
97-
98-
created_at: Optional[datetime] = None
99-
100-
inference_name: Optional[str] = None
101-
102-
inference_version: Optional[str] = None
103-
104-
is_foundational: Optional[bool] = None
105-
106-
metadata: Optional[object] = None
107-
108-
name: Optional[str] = None
109-
110-
parent_uuid: Optional[str] = None
111-
112-
provider: Optional[Literal["MODEL_PROVIDER_DIGITALOCEAN", "MODEL_PROVIDER_ANTHROPIC", "MODEL_PROVIDER_OPENAI"]] = (
113-
None
114-
)
115-
116-
updated_at: Optional[datetime] = None
117-
118-
upload_complete: Optional[bool] = None
119-
120-
url: Optional[str] = None
121-
122-
usecases: Optional[
123-
List[
124-
Literal[
125-
"MODEL_USECASE_UNKNOWN",
126-
"MODEL_USECASE_AGENT",
127-
"MODEL_USECASE_FINETUNED",
128-
"MODEL_USECASE_KNOWLEDGEBASE",
129-
"MODEL_USECASE_GUARDRAIL",
130-
"MODEL_USECASE_REASONING",
131-
"MODEL_USECASE_SERVERLESS",
132-
]
133-
]
134-
] = None
135-
136-
uuid: Optional[str] = None
137-
138-
version: Optional[AgentModelVersion] = None
139-
140-
14172
class AgentTemplateGuardrail(BaseModel):
14273
priority: Optional[int] = None
14374

14475
uuid: Optional[str] = None
14576

14677

147-
class AgentTemplateModelAgreement(BaseModel):
148-
description: Optional[str] = None
149-
150-
name: Optional[str] = None
151-
152-
url: Optional[str] = None
153-
154-
uuid: Optional[str] = None
155-
156-
157-
class AgentTemplateModelVersion(BaseModel):
158-
major: Optional[int] = None
159-
160-
minor: Optional[int] = None
161-
162-
patch: Optional[int] = None
163-
164-
165-
class AgentTemplateModel(BaseModel):
166-
agreement: Optional[AgentTemplateModelAgreement] = None
167-
168-
created_at: Optional[datetime] = None
169-
170-
inference_name: Optional[str] = None
171-
172-
inference_version: Optional[str] = None
173-
174-
is_foundational: Optional[bool] = None
175-
176-
metadata: Optional[object] = None
177-
178-
name: Optional[str] = None
179-
180-
parent_uuid: Optional[str] = None
181-
182-
provider: Optional[Literal["MODEL_PROVIDER_DIGITALOCEAN", "MODEL_PROVIDER_ANTHROPIC", "MODEL_PROVIDER_OPENAI"]] = (
183-
None
184-
)
185-
186-
updated_at: Optional[datetime] = None
187-
188-
upload_complete: Optional[bool] = None
189-
190-
url: Optional[str] = None
191-
192-
usecases: Optional[
193-
List[
194-
Literal[
195-
"MODEL_USECASE_UNKNOWN",
196-
"MODEL_USECASE_AGENT",
197-
"MODEL_USECASE_FINETUNED",
198-
"MODEL_USECASE_KNOWLEDGEBASE",
199-
"MODEL_USECASE_GUARDRAIL",
200-
"MODEL_USECASE_REASONING",
201-
"MODEL_USECASE_SERVERLESS",
202-
]
203-
]
204-
] = None
205-
206-
uuid: Optional[str] = None
207-
208-
version: Optional[AgentTemplateModelVersion] = None
209-
210-
21178
class AgentTemplate(BaseModel):
21279
created_at: Optional[datetime] = None
21380

@@ -225,7 +92,7 @@ class AgentTemplate(BaseModel):
22592

22693
max_tokens: Optional[int] = None
22794

228-
model: Optional[AgentTemplateModel] = None
95+
model: Optional[APIAgentModel] = None
22996

23097
name: Optional[str] = None
23198

@@ -276,7 +143,7 @@ class Agent(BaseModel):
276143
response.
277144
"""
278145

279-
model: Optional[AgentModel] = None
146+
model: Optional[APIAgentModel] = None
280147

281148
name: Optional[str] = None
282149

0 commit comments

Comments
 (0)