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/