Commit 160e0766 authored by Your Name's avatar Your Name

Fix user provider dict vs object config access in QwenProviderHandler

parent 6b127cb0
...@@ -745,6 +745,7 @@ class BaseProviderHandler: ...@@ -745,6 +745,7 @@ class BaseProviderHandler:
self.provider_id = provider_id self.provider_id = provider_id
self.api_key = api_key self.api_key = api_key
self.user_id = user_id self.user_id = user_id
self.user_provider_config = None
# Get error tracking config - use user-specific or default to global # Get error tracking config - use user-specific or default to global
if user_id is not None and provider_id in config.error_tracking: if user_id is not None and provider_id in config.error_tracking:
......
...@@ -572,11 +572,19 @@ class QwenProviderHandler(BaseProviderHandler): ...@@ -572,11 +572,19 @@ class QwenProviderHandler(BaseProviderHandler):
logger.info("QwenProviderHandler: Using API token authentication, fetching from models endpoint") logger.info("QwenProviderHandler: Using API token authentication, fetching from models endpoint")
# Check if models are already defined in provider configuration # Check if models are already defined in provider configuration
if self.provider_config.models and len(self.provider_config.models) > 0: # Handle both dict (user) and object (global) config formats
if hasattr(self, 'user_provider_config') and self.user_provider_config is not None:
# User provider config (dict)
models = self.user_provider_config.get('models', [])
else:
# Global provider config (object)
models = self.provider_config.models if hasattr(self.provider_config, 'models') else []
if models and len(models) > 0:
# Models are defined in configuration, use those instead of fetching # Models are defined in configuration, use those instead of fetching
logger.info("QwenProviderHandler: Models defined in configuration, using configured models") logger.info("QwenProviderHandler: Models defined in configuration, using configured models")
models = [] models = []
for model_config in self.provider_config.models: for model_config in models:
models.append(Model( models.append(Model(
id=model_config.get('name', ''), id=model_config.get('name', ''),
name=model_config.get('name', ''), name=model_config.get('name', ''),
......
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