Skip to content

Commit fb0e25a

Browse files
dineshyvdineshyv
committed
Sync updates from stainless branch: dineshyv/dev
1 parent 2e7de49 commit fb0e25a

Some content is hidden

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

61 files changed

+1449
-1219
lines changed

‎src/llama_stack_client/_client.py‎

Lines changed: 214 additions & 92 deletions
Large diffs are not rendered by default.

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

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
StepsResourceWithStreamingResponse,
2121
AsyncStepsResourceWithStreamingResponse,
2222
)
23-
from ...types import agent_create_params, agent_delete_params
23+
from ...types import agent_create_params
2424
from .session import (
2525
SessionResource,
2626
AsyncSessionResource,
@@ -115,7 +115,7 @@ def create(
115115
**(extra_headers or {}),
116116
}
117117
return self._post(
118-
"/v1/agents/create",
118+
"/v1/agents",
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
@@ -125,8 +125,8 @@ def create(
125125

126126
def delete(
127127
self,
128-
*,
129128
agent_id: str,
129+
*,
130130
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
131131
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
132132
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -146,6 +146,8 @@ def delete(
146146
147147
timeout: Override the client-level default timeout for this request, in seconds
148148
"""
149+
if not agent_id:
150+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
149151
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
150152
extra_headers = {
151153
**strip_not_given(
@@ -156,9 +158,8 @@ def delete(
156158
),
157159
**(extra_headers or {}),
158160
}
159-
return self._post(
160-
"/v1/agents/delete",
161-
body=maybe_transform({"agent_id": agent_id}, agent_delete_params.AgentDeleteParams),
161+
return self._delete(
162+
f"/v1/agents/{agent_id}",
162163
options=make_request_options(
163164
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
164165
),
@@ -231,7 +232,7 @@ async def create(
231232
**(extra_headers or {}),
232233
}
233234
return await self._post(
234-
"/v1/agents/create",
235+
"/v1/agents",
235236
body=await async_maybe_transform({"agent_config": agent_config}, agent_create_params.AgentCreateParams),
236237
options=make_request_options(
237238
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -241,8 +242,8 @@ async def create(
241242

242243
async def delete(
243244
self,
244-
*,
245245
agent_id: str,
246+
*,
246247
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
247248
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
248249
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -262,6 +263,8 @@ async def delete(
262263
263264
timeout: Override the client-level default timeout for this request, in seconds
264265
"""
266+
if not agent_id:
267+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
265268
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
266269
extra_headers = {
267270
**strip_not_given(
@@ -272,9 +275,8 @@ async def delete(
272275
),
273276
**(extra_headers or {}),
274277
}
275-
return await self._post(
276-
"/v1/agents/delete",
277-
body=await async_maybe_transform({"agent_id": agent_id}, agent_delete_params.AgentDeleteParams),
278+
return await self._delete(
279+
f"/v1/agents/{agent_id}",
278280
options=make_request_options(
279281
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
280282
),

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

Lines changed: 41 additions & 59 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
async_to_streamed_response_wrapper,
2222
)
2323
from ..._base_client import make_request_options
24-
from ...types.agents import session_create_params, session_delete_params, session_retrieve_params
24+
from ...types.agents import session_create_params, session_retrieve_params
2525
from ...types.agents.session import Session
2626
from ...types.agents.session_create_response import SessionCreateResponse
2727

@@ -50,8 +50,8 @@ def with_streaming_response(self) -> SessionResourceWithStreamingResponse:
5050

5151
def create(
5252
self,
53-
*,
5453
agent_id: str,
54+
*,
5555
session_name: str,
5656
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
5757
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
@@ -72,6 +72,8 @@ def create(
7272
7373
timeout: Override the client-level default timeout for this request, in seconds
7474
"""
75+
if not agent_id:
76+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
7577
extra_headers = {
7678
**strip_not_given(
7779
{
@@ -82,14 +84,8 @@ def create(
8284
**(extra_headers or {}),
8385
}
8486
return self._post(
85-
"/v1/agents/session/create",
86-
body=maybe_transform(
87-
{
88-
"agent_id": agent_id,
89-
"session_name": session_name,
90-
},
91-
session_create_params.SessionCreateParams,
92-
),
87+
f"/v1/agents/{agent_id}/session",
88+
body=maybe_transform({"session_name": session_name}, session_create_params.SessionCreateParams),
9389
options=make_request_options(
9490
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
9591
),
@@ -98,9 +94,9 @@ def create(
9894

9995
def retrieve(
10096
self,
97+
session_id: str,
10198
*,
10299
agent_id: str,
103-
session_id: str,
104100
turn_ids: List[str] | NotGiven = NOT_GIVEN,
105101
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
106102
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
@@ -121,6 +117,10 @@ def retrieve(
121117
122118
timeout: Override the client-level default timeout for this request, in seconds
123119
"""
120+
if not agent_id:
121+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
122+
if not session_id:
123+
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
124124
extra_headers = {
125125
**strip_not_given(
126126
{
@@ -130,30 +130,23 @@ def retrieve(
130130
),
131131
**(extra_headers or {}),
132132
}
133-
return self._post(
134-
"/v1/agents/session/get",
135-
body=maybe_transform({"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
133+
return self._get(
134+
f"/v1/agents/{agent_id}/session/{session_id}",
136135
options=make_request_options(
137136
extra_headers=extra_headers,
138137
extra_query=extra_query,
139138
extra_body=extra_body,
140139
timeout=timeout,
141-
query=maybe_transform(
142-
{
143-
"agent_id": agent_id,
144-
"session_id": session_id,
145-
},
146-
session_retrieve_params.SessionRetrieveParams,
147-
),
140+
query=maybe_transform({"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
148141
),
149142
cast_to=Session,
150143
)
151144

152145
def delete(
153146
self,
147+
session_id: str,
154148
*,
155149
agent_id: str,
156-
session_id: str,
157150
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
158151
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
159152
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -173,6 +166,10 @@ def delete(
173166
174167
timeout: Override the client-level default timeout for this request, in seconds
175168
"""
169+
if not agent_id:
170+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
171+
if not session_id:
172+
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
176173
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
177174
extra_headers = {
178175
**strip_not_given(
@@ -183,15 +180,8 @@ def delete(
183180
),
184181
**(extra_headers or {}),
185182
}
186-
return self._post(
187-
"/v1/agents/session/delete",
188-
body=maybe_transform(
189-
{
190-
"agent_id": agent_id,
191-
"session_id": session_id,
192-
},
193-
session_delete_params.SessionDeleteParams,
194-
),
183+
return self._delete(
184+
f"/v1/agents/{agent_id}/session/{session_id}",
195185
options=make_request_options(
196186
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
197187
),
@@ -221,8 +211,8 @@ def with_streaming_response(self) -> AsyncSessionResourceWithStreamingResponse:
221211

222212
async def create(
223213
self,
224-
*,
225214
agent_id: str,
215+
*,
226216
session_name: str,
227217
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
228218
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
@@ -243,6 +233,8 @@ async def create(
243233
244234
timeout: Override the client-level default timeout for this request, in seconds
245235
"""
236+
if not agent_id:
237+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
246238
extra_headers = {
247239
**strip_not_given(
248240
{
@@ -253,14 +245,8 @@ async def create(
253245
**(extra_headers or {}),
254246
}
255247
return await self._post(
256-
"/v1/agents/session/create",
257-
body=await async_maybe_transform(
258-
{
259-
"agent_id": agent_id,
260-
"session_name": session_name,
261-
},
262-
session_create_params.SessionCreateParams,
263-
),
248+
f"/v1/agents/{agent_id}/session",
249+
body=await async_maybe_transform({"session_name": session_name}, session_create_params.SessionCreateParams),
264250
options=make_request_options(
265251
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
266252
),
@@ -269,9 +255,9 @@ async def create(
269255

270256
async def retrieve(
271257
self,
258+
session_id: str,
272259
*,
273260
agent_id: str,
274-
session_id: str,
275261
turn_ids: List[str] | NotGiven = NOT_GIVEN,
276262
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
277263
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
@@ -292,6 +278,10 @@ async def retrieve(
292278
293279
timeout: Override the client-level default timeout for this request, in seconds
294280
"""
281+
if not agent_id:
282+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
283+
if not session_id:
284+
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
295285
extra_headers = {
296286
**strip_not_given(
297287
{
@@ -301,30 +291,25 @@ async def retrieve(
301291
),
302292
**(extra_headers or {}),
303293
}
304-
return await self._post(
305-
"/v1/agents/session/get",
306-
body=await async_maybe_transform({"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
294+
return await self._get(
295+
f"/v1/agents/{agent_id}/session/{session_id}",
307296
options=make_request_options(
308297
extra_headers=extra_headers,
309298
extra_query=extra_query,
310299
extra_body=extra_body,
311300
timeout=timeout,
312301
query=await async_maybe_transform(
313-
{
314-
"agent_id": agent_id,
315-
"session_id": session_id,
316-
},
317-
session_retrieve_params.SessionRetrieveParams,
302+
{"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams
318303
),
319304
),
320305
cast_to=Session,
321306
)
322307

323308
async def delete(
324309
self,
310+
session_id: str,
325311
*,
326312
agent_id: str,
327-
session_id: str,
328313
x_llama_stack_client_version: str | NotGiven = NOT_GIVEN,
329314
x_llama_stack_provider_data: str | NotGiven = NOT_GIVEN,
330315
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -344,6 +329,10 @@ async def delete(
344329
345330
timeout: Override the client-level default timeout for this request, in seconds
346331
"""
332+
if not agent_id:
333+
raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
334+
if not session_id:
335+
raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
347336
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
348337
extra_headers = {
349338
**strip_not_given(
@@ -354,15 +343,8 @@ async def delete(
354343
),
355344
**(extra_headers or {}),
356345
}
357-
return await self._post(
358-
"/v1/agents/session/delete",
359-
body=await async_maybe_transform(
360-
{
361-
"agent_id": agent_id,
362-
"session_id": session_id,
363-
},
364-
session_delete_params.SessionDeleteParams,
365-
),
346+
return await self._delete(
347+
f"/v1/agents/{agent_id}/session/{session_id}",
366348
options=make_request_options(
367349
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
368350
),

0 commit comments

Comments
 (0)