From 9140541179e1c2fe855acf1c2743e1800fd5052e Mon Sep 17 00:00:00 2001
From: abc <98614666+xtekky@users.noreply.github.com>
Date: Mon, 20 Nov 2023 18:40:55 +0000
Subject: ~ | improve compatibility

---
 g4f/Provider/deprecated/Lockchat.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'g4f/Provider/deprecated/Lockchat.py')

diff --git a/g4f/Provider/deprecated/Lockchat.py b/g4f/Provider/deprecated/Lockchat.py
index d93c9f8a..f885672d 100644
--- a/g4f/Provider/deprecated/Lockchat.py
+++ b/g4f/Provider/deprecated/Lockchat.py
@@ -38,6 +38,7 @@ class Lockchat(BaseProvider):
         for token in response.iter_lines():
             if b"The model: `gpt-4` does not exist" in token:
                 print("error, retrying...")
+                
                 Lockchat.create_completion(
                     model       = model,
                     messages    = messages,
@@ -47,5 +48,7 @@ class Lockchat(BaseProvider):
 
             if b"content" in token:
                 token = json.loads(token.decode("utf-8").split("data: ")[1])
-                if token := token["choices"][0]["delta"].get("content"):
+                token = token["choices"][0]["delta"].get("content")
+
+                if token:
                     yield (token)
\ No newline at end of file
-- 
cgit v1.2.3