add o1 to model

.changeset/swift-hats-explode.md

@@ -0,0 +1,5 @@
+---
+"@browserbasehq/stagehand": patch
+---
+
+add o1 model support

lib/cache/BaseCache.ts

@@ -52,13 +52,14 @@ export class BaseCache<T extends CacheEntry> {
     process.on("SIGINT", releaseLockAndExit);
     process.on("SIGTERM", releaseLockAndExit);
     process.on("uncaughtException", (err) => {
+      console.log("uncaught exception", err);
       this.logger({
         category: "base_cache",
         message: "uncaught exception",
         level: 2,
         auxiliary: {
           error: {
-            value: err.message,
+            value: JSON.stringify(err),
             type: "string",
           },
           trace: {

lib/handlers/actHandler.ts

@@ -84,7 +84,7 @@ export class StagehandActHandler {
     if (
       llmClient.modelName === "o1-mini" ||
       llmClient.modelName === "o1-preview" ||
-      llmClient.modelName.startsWith("o1-")
+      llmClient.modelName === "o1"
     ) {
       verifyLLmClient = this.llmProvider.getClient(
         "gpt-4o",

lib/llm/LLMProvider.ts

@@ -16,6 +16,7 @@ export class LLMProvider {
     "gpt-4o-2024-08-06": "openai",
     "o1-mini": "openai",
     "o1-preview": "openai",
+    o1: "openai",
     "claude-3-5-sonnet-latest": "anthropic",
     "claude-3-5-sonnet-20240620": "anthropic",
     "claude-3-5-sonnet-20241022": "anthropic",

lib/llm/OpenAIClient.ts

@@ -100,9 +100,19 @@ export class OpenAIClient extends LLMClient {
         });
       }
     }
+
+    if (this.modelName === "o1") {
+      delete options.temperature;
+      delete options.top_p;
+      delete options.frequency_penalty;
+      delete options.presence_penalty;
+    }
+
     if (
       options.temperature &&
-      (this.modelName === "o1-mini" || this.modelName === "o1-preview")
+      (this.modelName === "o1-mini" ||
+        this.modelName === "o1-preview" ||
+        this.modelName === "o1")
     ) {
       throw new Error("Temperature is not supported for o1 models");
     }
@@ -411,6 +421,22 @@ export class OpenAIClient extends LLMClient {
         );
       }
 
+      this.logger({
+        category: "openai",
+        message: "parsed response",
+        level: 1,
+        auxiliary: {
+          parsedData: {
+            value: JSON.stringify(parsedData),
+            type: "object",
+          },
+          requestId: {
+            value: requestId,
+            type: "string",
+          },
+        },
+      });
+
       return parsedData;
     }
 
@@ -437,6 +463,22 @@ export class OpenAIClient extends LLMClient {
       this.cache.set(cacheOptions, response, options.requestId);
     }
 
+    this.logger({
+      category: "openai",
+      message: "final response",
+      level: 1,
+      auxiliary: {
+        response: {
+          value: JSON.stringify(response),
+          type: "object",
+        },
+        requestId: {
+          value: requestId,
+          type: "string",
+        },
+      },
+    });
+
     // if the function was called with a response model, it would have returned earlier
     // so we can safely cast here to T, which defaults to ChatCompletion
     return response as T;

types/model.ts

@@ -12,6 +12,7 @@ export const AvailableModelSchema = z.enum([
   "claude-3-5-sonnet-20240620",
   "o1-mini",
   "o1-preview",
+  "o1",
 ]);
 
 export type AvailableModel = z.infer<typeof AvailableModelSchema>;