Chat GPT Constructor: Consider Model when caching

biolab · Dec 20, 2023 · 90a39f7 · 90a39f7
1 parent b4b9c17
commit 90a39f7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/orangecontrib/prototypes/widgets/owchatgptconstructor.py b/orangecontrib/prototypes/widgets/owchatgptconstructor.py
@@ -48,15 +48,15 @@ def ask_gpt(self, state) -> List:
             if state.is_interruption_requested():
                 raise Exception
 
-            args = (text.strip(),
+            args = (MODELS[self.model_index],
+                    text.strip(),
                     self.prompt_start.strip(),
                     self.prompt_end.strip())
             if args in self.cache:
                 answer = self.cache[args]
             else:
                 try:
-                    answer = run_gpt(self.access_key, MODELS[self.model_index],
-                                     *args)
+                    answer = run_gpt(self.access_key, *args)
                     self.cache[args] = answer
                 except Exception as ex:
                     answer = ex