zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-06-29 15:31:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: update variable completeness check to allow None parameter (#16389)

Browse Source
This commit is contained in:
buua436
2026-06-29 10:51:57 +08:00
committed by GitHub
parent 61ac1c1dff
commit b6dbb2f71e
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
agent/component/variable_assigner.py
View File

@@ -40,6 +40,7 @@ class VariableAssignerParam(ComponentParamBase):
class VariableAssigner(ComponentBase,ABC): class VariableAssigner(ComponentBase,ABC):
component_name = "VariableAssigner" component_name = "VariableAssigner"
_NO_PARAMETER_OPERATORS = {"clear", "remove_first", "remove_last"}
@timeout(int(os.environ.get("COMPONENT_EXEC_TIMEOUT", 10*60))) @timeout(int(os.environ.get("COMPONENT_EXEC_TIMEOUT", 10*60)))
def _invoke(self, **kwargs): def _invoke(self, **kwargs):
@@ -47,11 +48,14 @@ class VariableAssigner(ComponentBase,ABC):
return return
else: else:
for item in self._param.variables: for item in self._param.variables:
if any([not item.get("variable"), not item.get("operator"), not item.get("parameter")]): variable = item.get("variable")
operator = item.get("operator")
parameter = item.get("parameter")
if any([not variable, not operator]):
raise ValueError("Variable is not complete.")
if operator not in self._NO_PARAMETER_OPERATORS and parameter is None:
raise ValueError("Variable is not complete.") raise ValueError("Variable is not complete.")
variable=item["variable"]
operator=item["operator"]
parameter=item["parameter"]
variable_value=self._canvas.get_variable_value(variable) variable_value=self._canvas.get_variable_value(variable)
new_variable=self._operate(variable_value,operator,parameter) new_variable=self._operate(variable_value,operator,parameter)
self._canvas.set_variable_value(variable, new_variable) self._canvas.set_variable_value(variable, new_variable)