@olane/o-tool-registry 0.6.12 → 0.7.1

package/dist/src/vector-store/vector-memory.tool.d.ts

@@ -1,8 +1,9 @@
-import { oVirtualTool, ToolResult } from '@olane/o-tool';
+import { ToolResult } from '@olane/o-tool';
 import { oRequest } from '@olane/o-core';
-import { oToolConfig } from '@olane/o-tool';
-export declare abstract class VectorMemoryStorageTool extends oVirtualTool {
-    constructor(config: oToolConfig);
+import { oLaneTool } from '@olane/o-lane';
+import { oNodeToolConfig } from '@olane/o-node';
+export declare abstract class VectorMemoryStorageTool extends oLaneTool {
+    constructor(config: oNodeToolConfig);
     abstract _tool_search_similar(request: oRequest): Promise<ToolResult>;
     abstract _tool_add_documents(request: oRequest): Promise<ToolResult>;
     abstract _tool_delete_documents(request: oRequest): Promise<ToolResult>;

package/dist/src/vector-store/vector-memory.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vector-memory.tool.d.ts","sourceRoot":"","sources":["../../../src/vector-store/vector-memory.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,EAAY,QAAQ,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAG5C,8BAAsB,uBAAwB,SAAQ,YAAY;gBACpD,MAAM,EAAE,WAAW;IAS/B,QAAQ,CAAC,oBAAoB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACrE,QAAQ,CAAC,mBAAmB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACpE,QAAQ,CAAC,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACvE,QAAQ,CAAC,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;CACxE"}
+{"version":3,"file":"vector-memory.tool.d.ts","sourceRoot":"","sources":["../../../src/vector-store/vector-memory.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,EAAY,QAAQ,EAAE,MAAM,eAAe,CAAC;AAGnD,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAEhD,8BAAsB,uBAAwB,SAAQ,SAAS;gBACjD,MAAM,EAAE,eAAe;IASnC,QAAQ,CAAC,oBAAoB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACrE,QAAQ,CAAC,mBAAmB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACpE,QAAQ,CAAC,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACvE,QAAQ,CAAC,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;CACxE"}

package/dist/src/vector-store/vector-memory.tool.js

@@ -1,7 +1,7 @@
-import { oVirtualTool } from '@olane/o-tool';
 import { oAddress } from '@olane/o-core';
 import { VECTOR_STORE_PARAMS } from './methods/vector-store.methods.js';
-export class VectorMemoryStorageTool extends oVirtualTool {
+import { oLaneTool } from '@olane/o-lane';
+export class VectorMemoryStorageTool extends oLaneTool {
     constructor(config) {
         super({
             ...config,

package/dist/vector-store/index.d.ts

@@ -0,0 +1,3 @@
+export * from './langchain-memory.vector-store.tool.js';
+export * from './vector-memory.tool.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/vector-store/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vector-store/index.ts"],"names":[],"mappings":"AAAA,cAAc,yCAAyC,CAAC;AACxD,cAAc,yBAAyB,CAAC"}

package/dist/vector-store/index.js

@@ -0,0 +1,2 @@
+export * from './langchain-memory.vector-store.tool.js';
+export * from './vector-memory.tool.js';

package/dist/vector-store/langchain-memory.vector-store.tool.d.ts

@@ -0,0 +1,14 @@
+import { oRequest } from '@olane/o-core';
+import { oToolConfig } from '@olane/o-tool';
+import { VectorMemoryStorageTool } from './vector-memory.tool.js';
+export declare class LangchainMemoryVectorStoreTool extends VectorMemoryStorageTool {
+    private vectorStore;
+    constructor(config: oToolConfig);
+    private embeddingsTool;
+    initialize(): Promise<void>;
+    _tool_add_documents(request: oRequest): Promise<any>;
+    _tool_delete_documents(request: oRequest): Promise<any>;
+    _tool_update_documents(request: oRequest): Promise<any>;
+    _tool_search_similar(request: oRequest): Promise<any>;
+}
+//# sourceMappingURL=langchain-memory.vector-store.tool.d.ts.map

package/dist/vector-store/langchain-memory.vector-store.tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain-memory.vector-store.tool.d.ts","sourceRoot":"","sources":["../../src/vector-store/langchain-memory.vector-store.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAY,QAAQ,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AAKlE,qBAAa,8BAA+B,SAAQ,uBAAuB;IACzE,OAAO,CAAC,WAAW,CAAqB;gBAE5B,MAAM,EAAE,WAAW;IAO/B,OAAO,CAAC,cAAc;IAwBhB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAK3B,mBAAmB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC;IAOpD,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC;IAIvD,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC;IAIvD,oBAAoB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC;CAQ5D"}

package/dist/vector-store/langchain-memory.vector-store.tool.js

@@ -0,0 +1,60 @@
+import { MemoryVectorStore } from 'langchain/vectorstores/memory';
+import { oAddress } from '@olane/o-core';
+import { VectorMemoryStorageTool } from './vector-memory.tool.js';
+import { Document } from '@langchain/core/documents';
+import { VECTOR_STORE_PARAMS } from './methods/vector-store.methods.js';
+export class LangchainMemoryVectorStoreTool extends VectorMemoryStorageTool {
+    constructor(config) {
+        super({
+            ...config,
+            methods: VECTOR_STORE_PARAMS,
+        });
+    }
+    embeddingsTool() {
+        return {
+            embedDocuments: async (documents) => {
+                const response = await this.use(new oAddress('o://embeddings-text'), {
+                    method: 'embed_documents',
+                    params: {
+                        documents,
+                    },
+                });
+                return response.result.data;
+            },
+            embedQuery: async (document) => {
+                console.log('Embedding query: ', document);
+                const response = await this.use(new oAddress('o://embeddings-text'), {
+                    method: 'embed_query',
+                    params: {
+                        query: document,
+                    },
+                });
+                return response.result.data;
+            },
+        };
+    }
+    async initialize() {
+        await super.initialize();
+        this.vectorStore = new MemoryVectorStore(this.embeddingsTool());
+    }
+    async _tool_add_documents(request) {
+        const { documents } = request.params;
+        const docs = documents.map((doc) => new Document(doc));
+        const result = await this.vectorStore.addDocuments(docs);
+        return result;
+    }
+    async _tool_delete_documents(request) {
+        throw new Error('Not implemented');
+    }
+    async _tool_update_documents(request) {
+        throw new Error('Not implemented');
+    }
+    async _tool_search_similar(request) {
+        const params = request.params;
+        console.log('Search similar params: ', params);
+        const query = params.query;
+        const limit = params.limit;
+        const results = await this.vectorStore.similaritySearch(query, limit);
+        return results;
+    }
+}

package/dist/vector-store/methods/vector-store.methods.d.ts

@@ -0,0 +1,5 @@
+import { oMethod } from '@olane/o-protocol';
+export declare const VECTOR_STORE_PARAMS: {
+    [key: string]: oMethod;
+};
+//# sourceMappingURL=vector-store.methods.d.ts.map

package/dist/vector-store/methods/vector-store.methods.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-store.methods.d.ts","sourceRoot":"","sources":["../../../src/vector-store/methods/vector-store.methods.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAE5C,eAAO,MAAM,mBAAmB,EAAE;IAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CA2DzD,CAAC"}

package/dist/vector-store/methods/vector-store.methods.js

@@ -0,0 +1,60 @@
+export const VECTOR_STORE_PARAMS = {
+    add_documents: {
+        name: 'add_documents',
+        description: 'Add documents to the store',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'documents',
+                type: 'array',
+                value: '{pageContent: string, metadata: object}',
+                description: 'The documents to add',
+            },
+        ],
+    },
+    delete_documents: {
+        name: 'delete_documents',
+        description: 'Delete documents from the store',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'documents',
+                type: 'array',
+                value: 'object[]',
+                description: 'The documents to update',
+            },
+        ],
+    },
+    search_similar: {
+        name: 'search_similar',
+        description: 'Search for similar documents in the store',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'query',
+                type: 'string',
+                value: 'string',
+                description: 'The query to search for',
+            },
+            {
+                name: 'limit',
+                type: 'number',
+                value: 'number',
+                description: 'The limit of documents to return',
+            },
+        ],
+    },
+    update_documents: {
+        name: 'update_documents',
+        description: 'Update documents in the store',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'documents',
+                type: 'array',
+                value: 'object[]',
+                description: 'The documents to update',
+            },
+        ],
+    },
+};

package/dist/vector-store/vector-memory.tool.d.ts

@@ -0,0 +1,11 @@
+import { oVirtualTool, ToolResult } from '@olane/o-tool';
+import { oRequest } from '@olane/o-core';
+import { oToolConfig } from '@olane/o-tool';
+export declare abstract class VectorMemoryStorageTool extends oVirtualTool {
+    constructor(config: oToolConfig);
+    abstract _tool_search_similar(request: oRequest): Promise<ToolResult>;
+    abstract _tool_add_documents(request: oRequest): Promise<ToolResult>;
+    abstract _tool_delete_documents(request: oRequest): Promise<ToolResult>;
+    abstract _tool_update_documents(request: oRequest): Promise<ToolResult>;
+}
+//# sourceMappingURL=vector-memory.tool.d.ts.map

package/dist/vector-store/vector-memory.tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-memory.tool.d.ts","sourceRoot":"","sources":["../../src/vector-store/vector-memory.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,EAAY,QAAQ,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAG5C,8BAAsB,uBAAwB,SAAQ,YAAY;gBACpD,MAAM,EAAE,WAAW;IAS/B,QAAQ,CAAC,oBAAoB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACrE,QAAQ,CAAC,mBAAmB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACpE,QAAQ,CAAC,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IACvE,QAAQ,CAAC,sBAAsB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;CACxE"}

package/dist/vector-store/vector-memory.tool.js

@@ -0,0 +1,13 @@
+import { oVirtualTool } from '@olane/o-tool';
+import { oAddress } from '@olane/o-core';
+import { VECTOR_STORE_PARAMS } from './methods/vector-store.methods.js';
+export class VectorMemoryStorageTool extends oVirtualTool {
+    constructor(config) {
+        super({
+            ...config,
+            address: new oAddress('o://vector-store'),
+            methods: VECTOR_STORE_PARAMS,
+            description: 'Vector store tool for storing and searching documents',
+        });
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@olane/o-tool-registry",
-  "version": "0.6.12",
+  "version": "0.7.1",
   "type": "module",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",
@@ -55,13 +55,14 @@
     "typescript": "5.4.5"
   },
   "peerDependencies": {
-    "@olane/o-config": "^0.6.11",
-    "@olane/o-core": "^0.6.11",
-    "@olane/o-intelligence": "^0.6.7",
-    "@olane/o-mcp": "^0.6.7",
-    "@olane/o-protocol": "^0.6.11",
-    "@olane/o-tool": "^0.6.11",
-    "@olane/o-tools-common": "^0.6.11"
+    "@olane/o-config": "^0.7.0",
+    "@olane/o-core": "^0.7.0",
+    "@olane/o-intelligence": "^0.7.0",
+    "@olane/o-mcp": "^0.7.0",
+    "@olane/o-protocol": "^0.7.0",
+    "@olane/o-tool": "^0.7.0",
+    "@olane/o-tools-common": "^0.7.0",
+    "@olane/o-lane": "^0.7.0"
   },
   "dependencies": {
     "@huggingface/transformers": "^3.5.2",