mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2025-06-06 21:55:57 -04:00
Compare commits
2 commits
c5d8fd852c
...
872df4cdf7
Author | SHA1 | Date | |
---|---|---|---|
|
872df4cdf7 | ||
|
72264b5bac |
1 changed files with 8 additions and 0 deletions
|
@ -698,6 +698,14 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess
|
|||
|
||||
output['internal'][-1] = [text, updated_internal]
|
||||
output['visible'][-1] = [visible_text, updated_visible]
|
||||
|
||||
row_idx = len(output['internal']) - 1
|
||||
key = f"assistant_{row_idx}"
|
||||
current_idx = output['metadata'][key]['current_version_index']
|
||||
output['metadata'][key]['versions'][current_idx].update({
|
||||
'content': output['internal'][row_idx][1],
|
||||
'visible_content': output['visible'][row_idx][1]
|
||||
})
|
||||
|
||||
if is_stream:
|
||||
yield output
|
||||
|
|
Loading…
Add table
Reference in a new issue