add llm client option for AI SDK experimental telemetry config

lib/api.ts

@@ -18,7 +18,7 @@ import {
   ObserveOptions,
   ObserveResult,
 } from "../types/stagehand";
-import { AgentExecuteOptions, AgentResult } from ".";
+import { AgentExecuteOptions, AgentResult } from "../types/agent";
 import {
   StagehandAPIUnauthorizedError,
   StagehandHttpError,

lib/llm/LLMProvider.ts

@@ -172,6 +172,7 @@ export class LLMProvider {
         logger: this.logger,
         enableCaching: this.enableCaching,
         cache: this.cache,
+        telemetrySettings: clientOptions?.aiSdkTelemetrySettings,
       });
     }
 

lib/llm/aisdk.ts

@@ -15,30 +15,35 @@ import { LogLine } from "../../types/log";
 import { AvailableModel } from "../../types/model";
 import { ChatCompletion } from "openai/resources";
 import { LLMCache } from "../cache/LLMCache";
+import type { TelemetrySettings } from "ai";
 
 export class AISdkClient extends LLMClient {
   public type = "aisdk" as const;
   private model: LanguageModel;
   private logger?: (message: LogLine) => void;
   private cache: LLMCache | undefined;
   private enableCaching: boolean;
+  private telemetrySettings: TelemetrySettings | undefined;
 
   constructor({
     model,
     logger,
     enableCaching = false,
     cache,
+    telemetrySettings,
   }: {
     model: LanguageModel;
     logger?: (message: LogLine) => void;
     enableCaching?: boolean;
     cache?: LLMCache;
+    telemetrySettings?: TelemetrySettings;
   }) {
     super(model.modelId as AvailableModel);
     this.model = model;
     this.logger = logger;
     this.cache = cache;
     this.enableCaching = enableCaching;
+    this.telemetrySettings = telemetrySettings;
   }
 
   async createChatCompletion<T = ChatCompletion>({
@@ -161,6 +166,7 @@ export class AISdkClient extends LLMClient {
         model: this.model,
         messages: formattedMessages,
         schema: options.response_model.schema,
+        experimental_telemetry: this.telemetrySettings,
       });
 
       const result = {
@@ -227,6 +233,7 @@ export class AISdkClient extends LLMClient {
       model: this.model,
       messages: formattedMessages,
       tools,
+      experimental_telemetry: this.telemetrySettings,
     });
 
     const result = {

types/model.ts

@@ -1,4 +1,5 @@
 import type { ClientOptions as AnthropicClientOptions } from "@anthropic-ai/sdk";
+import type { TelemetrySettings } from "ai";
 import type { ClientOptions as OpenAIClientOptions } from "openai";
 import { z } from "zod";
 
@@ -44,7 +45,9 @@ export type ModelProvider =
   | "google"
   | "aisdk";
 
-export type ClientOptions = OpenAIClientOptions | AnthropicClientOptions;
+export type ClientOptions = (OpenAIClientOptions | AnthropicClientOptions) & {
+  aiSdkTelemetrySettings?: TelemetrySettings;
+};
 
 export interface AnthropicJsonSchemaObject {
   definitions?: {