Skip to content

Commit d5cef5a

Browse files
feat(api): deprecate toolgroup and tool_runtime apis
1 parent a83a98e commit d5cef5a

File tree

7 files changed

+348
-218
lines changed

7 files changed

+348
-218
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: 103
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-4ea81517e66f9dffe1f552b1d56fd745fd24dbf5523d2306fee7e19b96a86885.yml
3-
openapi_spec_hash: a30ec4b7a5399cb164fb818152a41b58
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-c53a119bf22adab585059b5426b8e4a5d9582783bce071ad600c22ab0ebc6578.yml
3+
openapi_spec_hash: c55f186ff8688e7ccca399cc7b4a87a0
44
config_hash: 39578cfdeb4a10121f2cb3fa3e4d5e20

src/llama_stack_client/resources/tool_runtime.py

Lines changed: 37 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99
from __future__ import annotations
1010

11+
import typing_extensions
1112
from typing import Dict, Type, Optional, cast
1213

1314
import httpx
@@ -51,6 +52,7 @@ def with_streaming_response(self) -> ToolRuntimeResourceWithStreamingResponse:
5152
"""
5253
return ToolRuntimeResourceWithStreamingResponse(self)
5354

55+
@typing_extensions.deprecated("deprecated")
5456
def invoke_tool(
5557
self,
5658
*,
@@ -92,6 +94,7 @@ def invoke_tool(
9294
cast_to=ToolInvocationResult,
9395
)
9496

97+
@typing_extensions.deprecated("deprecated")
9598
def list_tools(
9699
self,
97100
*,
@@ -160,6 +163,7 @@ def with_streaming_response(self) -> AsyncToolRuntimeResourceWithStreamingRespon
160163
"""
161164
return AsyncToolRuntimeResourceWithStreamingResponse(self)
162165

166+
@typing_extensions.deprecated("deprecated")
163167
async def invoke_tool(
164168
self,
165169
*,
@@ -201,6 +205,7 @@ async def invoke_tool(
201205
cast_to=ToolInvocationResult,
202206
)
203207

208+
@typing_extensions.deprecated("deprecated")
204209
async def list_tools(
205210
self,
206211
*,
@@ -253,45 +258,61 @@ class ToolRuntimeResourceWithRawResponse:
253258
def __init__(self, tool_runtime: ToolRuntimeResource) -> None:
254259
self._tool_runtime = tool_runtime
255260

256-
self.invoke_tool = to_raw_response_wrapper(
257-
tool_runtime.invoke_tool,
261+
self.invoke_tool = ( # pyright: ignore[reportDeprecated]
262+
to_raw_response_wrapper(
263+
tool_runtime.invoke_tool, # pyright: ignore[reportDeprecated],
264+
)
258265
)
259-
self.list_tools = to_raw_response_wrapper(
260-
tool_runtime.list_tools,
266+
self.list_tools = ( # pyright: ignore[reportDeprecated]
267+
to_raw_response_wrapper(
268+
tool_runtime.list_tools, # pyright: ignore[reportDeprecated],
269+
)
261270
)
262271

263272

264273
class AsyncToolRuntimeResourceWithRawResponse:
265274
def __init__(self, tool_runtime: AsyncToolRuntimeResource) -> None:
266275
self._tool_runtime = tool_runtime
267276

268-
self.invoke_tool = async_to_raw_response_wrapper(
269-
tool_runtime.invoke_tool,
277+
self.invoke_tool = ( # pyright: ignore[reportDeprecated]
278+
async_to_raw_response_wrapper(
279+
tool_runtime.invoke_tool, # pyright: ignore[reportDeprecated],
280+
)
270281
)
271-
self.list_tools = async_to_raw_response_wrapper(
272-
tool_runtime.list_tools,
282+
self.list_tools = ( # pyright: ignore[reportDeprecated]
283+
async_to_raw_response_wrapper(
284+
tool_runtime.list_tools, # pyright: ignore[reportDeprecated],
285+
)
273286
)
274287

275288

276289
class ToolRuntimeResourceWithStreamingResponse:
277290
def __init__(self, tool_runtime: ToolRuntimeResource) -> None:
278291
self._tool_runtime = tool_runtime
279292

280-
self.invoke_tool = to_streamed_response_wrapper(
281-
tool_runtime.invoke_tool,
293+
self.invoke_tool = ( # pyright: ignore[reportDeprecated]
294+
to_streamed_response_wrapper(
295+
tool_runtime.invoke_tool, # pyright: ignore[reportDeprecated],
296+
)
282297
)
283-
self.list_tools = to_streamed_response_wrapper(
284-
tool_runtime.list_tools,
298+
self.list_tools = ( # pyright: ignore[reportDeprecated]
299+
to_streamed_response_wrapper(
300+
tool_runtime.list_tools, # pyright: ignore[reportDeprecated],
301+
)
285302
)
286303

287304

288305
class AsyncToolRuntimeResourceWithStreamingResponse:
289306
def __init__(self, tool_runtime: AsyncToolRuntimeResource) -> None:
290307
self._tool_runtime = tool_runtime
291308

292-
self.invoke_tool = async_to_streamed_response_wrapper(
293-
tool_runtime.invoke_tool,
309+
self.invoke_tool = ( # pyright: ignore[reportDeprecated]
310+
async_to_streamed_response_wrapper(
311+
tool_runtime.invoke_tool, # pyright: ignore[reportDeprecated],
312+
)
294313
)
295-
self.list_tools = async_to_streamed_response_wrapper(
296-
tool_runtime.list_tools,
314+
self.list_tools = ( # pyright: ignore[reportDeprecated]
315+
async_to_streamed_response_wrapper(
316+
tool_runtime.list_tools, # pyright: ignore[reportDeprecated],
317+
)
297318
)

src/llama_stack_client/resources/toolgroups.py

Lines changed: 36 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ def with_streaming_response(self) -> ToolgroupsResourceWithStreamingResponse:
5252
"""
5353
return ToolgroupsResourceWithStreamingResponse(self)
5454

55+
@typing_extensions.deprecated("deprecated")
5556
def list(
5657
self,
5758
*,
@@ -75,6 +76,7 @@ def list(
7576
cast_to=cast(Type[ToolgroupListResponse], DataWrapper[ToolgroupListResponse]),
7677
)
7778

79+
@typing_extensions.deprecated("deprecated")
7880
def get(
7981
self,
8082
toolgroup_id: str,
@@ -211,6 +213,7 @@ def with_streaming_response(self) -> AsyncToolgroupsResourceWithStreamingRespons
211213
"""
212214
return AsyncToolgroupsResourceWithStreamingResponse(self)
213215

216+
@typing_extensions.deprecated("deprecated")
214217
async def list(
215218
self,
216219
*,
@@ -234,6 +237,7 @@ async def list(
234237
cast_to=cast(Type[ToolgroupListResponse], DataWrapper[ToolgroupListResponse]),
235238
)
236239

240+
@typing_extensions.deprecated("deprecated")
237241
async def get(
238242
self,
239243
toolgroup_id: str,
@@ -354,11 +358,15 @@ class ToolgroupsResourceWithRawResponse:
354358
def __init__(self, toolgroups: ToolgroupsResource) -> None:
355359
self._toolgroups = toolgroups
356360

357-
self.list = to_raw_response_wrapper(
358-
toolgroups.list,
361+
self.list = ( # pyright: ignore[reportDeprecated]
362+
to_raw_response_wrapper(
363+
toolgroups.list, # pyright: ignore[reportDeprecated],
364+
)
359365
)
360-
self.get = to_raw_response_wrapper(
361-
toolgroups.get,
366+
self.get = ( # pyright: ignore[reportDeprecated]
367+
to_raw_response_wrapper(
368+
toolgroups.get, # pyright: ignore[reportDeprecated],
369+
)
362370
)
363371
self.register = ( # pyright: ignore[reportDeprecated]
364372
to_raw_response_wrapper(
@@ -376,11 +384,15 @@ class AsyncToolgroupsResourceWithRawResponse:
376384
def __init__(self, toolgroups: AsyncToolgroupsResource) -> None:
377385
self._toolgroups = toolgroups
378386

379-
self.list = async_to_raw_response_wrapper(
380-
toolgroups.list,
387+
self.list = ( # pyright: ignore[reportDeprecated]
388+
async_to_raw_response_wrapper(
389+
toolgroups.list, # pyright: ignore[reportDeprecated],
390+
)
381391
)
382-
self.get = async_to_raw_response_wrapper(
383-
toolgroups.get,
392+
self.get = ( # pyright: ignore[reportDeprecated]
393+
async_to_raw_response_wrapper(
394+
toolgroups.get, # pyright: ignore[reportDeprecated],
395+
)
384396
)
385397
self.register = ( # pyright: ignore[reportDeprecated]
386398
async_to_raw_response_wrapper(
@@ -398,11 +410,15 @@ class ToolgroupsResourceWithStreamingResponse:
398410
def __init__(self, toolgroups: ToolgroupsResource) -> None:
399411
self._toolgroups = toolgroups
400412

401-
self.list = to_streamed_response_wrapper(
402-
toolgroups.list,
413+
self.list = ( # pyright: ignore[reportDeprecated]
414+
to_streamed_response_wrapper(
415+
toolgroups.list, # pyright: ignore[reportDeprecated],
416+
)
403417
)
404-
self.get = to_streamed_response_wrapper(
405-
toolgroups.get,
418+
self.get = ( # pyright: ignore[reportDeprecated]
419+
to_streamed_response_wrapper(
420+
toolgroups.get, # pyright: ignore[reportDeprecated],
421+
)
406422
)
407423
self.register = ( # pyright: ignore[reportDeprecated]
408424
to_streamed_response_wrapper(
@@ -420,11 +436,15 @@ class AsyncToolgroupsResourceWithStreamingResponse:
420436
def __init__(self, toolgroups: AsyncToolgroupsResource) -> None:
421437
self._toolgroups = toolgroups
422438

423-
self.list = async_to_streamed_response_wrapper(
424-
toolgroups.list,
439+
self.list = ( # pyright: ignore[reportDeprecated]
440+
async_to_streamed_response_wrapper(
441+
toolgroups.list, # pyright: ignore[reportDeprecated],
442+
)
425443
)
426-
self.get = async_to_streamed_response_wrapper(
427-
toolgroups.get,
444+
self.get = ( # pyright: ignore[reportDeprecated]
445+
async_to_streamed_response_wrapper(
446+
toolgroups.get, # pyright: ignore[reportDeprecated],
447+
)
428448
)
429449
self.register = ( # pyright: ignore[reportDeprecated]
430450
async_to_streamed_response_wrapper(

src/llama_stack_client/resources/tools.py

Lines changed: 37 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99
from __future__ import annotations
1010

11+
import typing_extensions
1112
from typing import Type, Optional, cast
1213

1314
import httpx
@@ -51,6 +52,7 @@ def with_streaming_response(self) -> ToolsResourceWithStreamingResponse:
5152
"""
5253
return ToolsResourceWithStreamingResponse(self)
5354

55+
@typing_extensions.deprecated("deprecated")
5456
def list(
5557
self,
5658
*,
@@ -87,6 +89,7 @@ def list(
8789
cast_to=cast(Type[ToolListResponse], DataWrapper[ToolListResponse]),
8890
)
8991

92+
@typing_extensions.deprecated("deprecated")
9093
def get(
9194
self,
9295
tool_name: str,
@@ -141,6 +144,7 @@ def with_streaming_response(self) -> AsyncToolsResourceWithStreamingResponse:
141144
"""
142145
return AsyncToolsResourceWithStreamingResponse(self)
143146

147+
@typing_extensions.deprecated("deprecated")
144148
async def list(
145149
self,
146150
*,
@@ -177,6 +181,7 @@ async def list(
177181
cast_to=cast(Type[ToolListResponse], DataWrapper[ToolListResponse]),
178182
)
179183

184+
@typing_extensions.deprecated("deprecated")
180185
async def get(
181186
self,
182187
tool_name: str,
@@ -215,45 +220,61 @@ class ToolsResourceWithRawResponse:
215220
def __init__(self, tools: ToolsResource) -> None:
216221
self._tools = tools
217222

218-
self.list = to_raw_response_wrapper(
219-
tools.list,
223+
self.list = ( # pyright: ignore[reportDeprecated]
224+
to_raw_response_wrapper(
225+
tools.list, # pyright: ignore[reportDeprecated],
226+
)
220227
)
221-
self.get = to_raw_response_wrapper(
222-
tools.get,
228+
self.get = ( # pyright: ignore[reportDeprecated]
229+
to_raw_response_wrapper(
230+
tools.get, # pyright: ignore[reportDeprecated],
231+
)
223232
)
224233

225234

226235
class AsyncToolsResourceWithRawResponse:
227236
def __init__(self, tools: AsyncToolsResource) -> None:
228237
self._tools = tools
229238

230-
self.list = async_to_raw_response_wrapper(
231-
tools.list,
239+
self.list = ( # pyright: ignore[reportDeprecated]
240+
async_to_raw_response_wrapper(
241+
tools.list, # pyright: ignore[reportDeprecated],
242+
)
232243
)
233-
self.get = async_to_raw_response_wrapper(
234-
tools.get,
244+
self.get = ( # pyright: ignore[reportDeprecated]
245+
async_to_raw_response_wrapper(
246+
tools.get, # pyright: ignore[reportDeprecated],
247+
)
235248
)
236249

237250

238251
class ToolsResourceWithStreamingResponse:
239252
def __init__(self, tools: ToolsResource) -> None:
240253
self._tools = tools
241254

242-
self.list = to_streamed_response_wrapper(
243-
tools.list,
255+
self.list = ( # pyright: ignore[reportDeprecated]
256+
to_streamed_response_wrapper(
257+
tools.list, # pyright: ignore[reportDeprecated],
258+
)
244259
)
245-
self.get = to_streamed_response_wrapper(
246-
tools.get,
260+
self.get = ( # pyright: ignore[reportDeprecated]
261+
to_streamed_response_wrapper(
262+
tools.get, # pyright: ignore[reportDeprecated],
263+
)
247264
)
248265

249266

250267
class AsyncToolsResourceWithStreamingResponse:
251268
def __init__(self, tools: AsyncToolsResource) -> None:
252269
self._tools = tools
253270

254-
self.list = async_to_streamed_response_wrapper(
255-
tools.list,
271+
self.list = ( # pyright: ignore[reportDeprecated]
272+
async_to_streamed_response_wrapper(
273+
tools.list, # pyright: ignore[reportDeprecated],
274+
)
256275
)
257-
self.get = async_to_streamed_response_wrapper(
258-
tools.get,
276+
self.get = ( # pyright: ignore[reportDeprecated]
277+
async_to_streamed_response_wrapper(
278+
tools.get, # pyright: ignore[reportDeprecated],
279+
)
259280
)

0 commit comments

Comments
 (0)