mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
import re
|
|
|
|
from litellm._logging import verbose_proxy_logger
|
|
from litellm.proxy._types import *
|
|
|
|
|
|
def route_in_additonal_public_routes(current_route: str):
|
|
"""
|
|
Helper to check if the user defined public_routes on config.yaml
|
|
|
|
Parameters:
|
|
- current_route: str - the route the user is trying to call
|
|
|
|
Returns:
|
|
- bool - True if the route is defined in public_routes
|
|
- bool - False if the route is not defined in public_routes
|
|
|
|
|
|
In order to use this the litellm config.yaml should have the following in general_settings:
|
|
|
|
```yaml
|
|
general_settings:
|
|
master_key: sk-1234
|
|
public_routes: ["LiteLLMRoutes.public_routes", "/spend/calculate"]
|
|
```
|
|
"""
|
|
|
|
# check if user is premium_user - if not do nothing
|
|
from litellm.proxy._types import LiteLLMRoutes
|
|
from litellm.proxy.proxy_server import general_settings, premium_user
|
|
|
|
try:
|
|
if premium_user is not True:
|
|
return False
|
|
# check if this is defined on the config
|
|
if general_settings is None:
|
|
return False
|
|
|
|
routes_defined = general_settings.get("public_routes", [])
|
|
if current_route in routes_defined:
|
|
return True
|
|
|
|
return False
|
|
except Exception as e:
|
|
verbose_proxy_logger.error(f"route_in_additonal_public_routes: {str(e)}")
|
|
return False
|
|
|
|
|
|
def is_openai_route(route: str) -> bool:
|
|
"""
|
|
Helper to checks if provided route is an OpenAI route
|
|
|
|
|
|
Returns:
|
|
- True: if route is an OpenAI route
|
|
- False: if route is not an OpenAI route
|
|
"""
|
|
|
|
if route in LiteLLMRoutes.openai_routes.value:
|
|
return True
|
|
|
|
# fuzzy match routes like "/v1/threads/thread_49EIN5QF32s4mH20M7GFKdlZ"
|
|
# Check for routes with placeholders
|
|
for openai_route in LiteLLMRoutes.openai_routes.value:
|
|
# Replace placeholders with regex pattern
|
|
# placeholders are written as "/threads/{thread_id}"
|
|
if "{" in openai_route:
|
|
pattern = re.sub(r"\{[^}]+\}", r"[^/]+", openai_route)
|
|
# Anchor the pattern to match the entire string
|
|
pattern = f"^{pattern}$"
|
|
if re.match(pattern, route):
|
|
return True
|
|
|
|
return False
|