perf: testing of gpt4free

RockChinQ · Oct 30, 2023 · 6d00504 · 6d00504
1 parent d0c9b50
commit 6d00504
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/free_one_api/impls/adapter/gpt4free.py b/free_one_api/impls/adapter/gpt4free.py
@@ -104,7 +104,9 @@ async def _select_provider(self):
                     ],
                     provider=self._use_provider
                 )
-                non_stream_tested = True
+
+                if 'Rock' in resp and '<' not in resp:
+                    non_stream_tested = True
             except Exception as e:
                 self._use_provider = None
         if non_stream_tested and self._use_stream_provider is not None:
@@ -158,6 +160,8 @@ async def _select_provider(self):
                     provider=provider
                 )
 
+                logging.debug("Testing provider %s, resp %s", provider, resp)
+
                 if 'Rock' in resp and '<' not in resp:
                     if self._use_provider is None:
                         self._use_provider = provider