modelfusion 0.136.0 → 0.137.0

package/index.cjs

@@ -42,6 +42,7 @@ __export(src_exports, {
   AzureOpenAIApiConfiguration: () => AzureOpenAIApiConfiguration,
   BaseUrlApiConfiguration: () => BaseUrlApiConfiguration,
   BaseUrlApiConfigurationWithDefaults: () => BaseUrlApiConfigurationWithDefaults,
+  CHAT_MODEL_CONTEXT_WINDOW_SIZES: () => CHAT_MODEL_CONTEXT_WINDOW_SIZES,
   COHERE_TEXT_EMBEDDING_MODELS: () => COHERE_TEXT_EMBEDDING_MODELS,
   COHERE_TEXT_GENERATION_MODELS: () => COHERE_TEXT_GENERATION_MODELS,
   ChatMLPrompt: () => ChatMLPromptTemplate_exports,
@@ -81,13 +82,9 @@ __export(src_exports, {
   NeuralChatPrompt: () => NeuralChatPromptTemplate_exports,
   NoSuchToolDefinitionError: () => NoSuchToolDefinitionError,
   OPENAI_CHAT_MESSAGE_BASE_TOKEN_COUNT: () => OPENAI_CHAT_MESSAGE_BASE_TOKEN_COUNT,
-  OPENAI_CHAT_MODELS: () => OPENAI_CHAT_MODELS,
   OPENAI_CHAT_PROMPT_BASE_TOKEN_COUNT: () => OPENAI_CHAT_PROMPT_BASE_TOKEN_COUNT,
-  OPENAI_IMAGE_MODELS: () => OPENAI_IMAGE_MODELS,
-  OPENAI_SPEECH_MODELS: () => OPENAI_SPEECH_MODELS,
   OPENAI_TEXT_EMBEDDING_MODELS: () => OPENAI_TEXT_EMBEDDING_MODELS,
   OPENAI_TEXT_GENERATION_MODELS: () => OPENAI_TEXT_GENERATION_MODELS,
-  OPENAI_TRANSCRIPTION_MODELS: () => OPENAI_TRANSCRIPTION_MODELS,
   ObjectFromTextGenerationModel: () => ObjectFromTextGenerationModel,
   ObjectFromTextStreamingModel: () => ObjectFromTextStreamingModel,
   ObjectGeneratorTool: () => ObjectGeneratorTool,
@@ -147,12 +144,6 @@ __export(src_exports, {
   ZodSchema: () => ZodSchema,
   api: () => ApiFacade_exports,
   automatic1111: () => Automatic1111Facade_exports,
-  calculateOpenAIChatCostInMillicents: () => calculateOpenAIChatCostInMillicents,
-  calculateOpenAICompletionCostInMillicents: () => calculateOpenAICompletionCostInMillicents,
-  calculateOpenAIEmbeddingCostInMillicents: () => calculateOpenAIEmbeddingCostInMillicents,
-  calculateOpenAIImageGenerationCostInMillicents: () => calculateOpenAIImageGenerationCostInMillicents,
-  calculateOpenAISpeechCostInMillicents: () => calculateOpenAISpeechCostInMillicents,
-  calculateOpenAITranscriptionCostInMillicents: () => calculateOpenAITranscriptionCostInMillicents,
   classify: () => classify,
   cohere: () => CohereFacade_exports,
   convertDataContentToBase64String: () => convertDataContentToBase64String,
@@ -183,9 +174,6 @@ __export(src_exports, {
   getOpenAICompletionModelInformation: () => getOpenAICompletionModelInformation,
   getRun: () => getRun,
   huggingface: () => HuggingFaceFacade_exports,
-  isOpenAIChatModel: () => isOpenAIChatModel,
-  isOpenAICompletionModel: () => isOpenAICompletionModel,
-  isOpenAIEmbeddingModel: () => isOpenAIEmbeddingModel,
   isPromptFunction: () => isPromptFunction,
   jsonObjectPrompt: () => jsonObjectPrompt,
   jsonToolCallPrompt: () => jsonToolCallPrompt,
@@ -197,6 +185,7 @@ __export(src_exports, {
   mistral: () => MistralFacade_exports,
   modelfusion: () => ModelFusionConfiguration_exports,
   ollama: () => OllamaFacade_exports,
+  openAITextEmbeddingResponseSchema: () => openAITextEmbeddingResponseSchema2,
   openai: () => OpenAIFacade_exports,
   openaicompatible: () => OpenAICompatibleFacade_exports,
   parseJSON: () => parseJSON,
@@ -288,6 +277,11 @@ var DefaultRun = class {
       this.functionEventSource.notify(event);
     }
   };
+  getSuccessfulModelCalls() {
+    return this.events.filter(
+      (event) => "model" in event && "result" in event && "status" in event.result && event.result.status === "success"
+    );
+  }
 };
 
 // src/core/ModelFusionConfiguration.ts
@@ -8402,134 +8396,45 @@ async function countOpenAIChatPromptTokens({
 }
 
 // src/model-provider/openai/OpenAIChatModel.ts
-var OPENAI_CHAT_MODELS = {
-  "gpt-4": {
-    contextWindowSize: 8192,
-    promptTokenCostInMillicents: 3,
-    completionTokenCostInMillicents: 6
-  },
-  "gpt-4-0314": {
-    contextWindowSize: 8192,
-    promptTokenCostInMillicents: 3,
-    completionTokenCostInMillicents: 6
-  },
-  "gpt-4-0613": {
-    contextWindowSize: 8192,
-    promptTokenCostInMillicents: 3,
-    completionTokenCostInMillicents: 6,
-    fineTunedPromptTokenCostInMillicents: null,
-    fineTunedCompletionTokenCostInMillicents: null
-  },
-  "gpt-4-turbo-preview": {
-    contextWindowSize: 128e3,
-    promptTokenCostInMillicents: 1,
-    completionTokenCostInMillicents: 3
-  },
-  "gpt-4-1106-preview": {
-    contextWindowSize: 128e3,
-    promptTokenCostInMillicents: 1,
-    completionTokenCostInMillicents: 3
-  },
-  "gpt-4-0125-preview": {
-    contextWindowSize: 128e3,
-    promptTokenCostInMillicents: 1,
-    completionTokenCostInMillicents: 3
-  },
-  "gpt-4-vision-preview": {
-    contextWindowSize: 128e3,
-    promptTokenCostInMillicents: 1,
-    completionTokenCostInMillicents: 3
-  },
-  "gpt-4-32k": {
-    contextWindowSize: 32768,
-    promptTokenCostInMillicents: 6,
-    completionTokenCostInMillicents: 12
-  },
-  "gpt-4-32k-0314": {
-    contextWindowSize: 32768,
-    promptTokenCostInMillicents: 6,
-    completionTokenCostInMillicents: 12
-  },
-  "gpt-4-32k-0613": {
-    contextWindowSize: 32768,
-    promptTokenCostInMillicents: 6,
-    completionTokenCostInMillicents: 12
-  },
-  "gpt-3.5-turbo": {
-    contextWindowSize: 4096,
-    promptTokenCostInMillicents: 0.15,
-    completionTokenCostInMillicents: 0.2,
-    fineTunedPromptTokenCostInMillicents: 0.3,
-    fineTunedCompletionTokenCostInMillicents: 0.6
-  },
-  "gpt-3.5-turbo-0125": {
-    contextWindowSize: 16385,
-    promptTokenCostInMillicents: 0.05,
-    completionTokenCostInMillicents: 0.15
-  },
-  "gpt-3.5-turbo-1106": {
-    contextWindowSize: 16385,
-    promptTokenCostInMillicents: 0.1,
-    completionTokenCostInMillicents: 0.2
-  },
-  "gpt-3.5-turbo-0301": {
-    contextWindowSize: 4096,
-    promptTokenCostInMillicents: 0.15,
-    completionTokenCostInMillicents: 0.2
-  },
-  "gpt-3.5-turbo-0613": {
-    contextWindowSize: 4096,
-    promptTokenCostInMillicents: 0.15,
-    completionTokenCostInMillicents: 0.2,
-    fineTunedPromptTokenCostInMillicents: 1.2,
-    fineTunedCompletionTokenCostInMillicents: 1.6
-  },
-  "gpt-3.5-turbo-16k": {
-    contextWindowSize: 16384,
-    promptTokenCostInMillicents: 0.3,
-    completionTokenCostInMillicents: 0.4
-  },
-  "gpt-3.5-turbo-16k-0613": {
-    contextWindowSize: 16384,
-    promptTokenCostInMillicents: 0.3,
-    completionTokenCostInMillicents: 0.4
-  }
+var CHAT_MODEL_CONTEXT_WINDOW_SIZES = {
+  "gpt-4": 8192,
+  "gpt-4-0314": 8192,
+  "gpt-4-0613": 8192,
+  "gpt-4-turbo-preview": 128e3,
+  "gpt-4-1106-preview": 128e3,
+  "gpt-4-0125-preview": 128e3,
+  "gpt-4-vision-preview": 128e3,
+  "gpt-4-32k": 32768,
+  "gpt-4-32k-0314": 32768,
+  "gpt-4-32k-0613": 32768,
+  "gpt-3.5-turbo": 4096,
+  "gpt-3.5-turbo-0125": 16385,
+  "gpt-3.5-turbo-1106": 16385,
+  "gpt-3.5-turbo-0301": 4096,
+  "gpt-3.5-turbo-0613": 4096,
+  "gpt-3.5-turbo-16k": 16384,
+  "gpt-3.5-turbo-16k-0613": 16384
 };
 function getOpenAIChatModelInformation(model) {
-  if (model in OPENAI_CHAT_MODELS) {
-    const baseModelInformation = OPENAI_CHAT_MODELS[model];
+  if (model in CHAT_MODEL_CONTEXT_WINDOW_SIZES) {
+    const contextWindowSize = CHAT_MODEL_CONTEXT_WINDOW_SIZES[model];
     return {
       baseModel: model,
       isFineTuned: false,
-      contextWindowSize: baseModelInformation.contextWindowSize,
-      promptTokenCostInMillicents: baseModelInformation.promptTokenCostInMillicents,
-      completionTokenCostInMillicents: baseModelInformation.completionTokenCostInMillicents
+      contextWindowSize
     };
   }
   const [_, baseModel, ___, ____, _____] = model.split(":");
   if (["gpt-3.5-turbo", "gpt-3.5-turbo-0613", "gpt-4-0613"].includes(baseModel)) {
-    const baseModelInformation = OPENAI_CHAT_MODELS[baseModel];
+    const contextWindowSize = CHAT_MODEL_CONTEXT_WINDOW_SIZES[baseModel];
     return {
       baseModel,
       isFineTuned: true,
-      contextWindowSize: baseModelInformation.contextWindowSize,
-      promptTokenCostInMillicents: baseModelInformation.fineTunedPromptTokenCostInMillicents,
-      completionTokenCostInMillicents: baseModelInformation.fineTunedCompletionTokenCostInMillicents
+      contextWindowSize
     };
   }
   throw new Error(`Unknown OpenAI chat base model ${baseModel}.`);
 }
-var isOpenAIChatModel = (model) => model in OPENAI_CHAT_MODELS || model.startsWith("ft:gpt-3.5-turbo-0613:") || model.startsWith("ft:gpt-3.5-turbo:");
-var calculateOpenAIChatCostInMillicents = ({
-  model,
-  response
-}) => {
-  const { promptTokenCostInMillicents, completionTokenCostInMillicents } = getOpenAIChatModelInformation(model);
-  if (promptTokenCostInMillicents == null || completionTokenCostInMillicents == null) {
-    return null;
-  }
-  return response.usage.prompt_tokens * promptTokenCostInMillicents + response.usage.completion_tokens * completionTokenCostInMillicents;
-};
 var OpenAIChatModel = class _OpenAIChatModel extends AbstractOpenAIChatModel {
   constructor(settings) {
     super(settings);
@@ -8627,22 +8532,12 @@ var OpenAIChatModel = class _OpenAIChatModel extends AbstractOpenAIChatModel {
 // src/model-provider/openai/OpenAICompletionModel.ts
 var OPENAI_TEXT_GENERATION_MODELS = {
   "gpt-3.5-turbo-instruct": {
-    contextWindowSize: 4097,
-    promptTokenCostInMillicents: 0.15,
-    completionTokenCostInMillicents: 0.2
+    contextWindowSize: 4097
   }
 };
 function getOpenAICompletionModelInformation(model) {
   return OPENAI_TEXT_GENERATION_MODELS[model];
 }
-var isOpenAICompletionModel = (model) => model in OPENAI_TEXT_GENERATION_MODELS;
-var calculateOpenAICompletionCostInMillicents = ({
-  model,
-  response
-}) => {
-  const modelInformation = getOpenAICompletionModelInformation(model);
-  return response.usage.prompt_tokens * modelInformation.promptTokenCostInMillicents + response.usage.completion_tokens * modelInformation.completionTokenCostInMillicents;
-};
 var OpenAICompletionModel = class _OpenAICompletionModel extends AbstractOpenAICompletionModel {
   constructor(settings) {
     super(settings);
@@ -8727,60 +8622,6 @@ __export(OpenAIFacade_exports, {
 
 // src/model-provider/openai/OpenAIImageGenerationModel.ts
 var import_zod27 = require("zod");
-var OPENAI_IMAGE_MODELS = {
-  "dall-e-2": {
-    getCost(settings) {
-      switch (settings.size ?? "1024x1024") {
-        case "1024x1024":
-          return 2e3;
-        case "512x512":
-          return 1800;
-        case "256x256":
-          return 1600;
-        default:
-          return null;
-      }
-    }
-  },
-  "dall-e-3": {
-    getCost(settings) {
-      switch (settings.quality ?? "standard") {
-        case "standard": {
-          switch (settings.size ?? "1024x1024") {
-            case "1024x1024":
-              return 4e3;
-            case "1024x1792":
-            case "1792x1024":
-              return 8e3;
-            default:
-              return null;
-          }
-        }
-        case "hd": {
-          switch (settings.size ?? "1024x1024") {
-            case "1024x1024":
-              return 8e3;
-            case "1024x1792":
-            case "1792x1024":
-              return 12e3;
-            default:
-              return null;
-          }
-        }
-      }
-    }
-  }
-};
-var calculateOpenAIImageGenerationCostInMillicents = ({
-  model,
-  settings
-}) => {
-  const cost = OPENAI_IMAGE_MODELS[model]?.getCost(settings);
-  if (cost == null) {
-    return null;
-  }
-  return (settings.numberOfGenerations ?? 1) * cost;
-};
 var OpenAIImageGenerationModel = class _OpenAIImageGenerationModel extends AbstractModel {
   constructor(settings) {
     super({ settings });
@@ -8884,25 +8725,6 @@ var OpenAIImageGenerationResponseFormat = {
 };
 
 // src/model-provider/openai/OpenAISpeechModel.ts
-var OPENAI_SPEECH_MODELS = {
-  "tts-1": {
-    costInMillicentsPerCharacter: 1.5
-    // = 1500 / 1000,
-  },
-  "tts-1-hd": {
-    costInMillicentsPerCharacter: 3
-    // = 3000 / 1000
-  }
-};
-var calculateOpenAISpeechCostInMillicents = ({
-  model,
-  input
-}) => {
-  if (!OPENAI_SPEECH_MODELS[model]) {
-    return null;
-  }
-  return input.length * OPENAI_SPEECH_MODELS[model].costInMillicentsPerCharacter;
-};
 var OpenAISpeechModel = class _OpenAISpeechModel extends AbstractModel {
   constructor(settings) {
     super({ settings });
@@ -8961,34 +8783,37 @@ var OpenAISpeechModel = class _OpenAISpeechModel extends AbstractModel {
 };
 
 // src/model-provider/openai/OpenAITextEmbeddingModel.ts
+var import_zod28 = __toESM(require("zod"), 1);
 var OPENAI_TEXT_EMBEDDING_MODELS = {
   "text-embedding-3-small": {
     contextWindowSize: 8192,
-    dimensions: 1536,
-    tokenCostInMillicents: 2e-3
+    dimensions: 1536
   },
   "text-embedding-3-large": {
     contextWindowSize: 8192,
-    dimensions: 3072,
-    tokenCostInMillicents: 0.013
+    dimensions: 3072
   },
   "text-embedding-ada-002": {
     contextWindowSize: 8192,
-    dimensions: 1536,
-    tokenCostInMillicents: 0.01
-  }
-};
-var isOpenAIEmbeddingModel = (model) => model in OPENAI_TEXT_EMBEDDING_MODELS;
-var calculateOpenAIEmbeddingCostInMillicents = ({
-  model,
-  responses
-}) => {
-  let amountInMilliseconds = 0;
-  for (const response of responses) {
-    amountInMilliseconds += response.usage.total_tokens * OPENAI_TEXT_EMBEDDING_MODELS[model].tokenCostInMillicents;
+    dimensions: 1536
   }
-  return amountInMilliseconds;
 };
+var openAITextEmbeddingResponseSchema2 = import_zod28.default.object({
+  object: import_zod28.default.literal("list"),
+  data: import_zod28.default.array(
+    import_zod28.default.object({
+      object: import_zod28.default.literal("embedding"),
+      embedding: import_zod28.default.array(import_zod28.default.number()),
+      index: import_zod28.default.number()
+    })
+  ),
+  model: import_zod28.default.string(),
+  usage: import_zod28.default.object({
+    prompt_tokens: import_zod28.default.number(),
+    total_tokens: import_zod28.default.number()
+  }).optional()
+  // for openai-compatible models
+});
 var OpenAITextEmbeddingModel = class _OpenAITextEmbeddingModel extends AbstractOpenAITextEmbeddingModel {
   constructor(settings) {
     super(settings);
@@ -9017,7 +8842,7 @@ var OpenAITextEmbeddingModel = class _OpenAITextEmbeddingModel extends AbstractO
 };
 
 // src/model-provider/openai/OpenAITranscriptionModel.ts
-var import_zod28 = require("zod");
+var import_zod29 = require("zod");
 
 // src/util/audio/getAudioFileExtension.ts
 function getAudioFileExtension(mimeType) {
@@ -9047,22 +8872,6 @@ function getAudioFileExtension(mimeType) {
 }
 
 // src/model-provider/openai/OpenAITranscriptionModel.ts
-var OPENAI_TRANSCRIPTION_MODELS = {
-  "whisper-1": {
-    costInMillicentsPerSecond: 10
-    // = 600 / 60,
-  }
-};
-var calculateOpenAITranscriptionCostInMillicents = ({
-  model,
-  response
-}) => {
-  if (model !== "whisper-1") {
-    return null;
-  }
-  const durationInSeconds = response.duration;
-  return Math.ceil(durationInSeconds) * OPENAI_TRANSCRIPTION_MODELS[model].costInMillicentsPerSecond;
-};
 var OpenAITranscriptionModel = class _OpenAITranscriptionModel extends AbstractModel {
   constructor(settings) {
     super({ settings });
@@ -9148,29 +8957,29 @@ var OpenAITranscriptionModel = class _OpenAITranscriptionModel extends AbstractM
     );
   }
 };
-var openAITranscriptionJsonSchema = import_zod28.z.object({
-  text: import_zod28.z.string()
+var openAITranscriptionJsonSchema = import_zod29.z.object({
+  text: import_zod29.z.string()
 });
-var openAITranscriptionVerboseJsonSchema = import_zod28.z.object({
-  task: import_zod28.z.literal("transcribe"),
-  language: import_zod28.z.string(),
-  duration: import_zod28.z.number(),
-  segments: import_zod28.z.array(
-    import_zod28.z.object({
-      id: import_zod28.z.number(),
-      seek: import_zod28.z.number(),
-      start: import_zod28.z.number(),
-      end: import_zod28.z.number(),
-      text: import_zod28.z.string(),
-      tokens: import_zod28.z.array(import_zod28.z.number()),
-      temperature: import_zod28.z.number(),
-      avg_logprob: import_zod28.z.number(),
-      compression_ratio: import_zod28.z.number(),
-      no_speech_prob: import_zod28.z.number(),
-      transient: import_zod28.z.boolean().optional()
+var openAITranscriptionVerboseJsonSchema = import_zod29.z.object({
+  task: import_zod29.z.literal("transcribe"),
+  language: import_zod29.z.string(),
+  duration: import_zod29.z.number(),
+  segments: import_zod29.z.array(
+    import_zod29.z.object({
+      id: import_zod29.z.number(),
+      seek: import_zod29.z.number(),
+      start: import_zod29.z.number(),
+      end: import_zod29.z.number(),
+      text: import_zod29.z.string(),
+      tokens: import_zod29.z.array(import_zod29.z.number()),
+      temperature: import_zod29.z.number(),
+      avg_logprob: import_zod29.z.number(),
+      compression_ratio: import_zod29.z.number(),
+      no_speech_prob: import_zod29.z.number(),
+      transient: import_zod29.z.boolean().optional()
     })
   ),
-  text: import_zod28.z.string()
+  text: import_zod29.z.string()
 });
 var OpenAITranscriptionResponseFormat = {
   json: {
@@ -9510,9 +9319,9 @@ var StabilityApiConfiguration = class extends BaseUrlApiConfigurationWithDefault
 };
 
 // src/model-provider/stability/StabilityError.ts
-var import_zod29 = require("zod");
-var stabilityErrorDataSchema = import_zod29.z.object({
-  message: import_zod29.z.string()
+var import_zod30 = require("zod");
+var stabilityErrorDataSchema = import_zod30.z.object({
+  message: import_zod30.z.string()
 });
 var failedStabilityCallResponseHandler = createJsonErrorResponseHandler({
   errorSchema: zodSchema(stabilityErrorDataSchema),
@@ -9527,7 +9336,7 @@ __export(StabilityFacade_exports, {
 });
 
 // src/model-provider/stability/StabilityImageGenerationModel.ts
-var import_zod30 = require("zod");
+var import_zod31 = require("zod");
 
 // src/model-provider/stability/StabilityImageGenerationPrompt.ts
 function mapBasicPromptToStabilityFormat() {
@@ -9616,12 +9425,12 @@ var StabilityImageGenerationModel = class _StabilityImageGenerationModel extends
     );
   }
 };
-var stabilityImageGenerationResponseSchema = import_zod30.z.object({
-  artifacts: import_zod30.z.array(
-    import_zod30.z.object({
-      base64: import_zod30.z.string(),
-      seed: import_zod30.z.number(),
-      finishReason: import_zod30.z.enum(["SUCCESS", "ERROR", "CONTENT_FILTERED"])
+var stabilityImageGenerationResponseSchema = import_zod31.z.object({
+  artifacts: import_zod31.z.array(
+    import_zod31.z.object({
+      base64: import_zod31.z.string(),
+      seed: import_zod31.z.number(),
+      finishReason: import_zod31.z.enum(["SUCCESS", "ERROR", "CONTENT_FILTERED"])
     })
   )
 });
@@ -9657,7 +9466,7 @@ __export(WhisperCppFacade_exports, {
 });
 
 // src/model-provider/whispercpp/WhisperCppTranscriptionModel.ts
-var import_zod31 = require("zod");
+var import_zod32 = require("zod");
 var WhisperCppTranscriptionModel = class _WhisperCppTranscriptionModel extends AbstractModel {
   constructor(settings) {
     super({ settings });
@@ -9728,9 +9537,9 @@ var WhisperCppTranscriptionModel = class _WhisperCppTranscriptionModel extends A
     );
   }
 };
-var whisperCppTranscriptionJsonSchema = import_zod31.z.union([
-  import_zod31.z.object({ text: import_zod31.z.string() }),
-  import_zod31.z.object({ error: import_zod31.z.string() })
+var whisperCppTranscriptionJsonSchema = import_zod32.z.union([
+  import_zod32.z.object({ text: import_zod32.z.string() }),
+  import_zod32.z.object({ error: import_zod32.z.string() })
 ]);
 var successfulResponseHandler = async ({ response, url, requestBodyValues }) => {
   const responseBody = await response.text();
@@ -10089,14 +9898,14 @@ var ToolExecutionError = class extends Error {
 };
 
 // src/tool/WebSearchTool.ts
-var import_zod32 = require("zod");
+var import_zod33 = require("zod");
 var RETURN_TYPE_SCHEMA = zodSchema(
-  import_zod32.z.object({
-    results: import_zod32.z.array(
-      import_zod32.z.object({
-        title: import_zod32.z.string(),
-        link: import_zod32.z.string().url(),
-        snippet: import_zod32.z.string()
+  import_zod33.z.object({
+    results: import_zod33.z.array(
+      import_zod33.z.object({
+        title: import_zod33.z.string(),
+        link: import_zod33.z.string().url(),
+        snippet: import_zod33.z.string()
       })
     )
   })
@@ -10104,8 +9913,8 @@ var RETURN_TYPE_SCHEMA = zodSchema(
 var createParameters = (description) => (
   // same schema, but with description:
   zodSchema(
-    import_zod32.z.object({
-      query: import_zod32.z.string().describe(description)
+    import_zod33.z.object({
+      query: import_zod33.z.string().describe(description)
     })
   )
 );
@@ -10579,13 +10388,13 @@ var VectorIndexRetriever = class _VectorIndexRetriever {
 };
 
 // src/vector-index/memory/MemoryVectorIndex.ts
-var import_zod33 = require("zod");
+var import_zod34 = require("zod");
 var jsonDataSchema = zodSchema(
-  import_zod33.z.array(
-    import_zod33.z.object({
-      id: import_zod33.z.string(),
-      vector: import_zod33.z.array(import_zod33.z.number()),
-      data: import_zod33.z.unknown()
+  import_zod34.z.array(
+    import_zod34.z.object({
+      id: import_zod34.z.string(),
+      vector: import_zod34.z.array(import_zod34.z.number()),
+      data: import_zod34.z.unknown()
     })
   )
 );
@@ -10684,6 +10493,7 @@ async function upsertIntoVectorIndex({
   AzureOpenAIApiConfiguration,
   BaseUrlApiConfiguration,
   BaseUrlApiConfigurationWithDefaults,
+  CHAT_MODEL_CONTEXT_WINDOW_SIZES,
   COHERE_TEXT_EMBEDDING_MODELS,
   COHERE_TEXT_GENERATION_MODELS,
   ChatMLPrompt,
@@ -10723,13 +10533,9 @@ async function upsertIntoVectorIndex({
   NeuralChatPrompt,
   NoSuchToolDefinitionError,
   OPENAI_CHAT_MESSAGE_BASE_TOKEN_COUNT,
-  OPENAI_CHAT_MODELS,
   OPENAI_CHAT_PROMPT_BASE_TOKEN_COUNT,
-  OPENAI_IMAGE_MODELS,
-  OPENAI_SPEECH_MODELS,
   OPENAI_TEXT_EMBEDDING_MODELS,
   OPENAI_TEXT_GENERATION_MODELS,
-  OPENAI_TRANSCRIPTION_MODELS,
   ObjectFromTextGenerationModel,
   ObjectFromTextStreamingModel,
   ObjectGeneratorTool,
@@ -10789,12 +10595,6 @@ async function upsertIntoVectorIndex({
   ZodSchema,
   api,
   automatic1111,
-  calculateOpenAIChatCostInMillicents,
-  calculateOpenAICompletionCostInMillicents,
-  calculateOpenAIEmbeddingCostInMillicents,
-  calculateOpenAIImageGenerationCostInMillicents,
-  calculateOpenAISpeechCostInMillicents,
-  calculateOpenAITranscriptionCostInMillicents,
   classify,
   cohere,
   convertDataContentToBase64String,
@@ -10825,9 +10625,6 @@ async function upsertIntoVectorIndex({
   getOpenAICompletionModelInformation,
   getRun,
   huggingface,
-  isOpenAIChatModel,
-  isOpenAICompletionModel,
-  isOpenAIEmbeddingModel,
   isPromptFunction,
   jsonObjectPrompt,
   jsonToolCallPrompt,
@@ -10839,6 +10636,7 @@ async function upsertIntoVectorIndex({
   mistral,
   modelfusion,
   ollama,
+  openAITextEmbeddingResponseSchema,
   openai,
   openaicompatible,
   parseJSON,