Skip to content

Commit 2fc9382

Browse files
feat(api): vector_db_id -> vector_store_id
1 parent 2ac7e1b commit 2fc9382

File tree

16 files changed

+571
-558
lines changed

16 files changed

+571
-558
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: 104
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-97b91eca4a3ff251edc02636b1a638866675d6c1abd46cd9fc18bc50a1de9656.yml
3-
openapi_spec_hash: 7302f1aa50090e3de78e34c184371267
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-35c6569e5e9fcc85084c9728eb7fc7c5908297fcc77043d621d25de3c850a990.yml
3+
openapi_spec_hash: 0f95bbeee16f3205d36ec34cfa62c711
44
config_hash: a3829dbdaa491194d01f399784d532cd

src/llama_stack_client/resources/moderations.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
import httpx
1414

1515
from ..types import moderation_create_params
16-
from .._types import Body, Query, Headers, NotGiven, SequenceNotStr, not_given
16+
from .._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
1717
from .._utils import maybe_transform, async_maybe_transform
1818
from .._compat import cached_property
1919
from .._resource import SyncAPIResource, AsyncAPIResource
@@ -53,7 +53,7 @@ def create(
5353
self,
5454
*,
5555
input: Union[str, SequenceNotStr[str]],
56-
model: str,
56+
model: str | Omit = omit,
5757
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
5858
# The extra values given here take precedence over values defined on the client or passed to this method.
5959
extra_headers: Headers | None = None,
@@ -70,7 +70,7 @@ def create(
7070
input: Input (or inputs) to classify. Can be a single string, an array of strings, or
7171
an array of multi-modal input objects similar to other models.
7272
73-
model: The content moderation model you would like to use.
73+
model: (Optional) The content moderation model you would like to use.
7474
7575
extra_headers: Send extra headers
7676
@@ -120,7 +120,7 @@ async def create(
120120
self,
121121
*,
122122
input: Union[str, SequenceNotStr[str]],
123-
model: str,
123+
model: str | Omit = omit,
124124
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
125125
# The extra values given here take precedence over values defined on the client or passed to this method.
126126
extra_headers: Headers | None = None,
@@ -137,7 +137,7 @@ async def create(
137137
input: Input (or inputs) to classify. Can be a single string, an array of strings, or
138138
an array of multi-modal input objects similar to other models.
139139
140-
model: The content moderation model you would like to use.
140+
model: (Optional) The content moderation model you would like to use.
141141
142142
extra_headers: Send extra headers
143143

src/llama_stack_client/resources/tool_runtime/rag_tool.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def insert(
5757
*,
5858
chunk_size_in_tokens: int,
5959
documents: Iterable[Document],
60-
vector_db_id: str,
60+
vector_store_id: str,
6161
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
6262
# The extra values given here take precedence over values defined on the client or passed to this method.
6363
extra_headers: Headers | None = None,
@@ -73,7 +73,7 @@ def insert(
7373
7474
documents: List of documents to index in the RAG system
7575
76-
vector_db_id: ID of the vector database to store the document embeddings
76+
vector_store_id: ID of the vector database to store the document embeddings
7777
7878
extra_headers: Send extra headers
7979
@@ -90,7 +90,7 @@ def insert(
9090
{
9191
"chunk_size_in_tokens": chunk_size_in_tokens,
9292
"documents": documents,
93-
"vector_db_id": vector_db_id,
93+
"vector_store_id": vector_store_id,
9494
},
9595
rag_tool_insert_params.RagToolInsertParams,
9696
),
@@ -104,7 +104,7 @@ def query(
104104
self,
105105
*,
106106
content: InterleavedContent,
107-
vector_db_ids: SequenceNotStr[str],
107+
vector_store_ids: SequenceNotStr[str],
108108
query_config: QueryConfig | Omit = omit,
109109
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
110110
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -119,7 +119,7 @@ def query(
119119
Args:
120120
content: The query content to search for in the indexed documents
121121
122-
vector_db_ids: List of vector database IDs to search within
122+
vector_store_ids: List of vector database IDs to search within
123123
124124
query_config: (Optional) Configuration parameters for the query operation
125125
@@ -136,7 +136,7 @@ def query(
136136
body=maybe_transform(
137137
{
138138
"content": content,
139-
"vector_db_ids": vector_db_ids,
139+
"vector_store_ids": vector_store_ids,
140140
"query_config": query_config,
141141
},
142142
rag_tool_query_params.RagToolQueryParams,
@@ -173,7 +173,7 @@ async def insert(
173173
*,
174174
chunk_size_in_tokens: int,
175175
documents: Iterable[Document],
176-
vector_db_id: str,
176+
vector_store_id: str,
177177
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
178178
# The extra values given here take precedence over values defined on the client or passed to this method.
179179
extra_headers: Headers | None = None,
@@ -189,7 +189,7 @@ async def insert(
189189
190190
documents: List of documents to index in the RAG system
191191
192-
vector_db_id: ID of the vector database to store the document embeddings
192+
vector_store_id: ID of the vector database to store the document embeddings
193193
194194
extra_headers: Send extra headers
195195
@@ -206,7 +206,7 @@ async def insert(
206206
{
207207
"chunk_size_in_tokens": chunk_size_in_tokens,
208208
"documents": documents,
209-
"vector_db_id": vector_db_id,
209+
"vector_store_id": vector_store_id,
210210
},
211211
rag_tool_insert_params.RagToolInsertParams,
212212
),
@@ -220,7 +220,7 @@ async def query(
220220
self,
221221
*,
222222
content: InterleavedContent,
223-
vector_db_ids: SequenceNotStr[str],
223+
vector_store_ids: SequenceNotStr[str],
224224
query_config: QueryConfig | Omit = omit,
225225
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
226226
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -235,7 +235,7 @@ async def query(
235235
Args:
236236
content: The query content to search for in the indexed documents
237237
238-
vector_db_ids: List of vector database IDs to search within
238+
vector_store_ids: List of vector database IDs to search within
239239
240240
query_config: (Optional) Configuration parameters for the query operation
241241
@@ -252,7 +252,7 @@ async def query(
252252
body=await async_maybe_transform(
253253
{
254254
"content": content,
255-
"vector_db_ids": vector_db_ids,
255+
"vector_store_ids": vector_store_ids,
256256
"query_config": query_config,
257257
},
258258
rag_tool_query_params.RagToolQueryParams,

src/llama_stack_client/resources/vector_io.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def insert(
5454
self,
5555
*,
5656
chunks: Iterable[vector_io_insert_params.Chunk],
57-
vector_db_id: str,
57+
vector_store_id: str,
5858
ttl_seconds: int | Omit = omit,
5959
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
6060
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -74,7 +74,7 @@ def insert(
7474
configure how Llama Stack formats the chunk during generation. If `embedding` is
7575
not provided, it will be computed later.
7676
77-
vector_db_id: The identifier of the vector database to insert the chunks into.
77+
vector_store_id: The identifier of the vector database to insert the chunks into.
7878
7979
ttl_seconds: The time to live of the chunks.
8080
@@ -92,7 +92,7 @@ def insert(
9292
body=maybe_transform(
9393
{
9494
"chunks": chunks,
95-
"vector_db_id": vector_db_id,
95+
"vector_store_id": vector_store_id,
9696
"ttl_seconds": ttl_seconds,
9797
},
9898
vector_io_insert_params.VectorIoInsertParams,
@@ -107,7 +107,7 @@ def query(
107107
self,
108108
*,
109109
query: InterleavedContent,
110-
vector_db_id: str,
110+
vector_store_id: str,
111111
params: Dict[str, Union[bool, float, str, Iterable[object], object, None]] | Omit = omit,
112112
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
113113
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -122,7 +122,7 @@ def query(
122122
Args:
123123
query: The query to search for.
124124
125-
vector_db_id: The identifier of the vector database to query.
125+
vector_store_id: The identifier of the vector database to query.
126126
127127
params: The parameters of the query.
128128
@@ -139,7 +139,7 @@ def query(
139139
body=maybe_transform(
140140
{
141141
"query": query,
142-
"vector_db_id": vector_db_id,
142+
"vector_store_id": vector_store_id,
143143
"params": params,
144144
},
145145
vector_io_query_params.VectorIoQueryParams,
@@ -175,7 +175,7 @@ async def insert(
175175
self,
176176
*,
177177
chunks: Iterable[vector_io_insert_params.Chunk],
178-
vector_db_id: str,
178+
vector_store_id: str,
179179
ttl_seconds: int | Omit = omit,
180180
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
181181
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -195,7 +195,7 @@ async def insert(
195195
configure how Llama Stack formats the chunk during generation. If `embedding` is
196196
not provided, it will be computed later.
197197
198-
vector_db_id: The identifier of the vector database to insert the chunks into.
198+
vector_store_id: The identifier of the vector database to insert the chunks into.
199199
200200
ttl_seconds: The time to live of the chunks.
201201
@@ -213,7 +213,7 @@ async def insert(
213213
body=await async_maybe_transform(
214214
{
215215
"chunks": chunks,
216-
"vector_db_id": vector_db_id,
216+
"vector_store_id": vector_store_id,
217217
"ttl_seconds": ttl_seconds,
218218
},
219219
vector_io_insert_params.VectorIoInsertParams,
@@ -228,7 +228,7 @@ async def query(
228228
self,
229229
*,
230230
query: InterleavedContent,
231-
vector_db_id: str,
231+
vector_store_id: str,
232232
params: Dict[str, Union[bool, float, str, Iterable[object], object, None]] | Omit = omit,
233233
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
234234
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -243,7 +243,7 @@ async def query(
243243
Args:
244244
query: The query to search for.
245245
246-
vector_db_id: The identifier of the vector database to query.
246+
vector_store_id: The identifier of the vector database to query.
247247
248248
params: The parameters of the query.
249249
@@ -260,7 +260,7 @@ async def query(
260260
body=await async_maybe_transform(
261261
{
262262
"query": query,
263-
"vector_db_id": vector_db_id,
263+
"vector_store_id": vector_store_id,
264264
"params": params,
265265
},
266266
vector_io_query_params.VectorIoQueryParams,

src/llama_stack_client/types/alpha/memory_retrieval_step.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class MemoryRetrievalStep(BaseModel):
2929
turn_id: str
3030
"""The ID of the turn."""
3131

32-
vector_db_ids: str
32+
vector_store_ids: str
3333
"""The IDs of the vector databases to retrieve context from."""
3434

3535
completed_at: Optional[datetime] = None

src/llama_stack_client/types/moderation_create_params.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,5 +24,5 @@ class ModerationCreateParams(TypedDict, total=False):
2424
objects similar to other models.
2525
"""
2626

27-
model: Required[str]
28-
"""The content moderation model you would like to use."""
27+
model: str
28+
"""(Optional) The content moderation model you would like to use."""

0 commit comments

Comments
 (0)