Bump version and remove explicit API key argument

torchbox · Mar 7, 2024 · 8cb06c2 · 8cb06c2
1 parent f2baff2
commit 8cb06c2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/nlbq/__init__.py b/nlbq/__init__.py
@@ -1,3 +1,3 @@
 """Natural language to BigQuery"""
 
-__version__ = "0.1.5"
+__version__ = "0.1.6"
diff --git a/nlbq/core.py b/nlbq/core.py
@@ -73,10 +73,7 @@ async def text_to_bq(self, question: str) -> tuple:
             {"role": "user", "content": self.prompt_template % question},
         ]
         resp = await client.chat.completions.create(
-            model=self.model,
-            messages=prompt_messages,
-            temperature=0,
-            api_key=settings.openai_api_key,
+            model=self.model, messages=prompt_messages, temperature=0
         )
 
         usage = resp["usage"]
@@ -121,9 +118,6 @@ async def answer(self, question, statement, results) -> str:
             },
         ]
         resp = await client.chat.completions.create(
-            model=self.model,
-            messages=prompt_messages,
-            temperature=0,
-            api_key=settings.openai_api_key,
+            model=self.model, messages=prompt_messages, temperature=0
         )
         return resp["choices"][0]["message"]["content"].strip()