feat: integrate o1

Author: monilpat

packages/core/src/generation.ts

@@ -158,7 +158,7 @@ export async function generateText({
         elizaLogger.debug(
             `Trimming context to max length of ${max_context_length} tokens.`
         );
-        context = await trimTokens(context, max_context_length, "gpt-4o");
+        context = await trimTokens(context, max_context_length, "o1");
 
         let response: string;
 

packages/core/src/models.ts

@@ -15,7 +15,7 @@ export const models: Models = {
         model: {
             [ModelClass.SMALL]: settings.SMALL_OPENAI_MODEL || "gpt-4o-mini",
             [ModelClass.MEDIUM]: settings.MEDIUM_OPENAI_MODEL || "gpt-4o",
-            [ModelClass.LARGE]: settings.LARGE_OPENAI_MODEL || "gpt-4o",
+            [ModelClass.LARGE]: settings.LARGE_OPENAI_MODEL || "o1",
             [ModelClass.EMBEDDING]: settings.EMBEDDING_OPENAI_MODEL || "text-embedding-3-small",
             [ModelClass.IMAGE]: settings.IMAGE_OPENAI_MODEL || "dall-e-3",
         },
@@ -228,7 +228,7 @@ export const models: Models = {
             [ModelClass.LARGE]:
                 settings.LARGE_REDPILL_MODEL ||
                 settings.REDPILL_MODEL ||
-                "gpt-4o",
+                "o1",
             [ModelClass.EMBEDDING]: "text-embedding-3-small",
         },
     },
@@ -424,7 +424,7 @@ export const models: Models = {
         model: {
             [ModelClass.SMALL]: settings.SMALL_NANOGPT_MODEL || "gpt-4o-mini",
             [ModelClass.MEDIUM]: settings.MEDIUM_NANOGPT_MODEL || "gpt-4o",
-            [ModelClass.LARGE]: settings.LARGE_NANOGPT_MODEL || "gpt-4o",
+            [ModelClass.LARGE]: settings.LARGE_NANOGPT_MODEL || "o1",
         }
     },
     [ModelProviderName.HYPERBOLIC]: {

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

@@ -33,7 +33,7 @@ describe("Model Provider Configuration", () => {
             const openAIModels = models[ModelProviderName.OPENAI].model;
             expect(openAIModels[ModelClass.SMALL]).toBe("gpt-4o-mini");
             expect(openAIModels[ModelClass.MEDIUM]).toBe("gpt-4o");
-            expect(openAIModels[ModelClass.LARGE]).toBe("gpt-4o");
+            expect(openAIModels[ModelClass.LARGE]).toBe("o1");
             expect(openAIModels[ModelClass.EMBEDDING]).toBe("text-embedding-3-small");
             expect(openAIModels[ModelClass.IMAGE]).toBe("dall-e-3");
         });