From c43ca52d112b72ddc7178cf3778a97b624d4036c Mon Sep 17 00:00:00 2001 From: eukub Date: Mon, 25 Dec 2023 20:47:42 +0300 Subject: [PATCH] =?UTF-8?q?=D1=81hanged=20concatenation=20of=20strings=20t?= =?UTF-8?q?o=20f-strings=20to=20improve=20readability=20and=20unify=20with?= =?UTF-8?q?=20the=20rest=20code?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ferret/conversation.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/ferret/conversation.py b/ferret/conversation.py index 23f90a8..91018ca 100644 --- a/ferret/conversation.py +++ b/ferret/conversation.py @@ -41,7 +41,7 @@ def get_prompt(self): messages.insert(0, (self.roles[0], "")) messages.insert(1, (self.roles[1], "Received.")) else: - messages[0] = (init_role, "\n" + init_msg) + messages[0] = (init_role, f"\n{init_msg}") if self.sep_style == SeparatorStyle.SINGLE: ret = self.system + self.sep @@ -49,9 +49,9 @@ def get_prompt(self): if message: if type(message) is tuple: message, _, _ = message - ret += role + ": " + message + self.sep + ret += f"{role}: {message}{self.sep}" else: - ret += role + ":" + ret += f"{role}:" elif self.sep_style == SeparatorStyle.TWO: seps = [self.sep, self.sep2] ret = self.system + seps[0] @@ -59,9 +59,9 @@ def get_prompt(self): if message: if type(message) is tuple: message, _, _ = message - ret += role + ": " + message + seps[i % 2] + ret += f"{role}: {message}{seps[i % 2]}" else: - ret += role + ":" + ret += f"{role}:" elif self.sep_style == SeparatorStyle.MPT: ret = self.system + self.sep for role, message in messages: @@ -88,7 +88,7 @@ def get_prompt(self): message = wrap_inst(message) ret += self.sep + message else: - ret += " " + message + " " + self.sep2 + ret += f" {message} {self.sep2}" else: ret += "" ret = ret.lstrip(self.sep)