From 3982f39424ea037aca1086d45c6f657b4bfc457c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=B2=98r=E1=B9=A8h=E0=B8=AA=E2=88=82ow?=
 <71973368+MrShadowDev@users.noreply.github.com>
Date: Mon, 23 Oct 2023 09:46:25 +0200
Subject: 'Refactored by Sourcery' (#1125)

Co-authored-by: Sourcery AI <>
---
 g4f/Provider/helper.py | 22 ++++++++++------------
 1 file changed, 10 insertions(+), 12 deletions(-)

(limited to 'g4f/Provider/helper.py')

diff --git a/g4f/Provider/helper.py b/g4f/Provider/helper.py
index 1b00ace5..519a024a 100644
--- a/g4f/Provider/helper.py
+++ b/g4f/Provider/helper.py
@@ -77,9 +77,8 @@ def get_cookies(domain_name=''):
     def g4f(domain_name):
         user_data_dir = user_config_dir("g4f")
         cookie_file = path.join(user_data_dir, "Default", "Cookies")
-        if not path.exists(cookie_file):
-            return []
-        return chrome(cookie_file, domain_name)
+        return [] if not path.exists(cookie_file) else chrome(cookie_file, domain_name)
+
     cookies = {}
     for cookie_fn in [g4f, chrome, chromium, opera, opera_gx, brave, edge, vivaldi, firefox]:
         try:
@@ -96,16 +95,15 @@ def get_cookies(domain_name=''):
 
 
 def format_prompt(messages: Messages, add_special_tokens=False) -> str:
-    if add_special_tokens or len(messages) > 1:
-        formatted = "\n".join(
-            [
-                "%s: %s" % ((message["role"]).capitalize(), message["content"])
-                for message in messages
-            ]
-        )
-        return f"{formatted}\nAssistant:"
-    else:
+    if not add_special_tokens and len(messages) <= 1:
         return messages[0]["content"]
+    formatted = "\n".join(
+        [
+            f'{message["role"].capitalize()}: {message["content"]}'
+            for message in messages
+        ]
+    )
+    return f"{formatted}\nAssistant:"
 
 
 def get_browser(user_data_dir: str = None):
-- 
cgit v1.2.3