openai_setup implementation fix

need to re-assign self.model after calling setup function
parkervg · Mar 17, 2024 · e117de4 · e117de4
1 parent 915e227
commit e117de4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/blendsql/models/remote/_openai.py b/blendsql/models/remote/_openai.py
@@ -75,7 +75,8 @@ def _load_model(self) -> Model:
         )
 
     def _setup(self, **kwargs) -> None:
-        return openai_setup()
+        openai_setup()
+        self.model = self._load_model()
 
 
 class OpenaiLLM(Model):
@@ -103,4 +104,5 @@ def _load_model(self) -> Model:
         )
 
     def _setup(self, **kwargs) -> None:
-        return openai_setup()
+        openai_setup()
+        self.model = self._load_model()