|
27 | 27 | import pytest
|
28 | 28 | from litellm.proxy.auth.route_checks import RouteChecks
|
29 | 29 | from litellm.proxy._types import LiteLLM_UserTable, LitellmUserRoles, UserAPIKeyAuth
|
| 30 | +from litellm.proxy.pass_through_endpoints.llm_passthrough_endpoints import ( |
| 31 | + router as llm_passthrough_router, |
| 32 | +) |
30 | 33 |
|
31 | 34 | # Replace the actual hash_token function with our mock
|
32 | 35 | import litellm.proxy.auth.route_checks
|
@@ -56,12 +59,21 @@ def test_is_llm_api_route():
|
56 | 59 | assert RouteChecks.is_llm_api_route("/vertex-ai/text") is True
|
57 | 60 | assert RouteChecks.is_llm_api_route("/gemini/generate") is True
|
58 | 61 | assert RouteChecks.is_llm_api_route("/cohere/generate") is True
|
| 62 | + assert RouteChecks.is_llm_api_route("/anthropic/messages") is True |
| 63 | + assert RouteChecks.is_llm_api_route("/anthropic/v1/messages") is True |
| 64 | + assert RouteChecks.is_llm_api_route("/azure/endpoint") is True |
59 | 65 |
|
60 | 66 | # check non-matching routes
|
61 | 67 | assert RouteChecks.is_llm_api_route("/some/random/route") is False
|
62 | 68 | assert RouteChecks.is_llm_api_route("/key/regenerate/82akk800000000jjsk") is False
|
63 | 69 | assert RouteChecks.is_llm_api_route("/key/82akk800000000jjsk/delete") is False
|
64 | 70 |
|
| 71 | + # check all routes in llm_passthrough_router, ensure they are considered llm api routes |
| 72 | + for route in llm_passthrough_router.routes: |
| 73 | + route_path = str(route.path) |
| 74 | + print("route_path", route_path) |
| 75 | + assert RouteChecks.is_llm_api_route(route_path) is True |
| 76 | + |
65 | 77 |
|
66 | 78 | # Test _route_matches_pattern
|
67 | 79 | def test_route_matches_pattern():
|
|
0 commit comments