pacnpal/Roo-Code
1
0
Fork 0
mirror of https://github.com/pacnpal/Roo-Code.git synced 2025-12-20 12:21:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add the o1 model

Browse Source
This commit is contained in:
Matt Rubens
2024-12-20 09:45:30 -05:00
parent 537514de44
commit 0aa5f1fb85
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/api/providers/openai-native.ts
View File

@@ -24,6 +24,7 @@ export class OpenAiNativeHandler implements ApiHandler {
async *createMessage(systemPrompt: string, messages: Anthropic.Messages.MessageParam[]): ApiStream { async *createMessage(systemPrompt: string, messages: Anthropic.Messages.MessageParam[]): ApiStream {
switch (this.getModel().id) { switch (this.getModel().id) {
case "o1":
case "o1-preview": case "o1-preview":
case "o1-mini": { case "o1-mini": {
// o1 doesnt support streaming, non-1 temp, or system prompt // o1 doesnt support streaming, non-1 temp, or system prompt

8
src/shared/api.ts
View File

@@ -481,6 +481,14 @@ export type OpenAiNativeModelId = keyof typeof openAiNativeModels
export const openAiNativeDefaultModelId: OpenAiNativeModelId = "gpt-4o" export const openAiNativeDefaultModelId: OpenAiNativeModelId = "gpt-4o"
export const openAiNativeModels = { export const openAiNativeModels = {
// don't support tool use yet // don't support tool use yet
"o1": {
maxTokens: 100_000,
contextWindow: 200_000,
supportsImages: true,
supportsPromptCache: false,
inputPrice: 15,
outputPrice: 60,
},
"o1-preview": { "o1-preview": {
maxTokens: 32_768, maxTokens: 32_768,
contextWindow: 128_000, contextWindow: 128_000,