letta-client 0.1.17__py3-none-any.whl → 0.1.21__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of letta-client might be problematic. Click here for more details.

Files changed (107) hide show
  1. letta_client/__init__.py +44 -116
  2. letta_client/agents/__init__.py +18 -42
  3. letta_client/agents/archival_memory/client.py +6 -6
  4. letta_client/agents/client.py +28 -18
  5. letta_client/agents/context/client.py +6 -4
  6. letta_client/agents/core_memory/client.py +152 -227
  7. letta_client/agents/memory_variables/__init__.py +2 -2
  8. letta_client/agents/memory_variables/client.py +15 -15
  9. letta_client/agents/memory_variables/types/__init__.py +2 -2
  10. letta_client/agents/memory_variables/types/{memory_variables_get_response.py → memory_variables_list_response.py} +1 -1
  11. letta_client/agents/messages/__init__.py +2 -22
  12. letta_client/agents/messages/client.py +39 -40
  13. letta_client/agents/messages/types/__init__.py +3 -21
  14. letta_client/agents/messages/types/letta_streaming_response.py +16 -139
  15. letta_client/agents/messages/types/message_update_content.py +6 -0
  16. letta_client/agents/messages/types/messages_list_response.py +2 -2
  17. letta_client/agents/sources/client.py +266 -5
  18. letta_client/agents/tools/client.py +25 -27
  19. letta_client/agents/types/__init__.py +15 -25
  20. letta_client/agents/types/agents_search_request_search_item.py +10 -78
  21. letta_client/agents/types/{agents_search_request_search_item_order_by.py → agents_search_request_search_item_direction.py} +7 -6
  22. letta_client/agents/types/agents_search_request_search_item_direction_direction.py +5 -0
  23. letta_client/agents/types/agents_search_request_search_item_direction_value.py +5 -0
  24. letta_client/agents/types/{agents_search_request_search_item_name.py → agents_search_request_search_item_one.py} +5 -4
  25. letta_client/agents/types/agents_search_request_search_item_one_operator.py +5 -0
  26. letta_client/agents/types/{agents_search_request_search_item_tags.py → agents_search_request_search_item_two.py} +2 -1
  27. letta_client/agents/types/{agents_search_request_search_item_version.py → agents_search_request_search_item_zero.py} +3 -2
  28. letta_client/blocks/client.py +45 -166
  29. letta_client/client.py +3 -3
  30. letta_client/core/client_wrapper.py +1 -1
  31. letta_client/jobs/client.py +4 -4
  32. letta_client/providers/client.py +10 -10
  33. letta_client/runs/client.py +44 -30
  34. letta_client/sources/client.py +12 -288
  35. letta_client/sources/files/client.py +6 -6
  36. letta_client/tag/client.py +6 -6
  37. letta_client/tools/client.py +69 -75
  38. letta_client/types/__init__.py +31 -99
  39. letta_client/types/agent_state.py +3 -7
  40. letta_client/types/{letta_schemas_openai_chat_completion_response_tool_call.py → assistant_message.py} +6 -4
  41. letta_client/types/assistant_message_content.py +6 -0
  42. letta_client/types/block.py +2 -6
  43. letta_client/types/block_update.py +1 -5
  44. letta_client/types/{assistant_message_output.py → chat_completion_message_tool_call.py} +5 -5
  45. letta_client/types/context_window_overview.py +4 -6
  46. letta_client/types/create_block.py +1 -5
  47. letta_client/types/{function_call_output.py → function.py} +1 -1
  48. letta_client/types/{function_schema.py → function_definition.py} +2 -1
  49. letta_client/types/{function_call_input.py → function_tool.py} +5 -3
  50. letta_client/types/job.py +1 -5
  51. letta_client/types/letta_message_union.py +9 -121
  52. letta_client/types/letta_usage_statistics.py +1 -0
  53. letta_client/types/llm_config.py +6 -0
  54. letta_client/types/{letta_schemas_message_message.py → message.py} +7 -8
  55. letta_client/types/message_create.py +3 -2
  56. letta_client/types/message_create_content.py +6 -0
  57. letta_client/types/passage.py +1 -5
  58. letta_client/types/reasoning_message.py +2 -1
  59. letta_client/types/run.py +1 -5
  60. letta_client/types/source.py +2 -6
  61. letta_client/types/{system_message_output.py → system_message.py} +6 -4
  62. letta_client/types/system_message_content.py +6 -0
  63. letta_client/types/{create_assistant_file_request.py → text_content.py} +5 -4
  64. letta_client/types/{letta_schemas_tool_tool.py → tool.py} +1 -1
  65. letta_client/types/{letta_schemas_letta_message_tool_call.py → tool_call.py} +1 -1
  66. letta_client/types/tool_call_message.py +2 -1
  67. letta_client/types/tool_call_message_tool_call.py +2 -2
  68. letta_client/types/tool_return_message.py +2 -1
  69. letta_client/types/tool_type.py +2 -1
  70. letta_client/types/{user_message_output.py → user_message.py} +6 -4
  71. letta_client/types/user_message_content.py +6 -0
  72. {letta_client-0.1.17.dist-info → letta_client-0.1.21.dist-info}/METADATA +2 -2
  73. {letta_client-0.1.17.dist-info → letta_client-0.1.21.dist-info}/RECORD +74 -99
  74. letta_client/agents/types/agents_search_request_search_item_name_operator.py +0 -5
  75. letta_client/agents/types/agents_search_request_search_item_order_by_direction.py +0 -5
  76. letta_client/agents/types/agents_search_request_search_item_order_by_value.py +0 -5
  77. letta_client/types/assistant_file.py +0 -33
  78. letta_client/types/assistant_message_input.py +0 -23
  79. letta_client/types/chat_completion_request.py +0 -49
  80. letta_client/types/chat_completion_request_function_call.py +0 -6
  81. letta_client/types/chat_completion_request_messages_item.py +0 -11
  82. letta_client/types/chat_completion_request_stop.py +0 -5
  83. letta_client/types/chat_completion_request_tool_choice.py +0 -8
  84. letta_client/types/chat_completion_response.py +0 -32
  85. letta_client/types/choice.py +0 -25
  86. letta_client/types/create_assistant_request.py +0 -57
  87. letta_client/types/delete_assistant_file_response.py +0 -28
  88. letta_client/types/delete_assistant_response.py +0 -28
  89. letta_client/types/letta_schemas_openai_chat_completion_request_tool.py +0 -21
  90. letta_client/types/letta_schemas_openai_chat_completion_request_tool_call.py +0 -24
  91. letta_client/types/letta_schemas_openai_chat_completion_request_tool_call_function.py +0 -20
  92. letta_client/types/letta_schemas_openai_chat_completion_response_message.py +0 -24
  93. letta_client/types/letta_schemas_openai_chat_completions_tool_call_function.py +0 -27
  94. letta_client/types/letta_schemas_openai_chat_completions_tool_call_input.py +0 -29
  95. letta_client/types/letta_schemas_openai_chat_completions_tool_call_output.py +0 -29
  96. letta_client/types/log_prob_token.py +0 -21
  97. letta_client/types/message_content_log_prob.py +0 -23
  98. letta_client/types/open_ai_assistant.py +0 -67
  99. letta_client/types/response_format.py +0 -19
  100. letta_client/types/system_message_input.py +0 -21
  101. letta_client/types/tool_call_function_output.py +0 -27
  102. letta_client/types/tool_function_choice.py +0 -21
  103. letta_client/types/tool_input.py +0 -21
  104. letta_client/types/tool_message.py +0 -21
  105. letta_client/types/user_message_input.py +0 -22
  106. letta_client/types/user_message_input_content.py +0 -5
  107. {letta_client-0.1.17.dist-info → letta_client-0.1.21.dist-info}/WHEEL +0 -0
letta_client/tag/client.py CHANGED
@@ -18,7 +18,7 @@ class TagClient:
18
18
  def list_tags(
19
19
  self,
20
20
  *,
21
- cursor: typing.Optional[str] = None,
21
+ after: typing.Optional[str] = None,
22
22
  limit: typing.Optional[int] = None,
23
23
  query_text: typing.Optional[str] = None,
24
24
  request_options: typing.Optional[RequestOptions] = None,
@@ -28,7 +28,7 @@ class TagClient:
28
28
 
29
29
  Parameters
30
30
  ----------
31
- cursor : typing.Optional[str]
31
+ after : typing.Optional[str]
32
32
 
33
33
  limit : typing.Optional[int]
34
34
 
@@ -55,7 +55,7 @@ class TagClient:
55
55
  "v1/tags/",
56
56
  method="GET",
57
57
  params={
58
- "cursor": cursor,
58
+ "after": after,
59
59
  "limit": limit,
60
60
  "query_text": query_text,
61
61
  },
@@ -93,7 +93,7 @@ class AsyncTagClient:
93
93
  async def list_tags(
94
94
  self,
95
95
  *,
96
- cursor: typing.Optional[str] = None,
96
+ after: typing.Optional[str] = None,
97
97
  limit: typing.Optional[int] = None,
98
98
  query_text: typing.Optional[str] = None,
99
99
  request_options: typing.Optional[RequestOptions] = None,
@@ -103,7 +103,7 @@ class AsyncTagClient:
103
103
 
104
104
  Parameters
105
105
  ----------
106
- cursor : typing.Optional[str]
106
+ after : typing.Optional[str]
107
107
 
108
108
  limit : typing.Optional[int]
109
109
 
@@ -138,7 +138,7 @@ class AsyncTagClient:
138
138
  "v1/tags/",
139
139
  method="GET",
140
140
  params={
141
- "cursor": cursor,
141
+ "after": after,
142
142
  "limit": limit,
143
143
  "query_text": query_text,
144
144
  },
letta_client/tools/client.py CHANGED
@@ -3,7 +3,7 @@
3
3
  import typing
4
4
  from ..core.client_wrapper import SyncClientWrapper
5
5
  from ..core.request_options import RequestOptions
6
- from ..types.letta_schemas_tool_tool import LettaSchemasToolTool
6
+ from ..types.tool import Tool
7
7
  from ..core.jsonable_encoder import jsonable_encoder
8
8
  from ..core.unchecked_base_model import construct_type
9
9
  from ..errors.unprocessable_entity_error import UnprocessableEntityError
@@ -23,7 +23,7 @@ class ToolsClient:
23
23
  def __init__(self, *, client_wrapper: SyncClientWrapper):
24
24
  self._client_wrapper = client_wrapper
25
25
 
26
- def get(self, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> LettaSchemasToolTool:
26
+ def retrieve(self, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Tool:
27
27
  """
28
28
  Get a tool by ID
29
29
 
@@ -36,7 +36,7 @@ class ToolsClient:
36
36
 
37
37
  Returns
38
38
  -------
39
- LettaSchemasToolTool
39
+ Tool
40
40
  Successful Response
41
41
 
42
42
  Examples
@@ -46,7 +46,7 @@ class ToolsClient:
46
46
  client = Letta(
47
47
  token="YOUR_TOKEN",
48
48
  )
49
- client.tools.get(
49
+ client.tools.retrieve(
50
50
  tool_id="tool_id",
51
51
  )
52
52
  """
@@ -58,9 +58,9 @@ class ToolsClient:
58
58
  try:
59
59
  if 200 <= _response.status_code < 300:
60
60
  return typing.cast(
61
- LettaSchemasToolTool,
61
+ Tool,
62
62
  construct_type(
63
- type_=LettaSchemasToolTool, # type: ignore
63
+ type_=Tool, # type: ignore
64
64
  object_=_response.json(),
65
65
  ),
66
66
  )
@@ -137,7 +137,7 @@ class ToolsClient:
137
137
  raise ApiError(status_code=_response.status_code, body=_response.text)
138
138
  raise ApiError(status_code=_response.status_code, body=_response_json)
139
139
 
140
- def update(
140
+ def modify(
141
141
  self,
142
142
  tool_id: str,
143
143
  *,
@@ -149,7 +149,7 @@ class ToolsClient:
149
149
  json_schema: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
150
150
  return_char_limit: typing.Optional[int] = OMIT,
151
151
  request_options: typing.Optional[RequestOptions] = None,
152
- ) -> LettaSchemasToolTool:
152
+ ) -> Tool:
153
153
  """
154
154
  Update an existing tool
155
155
 
@@ -183,7 +183,7 @@ class ToolsClient:
183
183
 
184
184
  Returns
185
185
  -------
186
- LettaSchemasToolTool
186
+ Tool
187
187
  Successful Response
188
188
 
189
189
  Examples
@@ -193,7 +193,7 @@ class ToolsClient:
193
193
  client = Letta(
194
194
  token="YOUR_TOKEN",
195
195
  )
196
- client.tools.update(
196
+ client.tools.modify(
197
197
  tool_id="tool_id",
198
198
  )
199
199
  """
@@ -218,9 +218,9 @@ class ToolsClient:
218
218
  try:
219
219
  if 200 <= _response.status_code < 300:
220
220
  return typing.cast(
221
- LettaSchemasToolTool,
221
+ Tool,
222
222
  construct_type(
223
- type_=LettaSchemasToolTool, # type: ignore
223
+ type_=Tool, # type: ignore
224
224
  object_=_response.json(),
225
225
  ),
226
226
  )
@@ -242,16 +242,16 @@ class ToolsClient:
242
242
  def list(
243
243
  self,
244
244
  *,
245
- cursor: typing.Optional[str] = None,
245
+ after: typing.Optional[str] = None,
246
246
  limit: typing.Optional[int] = None,
247
247
  request_options: typing.Optional[RequestOptions] = None,
248
- ) -> typing.List[LettaSchemasToolTool]:
248
+ ) -> typing.List[Tool]:
249
249
  """
250
250
  Get a list of all tools available to agents belonging to the org of the user
251
251
 
252
252
  Parameters
253
253
  ----------
254
- cursor : typing.Optional[str]
254
+ after : typing.Optional[str]
255
255
 
256
256
  limit : typing.Optional[int]
257
257
 
@@ -260,7 +260,7 @@ class ToolsClient:
260
260
 
261
261
  Returns
262
262
  -------
263
- typing.List[LettaSchemasToolTool]
263
+ typing.List[Tool]
264
264
  Successful Response
265
265
 
266
266
  Examples
@@ -276,7 +276,7 @@ class ToolsClient:
276
276
  "v1/tools/",
277
277
  method="GET",
278
278
  params={
279
- "cursor": cursor,
279
+ "after": after,
280
280
  "limit": limit,
281
281
  },
282
282
  request_options=request_options,
@@ -284,9 +284,9 @@ class ToolsClient:
284
284
  try:
285
285
  if 200 <= _response.status_code < 300:
286
286
  return typing.cast(
287
- typing.List[LettaSchemasToolTool],
287
+ typing.List[Tool],
288
288
  construct_type(
289
- type_=typing.List[LettaSchemasToolTool], # type: ignore
289
+ type_=typing.List[Tool], # type: ignore
290
290
  object_=_response.json(),
291
291
  ),
292
292
  )
@@ -316,7 +316,7 @@ class ToolsClient:
316
316
  json_schema: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
317
317
  return_char_limit: typing.Optional[int] = OMIT,
318
318
  request_options: typing.Optional[RequestOptions] = None,
319
- ) -> LettaSchemasToolTool:
319
+ ) -> Tool:
320
320
  """
321
321
  Create a new tool
322
322
 
@@ -348,7 +348,7 @@ class ToolsClient:
348
348
 
349
349
  Returns
350
350
  -------
351
- LettaSchemasToolTool
351
+ Tool
352
352
  Successful Response
353
353
 
354
354
  Examples
@@ -380,9 +380,9 @@ class ToolsClient:
380
380
  try:
381
381
  if 200 <= _response.status_code < 300:
382
382
  return typing.cast(
383
- LettaSchemasToolTool,
383
+ Tool,
384
384
  construct_type(
385
- type_=LettaSchemasToolTool, # type: ignore
385
+ type_=Tool, # type: ignore
386
386
  object_=_response.json(),
387
387
  ),
388
388
  )
@@ -412,7 +412,7 @@ class ToolsClient:
412
412
  json_schema: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
413
413
  return_char_limit: typing.Optional[int] = OMIT,
414
414
  request_options: typing.Optional[RequestOptions] = None,
415
- ) -> LettaSchemasToolTool:
415
+ ) -> Tool:
416
416
  """
417
417
  Create or update a tool
418
418
 
@@ -444,7 +444,7 @@ class ToolsClient:
444
444
 
445
445
  Returns
446
446
  -------
447
- LettaSchemasToolTool
447
+ Tool
448
448
  Successful Response
449
449
 
450
450
  Examples
@@ -476,9 +476,9 @@ class ToolsClient:
476
476
  try:
477
477
  if 200 <= _response.status_code < 300:
478
478
  return typing.cast(
479
- LettaSchemasToolTool,
479
+ Tool,
480
480
  construct_type(
481
- type_=LettaSchemasToolTool, # type: ignore
481
+ type_=Tool, # type: ignore
482
482
  object_=_response.json(),
483
483
  ),
484
484
  )
@@ -497,9 +497,7 @@ class ToolsClient:
497
497
  raise ApiError(status_code=_response.status_code, body=_response.text)
498
498
  raise ApiError(status_code=_response.status_code, body=_response_json)
499
499
 
500
- def add_base_tool(
501
- self, *, request_options: typing.Optional[RequestOptions] = None
502
- ) -> typing.List[LettaSchemasToolTool]:
500
+ def add_base_tool(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[Tool]:
503
501
  """
504
502
  Upsert base tools
505
503
 
@@ -510,7 +508,7 @@ class ToolsClient:
510
508
 
511
509
  Returns
512
510
  -------
513
- typing.List[LettaSchemasToolTool]
511
+ typing.List[Tool]
514
512
  Successful Response
515
513
 
516
514
  Examples
@@ -530,9 +528,9 @@ class ToolsClient:
530
528
  try:
531
529
  if 200 <= _response.status_code < 300:
532
530
  return typing.cast(
533
- typing.List[LettaSchemasToolTool],
531
+ typing.List[Tool],
534
532
  construct_type(
535
- type_=typing.List[LettaSchemasToolTool], # type: ignore
533
+ type_=typing.List[Tool], # type: ignore
536
534
  object_=_response.json(),
537
535
  ),
538
536
  )
@@ -753,7 +751,7 @@ class ToolsClient:
753
751
 
754
752
  def add_composio_tool(
755
753
  self, composio_action_name: str, *, request_options: typing.Optional[RequestOptions] = None
756
- ) -> LettaSchemasToolTool:
754
+ ) -> Tool:
757
755
  """
758
756
  Add a new Composio tool by action name (Composio refers to each tool as an `Action`)
759
757
 
@@ -766,7 +764,7 @@ class ToolsClient:
766
764
 
767
765
  Returns
768
766
  -------
769
- LettaSchemasToolTool
767
+ Tool
770
768
  Successful Response
771
769
 
772
770
  Examples
@@ -788,9 +786,9 @@ class ToolsClient:
788
786
  try:
789
787
  if 200 <= _response.status_code < 300:
790
788
  return typing.cast(
791
- LettaSchemasToolTool,
789
+ Tool,
792
790
  construct_type(
793
- type_=LettaSchemasToolTool, # type: ignore
791
+ type_=Tool, # type: ignore
794
792
  object_=_response.json(),
795
793
  ),
796
794
  )
@@ -814,9 +812,7 @@ class AsyncToolsClient:
814
812
  def __init__(self, *, client_wrapper: AsyncClientWrapper):
815
813
  self._client_wrapper = client_wrapper
816
814
 
817
- async def get(
818
- self, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None
819
- ) -> LettaSchemasToolTool:
815
+ async def retrieve(self, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> Tool:
820
816
  """
821
817
  Get a tool by ID
822
818
 
@@ -829,7 +825,7 @@ class AsyncToolsClient:
829
825
 
830
826
  Returns
831
827
  -------
832
- LettaSchemasToolTool
828
+ Tool
833
829
  Successful Response
834
830
 
835
831
  Examples
@@ -844,7 +840,7 @@ class AsyncToolsClient:
844
840
 
845
841
 
846
842
  async def main() -> None:
847
- await client.tools.get(
843
+ await client.tools.retrieve(
848
844
  tool_id="tool_id",
849
845
  )
850
846
 
@@ -859,9 +855,9 @@ class AsyncToolsClient:
859
855
  try:
860
856
  if 200 <= _response.status_code < 300:
861
857
  return typing.cast(
862
- LettaSchemasToolTool,
858
+ Tool,
863
859
  construct_type(
864
- type_=LettaSchemasToolTool, # type: ignore
860
+ type_=Tool, # type: ignore
865
861
  object_=_response.json(),
866
862
  ),
867
863
  )
@@ -946,7 +942,7 @@ class AsyncToolsClient:
946
942
  raise ApiError(status_code=_response.status_code, body=_response.text)
947
943
  raise ApiError(status_code=_response.status_code, body=_response_json)
948
944
 
949
- async def update(
945
+ async def modify(
950
946
  self,
951
947
  tool_id: str,
952
948
  *,
@@ -958,7 +954,7 @@ class AsyncToolsClient:
958
954
  json_schema: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
959
955
  return_char_limit: typing.Optional[int] = OMIT,
960
956
  request_options: typing.Optional[RequestOptions] = None,
961
- ) -> LettaSchemasToolTool:
957
+ ) -> Tool:
962
958
  """
963
959
  Update an existing tool
964
960
 
@@ -992,7 +988,7 @@ class AsyncToolsClient:
992
988
 
993
989
  Returns
994
990
  -------
995
- LettaSchemasToolTool
991
+ Tool
996
992
  Successful Response
997
993
 
998
994
  Examples
@@ -1007,7 +1003,7 @@ class AsyncToolsClient:
1007
1003
 
1008
1004
 
1009
1005
  async def main() -> None:
1010
- await client.tools.update(
1006
+ await client.tools.modify(
1011
1007
  tool_id="tool_id",
1012
1008
  )
1013
1009
 
@@ -1035,9 +1031,9 @@ class AsyncToolsClient:
1035
1031
  try:
1036
1032
  if 200 <= _response.status_code < 300:
1037
1033
  return typing.cast(
1038
- LettaSchemasToolTool,
1034
+ Tool,
1039
1035
  construct_type(
1040
- type_=LettaSchemasToolTool, # type: ignore
1036
+ type_=Tool, # type: ignore
1041
1037
  object_=_response.json(),
1042
1038
  ),
1043
1039
  )
@@ -1059,16 +1055,16 @@ class AsyncToolsClient:
1059
1055
  async def list(
1060
1056
  self,
1061
1057
  *,
1062
- cursor: typing.Optional[str] = None,
1058
+ after: typing.Optional[str] = None,
1063
1059
  limit: typing.Optional[int] = None,
1064
1060
  request_options: typing.Optional[RequestOptions] = None,
1065
- ) -> typing.List[LettaSchemasToolTool]:
1061
+ ) -> typing.List[Tool]:
1066
1062
  """
1067
1063
  Get a list of all tools available to agents belonging to the org of the user
1068
1064
 
1069
1065
  Parameters
1070
1066
  ----------
1071
- cursor : typing.Optional[str]
1067
+ after : typing.Optional[str]
1072
1068
 
1073
1069
  limit : typing.Optional[int]
1074
1070
 
@@ -1077,7 +1073,7 @@ class AsyncToolsClient:
1077
1073
 
1078
1074
  Returns
1079
1075
  -------
1080
- typing.List[LettaSchemasToolTool]
1076
+ typing.List[Tool]
1081
1077
  Successful Response
1082
1078
 
1083
1079
  Examples
@@ -1101,7 +1097,7 @@ class AsyncToolsClient:
1101
1097
  "v1/tools/",
1102
1098
  method="GET",
1103
1099
  params={
1104
- "cursor": cursor,
1100
+ "after": after,
1105
1101
  "limit": limit,
1106
1102
  },
1107
1103
  request_options=request_options,
@@ -1109,9 +1105,9 @@ class AsyncToolsClient:
1109
1105
  try:
1110
1106
  if 200 <= _response.status_code < 300:
1111
1107
  return typing.cast(
1112
- typing.List[LettaSchemasToolTool],
1108
+ typing.List[Tool],
1113
1109
  construct_type(
1114
- type_=typing.List[LettaSchemasToolTool], # type: ignore
1110
+ type_=typing.List[Tool], # type: ignore
1115
1111
  object_=_response.json(),
1116
1112
  ),
1117
1113
  )
@@ -1141,7 +1137,7 @@ class AsyncToolsClient:
1141
1137
  json_schema: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
1142
1138
  return_char_limit: typing.Optional[int] = OMIT,
1143
1139
  request_options: typing.Optional[RequestOptions] = None,
1144
- ) -> LettaSchemasToolTool:
1140
+ ) -> Tool:
1145
1141
  """
1146
1142
  Create a new tool
1147
1143
 
@@ -1173,7 +1169,7 @@ class AsyncToolsClient:
1173
1169
 
1174
1170
  Returns
1175
1171
  -------
1176
- LettaSchemasToolTool
1172
+ Tool
1177
1173
  Successful Response
1178
1174
 
1179
1175
  Examples
@@ -1213,9 +1209,9 @@ class AsyncToolsClient:
1213
1209
  try:
1214
1210
  if 200 <= _response.status_code < 300:
1215
1211
  return typing.cast(
1216
- LettaSchemasToolTool,
1212
+ Tool,
1217
1213
  construct_type(
1218
- type_=LettaSchemasToolTool, # type: ignore
1214
+ type_=Tool, # type: ignore
1219
1215
  object_=_response.json(),
1220
1216
  ),
1221
1217
  )
@@ -1245,7 +1241,7 @@ class AsyncToolsClient:
1245
1241
  json_schema: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
1246
1242
  return_char_limit: typing.Optional[int] = OMIT,
1247
1243
  request_options: typing.Optional[RequestOptions] = None,
1248
- ) -> LettaSchemasToolTool:
1244
+ ) -> Tool:
1249
1245
  """
1250
1246
  Create or update a tool
1251
1247
 
@@ -1277,7 +1273,7 @@ class AsyncToolsClient:
1277
1273
 
1278
1274
  Returns
1279
1275
  -------
1280
- LettaSchemasToolTool
1276
+ Tool
1281
1277
  Successful Response
1282
1278
 
1283
1279
  Examples
@@ -1317,9 +1313,9 @@ class AsyncToolsClient:
1317
1313
  try:
1318
1314
  if 200 <= _response.status_code < 300:
1319
1315
  return typing.cast(
1320
- LettaSchemasToolTool,
1316
+ Tool,
1321
1317
  construct_type(
1322
- type_=LettaSchemasToolTool, # type: ignore
1318
+ type_=Tool, # type: ignore
1323
1319
  object_=_response.json(),
1324
1320
  ),
1325
1321
  )
@@ -1338,9 +1334,7 @@ class AsyncToolsClient:
1338
1334
  raise ApiError(status_code=_response.status_code, body=_response.text)
1339
1335
  raise ApiError(status_code=_response.status_code, body=_response_json)
1340
1336
 
1341
- async def add_base_tool(
1342
- self, *, request_options: typing.Optional[RequestOptions] = None
1343
- ) -> typing.List[LettaSchemasToolTool]:
1337
+ async def add_base_tool(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[Tool]:
1344
1338
  """
1345
1339
  Upsert base tools
1346
1340
 
@@ -1351,7 +1345,7 @@ class AsyncToolsClient:
1351
1345
 
1352
1346
  Returns
1353
1347
  -------
1354
- typing.List[LettaSchemasToolTool]
1348
+ typing.List[Tool]
1355
1349
  Successful Response
1356
1350
 
1357
1351
  Examples
@@ -1379,9 +1373,9 @@ class AsyncToolsClient:
1379
1373
  try:
1380
1374
  if 200 <= _response.status_code < 300:
1381
1375
  return typing.cast(
1382
- typing.List[LettaSchemasToolTool],
1376
+ typing.List[Tool],
1383
1377
  construct_type(
1384
- type_=typing.List[LettaSchemasToolTool], # type: ignore
1378
+ type_=typing.List[Tool], # type: ignore
1385
1379
  object_=_response.json(),
1386
1380
  ),
1387
1381
  )
@@ -1628,7 +1622,7 @@ class AsyncToolsClient:
1628
1622
 
1629
1623
  async def add_composio_tool(
1630
1624
  self, composio_action_name: str, *, request_options: typing.Optional[RequestOptions] = None
1631
- ) -> LettaSchemasToolTool:
1625
+ ) -> Tool:
1632
1626
  """
1633
1627
  Add a new Composio tool by action name (Composio refers to each tool as an `Action`)
1634
1628
 
@@ -1641,7 +1635,7 @@ class AsyncToolsClient:
1641
1635
 
1642
1636
  Returns
1643
1637
  -------
1644
- LettaSchemasToolTool
1638
+ Tool
1645
1639
  Successful Response
1646
1640
 
1647
1641
  Examples
@@ -1671,9 +1665,9 @@ class AsyncToolsClient:
1671
1665
  try:
1672
1666
  if 200 <= _response.status_code < 300:
1673
1667
  return typing.cast(
1674
- LettaSchemasToolTool,
1668
+ Tool,
1675
1669
  construct_type(
1676
- type_=LettaSchemasToolTool, # type: ignore
1670
+ type_=Tool, # type: ignore
1677
1671
  object_=_response.json(),
1678
1672
  ),
1679
1673
  )