Skip to content

Commit c57d9ec

Browse files
ashwinbashwinb
authored
Sync updates from stainless branch: ashwinb/dev (#104)
Update for client SDK, new docs and simpler OpenAPI spec. - client-version and provider-data headers are _not_ added to the spec anymore. we don't want these parameters to pollute every single API method doc. - we can just work by initializing the client with these headers once - enums are flattened in the OpenAPI spec (no separate schema type registered for them)
1 parent 2fe2869 commit c57d9ec

File tree

118 files changed

+655
-3417
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

118 files changed

+655
-3417
lines changed

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

Lines changed: 0 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@
3232
from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
3333
from ..._utils import (
3434
maybe_transform,
35-
strip_not_given,
3635
async_maybe_transform,
3736
)
3837
from ..._compat import cached_property
@@ -86,8 +85,6 @@ def create(
8685
self,
8786
*,
8887
agent_config: AgentConfig,
89-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
90-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
9188
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
9289
# The extra values given here take precedence over values defined on the client or passed to this method.
9390
extra_headers: Headers | None = None,
@@ -105,15 +102,6 @@ def create(
105102
106103
timeout: Override the client-level default timeout for this request, in seconds
107104
"""
108-
extra_headers = {
109-
**strip_not_given(
110-
{
111-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
112-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
113-
}
114-
),
115-
**(extra_headers or {}),
116-
}
117105
return self._post(
118106
"/v1/agents",
119107
body=maybe_transform({"agent_config": agent_config}, agent_create_params.AgentCreateParams),
@@ -127,8 +115,6 @@ def delete(
127115
self,
128116
agent_id: str,
129117
*,
130-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
131-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
132118
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
133119
# The extra values given here take precedence over values defined on the client or passed to this method.
134120
extra_headers: Headers | None = None,
@@ -149,15 +135,6 @@ def delete(
149135
if not agent_id:
150136
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
151137
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
152-
extra_headers = {
153-
**strip_not_given(
154-
{
155-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
156-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
157-
}
158-
),
159-
**(extra_headers or {}),
160-
}
161138
return self._delete(
162139
f"/v1/agents/{agent_id}",
163140
options=make_request_options(
@@ -203,8 +180,6 @@ async def create(
203180
self,
204181
*,
205182
agent_config: AgentConfig,
206-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
207-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
208183
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
209184
# The extra values given here take precedence over values defined on the client or passed to this method.
210185
extra_headers: Headers | None = None,
@@ -222,15 +197,6 @@ async def create(
222197
223198
timeout: Override the client-level default timeout for this request, in seconds
224199
"""
225-
extra_headers = {
226-
**strip_not_given(
227-
{
228-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
229-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
230-
}
231-
),
232-
**(extra_headers or {}),
233-
}
234200
return await self._post(
235201
"/v1/agents",
236202
body=await async_maybe_transform({"agent_config": agent_config}, agent_create_params.AgentCreateParams),
@@ -244,8 +210,6 @@ async def delete(
244210
self,
245211
agent_id: str,
246212
*,
247-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
248-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
249213
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
250214
# The extra values given here take precedence over values defined on the client or passed to this method.
251215
extra_headers: Headers | None = None,
@@ -266,15 +230,6 @@ async def delete(
266230
if not agent_id:
267231
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
268232
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
269-
extra_headers = {
270-
**strip_not_given(
271-
{
272-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
273-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
274-
}
275-
),
276-
**(extra_headers or {}),
277-
}
278233
return await self._delete(
279234
f"/v1/agents/{agent_id}",
280235
options=make_request_options(

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

Lines changed: 0 additions & 67 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
1010
from ..._utils import (
1111
maybe_transform,
12-
strip_not_given,
1312
async_maybe_transform,
1413
)
1514
from ..._compat import cached_property
@@ -53,8 +52,6 @@ def create(
5352
agent_id: str,
5453
*,
5554
session_name: str,
56-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
57-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
5855
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
5956
# The extra values given here take precedence over values defined on the client or passed to this method.
6057
extra_headers: Headers | None = None,
@@ -74,15 +71,6 @@ def create(
7471
"""
7572
if not agent_id:
7673
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
77-
extra_headers = {
78-
**strip_not_given(
79-
{
80-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
81-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
82-
}
83-
),
84-
**(extra_headers or {}),
85-
}
8674
return self._post(
8775
f"/v1/agents/{agent_id}/session",
8876
body=maybe_transform({"session_name": session_name}, session_create_params.SessionCreateParams),
@@ -98,8 +86,6 @@ def retrieve(
9886
*,
9987
agent_id: str,
10088
turn_ids: List[str] | NotGiven = NOT_GIVEN,
101-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
102-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
10389
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
10490
# The extra values given here take precedence over values defined on the client or passed to this method.
10591
extra_headers: Headers | None = None,
@@ -121,15 +107,6 @@ def retrieve(
121107
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
122108
if not session_id:
123109
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
124-
extra_headers = {
125-
**strip_not_given(
126-
{
127-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
128-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
129-
}
130-
),
131-
**(extra_headers or {}),
132-
}
133110
return self._get(
134111
f"/v1/agents/{agent_id}/session/{session_id}",
135112
options=make_request_options(
@@ -147,8 +124,6 @@ def delete(
147124
session_id: str,
148125
*,
149126
agent_id: str,
150-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
151-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
152127
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
153128
# The extra values given here take precedence over values defined on the client or passed to this method.
154129
extra_headers: Headers | None = None,
@@ -171,15 +146,6 @@ def delete(
171146
if not session_id:
172147
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
173148
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
174-
extra_headers = {
175-
**strip_not_given(
176-
{
177-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
178-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
179-
}
180-
),
181-
**(extra_headers or {}),
182-
}
183149
return self._delete(
184150
f"/v1/agents/{agent_id}/session/{session_id}",
185151
options=make_request_options(
@@ -214,8 +180,6 @@ async def create(
214180
agent_id: str,
215181
*,
216182
session_name: str,
217-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
218-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
219183
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
220184
# The extra values given here take precedence over values defined on the client or passed to this method.
221185
extra_headers: Headers | None = None,
@@ -235,15 +199,6 @@ async def create(
235199
"""
236200
if not agent_id:
237201
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
238-
extra_headers = {
239-
**strip_not_given(
240-
{
241-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
242-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
243-
}
244-
),
245-
**(extra_headers or {}),
246-
}
247202
return await self._post(
248203
f"/v1/agents/{agent_id}/session",
249204
body=await async_maybe_transform({"session_name": session_name}, session_create_params.SessionCreateParams),
@@ -259,8 +214,6 @@ async def retrieve(
259214
*,
260215
agent_id: str,
261216
turn_ids: List[str] | NotGiven = NOT_GIVEN,
262-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
263-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
264217
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
265218
# The extra values given here take precedence over values defined on the client or passed to this method.
266219
extra_headers: Headers | None = None,
@@ -282,15 +235,6 @@ async def retrieve(
282235
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
283236
if not session_id:
284237
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
285-
extra_headers = {
286-
**strip_not_given(
287-
{
288-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
289-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
290-
}
291-
),
292-
**(extra_headers or {}),
293-
}
294238
return await self._get(
295239
f"/v1/agents/{agent_id}/session/{session_id}",
296240
options=make_request_options(
@@ -310,8 +254,6 @@ async def delete(
310254
session_id: str,
311255
*,
312256
agent_id: str,
313-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
314-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
315257
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
316258
# The extra values given here take precedence over values defined on the client or passed to this method.
317259
extra_headers: Headers | None = None,
@@ -334,15 +276,6 @@ async def delete(
334276
if not session_id:
335277
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
336278
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
337-
extra_headers = {
338-
**strip_not_given(
339-
{
340-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
341-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
342-
}
343-
),
344-
**(extra_headers or {}),
345-
}
346279
return await self._delete(
347280
f"/v1/agents/{agent_id}/session/{session_id}",
348281
options=make_request_options(

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

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
import httpx
66

77
from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
8-
from ..._utils import strip_not_given
98
from ..._compat import cached_property
109
from ..._resource import SyncAPIResource, AsyncAPIResource
1110
from ..._response import (
@@ -47,8 +46,6 @@ def retrieve(
4746
agent_id: str,
4847
session_id: str,
4948
turn_id: str,
50-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
51-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
5249
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
5350
# The extra values given here take precedence over values defined on the client or passed to this method.
5451
extra_headers: Headers | None = None,
@@ -74,15 +71,6 @@ def retrieve(
7471
raise ValueError(f"Expected a non-empty value for `turn_id` but received {turn_id!r}")
7572
if not step_id:
7673
raise ValueError(f"Expected a non-empty value for `step_id` but received {step_id!r}")
77-
extra_headers = {
78-
**strip_not_given(
79-
{
80-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
81-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
82-
}
83-
),
84-
**(extra_headers or {}),
85-
}
8674
return self._get(
8775
f"/v1/agents/{agent_id}/session/{session_id}/turn/{turn_id}/step/{step_id}",
8876
options=make_request_options(
@@ -119,8 +107,6 @@ async def retrieve(
119107
agent_id: str,
120108
session_id: str,
121109
turn_id: str,
122-
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
123-
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
124110
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
125111
# The extra values given here take precedence over values defined on the client or passed to this method.
126112
extra_headers: Headers | None = None,
@@ -146,15 +132,6 @@ async def retrieve(
146132
raise ValueError(f"Expected a non-empty value for `turn_id` but received {turn_id!r}")
147133
if not step_id:
148134
raise ValueError(f"Expected a non-empty value for `step_id` but received {step_id!r}")
149-
extra_headers = {
150-
**strip_not_given(
151-
{
152-
"X-LlamaStack-Client-Version": x_llama_stack_client_version,
153-
"X-LlamaStack-Provider-Data": x_llama_stack_provider_data,
154-
}
155-
),
156-
**(extra_headers or {}),
157-
}
158135
return await self._get(
159136
f"/v1/agents/{agent_id}/session/{session_id}/turn/{turn_id}/step/{step_id}",
160137
options=make_request_options(

0 commit comments

Comments
 (0)