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):
logging.info("ClaudeProviderHandler: Starting model list retrieval")
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()
try:
......
......@@ -11480,7 +11480,7 @@ async def user_chat_completions_by_username(request: Request, username: str, bod
token_id = getattr(request.state, 'token_id', None)
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
handler = get_user_handler('request', target_user_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