@launchdarkly/server-sdk-ai 0.9.8 → 0.10.0

package/src/api/config/VercelAISDK.ts

@@ -0,0 +1,20 @@
+import { type LDMessage } from './LDAIConfig';
+
+export type VercelAISDKProvider<TMod> = (modelName: string) => TMod;
+
+export interface VercelAISDKMapOptions {
+  nonInterpolatedMessages?: LDMessage[] | undefined;
+}
+
+export interface VercelAISDKConfig<TMod> {
+  model: TMod;
+  messages?: LDMessage[] | undefined;
+  maxTokens?: number | undefined;
+  temperature?: number | undefined;
+  topP?: number | undefined;
+  topK?: number | undefined;
+  presencePenalty?: number | undefined;
+  frequencyPenalty?: number | undefined;
+  stopSequences?: string[] | undefined;
+  seed?: number | undefined;
+}

package/src/api/config/index.ts

@@ -1,2 +1,3 @@
 export * from './LDAIConfig';
+export * from './VercelAISDK';
 export { LDAIConfigTracker } from './LDAIConfigTracker';

package/src/api/metrics/VercelAISDKTokenUsage.ts

@@ -0,0 +1,13 @@
+import { LDTokenUsage } from './LDTokenUsage';
+
+export function createVercelAISDKTokenUsage(data: {
+  totalTokens?: number;
+  promptTokens?: number;
+  completionTokens?: number;
+}): LDTokenUsage {
+  return {
+    total: data.totalTokens ?? 0,
+    input: data.promptTokens ?? 0,
+    output: data.completionTokens ?? 0,
+  };
+}

package/src/api/metrics/index.ts

@@ -1,3 +1,5 @@
 export * from './BedrockTokenUsage';
+export * from './OpenAiUsage';
 export * from './LDFeedbackKind';
 export * from './LDTokenUsage';
+export * from './VercelAISDKTokenUsage';