Skip to content

Commit 8e5d69c

Browse files
committed
feat: integrate o1
1 parent 81d0273 commit 8e5d69c

File tree

4 files changed

+21934
-16984
lines changed

4 files changed

+21934
-16984
lines changed

packages/core/src/generation.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ export async function generateText({
158158
elizaLogger.debug(
159159
`Trimming context to max length of ${max_context_length} tokens.`
160160
);
161-
context = await trimTokens(context, max_context_length, "gpt-4o");
161+
context = await trimTokens(context, max_context_length, "o1");
162162

163163
let response: string;
164164

packages/core/src/models.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ export const models: Models = {
1515
model: {
1616
[ModelClass.SMALL]: settings.SMALL_OPENAI_MODEL || "gpt-4o-mini",
1717
[ModelClass.MEDIUM]: settings.MEDIUM_OPENAI_MODEL || "gpt-4o",
18-
[ModelClass.LARGE]: settings.LARGE_OPENAI_MODEL || "gpt-4o",
18+
[ModelClass.LARGE]: settings.LARGE_OPENAI_MODEL || "o1",
1919
[ModelClass.EMBEDDING]: settings.EMBEDDING_OPENAI_MODEL || "text-embedding-3-small",
2020
[ModelClass.IMAGE]: settings.IMAGE_OPENAI_MODEL || "dall-e-3",
2121
},
@@ -228,7 +228,7 @@ export const models: Models = {
228228
[ModelClass.LARGE]:
229229
settings.LARGE_REDPILL_MODEL ||
230230
settings.REDPILL_MODEL ||
231-
"gpt-4o",
231+
"o1",
232232
[ModelClass.EMBEDDING]: "text-embedding-3-small",
233233
},
234234
},
@@ -424,7 +424,7 @@ export const models: Models = {
424424
model: {
425425
[ModelClass.SMALL]: settings.SMALL_NANOGPT_MODEL || "gpt-4o-mini",
426426
[ModelClass.MEDIUM]: settings.MEDIUM_NANOGPT_MODEL || "gpt-4o",
427-
[ModelClass.LARGE]: settings.LARGE_NANOGPT_MODEL || "gpt-4o",
427+
[ModelClass.LARGE]: settings.LARGE_NANOGPT_MODEL || "o1",
428428
}
429429
},
430430
[ModelProviderName.HYPERBOLIC]: {

packages/core/src/tests/models.test.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ describe("Model Provider Configuration", () => {
3333
const openAIModels = models[ModelProviderName.OPENAI].model;
3434
expect(openAIModels[ModelClass.SMALL]).toBe("gpt-4o-mini");
3535
expect(openAIModels[ModelClass.MEDIUM]).toBe("gpt-4o");
36-
expect(openAIModels[ModelClass.LARGE]).toBe("gpt-4o");
36+
expect(openAIModels[ModelClass.LARGE]).toBe("o1");
3737
expect(openAIModels[ModelClass.EMBEDDING]).toBe("text-embedding-3-small");
3838
expect(openAIModels[ModelClass.IMAGE]).toBe("dall-e-3");
3939
});

0 commit comments

Comments
 (0)