Skip to content

Commit 5ed0c9c

Browse files
anubhav756anubhav756
committed
fix: Revert warnings to prefer auth_tokens over auth_headers
1 parent ea68bfc commit 5ed0c9c

File tree

1 file changed

+32
-32
lines changed
  • packages/toolbox-langchain/src/toolbox_langchain

1 file changed

+32
-32
lines changed

‎packages/toolbox-langchain/src/toolbox_langchain/client.py

Lines changed: 32 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -59,31 +59,31 @@ async def aload_tool(
5959
Returns:
6060
A tool loaded from the Toolbox.
6161
"""
62-
if auth_headers:
62+
if auth_tokens:
6363
if auth_token_getters:
6464
warn(
65-
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
65+
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
6666
DeprecationWarning,
6767
)
6868
else:
6969
warn(
70-
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
70+
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
7171
DeprecationWarning,
7272
)
73-
auth_token_getters = auth_headers
73+
auth_token_getters = auth_tokens
7474

75-
if auth_tokens:
75+
if auth_headers:
7676
if auth_token_getters:
7777
warn(
78-
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
78+
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
7979
DeprecationWarning,
8080
)
8181
else:
8282
warn(
83-
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
83+
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
8484
DeprecationWarning,
8585
)
86-
auth_token_getters = auth_tokens
86+
auth_token_getters = auth_headers
8787

8888
coro = self.__core_sync_client._async_client.load_tool(
8989
name=tool_name,
@@ -141,31 +141,31 @@ async def aload_toolset(
141141
Returns:
142142
A list of all tools loaded from the Toolbox.
143143
"""
144-
if auth_headers:
144+
if auth_tokens:
145145
if auth_token_getters:
146146
warn(
147-
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
147+
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
148148
DeprecationWarning,
149149
)
150150
else:
151151
warn(
152-
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
152+
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
153153
DeprecationWarning,
154154
)
155-
auth_token_getters = auth_headers
155+
auth_token_getters = auth_tokens
156156

157-
if auth_tokens:
157+
if auth_headers:
158158
if auth_token_getters:
159159
warn(
160-
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
160+
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
161161
DeprecationWarning,
162162
)
163163
else:
164164
warn(
165-
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
165+
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
166166
DeprecationWarning,
167167
)
168-
auth_token_getters = auth_tokens
168+
auth_token_getters = auth_headers
169169

170170
coro = self.__core_sync_client._async_client.load_toolset(
171171
name=toolset_name,
@@ -222,31 +222,31 @@ def load_tool(
222222
Returns:
223223
A tool loaded from the Toolbox.
224224
"""
225-
if auth_headers:
225+
if auth_tokens:
226226
if auth_token_getters:
227227
warn(
228-
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
228+
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
229229
DeprecationWarning,
230230
)
231231
else:
232232
warn(
233-
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
233+
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
234234
DeprecationWarning,
235235
)
236-
auth_token_getters = auth_headers
236+
auth_token_getters = auth_tokens
237237

238-
if auth_tokens:
238+
if auth_headers:
239239
if auth_token_getters:
240240
warn(
241-
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
241+
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
242242
DeprecationWarning,
243243
)
244244
else:
245245
warn(
246-
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
246+
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
247247
DeprecationWarning,
248248
)
249-
auth_token_getters = auth_tokens
249+
auth_token_getters = auth_headers
250250

251251
core_sync_tool = self.__core_sync_client.load_tool(
252252
name=tool_name,
@@ -286,31 +286,31 @@ def load_toolset(
286286
Returns:
287287
A list of all tools loaded from the Toolbox.
288288
"""
289-
if auth_headers:
289+
if auth_tokens:
290290
if auth_token_getters:
291291
warn(
292-
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
292+
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
293293
DeprecationWarning,
294294
)
295295
else:
296296
warn(
297-
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
297+
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
298298
DeprecationWarning,
299299
)
300-
auth_token_getters = auth_headers
300+
auth_token_getters = auth_tokens
301301

302-
if auth_tokens:
302+
if auth_headers:
303303
if auth_token_getters:
304304
warn(
305-
"Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.",
305+
"Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.",
306306
DeprecationWarning,
307307
)
308308
else:
309309
warn(
310-
"Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.",
310+
"Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.",
311311
DeprecationWarning,
312312
)
313-
auth_token_getters = auth_tokens
313+
auth_token_getters = auth_headers
314314

315315
core_sync_tools = self.__core_sync_client.load_toolset(
316316
name=toolset_name,

0 commit comments

Comments
 (0)