@smyslenny/agent-memory 4.3.0 → 4.3.1

package/dist/bin/agent-memory.js

@@ -692,7 +692,9 @@ function createLocalHttpEmbeddingProvider(opts) {
   };
 }
 function createGeminiEmbeddingProvider(opts) {
-  const id = stableProviderId(`gemini:${opts.model}`, `${opts.model}|${opts.dimension}`);
+  const baseUrl = trimTrailingSlashes(opts.baseUrl || GEMINI_DEFAULT_BASE_URL);
+  const descriptorInput = `${baseUrl}|${opts.model}|${opts.dimension}`;
+  const id = stableProviderId(`gemini:${opts.model}`, descriptorInput);
   return {
     id,
     model: opts.model,
@@ -700,7 +702,7 @@ function createGeminiEmbeddingProvider(opts) {
     async embed(texts) {
       if (texts.length === 0) return [];
       const fetchFn = getFetch(opts.fetchImpl);
-      const url = `https://generativelanguage.googleapis.com/v1beta/models/${opts.model}:batchEmbedContents?key=${opts.apiKey}`;
+      const url = `${baseUrl}/v1beta/models/${opts.model}:batchEmbedContents?key=${opts.apiKey}`;
       const requests = texts.map((text) => ({
         model: `models/${opts.model}`,
         content: { parts: [{ text }] },
@@ -732,9 +734,11 @@ function createGeminiEmbeddingProvider(opts) {
 function normalizeEmbeddingBaseUrl(baseUrl) {
   return trimTrailingSlashes(baseUrl);
 }
+var GEMINI_DEFAULT_BASE_URL;
 var init_embedding = __esm({
   "src/search/embedding.ts"() {
     "use strict";
+    GEMINI_DEFAULT_BASE_URL = "https://generativelanguage.googleapis.com";
   }
 });
 
@@ -790,6 +794,7 @@ function createEmbeddingProvider(input, opts) {
       model: normalized.model,
       dimension: normalized.dimension,
       apiKey: normalized.apiKey,
+      baseUrl: normalized.baseUrl || void 0,
       fetchImpl: opts?.fetchImpl
     });
   }