lint: fix lint

Sinaptik-AI · Mar 27, 2024 · 9c39fd7 · 9c39fd7
1 parent 46bda2c
commit 9c39fd7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pandasai/llm/openai.py b/pandasai/llm/openai.py
@@ -69,8 +69,10 @@ def __init__(
 
         if not self.api_token:
             raise APIKeyNotFoundError("OpenAI API key is required")
-
-        self.api_base = kwargs.get("api_base") or os.getenv("OPENAI_API_BASE") or self.api_base
+
+        self.api_base = (
+            kwargs.get("api_base") or os.getenv("OPENAI_API_BASE") or self.api_base
+        )
         self.openai_proxy = kwargs.get("openai_proxy") or os.getenv("OPENAI_PROXY")
         if self.openai_proxy:
             openai.proxy = {"http": self.openai_proxy, "https": self.openai_proxy}