From e117de482d05b719d2193ca11acbc01e738f9f13 Mon Sep 17 00:00:00 2001 From: parkervg Date: Sat, 16 Mar 2024 21:19:46 -0400 Subject: [PATCH] openai_setup implementation fix need to re-assign self.model after calling setup function --- blendsql/models/remote/_openai.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/blendsql/models/remote/_openai.py b/blendsql/models/remote/_openai.py index b8911cb2..78a35d7c 100644 --- 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()