Fix superbooga when using regenerate (#3362)

This commit is contained in:
Hans Raaf 2023-08-09 04:26:28 +02:00 committed by GitHub
parent 901b028d55
commit f4caaf337a
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: 4AEE18F83AFDEB23

View file

@ -96,7 +96,8 @@ def apply_settings(chunk_count, chunk_count_initial, time_weight):
def custom_generate_chat_prompt(user_input, state, **kwargs):
global chat_collector
history = state['history']
# get history as being modified when using regenerate.
history = kwargs['history']
if state['mode'] == 'instruct':
results = collector.get_sorted(user_input, n_results=params['chunk_count'])