youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 01:48:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: update model attribute references from 'model' to 'model_name' across multiple files and introduce new fetch_model_config function in llm_utils.py

Browse Source
This commit is contained in:
Novice
2026-03-23 21:17:43 +08:00
parent ee81ea882b
commit ed1bd338f1
23 changed files with 640 additions and 175 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/dify_graph/nodes/llm/llm_utils.py
View File

@ -47,6 +47,20 @@ from .exc import (
from .protocols import TemplateRenderer
def fetch_model_config(
*, tenant_id: str, node_data_model: ModelConfig
) -> tuple[ModelInstance, Any]:
from core.app.llm.model_access import build_dify_model_access
from core.app.llm.model_access import fetch_model_config as _fetch
credentials_provider, model_factory = build_dify_model_access(tenant_id)
return _fetch(
node_data_model=node_data_model,
credentials_provider=credentials_provider,
model_factory=model_factory,
)
def fetch_model_schema(*, model_instance: ModelInstance) -> AIModelEntity:
model_schema = cast(LargeLanguageModel, model_instance.model_type_instance).get_model_schema(
model_instance.model_name,

2
api/dify_graph/nodes/llm/node.py
View File

@ -1985,7 +1985,7 @@ class LLMNode(Node[LLMNodeData]):
try:
model_type_instance = model_instance.model_type_instance
model_schema = model_type_instance.get_model_schema(
model_instance.model,
model_instance.model_name,
model_instance.credentials,
)
return model_schema.features if model_schema and model_schema.features else []