From 82bd6f91808a383781807262c4ae1f3de9740531 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <heiner@lohaus.eu>
Date: Wed, 20 Sep 2023 06:12:34 +0200
Subject: Cache "snlm0e" in Bard Improve error handling in ChatgptLogin Fix
 async example in readme

---
 g4f/Provider/Vitalentum.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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

diff --git a/g4f/Provider/Vitalentum.py b/g4f/Provider/Vitalentum.py
index 31ad8b80..d5265428 100644
--- a/g4f/Provider/Vitalentum.py
+++ b/g4f/Provider/Vitalentum.py
@@ -46,7 +46,9 @@ class Vitalentum(AsyncGeneratorProvider):
                 response.raise_for_status()
                 async for line in response.content:
                     line = line.decode()
-                    if line.startswith("data: ") and not line.startswith("data: [DONE]"):
+                    if line.startswith("data: "):
+                        if line.startswith("data: [DONE]"):
+                            break
                         line = json.loads(line[6:-1])
                         content = line["choices"][0]["delta"].get("content")
                         if content:
-- 
cgit v1.2.3