@agentskit/memory 0.7.0 → 0.8.1

package/dist/index.d.ts

@@ -265,6 +265,57 @@ interface SupabaseVectorStoreConfig {
  */
 declare function supabaseVectorStore(config: SupabaseVectorStoreConfig): VectorMemory;
 
+interface WeaviateConfig {
+    /** Cluster URL, e.g. `https://my-cluster.weaviate.network`. */
+    url: string;
+    /** Optional API key (Weaviate Cloud Services). */
+    apiKey?: string;
+    /** Class name in the Weaviate schema. */
+    className: string;
+    topK?: number;
+    fetch?: typeof globalThis.fetch;
+}
+declare function weaviateVectorStore(config: WeaviateConfig): VectorMemory;
+
+interface MilvusConfig {
+    /** Milvus REST endpoint, e.g. `https://in03-xxx.api.gcp-us-west1.zillizcloud.com`. */
+    url: string;
+    /** API key / Zilliz Cloud token (Bearer). */
+    token?: string;
+    collection: string;
+    /** Vector field name in the schema. Default `vector`. */
+    vectorField?: string;
+    topK?: number;
+    fetch?: typeof globalThis.fetch;
+}
+declare function milvusVectorStore(config: MilvusConfig): VectorMemory;
+
+/**
+ * MongoDB Atlas Vector Search adapter. Caller injects a typed collection
+ * shape (drop-in for the official `mongodb` driver's `Collection` type) so
+ * we don't bundle a driver. Atlas' \`$vectorSearch\` aggregation runs
+ * server-side; we just translate \`store\` / \`search\` / \`delete\` to
+ * insertMany + aggregate + deleteMany.
+ */
+interface MongoCollectionLike {
+    insertMany(docs: Array<Record<string, unknown>>, options?: unknown): Promise<unknown>;
+    deleteMany(filter: Record<string, unknown>): Promise<unknown>;
+    aggregate<T = Record<string, unknown>>(pipeline: Array<Record<string, unknown>>): {
+        toArray(): Promise<T[]>;
+    };
+}
+interface MongoAtlasVectorConfig {
+    collection: MongoCollectionLike;
+    /** Atlas Search index name on the embedding field. */
+    indexName: string;
+    /** Field that holds the embedding vector. Default `embedding`. */
+    vectorField?: string;
+    /** numCandidates for $vectorSearch. Default `topK * 10`. */
+    numCandidates?: number;
+    topK?: number;
+}
+declare function mongoAtlasVectorStore(config: MongoAtlasVectorConfig): VectorMemory;
+
 /**
  * Evaluate a `VectorFilter` against a metadata record. Used by in-memory /
  * file-backed vector stores. Backends with native filter languages (pgvector,
@@ -361,4 +412,4 @@ interface HierarchicalMemory extends ChatMemory {
  */
 declare function createHierarchicalMemory(options: HierarchicalMemoryOptions): HierarchicalMemory;
 
-export { type ChromaConfig, type EncryptedEnvelope, type EncryptedMemoryOptions, type FileVectorMemoryConfig, type GraphEdge, type GraphMemory, type GraphNode, type GraphQuery, type HierarchicalMemory, type HierarchicalMemoryOptions, type HierarchicalRecall, type PersonalizationProfile, type PersonalizationStore, type PgVectorConfig, type PgVectorRunner, type PineconeConfig, type QdrantConfig, type RedisChatMemoryConfig, type RedisClientAdapter, type RedisConnectionConfig, type RedisVectorMemoryConfig, type SqliteChatMemoryConfig, type SupabaseVectorStoreConfig, type TursoChatMemoryConfig, type UpstashVectorConfig, type VectorStore, type VectorStoreDocument, type VectorStoreResult, chroma, createEncryptedMemory, createHierarchicalMemory, createInMemoryGraph, createInMemoryPersonalization, fileChatMemory, fileVectorMemory, matchesFilter, pgvector, pinecone, qdrant, redisChatMemory, redisVectorMemory, renderProfileContext, sqliteChatMemory, supabaseVectorStore, tursoChatMemory, upstashVector };
+export { type ChromaConfig, type EncryptedEnvelope, type EncryptedMemoryOptions, type FileVectorMemoryConfig, type GraphEdge, type GraphMemory, type GraphNode, type GraphQuery, type HierarchicalMemory, type HierarchicalMemoryOptions, type HierarchicalRecall, type MilvusConfig, type MongoAtlasVectorConfig, type MongoCollectionLike, type PersonalizationProfile, type PersonalizationStore, type PgVectorConfig, type PgVectorRunner, type PineconeConfig, type QdrantConfig, type RedisChatMemoryConfig, type RedisClientAdapter, type RedisConnectionConfig, type RedisVectorMemoryConfig, type SqliteChatMemoryConfig, type SupabaseVectorStoreConfig, type TursoChatMemoryConfig, type UpstashVectorConfig, type VectorStore, type VectorStoreDocument, type VectorStoreResult, type WeaviateConfig, chroma, createEncryptedMemory, createHierarchicalMemory, createInMemoryGraph, createInMemoryPersonalization, fileChatMemory, fileVectorMemory, matchesFilter, milvusVectorStore, mongoAtlasVectorStore, pgvector, pinecone, qdrant, redisChatMemory, redisVectorMemory, renderProfileContext, sqliteChatMemory, supabaseVectorStore, tursoChatMemory, upstashVector, weaviateVectorStore };

package/dist/index.js

@@ -904,6 +904,167 @@ function supabaseVectorStore(config) {
   };
 }
 
+// src/vector/weaviate.ts
+async function call5(config, method, path, body) {
+  const fetchImpl = config.fetch ?? globalThis.fetch;
+  const response = await fetchImpl(`${config.url}${path}`, {
+    method,
+    headers: {
+      "content-type": "application/json",
+      ...config.apiKey ? { authorization: `Bearer ${config.apiKey}` } : {}
+    },
+    body: body === void 0 ? void 0 : JSON.stringify(body)
+  });
+  const text = await response.text();
+  if (!response.ok) throw new Error(`weaviate ${response.status}: ${text.slice(0, 200)}`);
+  return text.length > 0 ? JSON.parse(text) : {};
+}
+function weaviateVectorStore(config) {
+  const defaultTopK = Math.max(1, config.topK ?? 10);
+  const className = config.className;
+  return {
+    async store(docs) {
+      if (docs.length === 0) return;
+      await call5(config, "POST", "/v1/batch/objects", {
+        objects: docs.map((d) => ({
+          class: className,
+          id: d.id,
+          properties: { content: d.content, ...d.metadata ?? {} },
+          vector: d.embedding
+        }))
+      });
+    },
+    async search(embedding, options = {}) {
+      const topK = options.topK ?? defaultTopK;
+      const threshold = options.threshold ?? 0;
+      const query = `{
+        Get {
+          ${className}(nearVector: { vector: [${embedding.join(",")}] }, limit: ${topK}) {
+            content
+            _additional { id certainty }
+          }
+        }
+      }`;
+      const result = await call5(config, "POST", "/v1/graphql", { query });
+      const rows = result.data?.Get?.[className] ?? [];
+      return rows.map((row) => ({
+        id: String(row._additional?.id ?? ""),
+        content: String(row.content ?? ""),
+        score: row._additional?.certainty ?? 0,
+        metadata: row
+      })).filter((r) => (r.score ?? 0) >= threshold);
+    },
+    async delete(ids) {
+      for (const id of ids) {
+        await call5(config, "DELETE", `/v1/objects/${className}/${id}`);
+      }
+    }
+  };
+}
+
+// src/vector/milvus.ts
+async function call6(config, path, body) {
+  const fetchImpl = config.fetch ?? globalThis.fetch;
+  const response = await fetchImpl(`${config.url}${path}`, {
+    method: "POST",
+    headers: {
+      "content-type": "application/json",
+      ...config.token ? { authorization: `Bearer ${config.token}` } : {}
+    },
+    body: JSON.stringify(body)
+  });
+  const text = await response.text();
+  if (!response.ok) throw new Error(`milvus ${response.status}: ${text.slice(0, 200)}`);
+  return text.length > 0 ? JSON.parse(text) : {};
+}
+function milvusVectorStore(config) {
+  const defaultTopK = Math.max(1, config.topK ?? 10);
+  const vectorField = config.vectorField ?? "vector";
+  return {
+    async store(docs) {
+      if (docs.length === 0) return;
+      await call6(config, "/v2/vectordb/entities/upsert", {
+        collectionName: config.collection,
+        data: docs.map((d) => ({
+          id: d.id,
+          [vectorField]: d.embedding,
+          content: d.content,
+          metadata: d.metadata ?? {}
+        }))
+      });
+    },
+    async search(embedding, options = {}) {
+      const topK = options.topK ?? defaultTopK;
+      const threshold = options.threshold ?? 0;
+      const result = await call6(config, "/v2/vectordb/entities/search", {
+        collectionName: config.collection,
+        data: [embedding],
+        annsField: vectorField,
+        limit: topK,
+        outputFields: ["content", "metadata"]
+      });
+      return (result.data ?? []).map((m) => ({
+        id: String(m.id),
+        content: String(m.content ?? ""),
+        score: 1 - m.distance,
+        metadata: m.metadata
+      })).filter((r) => (r.score ?? 0) >= threshold);
+    },
+    async delete(ids) {
+      if (ids.length === 0) return;
+      await call6(config, "/v2/vectordb/entities/delete", {
+        collectionName: config.collection,
+        filter: `id in [${ids.map((id) => `"${id}"`).join(",")}]`
+      });
+    }
+  };
+}
+
+// src/vector/mongo-atlas.ts
+function mongoAtlasVectorStore(config) {
+  const defaultTopK = Math.max(1, config.topK ?? 10);
+  const vectorField = config.vectorField ?? "embedding";
+  return {
+    async store(docs) {
+      if (docs.length === 0) return;
+      await config.collection.insertMany(docs.map((d) => ({
+        _id: d.id,
+        content: d.content,
+        [vectorField]: d.embedding,
+        metadata: d.metadata ?? {}
+      })));
+    },
+    async search(embedding, options = {}) {
+      const topK = options.topK ?? defaultTopK;
+      const threshold = options.threshold ?? 0;
+      const numCandidates = config.numCandidates ?? topK * 10;
+      const cursor = config.collection.aggregate([
+        {
+          $vectorSearch: {
+            index: config.indexName,
+            path: vectorField,
+            queryVector: embedding,
+            numCandidates,
+            limit: topK
+          }
+        },
+        { $project: { content: 1, metadata: 1, score: { $meta: "vectorSearchScore" } } }
+      ]);
+      const rows = await cursor.toArray();
+      return rows.map((row) => ({
+        id: String(row._id),
+        content: String(row.content ?? ""),
+        score: row.score,
+        metadata: row.metadata
+      })).filter((r) => (r.score ?? 0) >= threshold);
+    },
+    async delete(ids) {
+      if (ids.length === 0) return;
+      await config.collection.deleteMany({ _id: { $in: ids } });
+    }
+  };
+}
+
 // src/encrypted.ts
 function toBase64(bytes) {
   if (typeof Buffer !== "undefined") return Buffer.from(bytes).toString("base64");
@@ -1046,6 +1207,6 @@ function createHierarchicalMemory(options) {
   };
 }
 
-export { chroma, createEncryptedMemory, createHierarchicalMemory, createInMemoryGraph, createInMemoryPersonalization, fileChatMemory, fileVectorMemory, matchesFilter, pgvector, pinecone, qdrant, redisChatMemory, redisVectorMemory, renderProfileContext, sqliteChatMemory, supabaseVectorStore, tursoChatMemory, upstashVector };
+export { chroma, createEncryptedMemory, createHierarchicalMemory, createInMemoryGraph, createInMemoryPersonalization, fileChatMemory, fileVectorMemory, matchesFilter, milvusVectorStore, mongoAtlasVectorStore, pgvector, pinecone, qdrant, redisChatMemory, redisVectorMemory, renderProfileContext, sqliteChatMemory, supabaseVectorStore, tursoChatMemory, upstashVector, weaviateVectorStore };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map