Attributes

skllm/llm/gpt/mixin.py

@@ -66,18 +66,20 @@ def _set_keys(self, key: Optional[str] = None, org: Optional[str] = None) -> Non
         """
         Set the OpenAI key and organization.
         """
-        self.openai_key = key
-        self.openai_org = org
+
+        self.key = key
+        self.org = org
+
 
     def _get_openai_key(self) -> str:
         """
         Get the OpenAI key from the class or the config file.
 
         Returns
         -------
-        openai_key: str
+        key: str
         """
-        key = self.openai_key
+        key = self.key
         if key is None:
             key = _Config.get_openai_key()
         if key is None:
@@ -90,14 +92,14 @@ def _get_openai_org(self) -> str:
 
         Returns
         -------
-        openai_org: str
+        org: str
         """
-        key = self.openai_org
-        if key is None:
-            key = _Config.get_openai_org()
-        if key is None:
+        org = self.org
+        if org is None:
+            org = _Config.get_openai_org()
+        if org is None:
             raise RuntimeError("OpenAI organization was not found")
-        return key
+        return org
 
 
 class GPTTextCompletionMixin(GPTMixin, BaseTextCompletionMixin):
@@ -262,4 +264,4 @@ def _tune(self, X, y):
         self.openai_model = job.fine_tuned_model
         self.model = self.openai_model  # openai_model is probably not required anymore
         delete_file(client, job.training_file)
-        print(f"Finished training.")
+        print(f"Finished training.")

skllm/models/gpt/classification/few_shot.py

@@ -129,7 +129,7 @@ def __init__(
             metric used for similarity search, by default "euclidean"
         """
         if vectorizer is None:
-            vectorizer = GPTVectorizer(model="text-embedding-ada-002")
+            vectorizer = GPTVectorizer(model="text-embedding-ada-002", key=key, org=org)
         super().__init__(
             model=model,
             default_label=default_label,