@@ -45,7 +45,7 @@ async def test_load_toolset_and_run(self):
4545 toolset = ToolboxToolset (
4646 server_url = "http://localhost:5000" ,
4747 toolset_name = "my-toolset" ,
48- credentials = CredentialStrategy .TOOLBOX_IDENTITY (),
48+ credentials = CredentialStrategy .toolbox_identity (),
4949 )
5050
5151 try :
@@ -71,7 +71,7 @@ async def test_partial_loading_by_names(self):
7171 toolset = ToolboxToolset (
7272 server_url = "http://localhost:5000" ,
7373 tool_names = ["get-n-rows" ],
74- credentials = CredentialStrategy .TOOLBOX_IDENTITY (),
74+ credentials = CredentialStrategy .toolbox_identity (),
7575 )
7676 try :
7777 tools = await toolset .get_tools ()
@@ -92,7 +92,7 @@ async def test_bound_params_e2e(self):
9292 server_url = "http://localhost:5000" ,
9393 tool_names = ["get-n-rows" ],
9494 bound_params = {"num_rows" : "2" },
95- credentials = CredentialStrategy .TOOLBOX_IDENTITY (),
95+ credentials = CredentialStrategy .toolbox_identity (),
9696 )
9797 try :
9898 tools = await toolset .get_tools ()
@@ -108,7 +108,7 @@ async def test_3lo_flow_simulation(self):
108108 server_url = "http://localhost:5000" ,
109109 # We use an existing toolset that contains auth tools or just any toolset if we can pick specific tools
110110 toolset_name = "my-toolset" ,
111- credentials = CredentialStrategy .USER_IDENTITY (
111+ credentials = CredentialStrategy .user_identity (
112112 client_id = "test-client-id" , client_secret = "test-client-secret"
113113 ),
114114 )
@@ -171,7 +171,7 @@ async def test_manual_token_integration(self):
171171 toolset = ToolboxToolset (
172172 server_url = "http://localhost:5000" ,
173173 toolset_name = "my-toolset" ,
174- credentials = CredentialStrategy .MANUAL_TOKEN (token = "fake-manual-token" ),
174+ credentials = CredentialStrategy .manual_token (token = "fake-manual-token" ),
175175 )
176176 try :
177177 tools = await toolset .get_tools ()
@@ -190,7 +190,7 @@ async def test_manual_creds_integration(self):
190190 toolset = ToolboxToolset (
191191 server_url = "http://localhost:5000" ,
192192 toolset_name = "my-toolset" ,
193- credentials = CredentialStrategy .MANUAL_CREDS (credentials = mock_creds ),
193+ credentials = CredentialStrategy .manual_creds (credentials = mock_creds ),
194194 )
195195 try :
196196 tools = await toolset .get_tools ()
@@ -211,7 +211,7 @@ async def test_header_collision(self):
211211 server_url = "http://localhost:5000" ,
212212 toolset_name = "my-toolset" ,
213213 additional_headers = {"Authorization" : manual_override },
214- credentials = CredentialStrategy .MANUAL_TOKEN (token = "strategy-token" ),
214+ credentials = CredentialStrategy .manual_token (token = "strategy-token" ),
215215 )
216216
217217 # We need to inspect the client to see what it chose.
0 commit comments