Skip to content

Commit e783720

Browse files
yanxi0830yanxi0830
authored
sync v1 prefix sdk (#84)
# What does this PR do? - update SDK to reflect llamastack/llama-stack@b78e667 ## Test Plan ``` LLAMA_STACK_BASE_URL="http://localhost:5000" pytest -v tests/client-sdk/inference/test_inference.py ``` <img width="1306" alt="image" src="https://github.com/user-attachments/assets/e3d9002b-8fe7-4a1e-8ba0-256f8298a7ee" /> - these are pre-existing errors, doesn't impact v1 sdk sync ## Sources Please link relevant resources if necessary. ## Before submitting - [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case). - [ ] Ran pre-commit to handle lint / formatting issues. - [ ] Read the [contributor guideline](https://github.com/meta-llama/llama-stack/blob/main/CONTRIBUTING.md), Pull Request section? - [ ] Updated relevant documentation. - [ ] Wrote necessary unit or integration tests.
1 parent c220a4b commit e783720

38 files changed

+310
-211
lines changed

‎src/llama_stack_client/_client.py‎

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2-
32
from __future__ import annotations
4-
import json
53

4+
import json
65
import os
76
from typing import Any, Union, Mapping
87
from typing_extensions import Self, override
@@ -128,18 +127,18 @@ def __init__(
128127
if base_url is None:
129128
base_url = f"http://any-hosted-llama-stack.com"
130129

131-
custom_headers = default_headers or {}
132-
custom_headers["X-LlamaStack-Client-Version"] = __version__
133130
if provider_data is not None:
134-
custom_headers["X-LlamaStack-Provider-Data"] = json.dumps(provider_data)
131+
if default_headers is None:
132+
default_headers = {}
133+
default_headers["X-LlamaStack-ProviderData"] = json.dumps(provider_data)
135134

136135
super().__init__(
137136
version=__version__,
138137
base_url=base_url,
139138
max_retries=max_retries,
140139
timeout=timeout,
141140
http_client=http_client,
142-
custom_headers=custom_headers,
141+
custom_headers=default_headers,
143142
custom_query=default_query,
144143
_strict_response_validation=_strict_response_validation,
145144
)
@@ -325,18 +324,18 @@ def __init__(
325324
if base_url is None:
326325
base_url = f"http://any-hosted-llama-stack.com"
327326

328-
custom_headers = default_headers or {}
329-
custom_headers["X-LlamaStack-Client-Version"] = __version__
330327
if provider_data is not None:
331-
custom_headers["X-LlamaStack-Provider-Data"] = json.dumps(provider_data)
328+
if default_headers is None:
329+
default_headers = {}
330+
default_headers["X-LlamaStack-ProviderData"] = json.dumps(provider_data)
332331

333332
super().__init__(
334333
version=__version__,
335334
base_url=base_url,
336335
max_retries=max_retries,
337336
timeout=timeout,
338337
http_client=http_client,
339-
custom_headers=custom_headers,
338+
custom_headers=default_headers,
340339
custom_query=default_query,
341340
_strict_response_validation=_strict_response_validation,
342341
)

‎src/llama_stack_client/_version.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
22

33
__title__ = "llama_stack_client"
4-
__version__ = "0.0.63"
4+
__version__ = "0.0.1-alpha.0"

‎src/llama_stack_client/resources/agents/agents.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def create(
115115
**(extra_headers or {}),
116116
}
117117
return self._post(
118-
"/alpha/agents/create",
118+
"/v1/agents/create",
119119
body=maybe_transform({"agent_config": agent_config}, agent_create_params.AgentCreateParams),
120120
options=make_request_options(
121121
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -157,7 +157,7 @@ def delete(
157157
**(extra_headers or {}),
158158
}
159159
return self._post(
160-
"/alpha/agents/delete",
160+
"/v1/agents/delete",
161161
body=maybe_transform({"agent_id": agent_id}, agent_delete_params.AgentDeleteParams),
162162
options=make_request_options(
163163
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -231,7 +231,7 @@ async def create(
231231
**(extra_headers or {}),
232232
}
233233
return await self._post(
234-
"/alpha/agents/create",
234+
"/v1/agents/create",
235235
body=await async_maybe_transform({"agent_config": agent_config}, agent_create_params.AgentCreateParams),
236236
options=make_request_options(
237237
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -273,7 +273,7 @@ async def delete(
273273
**(extra_headers or {}),
274274
}
275275
return await self._post(
276-
"/alpha/agents/delete",
276+
"/v1/agents/delete",
277277
body=await async_maybe_transform({"agent_id": agent_id}, agent_delete_params.AgentDeleteParams),
278278
options=make_request_options(
279279
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout

‎src/llama_stack_client/resources/agents/session.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def create(
8282
**(extra_headers or {}),
8383
}
8484
return self._post(
85-
"/alpha/agents/session/create",
85+
"/v1/agents/session/create",
8686
body=maybe_transform(
8787
{
8888
"agent_id": agent_id,
@@ -131,7 +131,7 @@ def retrieve(
131131
**(extra_headers or {}),
132132
}
133133
return self._post(
134-
"/alpha/agents/session/get",
134+
"/v1/agents/session/get",
135135
body=maybe_transform({"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
136136
options=make_request_options(
137137
extra_headers=extra_headers,
@@ -184,7 +184,7 @@ def delete(
184184
**(extra_headers or {}),
185185
}
186186
return self._post(
187-
"/alpha/agents/session/delete",
187+
"/v1/agents/session/delete",
188188
body=maybe_transform(
189189
{
190190
"agent_id": agent_id,
@@ -253,7 +253,7 @@ async def create(
253253
**(extra_headers or {}),
254254
}
255255
return await self._post(
256-
"/alpha/agents/session/create",
256+
"/v1/agents/session/create",
257257
body=await async_maybe_transform(
258258
{
259259
"agent_id": agent_id,
@@ -302,7 +302,7 @@ async def retrieve(
302302
**(extra_headers or {}),
303303
}
304304
return await self._post(
305-
"/alpha/agents/session/get",
305+
"/v1/agents/session/get",
306306
body=await async_maybe_transform({"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
307307
options=make_request_options(
308308
extra_headers=extra_headers,
@@ -355,7 +355,7 @@ async def delete(
355355
**(extra_headers or {}),
356356
}
357357
return await self._post(
358-
"/alpha/agents/session/delete",
358+
"/v1/agents/session/delete",
359359
body=await async_maybe_transform(
360360
{
361361
"agent_id": agent_id,

‎src/llama_stack_client/resources/agents/steps.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def retrieve(
8181
**(extra_headers or {}),
8282
}
8383
return self._get(
84-
"/alpha/agents/step/get",
84+
"/v1/agents/step/get",
8585
options=make_request_options(
8686
extra_headers=extra_headers,
8787
extra_query=extra_query,
@@ -157,7 +157,7 @@ async def retrieve(
157157
**(extra_headers or {}),
158158
}
159159
return await self._get(
160-
"/alpha/agents/step/get",
160+
"/v1/agents/step/get",
161161
options=make_request_options(
162162
extra_headers=extra_headers,
163163
extra_query=extra_query,

‎src/llama_stack_client/resources/agents/turn.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ def create(
176176
return cast(
177177
TurnCreateResponse,
178178
self._post(
179-
"/alpha/agents/turn/create",
179+
"/v1/agents/turn/create",
180180
body=maybe_transform(
181181
{
182182
"agent_id": agent_id,
@@ -234,7 +234,7 @@ def retrieve(
234234
**(extra_headers or {}),
235235
}
236236
return self._get(
237-
"/alpha/agents/turn/get",
237+
"/v1/agents/turn/get",
238238
options=make_request_options(
239239
extra_headers=extra_headers,
240240
extra_query=extra_query,
@@ -398,7 +398,7 @@ async def create(
398398
return cast(
399399
TurnCreateResponse,
400400
await self._post(
401-
"/alpha/agents/turn/create",
401+
"/v1/agents/turn/create",
402402
body=await async_maybe_transform(
403403
{
404404
"agent_id": agent_id,
@@ -456,7 +456,7 @@ async def retrieve(
456456
**(extra_headers or {}),
457457
}
458458
return await self._get(
459-
"/alpha/agents/turn/get",
459+
"/v1/agents/turn/get",
460460
options=make_request_options(
461461
extra_headers=extra_headers,
462462
extra_query=extra_query,

‎src/llama_stack_client/resources/batch_inference.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def chat_completion(
100100
**(extra_headers or {}),
101101
}
102102
return self._post(
103-
"/alpha/batch-inference/chat-completion",
103+
"/v1/batch-inference/chat-completion",
104104
body=maybe_transform(
105105
{
106106
"messages_batch": messages_batch,
@@ -155,7 +155,7 @@ def completion(
155155
**(extra_headers or {}),
156156
}
157157
return self._post(
158-
"/alpha/batch-inference/completion",
158+
"/v1/batch-inference/completion",
159159
body=maybe_transform(
160160
{
161161
"content_batch": content_batch,
@@ -241,7 +241,7 @@ async def chat_completion(
241241
**(extra_headers or {}),
242242
}
243243
return await self._post(
244-
"/alpha/batch-inference/chat-completion",
244+
"/v1/batch-inference/chat-completion",
245245
body=await async_maybe_transform(
246246
{
247247
"messages_batch": messages_batch,
@@ -296,7 +296,7 @@ async def completion(
296296
**(extra_headers or {}),
297297
}
298298
return await self._post(
299-
"/alpha/batch-inference/completion",
299+
"/v1/batch-inference/completion",
300300
body=await async_maybe_transform(
301301
{
302302
"content_batch": content_batch,

‎src/llama_stack_client/resources/datasetio.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def append_rows(
8282
**(extra_headers or {}),
8383
}
8484
return self._post(
85-
"/alpha/datasetio/append-rows",
85+
"/v1/datasetio/append-rows",
8686
body=maybe_transform(
8787
{
8888
"dataset_id": dataset_id,
@@ -132,7 +132,7 @@ def get_rows_paginated(
132132
**(extra_headers or {}),
133133
}
134134
return self._get(
135-
"/alpha/datasetio/get-rows-paginated",
135+
"/v1/datasetio/get-rows-paginated",
136136
options=make_request_options(
137137
extra_headers=extra_headers,
138138
extra_query=extra_query,
@@ -207,7 +207,7 @@ async def append_rows(
207207
**(extra_headers or {}),
208208
}
209209
return await self._post(
210-
"/alpha/datasetio/append-rows",
210+
"/v1/datasetio/append-rows",
211211
body=await async_maybe_transform(
212212
{
213213
"dataset_id": dataset_id,
@@ -257,7 +257,7 @@ async def get_rows_paginated(
257257
**(extra_headers or {}),
258258
}
259259
return await self._get(
260-
"/alpha/datasetio/get-rows-paginated",
260+
"/v1/datasetio/get-rows-paginated",
261261
options=make_request_options(
262262
extra_headers=extra_headers,
263263
extra_query=extra_query,

‎src/llama_stack_client/resources/datasets.py‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def retrieve(
8383
**(extra_headers or {}),
8484
}
8585
return self._get(
86-
"/alpha/datasets/get",
86+
"/v1/datasets/get",
8787
options=make_request_options(
8888
extra_headers=extra_headers,
8989
extra_query=extra_query,
@@ -127,7 +127,7 @@ def list(
127127
**(extra_headers or {}),
128128
}
129129
return self._get(
130-
"/alpha/datasets/list",
130+
"/v1/datasets/list",
131131
options=make_request_options(
132132
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
133133
),
@@ -173,7 +173,7 @@ def register(
173173
**(extra_headers or {}),
174174
}
175175
return self._post(
176-
"/alpha/datasets/register",
176+
"/v1/datasets/register",
177177
body=maybe_transform(
178178
{
179179
"dataset_id": dataset_id,
@@ -225,7 +225,7 @@ def unregister(
225225
**(extra_headers or {}),
226226
}
227227
return self._post(
228-
"/alpha/datasets/unregister",
228+
"/v1/datasets/unregister",
229229
body=maybe_transform({"dataset_id": dataset_id}, dataset_unregister_params.DatasetUnregisterParams),
230230
options=make_request_options(
231231
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -287,7 +287,7 @@ async def retrieve(
287287
**(extra_headers or {}),
288288
}
289289
return await self._get(
290-
"/alpha/datasets/get",
290+
"/v1/datasets/get",
291291
options=make_request_options(
292292
extra_headers=extra_headers,
293293
extra_query=extra_query,
@@ -333,7 +333,7 @@ async def list(
333333
**(extra_headers or {}),
334334
}
335335
return await self._get(
336-
"/alpha/datasets/list",
336+
"/v1/datasets/list",
337337
options=make_request_options(
338338
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
339339
),
@@ -379,7 +379,7 @@ async def register(
379379
**(extra_headers or {}),
380380
}
381381
return await self._post(
382-
"/alpha/datasets/register",
382+
"/v1/datasets/register",
383383
body=await async_maybe_transform(
384384
{
385385
"dataset_id": dataset_id,
@@ -431,7 +431,7 @@ async def unregister(
431431
**(extra_headers or {}),
432432
}
433433
return await self._post(
434-
"/alpha/datasets/unregister",
434+
"/v1/datasets/unregister",
435435
body=await async_maybe_transform(
436436
{"dataset_id": dataset_id}, dataset_unregister_params.DatasetUnregisterParams
437437
),

‎src/llama_stack_client/resources/eval/eval.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def evaluate_rows(
9696
**(extra_headers or {}),
9797
}
9898
return self._post(
99-
"/alpha/eval/evaluate-rows",
99+
"/v1/eval/evaluate-rows",
100100
body=maybe_transform(
101101
{
102102
"input_rows": input_rows,
@@ -146,7 +146,7 @@ def run_eval(
146146
**(extra_headers or {}),
147147
}
148148
return self._post(
149-
"/alpha/eval/run-eval",
149+
"/v1/eval/run-eval",
150150
body=maybe_transform(
151151
{
152152
"task_config": task_config,
@@ -221,7 +221,7 @@ async def evaluate_rows(
221221
**(extra_headers or {}),
222222
}
223223
return await self._post(
224-
"/alpha/eval/evaluate-rows",
224+
"/v1/eval/evaluate-rows",
225225
body=await async_maybe_transform(
226226
{
227227
"input_rows": input_rows,
@@ -271,7 +271,7 @@ async def run_eval(
271271
**(extra_headers or {}),
272272
}
273273
return await self._post(
274-
"/alpha/eval/run-eval",
274+
"/v1/eval/run-eval",
275275
body=await async_maybe_transform(
276276
{
277277
"task_config": task_config,

0 commit comments

Comments
 (0)