Pārlūkot izejas kodu

fix component rewrite bug (#2549)

### What problem does this PR solve?

#2545

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.12.0
Kevin Hu pirms 1 gada
vecāks
revīzija
100c60017f
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 2 papildinājumiem un 0 dzēšanām
  1. 2
    0
      agent/component/rewrite.py

+ 2
- 0
agent/component/rewrite.py Parādīt failu

@@ -65,6 +65,8 @@ class RewriteQuestion(Generate, ABC):
chat_mdl = LLMBundle(self._canvas.get_tenant_id(), LLMType.CHAT, self._param.llm_id)
ans = chat_mdl.chat(self._param.get_prompt(), [{"role": "user", "content": q}],
self._param.gen_conf())
self._canvas.history.pop()
self._canvas.history.append(("user", ans))

print(ans, ":::::::::::::::::::::::::::::::::")
return RewriteQuestion.be_output(ans)

Notiek ielāde…
Atcelt
Saglabāt