Kevin Hu commited on
Commit
6d13ea7
·
1 Parent(s): c14e2e5

Remove vector stored in component output. (#3908)

Browse files

### What problem does this PR solve?

Close #3805

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. agent/component/base.py +4 -1
agent/component/base.py CHANGED
@@ -383,6 +383,9 @@ class ComponentBase(ABC):
383
  "params": {}
384
  }
385
  """
 
 
 
386
  return """{{
387
  "component_name": "{}",
388
  "params": {},
@@ -390,7 +393,7 @@ class ComponentBase(ABC):
390
  "inputs": {}
391
  }}""".format(self.component_name,
392
  self._param,
393
- json.dumps(json.loads(str(self._param)).get("output", {}), ensure_ascii=False),
394
  json.dumps(json.loads(str(self._param)).get("inputs", []), ensure_ascii=False)
395
  )
396
 
 
383
  "params": {}
384
  }
385
  """
386
+ out = json.loads(str(self._param)).get("output", {})
387
+ if isinstance(out, dict) and "vector" in out:
388
+ del out["vector"]
389
  return """{{
390
  "component_name": "{}",
391
  "params": {},
 
393
  "inputs": {}
394
  }}""".format(self.component_name,
395
  self._param,
396
+ json.dumps(out, ensure_ascii=False),
397
  json.dumps(json.loads(str(self._param)).get("inputs", []), ensure_ascii=False)
398
  )
399