Merge pull request #59 from iryna-kondr/fix-gpt

fix gpt prediction function

Author: iryna-kondr

pyproject.toml

@@ -11,7 +11,7 @@ dependencies = [
   "google-cloud-aiplatform>=1.27.0"
 ]
 name = "scikit-llm"
-version = "0.4.0"
+version = "0.4.1"
 authors = [
   { name="Oleg Kostromin", email="[email protected]" },
   { name="Iryna Kondrashchenko", email="[email protected]" },

skllm/models/gpt/gpt.py

@@ -149,6 +149,10 @@ def _get_prompt(self, x: str) -> str:
     def _build_label(self, label: str):
         return label
 
+    def _predict_single(self, x):
+        completion = self._get_chat_completion(x)
+        return completion["choices"][0]["message"]["content"]
+
     def fit(
         self,
         X: Union[np.ndarray, pd.Series, List[str]],