vanna 0.2.0__tar.gz → 0.2.1__tar.gz

{vanna-0.2.0 → vanna-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vanna
-Version: 0.2.0
+Version: 0.2.1
 Summary: Generate SQL queries from natural language
 Author-email: Zain Hoda <zain@vanna.ai>
 Requires-Python: >=3.9

{vanna-0.2.0 → vanna-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "vanna"
-version = "0.2.0"
+version = "0.2.1"
 authors = [
   { name="Zain Hoda", email="zain@vanna.ai" },
 ]

{vanna-0.2.0 → vanna-0.2.1}/src/vanna/openai/openai_chat.py

@@ -9,14 +9,6 @@ class OpenAI_Chat(VannaBase):
     def __init__(self, client=None, config=None):
         VannaBase.__init__(self, config=config)
 
-        if client is not None:
-            self.client = client
-            return
-
-        if config is None and client is None:
-            self.client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-            return
-
         # default parameters - can be overrided using config
         self.temperature = 0.7
         self.max_tokens = 500
@@ -42,6 +34,14 @@ class OpenAI_Chat(VannaBase):
                 "Passing api_version is now deprecated. Please pass an OpenAI client instead."
             )
 
+        if client is not None:
+            self.client = client
+            return
+
+        if config is None and client is None:
+            self.client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+            return
+      
         if "api_key" in config:
             self.client = OpenAI(api_key=config["api_key"])