From f41efef0aefe84d8ed5765f524dd673095d8627d Mon Sep 17 00:00:00 2001
From: Abobo <113653079+Abobo8964@users.noreply.github.com>
Date: Fri, 15 Mar 2024 20:30:15 +0800
Subject: Error checking for Vercel (#1704)

Error checking is done for Vercel's create_completion to avoid interruption caused by UnicodeDecodeError.
---
 g4f/Provider/Vercel.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'g4f')

diff --git a/g4f/Provider/Vercel.py b/g4f/Provider/Vercel.py
index f8faeeaf..26fc5211 100644
--- a/g4f/Provider/Vercel.py
+++ b/g4f/Provider/Vercel.py
@@ -63,7 +63,10 @@ class Vercel(AbstractProvider):
             if not response.ok:
                 continue
             for token in response.iter_content(chunk_size=None):
-                yield token.decode()
+                try:
+                    yield token.decode()
+                except UnicodeDecodeError:
+                    pass
             break
         raise_for_status(response)
 
-- 
cgit v1.2.3