feat: Added gpt-4o-mini support

Upsonic · Oct 14, 2024 · 89a4171 · 89a4171
1 parent 9ed820c
commit 89a4171
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/upsonic_on_prem/api/utils/ai/ai.py b/upsonic_on_prem/api/utils/ai/ai.py
@@ -229,6 +229,8 @@ def completion(self, input_text, model):
             result = self.gpt(input_text, model=model)
         elif model == "gpt-4o":
             result = self.gpt(input_text, model=model)
+        elif model == "gpt-4o-mini":
+            result = self.gpt(input_text, model=model)
 
         if active_ai_history:
             save_ai_call(input_text, result, model)