fix: use model provided by user in prompt generator (#22541) (#22542)

Co-authored-by: stream <stream@dify.ai>
9 months ago · 4b2baeea65
parent 93c27b134d
commit 4b2baeea65
1 changed files with 3 additions and 1 deletions
--- a/api/core/llm_generator/llm_generator.py
+++ b/api/core/llm_generator/llm_generator.py
@ -148,9 +148,11 @@ class LLMGenerator:

            model_manager = ModelManager()

-            model_instance = model_manager.get_default_model_instance(
+            model_instance = model_manager.get_model_instance(
                tenant_id=tenant_id,
                model_type=ModelType.LLM,
+                provider=model_config.get("provider", ""),
+                model=model_config.get("name", ""),
            )

            try: