From 001ecf95b263c6a78face532d297334d73663600 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Tue, 31 Jan 2023 08:14:16 -0300 Subject: [PATCH] Update server.py --- server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server.py b/server.py index 02ef8eb..37e45cc 100644 --- a/server.py +++ b/server.py @@ -364,6 +364,8 @@ if args.chat or args.cai_chat: def extract_message_from_reply(question, reply, current, other, check, extensions=False): next_character_found = False + substring_found = False + previous_idx = [m.start() for m in re.finditer(f"(^|\n){current}:", question)] idx = [m.start() for m in re.finditer(f"(^|\n){current}:", reply)] idx = idx[len(previous_idx)-1] @@ -372,8 +374,6 @@ if args.chat or args.cai_chat: reply = reply[idx + 1 + len(apply_extensions(f"{current}:", "bot_prefix")):] else: reply = reply[idx + 1 + len(f"{current}:"):] - - substring_found = False if check: reply = reply.split('\n')[0].strip()