Skip to content

Commit b260e6b

Browse files
committed
Remove unused code and fix performance issue
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
1 parent db4fd19 commit b260e6b

2 files changed

Lines changed: 7 additions & 6 deletions

File tree

agent/canvas.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -298,8 +298,6 @@ def run(self, **kwargs):
298298
for kk, vv in kwargs["webhook_payload"].items():
299299
self.components[k]["obj"].set_output(kk, vv)
300300

301-
self.components[k]["obj"].reset(True)
302-
303301
for k in kwargs.keys():
304302
if k in ["query", "user_id", "files"] and kwargs[k]:
305303
if k == "files":

agent/component/base.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -463,12 +463,15 @@ def error(self):
463463
return self._param.outputs.get("_ERROR", {}).get("value")
464464

465465
def reset(self, only_output=False):
466-
for k in self._param.outputs.keys():
467-
self._param.outputs[k]["value"] = None
466+
outputs: dict = self._param.outputs # for better performance
467+
for k in outputs.keys():
468+
outputs[k]["value"] = None
468469
if only_output:
469470
return
470-
for k in self._param.inputs.keys():
471-
self._param.inputs[k]["value"] = None
471+
472+
inputs: dict = self._param.inputs # for better performance
473+
for k in inputs.keys():
474+
inputs[k]["value"] = None
472475
self._param.debug_inputs = {}
473476

474477
def get_input(self, key: str=None) -> Union[Any, dict[str, Any]]:

0 commit comments

Comments
 (0)