opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-05 00:31:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): degrade unavailable suggestion models

Browse Source
This commit is contained in:
Yansong Zhang 2026-09-03 11:46:57 +08:00
parent f7fb1abf4f
commit 65bfdd3e05
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/llm_generator/llm_generator.py
View File

@ -38,7 +38,7 @@ from graphon.enums import WorkflowNodeExecutionMetadataKey
from graphon.model_runtime.entities.llm_entities import LLMResult
from graphon.model_runtime.entities.message_entities import PromptMessage, SystemPromptMessage, UserPromptMessage
from graphon.model_runtime.entities.model_entities import ModelType, ParameterType
from graphon.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
from graphon.model_runtime.errors.invoke import InvokeError
from models import App, Message, WorkflowNodeExecutionModel
from models.workflow import Workflow
@ -344,7 +344,8 @@ class LLMGenerator:
tenant_id=tenant_id,
model_type=ModelType.LLM,
)
except InvokeAuthorizationError:
except Exception:
logger.exception("Failed to resolve the suggested-questions model")
return []
prompt_messages: list[PromptMessage] = [UserPromptMessage(content=prompt)]

8
api/tests/unit_tests/core/llm_generator/test_llm_generator.py
View File

@ -317,6 +317,14 @@ class TestLLMGenerator:
questions = LLMGenerator.generate_suggested_questions_after_answer("tenant_id", "histories")
assert questions == []
def test_generate_suggested_questions_after_answer_model_resolution_error(self, mock_model_instance):
with patch("core.llm_generator.llm_generator.ModelManager.for_tenant") as mock_manager:
mock_manager.return_value.get_default_model_instance.side_effect = ValueError("unsupported default model")
questions = LLMGenerator.generate_suggested_questions_after_answer("tenant_id", "histories")
assert questions == []
def test_generate_suggested_questions_after_answer_invoke_error(self, mock_model_instance):
mock_model_instance.invoke_llm.side_effect = InvokeError("Invoke failed")
questions = LLMGenerator.generate_suggested_questions_after_answer("tenant_id", "histories")