From 90d16c35e2e8dae79f42599bda41d37065200dfd Mon Sep 17 00:00:00 2001
From: jelveh <nj@puter.com>
Date: Mon, 14 Apr 2025 12:36:07 -0700
Subject: [PATCH] Add missing openai models

---
 .../modules/puterai/OpenAICompletionService.js | 18 ++++++++++++++++++
 src/puter-js/src/modules/AI.js                 |  6 ++++++
 2 files changed, 24 insertions(+)

diff --git a/src/backend/src/modules/puterai/OpenAICompletionService.js b/src/backend/src/modules/puterai/OpenAICompletionService.js
index 68a396fc7..2aeee91ca 100644
--- a/src/backend/src/modules/puterai/OpenAICompletionService.js
+++ b/src/backend/src/modules/puterai/OpenAICompletionService.js
@@ -142,6 +142,24 @@ class OpenAICompletionService extends BaseService {
                     output: 1200,
                 }
             },
+            {
+                id: 'o1-pro',
+                cost: {
+                    currency: 'usd-cents',
+                    tokens: 1_000_000,
+                    input: 15000,
+                    output: 60000,
+                }
+            },
+            {
+                id: 'o1',
+                cost: {
+                    currency: 'usd-cents',
+                    tokens: 1_000_000,
+                    input: 1500,
+                    output: 6000,
+                }
+            },
             {
                 id: 'o3-mini',
                 cost: {
diff --git a/src/puter-js/src/modules/AI.js b/src/puter-js/src/modules/AI.js
index 591824e31..7cc71cf59 100644
--- a/src/puter-js/src/modules/AI.js
+++ b/src/puter-js/src/modules/AI.js
@@ -309,6 +309,12 @@ class AI{
             requestParams.model = 'openrouter:openai/o1-mini';
         }
 
+        // if a model is prepended with "openai/", remove it
+        if (requestParams.model && requestParams.model.startsWith('openai/')) {
+            requestParams.model = requestParams.model.replace('openai/', '');
+            driver = 'openai-completion';
+        }
+
         // if model starts with: 
         //      meta-llama/
         //      google/