@@ -319,8 +319,8 @@ async def mock_stream(): # type: ignore[no-untyped-def]
319319 assert call_args .kwargs ["completion" ].usage .total_tokens == 15
320320
321321
322- @patch ("any_llm.providers.platform.utils.requests .post" )
323- @patch ("any_llm.providers.platform.utils.requests .get" )
322+ @patch ("any_llm.providers.platform.utils.httpx .post" )
323+ @patch ("any_llm.providers.platform.utils.httpx .get" )
324324@patch ("any_llm.providers.platform.utils._solve_challenge" )
325325def test_get_provider_key_success (
326326 mock_solve_challenge : Mock ,
@@ -363,7 +363,7 @@ def test_get_provider_key_success(
363363 mock_decrypt .assert_called_once ()
364364
365365
366- @patch ("any_llm.providers.platform.utils.requests .post" )
366+ @patch ("any_llm.providers.platform.utils.httpx .post" )
367367def test_get_provider_key_invalid_api_key_format (mock_post : Mock ) -> None :
368368 """Test error handling when ANY_LLM_KEY has invalid format."""
369369 invalid_key = "INVALID_KEY_FORMAT"
@@ -374,7 +374,7 @@ def test_get_provider_key_invalid_api_key_format(mock_post: Mock) -> None:
374374 mock_post .assert_not_called ()
375375
376376
377- @patch ("any_llm.providers.platform.utils.requests .post" )
377+ @patch ("any_llm.providers.platform.utils.httpx .post" )
378378def test_get_provider_key_challenge_creation_failure (mock_post : Mock ) -> None :
379379 """Test error handling when challenge creation fails."""
380380 any_llm_key = "ANY.v1.kid123.fingerprint456-YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXoxMjM0NTY="
@@ -390,8 +390,8 @@ def test_get_provider_key_challenge_creation_failure(mock_post: Mock) -> None:
390390 mock_post .assert_called_once ()
391391
392392
393- @patch ("any_llm.providers.platform.utils.requests .post" )
394- @patch ("any_llm.providers.platform.utils.requests .get" )
393+ @patch ("any_llm.providers.platform.utils.httpx .post" )
394+ @patch ("any_llm.providers.platform.utils.httpx .get" )
395395@patch ("any_llm.providers.platform.utils._solve_challenge" )
396396def test_get_provider_key_fetch_failure (
397397 mock_solve_challenge : Mock ,
@@ -425,8 +425,8 @@ def test_get_provider_key_fetch_failure(
425425
426426
427427@pytest .mark .asyncio
428- @patch ("any_llm.providers.platform.utils.requests .post" )
429- @patch ("any_llm.providers.platform.utils.requests .get" )
428+ @patch ("any_llm.providers.platform.utils.httpx .post" )
429+ @patch ("any_llm.providers.platform.utils.httpx .get" )
430430@patch ("any_llm.providers.platform.utils._solve_challenge" )
431431async def test_post_completion_usage_event_success (
432432 mock_solve_challenge : Mock ,
@@ -515,7 +515,7 @@ async def test_post_completion_usage_event_success(
515515
516516
517517@pytest .mark .asyncio
518- @patch ("any_llm.providers.platform.utils.requests .post" )
518+ @patch ("any_llm.providers.platform.utils.httpx .post" )
519519async def test_post_completion_usage_event_invalid_key_format (mock_post : Mock ) -> None :
520520 """Test error handling when ANY_LLM_KEY has invalid format."""
521521 invalid_key = "INVALID_KEY_FORMAT"
0 commit comments