upgrades openai API

Author: WolfgangFahl

ngwidgets/__init__.py

@@ -1 +1 @@
-__version__ = "0.21.2"
+__version__ = "0.22.0"

ngwidgets/llm.py

@@ -13,7 +13,7 @@
 from pathlib import Path
 from typing import List, Optional
 
-import openai
+from openai import OpenAI
 
 from ngwidgets.yamlable import lod_storable
 
@@ -127,6 +127,7 @@ def __init__(
             model(str): the model to use
             prompt_filepath(str): the filepath for the prompt logging
         """
+        self.client=None
         self.model = model
         self.token_size_limit = LLM.MODEL_SIZE_LIMITS.get(
             model, 4096
@@ -135,7 +136,7 @@ def __init__(
         openai_api_key = None
         if api_key:
             # If an API key is provided during object creation, set it.
-            openai.api_key = api_key
+            openai_api_key = api_key
         else:
             # Load the API key from the environment or a JSON file
             openai_api_key = os.getenv("OPENAI_API_KEY")
@@ -153,8 +154,8 @@ def __init__(
                 )
             else:
                 return
-        # set the global api key
-        openai.api_key = openai_api_key
+        # set the client using the  api key
+        self.client=OpenAI(api_key=openai_api_key)
         # If prompts_filepath is None, use default path in the user's home directory with the current date
         if prompts_filepath is None:
             # Format: Year-Month-Day
@@ -187,7 +188,7 @@ def available(self):
         Returns:
             bool: True if the Large Language Model is available
         """
-        return openai.api_key is not None
+        return self.client is not None
 
     def ask(self, prompt_text: str, model: str = None, temperature: float = 0.7) -> str:
         """
@@ -211,9 +212,14 @@ def ask(self, prompt_text: str, model: str = None, temperature: float = 0.7) ->
         start_time = datetime.now()
 
         # Interact with the API
-        chat_completion = openai.chat.completions.create(
+        chat_completion = self.client.chat.completions.create(
             model=model,
-            messages=[{"role": "user", "content": prompt_text}],
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt_text
+                }
+            ],
             temperature=temperature,  # Include the temperature parameter here
         )
         result = chat_completion.choices[0].message.content
@@ -278,7 +284,7 @@ def analyze_image(self, image_path: str, auth: dict, prompt_text: str) -> str:
             }
         ]
 
-        chat_completion = openai.chat.completions.create(
+        chat_completion = self.client.chat.completions.create(
             model=self.model, messages=messages
         )
 

ngwidgets/version.py

@@ -17,7 +17,7 @@ class Version:
     name = "ngwidgets"
     version = ngwidgets.__version__
     date = "2023-09-10"
-    updated = "2025-02-01"
+    updated = "2025-02-15"
     description = "NiceGUI widgets"
 
     authors = "Wolfgang Fahl"

tests/test_open_ai.py

@@ -6,8 +6,7 @@
 
 import unittest
 
-import openai
-
+from openai import OpenAI
 from ngwidgets.basetest import Basetest
 from ngwidgets.llm import LLM
 
@@ -24,7 +23,7 @@ def setUp(self, debug=True, profile=True):
     @unittest.skipIf(Basetest.inPublicCI(), "chatgpt")
     def testOpenAI(self):
         # list models
-        models = openai.Model.list()
+        models = self.llm.client.models.list()
         for model in models.data:
             if self.debug:
                 print(model)
@@ -35,8 +34,8 @@ def testChatGpt(self):
         """
         test the chatgpt interface
         """
-        result = self.llm.ask("What's your model version?")
+        result = self.llm.ask("Who was the first man on the moon?")
         debug = True
         if debug:
             print(result)
-        self.assertTrue("OpenAI" in result)
+        self.assertTrue("Armstrong" in result)