mem0ai 2.1.29 → 2.1.30

package/dist/oss/index.d.mts

@@ -106,14 +106,17 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
             apiKey: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodAny]>>;
+            baseURL: z.ZodOptional<z.ZodString>;
         }, "strip", z.ZodTypeAny, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }>;
     }, "strip", z.ZodTypeAny, {
         provider: string;
@@ -121,6 +124,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }, {
         provider: string;
@@ -128,6 +132,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }>;
     vectorStore: z.ZodObject<{
@@ -170,14 +175,17 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodAny]>>;
             modelProperties: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+            baseURL: z.ZodOptional<z.ZodString>;
         }, "strip", z.ZodTypeAny, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }>;
     }, "strip", z.ZodTypeAny, {
         provider: string;
@@ -185,6 +193,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }, {
         provider: string;
@@ -192,6 +201,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }>;
     historyDbPath: z.ZodOptional<z.ZodString>;
@@ -266,6 +276,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     vectorStore: {
@@ -284,6 +295,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     version?: string | undefined;
@@ -315,6 +327,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     vectorStore: {
@@ -333,6 +346,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     version?: string | undefined;
@@ -704,6 +718,30 @@ declare class LangchainVectorStore implements VectorStore {
     initialize(): Promise<void>;
 }
 
+interface VectorizeConfig extends VectorStoreConfig {
+    apiKey?: string;
+    indexName: string;
+    accountId: string;
+}
+declare class VectorizeDB implements VectorStore {
+    private client;
+    private dimensions;
+    private indexName;
+    private accountId;
+    constructor(config: VectorizeConfig);
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    delete(vectorId: string): Promise<void>;
+    deleteCol(): Promise<void>;
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+    private generateUUID;
+    getUserId(): Promise<string>;
+    setUserId(userId: string): Promise<void>;
+    initialize(): Promise<void>;
+}
+
 interface HistoryManager {
     addHistory(memoryId: string, previousValue: string | null, newValue: string | null, action: string, createdAt?: string, updatedAt?: string, isDeleted?: number): Promise<void>;
     getHistory(memoryId: string): Promise<any[]>;
@@ -724,4 +762,4 @@ declare class HistoryManagerFactory {
     static create(provider: string, config: HistoryStoreConfig): HistoryManager;
 }
 
-export { type AddMemoryOptions, AnthropicLLM, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };
+export { type AddMemoryOptions, AnthropicLLM, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult, VectorizeDB };

package/dist/oss/index.d.ts

@@ -106,14 +106,17 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
             apiKey: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodAny]>>;
+            baseURL: z.ZodOptional<z.ZodString>;
         }, "strip", z.ZodTypeAny, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }>;
     }, "strip", z.ZodTypeAny, {
         provider: string;
@@ -121,6 +124,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }, {
         provider: string;
@@ -128,6 +132,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }>;
     vectorStore: z.ZodObject<{
@@ -170,14 +175,17 @@ declare const MemoryConfigSchema: z.ZodObject<{
             apiKey: z.ZodOptional<z.ZodString>;
             model: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodAny]>>;
             modelProperties: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+            baseURL: z.ZodOptional<z.ZodString>;
         }, "strip", z.ZodTypeAny, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }, {
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         }>;
     }, "strip", z.ZodTypeAny, {
         provider: string;
@@ -185,6 +193,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }, {
         provider: string;
@@ -192,6 +201,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     }>;
     historyDbPath: z.ZodOptional<z.ZodString>;
@@ -266,6 +276,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     vectorStore: {
@@ -284,6 +295,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     version?: string | undefined;
@@ -315,6 +327,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     vectorStore: {
@@ -333,6 +346,7 @@ declare const MemoryConfigSchema: z.ZodObject<{
             modelProperties?: Record<string, any> | undefined;
             apiKey?: string | undefined;
             model?: any;
+            baseURL?: string | undefined;
         };
     };
     version?: string | undefined;
@@ -704,6 +718,30 @@ declare class LangchainVectorStore implements VectorStore {
     initialize(): Promise<void>;
 }
 
+interface VectorizeConfig extends VectorStoreConfig {
+    apiKey?: string;
+    indexName: string;
+    accountId: string;
+}
+declare class VectorizeDB implements VectorStore {
+    private client;
+    private dimensions;
+    private indexName;
+    private accountId;
+    constructor(config: VectorizeConfig);
+    insert(vectors: number[][], ids: string[], payloads: Record<string, any>[]): Promise<void>;
+    search(query: number[], limit?: number, filters?: SearchFilters): Promise<VectorStoreResult[]>;
+    get(vectorId: string): Promise<VectorStoreResult | null>;
+    update(vectorId: string, vector: number[], payload: Record<string, any>): Promise<void>;
+    delete(vectorId: string): Promise<void>;
+    deleteCol(): Promise<void>;
+    list(filters?: SearchFilters, limit?: number): Promise<[VectorStoreResult[], number]>;
+    private generateUUID;
+    getUserId(): Promise<string>;
+    setUserId(userId: string): Promise<void>;
+    initialize(): Promise<void>;
+}
+
 interface HistoryManager {
     addHistory(memoryId: string, previousValue: string | null, newValue: string | null, action: string, createdAt?: string, updatedAt?: string, isDeleted?: number): Promise<void>;
     getHistory(memoryId: string): Promise<any[]>;
@@ -724,4 +762,4 @@ declare class HistoryManagerFactory {
     static create(provider: string, config: HistoryStoreConfig): HistoryManager;
 }
 
-export { type AddMemoryOptions, AnthropicLLM, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult };
+export { type AddMemoryOptions, AnthropicLLM, AzureOpenAIEmbedder, type DeleteAllMemoryOptions, type Embedder, EmbedderFactory, type EmbeddingConfig, type Entity, type GetAllMemoryOptions, GoogleEmbedder, GoogleLLM, type GraphStoreConfig, GroqLLM, HistoryManagerFactory, type HistoryStoreConfig, type LLM, type LLMConfig, LLMFactory, type LLMResponse, LangchainEmbedder, LangchainLLM, LangchainVectorStore, Memory, type MemoryConfig, MemoryConfigSchema, type MemoryItem, MemoryVectorStore, type Message, MistralLLM, type MultiModalMessages, type Neo4jConfig, OllamaEmbedder, OllamaLLM, OpenAIEmbedder, OpenAILLM, OpenAIStructuredLLM, Qdrant, RedisDB, type SearchFilters, type SearchMemoryOptions, type SearchResult, SupabaseDB, type VectorStore, type VectorStoreConfig, VectorStoreFactory, type VectorStoreResult, VectorizeDB };

package/dist/oss/index.js

@@ -53,7 +53,8 @@ __export(index_exports, {
   Qdrant: () => Qdrant,
   RedisDB: () => RedisDB,
   SupabaseDB: () => SupabaseDB,
-  VectorStoreFactory: () => VectorStoreFactory
+  VectorStoreFactory: () => VectorStoreFactory,
+  VectorizeDB: () => VectorizeDB
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -70,7 +71,8 @@ var MemoryConfigSchema = import_zod.z.object({
     config: import_zod.z.object({
       modelProperties: import_zod.z.record(import_zod.z.string(), import_zod.z.any()).optional(),
       apiKey: import_zod.z.string().optional(),
-      model: import_zod.z.union([import_zod.z.string(), import_zod.z.any()]).optional()
+      model: import_zod.z.union([import_zod.z.string(), import_zod.z.any()]).optional(),
+      baseURL: import_zod.z.string().optional()
     })
   }),
   vectorStore: import_zod.z.object({
@@ -86,7 +88,8 @@ var MemoryConfigSchema = import_zod.z.object({
     config: import_zod.z.object({
       apiKey: import_zod.z.string().optional(),
       model: import_zod.z.union([import_zod.z.string(), import_zod.z.any()]).optional(),
-      modelProperties: import_zod.z.record(import_zod.z.string(), import_zod.z.any()).optional()
+      modelProperties: import_zod.z.record(import_zod.z.string(), import_zod.z.any()).optional(),
+      baseURL: import_zod.z.string().optional()
     })
   }),
   historyDbPath: import_zod.z.string().optional(),
@@ -892,6 +895,377 @@ var Qdrant = class {
   }
 };
 
+// src/oss/src/vector_stores/vectorize.ts
+var import_cloudflare = __toESM(require("cloudflare"));
+var VectorizeDB = class {
+  constructor(config) {
+    this.client = null;
+    this.client = new import_cloudflare.default({ apiToken: config.apiKey });
+    this.dimensions = config.dimension || 1536;
+    this.indexName = config.indexName;
+    this.accountId = config.accountId;
+    this.initialize().catch(console.error);
+  }
+  async insert(vectors, ids, payloads) {
+    var _a2;
+    try {
+      const vectorObjects = vectors.map(
+        (vector, index) => ({
+          id: ids[index],
+          values: vector,
+          metadata: payloads[index] || {}
+        })
+      );
+      const ndjsonPayload = vectorObjects.map((v) => JSON.stringify(v)).join("\n");
+      const response = await fetch(
+        `https://api.cloudflare.com/client/v4/accounts/${this.accountId}/vectorize/v2/indexes/${this.indexName}/insert`,
+        {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/x-ndjson",
+            Authorization: `Bearer ${(_a2 = this.client) == null ? void 0 : _a2.apiToken}`
+          },
+          body: ndjsonPayload
+        }
+      );
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(
+          `Failed to insert vectors: ${response.status} ${errorText}`
+        );
+      }
+    } catch (error) {
+      console.error("Error inserting vectors:", error);
+      throw new Error(
+        `Failed to insert vectors: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async search(query, limit = 5, filters) {
+    var _a2, _b;
+    try {
+      const result = await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.query(
+        this.indexName,
+        {
+          account_id: this.accountId,
+          vector: query,
+          filter: filters,
+          returnMetadata: "all",
+          topK: limit
+        }
+      ));
+      return ((_b = result == null ? void 0 : result.matches) == null ? void 0 : _b.map((match) => ({
+        id: match.id,
+        payload: match.metadata,
+        score: match.score
+      }))) || [];
+    } catch (error) {
+      console.error("Error searching vectors:", error);
+      throw new Error(
+        `Failed to search vectors: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async get(vectorId) {
+    var _a2;
+    try {
+      const result = await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.getByIds(
+        this.indexName,
+        {
+          account_id: this.accountId,
+          ids: [vectorId]
+        }
+      ));
+      if (!(result == null ? void 0 : result.length)) return null;
+      return {
+        id: vectorId,
+        payload: result[0].metadata
+      };
+    } catch (error) {
+      console.error("Error getting vector:", error);
+      throw new Error(
+        `Failed to get vector: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async update(vectorId, vector, payload) {
+    var _a2;
+    try {
+      const data = {
+        id: vectorId,
+        values: vector,
+        metadata: payload
+      };
+      const response = await fetch(
+        `https://api.cloudflare.com/client/v4/accounts/${this.accountId}/vectorize/v2/indexes/${this.indexName}/upsert`,
+        {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/x-ndjson",
+            Authorization: `Bearer ${(_a2 = this.client) == null ? void 0 : _a2.apiToken}`
+          },
+          body: JSON.stringify(data) + "\n"
+          // ndjson format
+        }
+      );
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(
+          `Failed to update vector: ${response.status} ${errorText}`
+        );
+      }
+    } catch (error) {
+      console.error("Error updating vector:", error);
+      throw new Error(
+        `Failed to update vector: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async delete(vectorId) {
+    var _a2;
+    try {
+      await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.deleteByIds(this.indexName, {
+        account_id: this.accountId,
+        ids: [vectorId]
+      }));
+    } catch (error) {
+      console.error("Error deleting vector:", error);
+      throw new Error(
+        `Failed to delete vector: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async deleteCol() {
+    var _a2;
+    try {
+      await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.delete(this.indexName, {
+        account_id: this.accountId
+      }));
+    } catch (error) {
+      console.error("Error deleting collection:", error);
+      throw new Error(
+        `Failed to delete collection: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async list(filters, limit = 20) {
+    var _a2, _b;
+    try {
+      const result = await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.query(
+        this.indexName,
+        {
+          account_id: this.accountId,
+          vector: Array(this.dimensions).fill(0),
+          // Dummy vector for listing
+          filter: filters,
+          topK: limit,
+          returnMetadata: "all"
+        }
+      ));
+      const matches = ((_b = result == null ? void 0 : result.matches) == null ? void 0 : _b.map((match) => ({
+        id: match.id,
+        payload: match.metadata,
+        score: match.score
+      }))) || [];
+      return [matches, matches.length];
+    } catch (error) {
+      console.error("Error listing vectors:", error);
+      throw new Error(
+        `Failed to list vectors: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  generateUUID() {
+    return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(
+      /[xy]/g,
+      function(c) {
+        const r = Math.random() * 16 | 0;
+        const v = c === "x" ? r : r & 3 | 8;
+        return v.toString(16);
+      }
+    );
+  }
+  async getUserId() {
+    var _a2, _b, _c;
+    try {
+      let found = false;
+      for await (const index of this.client.vectorize.indexes.list({
+        account_id: this.accountId
+      })) {
+        if (index.name === "memory_migrations") {
+          found = true;
+        }
+      }
+      if (!found) {
+        await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.create({
+          account_id: this.accountId,
+          name: "memory_migrations",
+          config: {
+            dimensions: 1,
+            metric: "cosine"
+          }
+        }));
+      }
+      const result = await ((_b = this.client) == null ? void 0 : _b.vectorize.indexes.query(
+        "memory_migrations",
+        {
+          account_id: this.accountId,
+          vector: [0],
+          topK: 1,
+          returnMetadata: "all"
+        }
+      ));
+      if (result.matches.length > 0) {
+        return result.matches[0].metadata.userId;
+      }
+      const randomUserId = Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
+      const data = {
+        id: this.generateUUID(),
+        values: [0],
+        metadata: { userId: randomUserId }
+      };
+      await fetch(
+        `https://api.cloudflare.com/client/v4/accounts/${this.accountId}/vectorize/v2/indexes/memory_migrations/upsert`,
+        {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/x-ndjson",
+            Authorization: `Bearer ${(_c = this.client) == null ? void 0 : _c.apiToken}`
+          },
+          body: JSON.stringify(data) + "\n"
+          // ndjson format
+        }
+      );
+      return randomUserId;
+    } catch (error) {
+      console.error("Error getting user ID:", error);
+      throw new Error(
+        `Failed to get user ID: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async setUserId(userId) {
+    var _a2, _b;
+    try {
+      const result = await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.query(
+        "memory_migrations",
+        {
+          account_id: this.accountId,
+          vector: [0],
+          topK: 1,
+          returnMetadata: "all"
+        }
+      ));
+      const pointId = result.matches.length > 0 ? result.matches[0].id : this.generateUUID();
+      const data = {
+        id: pointId,
+        values: [0],
+        metadata: { userId }
+      };
+      await fetch(
+        `https://api.cloudflare.com/client/v4/accounts/${this.accountId}/vectorize/v2/indexes/memory_migrations/upsert`,
+        {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/x-ndjson",
+            Authorization: `Bearer ${(_b = this.client) == null ? void 0 : _b.apiToken}`
+          },
+          body: JSON.stringify(data) + "\n"
+          // ndjson format
+        }
+      );
+    } catch (error) {
+      console.error("Error setting user ID:", error);
+      throw new Error(
+        `Failed to set user ID: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
+  async initialize() {
+    var _a2, _b, _c, _d, _e;
+    try {
+      let indexFound = false;
+      for await (const idx of this.client.vectorize.indexes.list({
+        account_id: this.accountId
+      })) {
+        if (idx.name === this.indexName) {
+          indexFound = true;
+          break;
+        }
+      }
+      if (!indexFound) {
+        try {
+          await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.create({
+            account_id: this.accountId,
+            name: this.indexName,
+            config: {
+              dimensions: this.dimensions,
+              metric: "cosine"
+            }
+          }));
+          const properties2 = ["userId", "agentId", "runId"];
+          for (const propertyName of properties2) {
+            await ((_b = this.client) == null ? void 0 : _b.vectorize.indexes.metadataIndex.create(
+              this.indexName,
+              {
+                account_id: this.accountId,
+                indexType: "string",
+                propertyName
+              }
+            ));
+          }
+        } catch (err) {
+          throw new Error(err);
+        }
+      }
+      const metadataIndexes = await ((_c = this.client) == null ? void 0 : _c.vectorize.indexes.metadataIndex.list(
+        this.indexName,
+        {
+          account_id: this.accountId
+        }
+      ));
+      const existingMetadataIndexes = /* @__PURE__ */ new Set();
+      for (const metadataIndex of (metadataIndexes == null ? void 0 : metadataIndexes.metadataIndexes) || []) {
+        existingMetadataIndexes.add(metadataIndex.propertyName);
+      }
+      const properties = ["userId", "agentId", "runId"];
+      for (const propertyName of properties) {
+        if (!existingMetadataIndexes.has(propertyName)) {
+          await ((_d = this.client) == null ? void 0 : _d.vectorize.indexes.metadataIndex.create(
+            this.indexName,
+            {
+              account_id: this.accountId,
+              indexType: "string",
+              propertyName
+            }
+          ));
+        }
+      }
+      let found = false;
+      for await (const index of this.client.vectorize.indexes.list({
+        account_id: this.accountId
+      })) {
+        if (index.name === "memory_migrations") {
+          found = true;
+          break;
+        }
+      }
+      if (!found) {
+        await ((_e = this.client) == null ? void 0 : _e.vectorize.indexes.create({
+          account_id: this.accountId,
+          name: "memory_migrations",
+          config: {
+            dimensions: 1,
+            metric: "cosine"
+          }
+        }));
+      }
+    } catch (err) {
+      throw new Error(err);
+    }
+  }
+};
+
 // src/oss/src/vector_stores/redis.ts
 var import_redis = require("redis");
 var DEFAULT_FIELDS = [
@@ -2797,6 +3171,8 @@ var VectorStoreFactory = class {
         return new SupabaseDB(config);
       case "langchain":
         return new LangchainVectorStore(config);
+      case "vectorize":
+        return new VectorizeDB(config);
       default:
         throw new Error(`Unsupported vector store provider: ${provider}`);
     }
@@ -4263,6 +4639,7 @@ ${parsedMessages}`] : getFactRetrievalMessages(parsedMessages);
   Qdrant,
   RedisDB,
   SupabaseDB,
-  VectorStoreFactory
+  VectorStoreFactory,
+  VectorizeDB
 });
 //# sourceMappingURL=index.js.map