@agentskit/memory 0.7.0 → 0.8.1

package/dist/index.cjs

@@ -906,6 +906,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");
@@ -1056,6 +1217,8 @@ exports.createInMemoryPersonalization = createInMemoryPersonalization;
 exports.fileChatMemory = fileChatMemory;
 exports.fileVectorMemory = fileVectorMemory;
 exports.matchesFilter = matchesFilter;
+exports.milvusVectorStore = milvusVectorStore;
+exports.mongoAtlasVectorStore = mongoAtlasVectorStore;
 exports.pgvector = pgvector;
 exports.pinecone = pinecone;
 exports.qdrant = qdrant;
@@ -1066,5 +1229,6 @@ exports.sqliteChatMemory = sqliteChatMemory;
 exports.supabaseVectorStore = supabaseVectorStore;
 exports.tursoChatMemory = tursoChatMemory;
 exports.upstashVector = upstashVector;
+exports.weaviateVectorStore = weaviateVectorStore;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map