Pārlūkot izejas kodu

Fix: set output for Message template (#8064)

### What problem does this PR solve?
now Streamning logic is not match with none streaming logic, which may
introduce down stream can not find upstream components.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.19.1
Stephen Hu pirms 5 mēnešiem
vecāks
revīzija
640fca7dc9
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 3 papildinājumiem un 1 dzēšanām
  1. 3
    1
      agent/component/message.py

+ 3
- 1
agent/component/message.py Parādīt failu

@@ -40,7 +40,9 @@ class Message(ComponentBase, ABC):
if kwargs.get("stream"):
return partial(self.stream_output)

return Message.be_output(random.choice(self._param.messages))
res = Message.be_output(random.choice(self._param.messages))
self.set_output(res)
return res

def stream_output(self):
res = None

Notiek ielāde…
Atcelt
Saglabāt