Update haystack/nodes/prompt/invocation_layer/open_ai.py

Co-authored-by: Madeesh Kannan <[email protected]>
deepset-ai · Jan 30, 2024 · 84d12fd · 84d12fd
1 parent ce1f489
commit 84d12fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/haystack/nodes/prompt/invocation_layer/open_ai.py b/haystack/nodes/prompt/invocation_layer/open_ai.py
@@ -65,7 +65,7 @@ def __init__(
                 f"api_key {api_key} must be a valid OpenAI key. Visit https://openai.com/api/ to get one."
             )
         self.api_key = api_key
-        self.api_base = "https://api.openai.com/v1" if api_base is None else api_base
+        self.api_base = api_base
         self.openai_organization = openai_organization
         self.timeout = timeout