Browse Source
fix categorize agent input content not format error (#4842)
### What problem does this PR solve? Fix categorize agent input content not format error ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Co-authored-by: wangrui <wangrui@haima.me>tags/v0.17.0
WANGRUI-ZB
8 months ago
1 changed files with 1 additions and 0 deletions
+ 1
- 0
agent/component/categorize.py
View File
| def _run(self, history, **kwargs): | def _run(self, history, **kwargs): | ||||
| input = self.get_input() | input = self.get_input() | ||||
| input = " - ".join(input["content"]) if "content" in input else "" | |||||
| chat_mdl = LLMBundle(self._canvas.get_tenant_id(), LLMType.CHAT, self._param.llm_id) | chat_mdl = LLMBundle(self._canvas.get_tenant_id(), LLMType.CHAT, self._param.llm_id) | ||||
| ans = chat_mdl.chat(self._param.get_prompt(input), [{"role": "user", "content": "\nCategory: "}], | ans = chat_mdl.chat(self._param.get_prompt(input), [{"role": "user", "content": "\nCategory: "}], | ||||
| self._param.gen_conf()) | self._param.gen_conf()) |
Loading…