Ver código fonte

fix: query will be None if the query_prompt_template not exists (#11031)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.12.0
-LAN- 11 meses atrás
pai
commit
60b5dac3ab
Nenhuma conta vinculada ao e-mail do autor do commit
1 arquivos alterados com 8 adições e 3 exclusões
  1. 8
    3
      api/core/workflow/nodes/llm/node.py

+ 8
- 3
api/core/workflow/nodes/llm/node.py Ver arquivo

ObjectSegment, ObjectSegment,
StringSegment, StringSegment,
) )
from core.workflow.constants import SYSTEM_VARIABLE_NODE_ID
from core.workflow.entities.node_entities import NodeRunMetadataKey, NodeRunResult from core.workflow.entities.node_entities import NodeRunMetadataKey, NodeRunResult
from core.workflow.entities.variable_entities import VariableSelector from core.workflow.entities.variable_entities import VariableSelector
from core.workflow.entities.variable_pool import VariablePool from core.workflow.entities.variable_pool import VariablePool
# fetch memory # fetch memory
memory = self._fetch_memory(node_data_memory=self.node_data.memory, model_instance=model_instance) memory = self._fetch_memory(node_data_memory=self.node_data.memory, model_instance=model_instance)


# fetch prompt messages
query = None
if self.node_data.memory: if self.node_data.memory:
query = self.node_data.memory.query_prompt_template query = self.node_data.memory.query_prompt_template
else:
query = None
if query is None and (
query_variable := self.graph_runtime_state.variable_pool.get(
(SYSTEM_VARIABLE_NODE_ID, SystemVariableKey.QUERY)
)
):
query = query_variable.text


prompt_messages, stop = self._fetch_prompt_messages( prompt_messages, stop = self._fetch_prompt_messages(
user_query=query, user_query=query,

Carregando…
Cancelar
Salvar