add ignored provider - chatbase

Lin-jun-xiang · Nov 9, 2023 · a5d1c05 · a5d1c05
1 parent 4d962d2
commit a5d1c05
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/docGPT/docGPT.py b/docGPT/docGPT.py
@@ -225,7 +225,8 @@ def _call(
             return g4f.ChatCompletion.create(
                 model="gpt-3.5-turbo",
                 messages=[{"role": "user", "content": prompt}],
-                provider=provider
+                provider=provider,
+                ignored=["ChatBase"]
             )
         except Exception as e:
             module_logger.info(f'{__file__}: call gpt4free error - {e}')
@@ -236,7 +237,8 @@ async def _test_provider(self, provider: g4f.Provider) -> str:
             await g4f.ChatCompletion.create_async(
                 model="gpt-3.5-turbo",
                 messages=[{"role": "user", "content": 'Hi, this is test'}],
-                provider=provider
+                provider=provider,
+                ignored=["ChatBase"]
             )
             return provider_name
         except Exception as e: