AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 00:15:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: catch non-begin component output (#7827)

Browse Source 
### What problem does this PR solve?

Catch non-begin component output

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Yongteng Lei 2025-05-23 20:07:39 +08:00 committed by GitHub
parent d285a12b85
commit 6ca1aef52e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
agent/component/code.py
View File

@ -79,7 +79,7 @@ class Code(ComponentBase, ABC):
def _run(self, history, **kwargs): def _run(self, history, **kwargs):
arguments = {} arguments = {}
for input in self._param.arguments: for input in self._param.arguments:
assert "@" in input["component_id"], "Each code argument should bind to a specific compontent" if "@" in input["component_id"]:
component_id = input["component_id"].split("@")[0] component_id = input["component_id"].split("@")[0]
refered_component_key = input["component_id"].split("@")[1] refered_component_key = input["component_id"].split("@")[1]
refered_component = self._canvas.get_component(component_id)["obj"] refered_component = self._canvas.get_component(component_id)["obj"]
@ -88,6 +88,14 @@ class Code(ComponentBase, ABC):
if param["key"] == refered_component_key: if param["key"] == refered_component_key:
if "value" in param: if "value" in param:
arguments[input["name"]] = param["value"] arguments[input["name"]] = param["value"]
else:
cpn = self._canvas.get_component(input["component_id"])["obj"]
if cpn.component_name.lower() == "answer":
arguments[input["name"]] = self._canvas.get_history(1)[0]["content"]
continue
_, out = cpn.output(allow_partial=False)
if not out.empty:
arguments[input["name"]] = "\n".join(out["content"])
return self._execute_code( return self._execute_code(
language=self._param.lang, language=self._param.lang,