Commit 727fca41 authored by Your Name's avatar Your Name

Fix user provider id checking in chat completions endpoint

parent 712972e7
...@@ -1649,6 +1649,12 @@ class ClaudeProviderHandler(BaseProviderHandler): ...@@ -1649,6 +1649,12 @@ class ClaudeProviderHandler(BaseProviderHandler):
logging.info("ClaudeProviderHandler: Starting model list retrieval") logging.info("ClaudeProviderHandler: Starting model list retrieval")
logging.info("=" * 80) logging.info("=" * 80)
# First try to load from cache
cached_models = self._load_models_cache()
if cached_models:
logging.info(f"ClaudeProviderHandler: ✓ Returning {len(cached_models)} models from cache")
return cached_models
await self.apply_rate_limit() await self.apply_rate_limit()
try: try:
......
...@@ -11480,7 +11480,7 @@ async def user_chat_completions_by_username(request: Request, username: str, bod ...@@ -11480,7 +11480,7 @@ async def user_chat_completions_by_username(request: Request, username: str, bod
token_id = getattr(request.state, 'token_id', None) token_id = getattr(request.state, 'token_id', None)
return await handler.handle_rotation_request(actual_model, body_dict, authenticated_user_id, token_id) return await handler.handle_rotation_request(actual_model, body_dict, authenticated_user_id, token_id)
if provider_id == "user-provider" or provider_id in (p['provider_id'] for p in (get_user_handler('request', target_user_id).user_providers.values() if target_user_id else [])): if provider_id == "user-provider" or provider_id in (get_user_handler('request', target_user_id).user_providers.keys() if target_user_id else []):
# Check if this is a user provider # Check if this is a user provider
handler = get_user_handler('request', target_user_id) handler = get_user_handler('request', target_user_id)
provider_name = actual_model if provider_id == "user-provider" else provider_id provider_name = actual_model if provider_id == "user-provider" else provider_id
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment