opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge c74af1e065 into 2c919efa69

This commit is contained in:
leon.Huang 2025-12-29 16:33:41 +08:00 committed by GitHub
parent 2c919efa69 c74af1e065
commit 465859f632
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/workflow/nodes/agent/agent_node.py
View File

@ -303,9 +303,10 @@ class AgentNode(Node[AgentNodeData]):
if parameter.type == AgentStrategyParameter.AgentStrategyParameterType.MODEL_SELECTOR:
value = cast(dict[str, Any], value)
model_instance, model_schema = self._fetch_model(value)
# memory config
# memory config - only for LLM models
history_prompt_messages = []
if node_data.memory:
model_type_str = value.get("model_type", ModelType.LLM.value)
if node_data.memory and model_type_str == ModelType.LLM.value:
memory = self._fetch_memory(model_instance)
if memory:
prompt_messages = memory.get_history_prompt_messages(
@ -415,12 +416,13 @@ class AgentNode(Node[AgentNodeData]):
def _fetch_model(self, value: dict[str, Any]) -> tuple[ModelInstance, AIModelEntity | None]:
provider_manager = ProviderManager()
model_type = ModelType(value.get("model_type", ModelType.LLM.value))
provider_model_bundle = provider_manager.get_provider_model_bundle(
tenant_id=self.tenant_id, provider=value.get("provider", ""), model_type=ModelType.LLM
tenant_id=self.tenant_id, provider=value.get("provider", ""), model_type=model_type
)
model_name = value.get("model", "")
model_credentials = provider_model_bundle.configuration.get_current_credentials(
model_type=ModelType.LLM, model=model_name
model_type=model_type, model=model_name
)
provider_name = provider_model_bundle.configuration.provider.provider
model_type_instance = provider_model_bundle.model_type_instance