Commit f69d5798 authored by Your Name's avatar Your Name

Add debug logging to model resolution

parent 5b3d49ba
...@@ -121,16 +121,22 @@ def get_resolved_model_name(requested_model: str, current_manager = None) -> str ...@@ -121,16 +121,22 @@ def get_resolved_model_name(requested_model: str, current_manager = None) -> str
# If requested_model is "default" or empty, get the actual loaded model # If requested_model is "default" or empty, get the actual loaded model
if requested_model in ("default", "", None) or not requested_model: if requested_model in ("default", "", None) or not requested_model:
# Try default_model first # Try default_model first
if hasattr(current_manager, 'default_model') and current_manager.default_model and current_manager.default_model != "default": default_model = getattr(current_manager, 'default_model', None)
return current_manager.default_model print(f"DEBUG resolve: default_model = {default_model}, models = {list(current_manager.models.keys())}")
if default_model and default_model != "default":
print(f"DEBUG resolve: returning default_model: {default_model}")
return default_model
# Otherwise return the first model that is not a special key (default, image:, audio:) # Otherwise return the first model that is not a special key (default, image:, audio:)
for key in current_manager.models.keys(): for key in current_manager.models.keys():
# Skip special model keys # Skip special model keys
if key in ("default", "image", "audio") or key.startswith("image:") or key.startswith("audio:"): if key in ("default", "image", "audio") or key.startswith("image:") or key.startswith("audio:"):
continue continue
print(f"DEBUG resolve: returning first non-special key: {key}")
return key return key
# Fallback to first model if all are special keys # Fallback to first model if all are special keys
return list(current_manager.models.keys())[0] fallback = list(current_manager.models.keys())[0]
print(f"DEBUG resolve: fallback to first: {fallback}")
return fallback
# Check if the model is loaded in the manager # Check if the model is loaded in the manager
for key, model in current_manager.models.items(): for key, model in current_manager.models.items():
if requested_model == key or requested_model in key: if requested_model == key or requested_model in key:
......
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