@smyslenny/agent-memory 4.3.0 → 4.3.1

package/dist/mcp/server.js

@@ -691,7 +691,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,
@@ -699,7 +701,7 @@ function createGeminiEmbeddingProvider(opts) {
     async embed(texts) {
       if (texts.length === 0) return [];
       const fetchFn = getFetch(opts.fetchImpl);
-      const url2 = `https://generativelanguage.googleapis.com/v1beta/models/${opts.model}:batchEmbedContents?key=${opts.apiKey}`;
+      const url2 = `${baseUrl}/v1beta/models/${opts.model}:batchEmbedContents?key=${opts.apiKey}`;
       const requests = texts.map((text) => ({
         model: `models/${opts.model}`,
         content: { parts: [{ text }] },
@@ -731,9 +733,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";
   }
 });
 
@@ -789,6 +793,7 @@ function createEmbeddingProvider(input, opts) {
       model: normalized.model,
       dimension: normalized.dimension,
       apiKey: normalized.apiKey,
+      baseUrl: normalized.baseUrl || void 0,
       fetchImpl: opts?.fetchImpl
     });
   }