refac/fix

2026-03-08 17:26:29 -05:00
parent d513eb8c4d
commit c97767424f
1 changed files with 2 additions and 2 deletions
@@ -1732,8 +1732,8 @@ async def chat_completion(
        }

        # Check base model existence for custom models
-        if model_info_params.get("base_model_id"):
-            base_model_id = model_info_params.get("base_model_id")
+        if model_info and model_info.base_model_id:
+            base_model_id = model_info.base_model_id
            if base_model_id not in request.app.state.MODELS:
                if ENABLE_CUSTOM_MODEL_FALLBACK:
                    default_models = (