Skip to content
This repository was archived by the owner on Aug 14, 2025. It is now read-only.

Commit 303f8d9

Browse files
stainless-app[bot]stainless-app[bot]
committed
feat(api): update via SDK Studio
1 parent 747f025 commit 303f8d9

File tree

15 files changed

+79
-24
lines changed

15 files changed

+79
-24
lines changed

‎.stats.yml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 105
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-df7a19394e9124c18ec4e888e2856d22b5ebfd6fe6fe6e929ff6cfadb2ae7e2a.yml
3-
openapi_spec_hash: 9428682672fdd7e2afee7af9ef849dc9
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-001fdbbf185bc9fc18521e55461f50dbd77b2f6bdf4b9565fe18a5ade77004dd.yml
3+
openapi_spec_hash: f410f7b9ebf460becbfc1427256f68e8
44
config_hash: e7d2a81bdf0160da1d41afbaac55d4d2

‎src/llama_stack_client/resources/vector_dbs.py‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,7 @@ def register(
110110
embedding_dimension: int | NotGiven = NOT_GIVEN,
111111
provider_id: str | NotGiven = NOT_GIVEN,
112112
provider_vector_db_id: str | NotGiven = NOT_GIVEN,
113+
vector_db_name: str | NotGiven = NOT_GIVEN,
113114
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
114115
# The extra values given here take precedence over values defined on the client or passed to this method.
115116
extra_headers: Headers | None = None,
@@ -131,6 +132,8 @@ def register(
131132
132133
provider_vector_db_id: The identifier of the vector database in the provider.
133134
135+
vector_db_name: The name of the vector database.
136+
134137
extra_headers: Send extra headers
135138
136139
extra_query: Add additional query parameters to the request
@@ -148,6 +151,7 @@ def register(
148151
"embedding_dimension": embedding_dimension,
149152
"provider_id": provider_id,
150153
"provider_vector_db_id": provider_vector_db_id,
154+
"vector_db_name": vector_db_name,
151155
},
152156
vector_db_register_params.VectorDBRegisterParams,
153157
),
@@ -276,6 +280,7 @@ async def register(
276280
embedding_dimension: int | NotGiven = NOT_GIVEN,
277281
provider_id: str | NotGiven = NOT_GIVEN,
278282
provider_vector_db_id: str | NotGiven = NOT_GIVEN,
283+
vector_db_name: str | NotGiven = NOT_GIVEN,
279284
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
280285
# The extra values given here take precedence over values defined on the client or passed to this method.
281286
extra_headers: Headers | None = None,
@@ -297,6 +302,8 @@ async def register(
297302
298303
provider_vector_db_id: The identifier of the vector database in the provider.
299304
305+
vector_db_name: The name of the vector database.
306+
300307
extra_headers: Send extra headers
301308
302309
extra_query: Add additional query parameters to the request
@@ -314,6 +321,7 @@ async def register(
314321
"embedding_dimension": embedding_dimension,
315322
"provider_id": provider_id,
316323
"provider_vector_db_id": provider_vector_db_id,
324+
"vector_db_name": vector_db_name,
317325
},
318326
vector_db_register_params.VectorDBRegisterParams,
319327
),

‎src/llama_stack_client/resources/vector_stores/vector_stores.py‎

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,6 @@ def create(
7474
file_ids: List[str] | NotGiven = NOT_GIVEN,
7575
metadata: Dict[str, Union[bool, float, str, Iterable[object], object, None]] | NotGiven = NOT_GIVEN,
7676
provider_id: str | NotGiven = NOT_GIVEN,
77-
provider_vector_db_id: str | NotGiven = NOT_GIVEN,
7877
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
7978
# The extra values given here take precedence over values defined on the client or passed to this method.
8079
extra_headers: Headers | None = None,
@@ -104,8 +103,6 @@ def create(
104103
105104
provider_id: The ID of the provider to use for this vector store.
106105
107-
provider_vector_db_id: The provider-specific vector database ID.
108-
109106
extra_headers: Send extra headers
110107
111108
extra_query: Add additional query parameters to the request
@@ -126,7 +123,6 @@ def create(
126123
"file_ids": file_ids,
127124
"metadata": metadata,
128125
"provider_id": provider_id,
129-
"provider_vector_db_id": provider_vector_db_id,
130126
},
131127
vector_store_create_params.VectorStoreCreateParams,
132128
),
@@ -411,7 +407,6 @@ async def create(
411407
file_ids: List[str] | NotGiven = NOT_GIVEN,
412408
metadata: Dict[str, Union[bool, float, str, Iterable[object], object, None]] | NotGiven = NOT_GIVEN,
413409
provider_id: str | NotGiven = NOT_GIVEN,
414-
provider_vector_db_id: str | NotGiven = NOT_GIVEN,
415410
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
416411
# The extra values given here take precedence over values defined on the client or passed to this method.
417412
extra_headers: Headers | None = None,
@@ -441,8 +436,6 @@ async def create(
441436
442437
provider_id: The ID of the provider to use for this vector store.
443438
444-
provider_vector_db_id: The provider-specific vector database ID.
445-
446439
extra_headers: Send extra headers
447440
448441
extra_query: Add additional query parameters to the request
@@ -463,7 +456,6 @@ async def create(
463456
"file_ids": file_ids,
464457
"metadata": metadata,
465458
"provider_id": provider_id,
466-
"provider_vector_db_id": provider_vector_db_id,
467459
},
468460
vector_store_create_params.VectorStoreCreateParams,
469461
),

‎src/llama_stack_client/types/post_training/job_artifacts_response.py‎

Lines changed: 28 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,38 @@
11
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
22

3-
from typing import List
3+
from typing import List, Optional
4+
from datetime import datetime
45

56
from ..._models import BaseModel
67

7-
__all__ = ["JobArtifactsResponse"]
8+
__all__ = ["JobArtifactsResponse", "Checkpoint", "CheckpointTrainingMetrics"]
9+
10+
11+
class CheckpointTrainingMetrics(BaseModel):
12+
epoch: int
13+
14+
perplexity: float
15+
16+
train_loss: float
17+
18+
validation_loss: float
19+
20+
21+
class Checkpoint(BaseModel):
22+
created_at: datetime
23+
24+
epoch: int
25+
26+
identifier: str
27+
28+
path: str
29+
30+
post_training_job_id: str
31+
32+
training_metrics: Optional[CheckpointTrainingMetrics] = None
833

934

1035
class JobArtifactsResponse(BaseModel):
11-
checkpoints: List[object]
36+
checkpoints: List[Checkpoint]
1237

1338
job_uuid: str

‎src/llama_stack_client/types/post_training/job_status_response.py‎

Lines changed: 26 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,35 @@
66

77
from ..._models import BaseModel
88

9-
__all__ = ["JobStatusResponse"]
9+
__all__ = ["JobStatusResponse", "Checkpoint", "CheckpointTrainingMetrics"]
10+
11+
12+
class CheckpointTrainingMetrics(BaseModel):
13+
epoch: int
14+
15+
perplexity: float
16+
17+
train_loss: float
18+
19+
validation_loss: float
20+
21+
22+
class Checkpoint(BaseModel):
23+
created_at: datetime
24+
25+
epoch: int
26+
27+
identifier: str
28+
29+
path: str
30+
31+
post_training_job_id: str
32+
33+
training_metrics: Optional[CheckpointTrainingMetrics] = None
1034

1135

1236
class JobStatusResponse(BaseModel):
13-
checkpoints: List[object]
37+
checkpoints: List[Checkpoint]
1438

1539
job_uuid: str
1640

‎src/llama_stack_client/types/shared/query_config.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ class QueryConfig(BaseModel):
5555
query_generator_config: QueryGeneratorConfig
5656
"""Configuration for the query generator."""
5757

58-
mode: Optional[str] = None
58+
mode: Optional[Literal["vector", "keyword", "hybrid"]] = None
5959
"""Search mode for retrieval—either "vector", "keyword", or "hybrid".
6060
6161
Default "vector".

‎src/llama_stack_client/types/shared_params/query_config.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ class QueryConfig(TypedDict, total=False):
5555
query_generator_config: Required[QueryGeneratorConfig]
5656
"""Configuration for the query generator."""
5757

58-
mode: str
58+
mode: Literal["vector", "keyword", "hybrid"]
5959
"""Search mode for retrieval—either "vector", "keyword", or "hybrid".
6060
6161
Default "vector".

‎src/llama_stack_client/types/vector_db_list_response.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,5 +21,7 @@ class VectorDBListResponseItem(BaseModel):
2121

2222
provider_resource_id: Optional[str] = None
2323

24+
vector_db_name: Optional[str] = None
25+
2426

2527
VectorDBListResponse: TypeAlias = List[VectorDBListResponseItem]

‎src/llama_stack_client/types/vector_db_register_params.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,3 +22,6 @@ class VectorDBRegisterParams(TypedDict, total=False):
2222

2323
provider_vector_db_id: str
2424
"""The identifier of the vector database in the provider."""
25+
26+
vector_db_name: str
27+
"""The name of the vector database."""

‎src/llama_stack_client/types/vector_db_register_response.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,3 +20,5 @@ class VectorDBRegisterResponse(BaseModel):
2020
type: Literal["vector_db"]
2121

2222
provider_resource_id: Optional[str] = None
23+
24+
vector_db_name: Optional[str] = None

0 commit comments

Comments
 (0)