Merge pull request #399 from devchat-ai/fix_timeout_error

Update OpenAI API base URL handling
devchat-ai · May 30, 2024 · 02f0ebc · 02f0ebc
2 parents 2aea6e1 + c6f24ba
commit 02f0ebc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/devchat/openai/openai_chat.py b/devchat/openai/openai_chat.py
@@ -99,7 +99,10 @@ def stream_response(self, prompt: OpenAIPrompt) -> Iterator:
         api_key = os.environ.get("OPENAI_API_KEY", None)
         base_url = os.environ.get("OPENAI_API_BASE", "https://api.openai.com/v1/")
 
-        if not os.environ.get("USE_TIKTOKEN", False) and base_url != "https://api.openai.com/v1/":
+        if (
+            not os.environ.get("USE_TIKTOKEN", False)
+            and base_url.find("https://api.openai.com/v1") == -1
+        ):
             config_params = self.config.dict(exclude_unset=True)
             if prompt.get_functions():
                 config_params["functions"] = prompt.get_functions()