Sfoglia il codice sorgente

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 5 mesi fa
parent
commit
640fca7dc9
Nessun account collegato all'indirizzo email del committer
1 ha cambiato i file con 3 aggiunte e 1 eliminazioni
  1. 3
    1
      agent/component/message.py

+ 3
- 1
agent/component/message.py Vedi File

if kwargs.get("stream"): if kwargs.get("stream"):
return partial(self.stream_output) 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): def stream_output(self):
res = None res = None

Loading…
Annulla
Salva