@nodetool-ai/agents 0.7.0-rc.11

package/dist/tools/vector-tools.d.ts

@@ -0,0 +1,224 @@
+/**
+ * Vector database search tools.
+ *
+ * Port of src/nodetool/agents/tools/chroma_tools.py
+ */
+import type { ProcessingContext } from "@nodetool-ai/runtime";
+import { Tool } from "./base-tool.js";
+export interface VecCollection {
+    query(params: {
+        queryTexts: string[];
+        nResults: number;
+        whereDocument?: Record<string, unknown>;
+        include?: string[];
+    }): Promise<{
+        ids: string[][];
+        documents: (string | null)[][];
+    }>;
+    add(params: {
+        ids: string[];
+        documents: string[];
+        metadatas?: Record<string, unknown>[] | null;
+    }): Promise<void>;
+}
+export declare class VecTextSearchTool extends Tool {
+    readonly name = "vector_text_search";
+    readonly description = "Search all vector database collections for similar text using semantic search";
+    readonly inputSchema: {
+        type: string;
+        properties: {
+            text: {
+                type: string;
+                description: string;
+            };
+            n_results: {
+                type: string;
+                description: string;
+                default: number;
+            };
+        };
+        required: string[];
+    };
+    private collection;
+    constructor(collection: VecCollection);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<Record<string, string>>;
+    userMessage(params: Record<string, unknown>): string;
+}
+export declare class VecIndexTool extends Tool {
+    readonly name = "vector_index";
+    readonly description = "Index a text chunk into a vector database collection";
+    readonly inputSchema: {
+        type: string;
+        properties: {
+            text: {
+                type: string;
+                description: string;
+            };
+            source_id: {
+                type: string;
+                description: string;
+            };
+            metadata: {
+                type: string;
+                description: string;
+                default: {};
+            };
+        };
+        required: string[];
+    };
+    private collection;
+    constructor(collection: VecCollection);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<Record<string, string>>;
+    userMessage(params: Record<string, unknown>): string;
+}
+export declare class VecHybridSearchTool extends Tool {
+    readonly name = "vector_hybrid_search";
+    readonly description = "Search all vector database collections using both semantic and keyword-based search";
+    readonly inputSchema: {
+        type: string;
+        properties: {
+            text: {
+                type: string;
+                description: string;
+            };
+            n_results: {
+                type: string;
+                description: string;
+                default: number;
+            };
+            k_constant: {
+                type: string;
+                description: string;
+                default: number;
+            };
+            min_keyword_length: {
+                type: string;
+                description: string;
+                default: number;
+            };
+        };
+        required: string[];
+    };
+    private collection;
+    constructor(collection: VecCollection);
+    private getKeywordQuery;
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<Record<string, string>>;
+    userMessage(params: Record<string, unknown>): string;
+}
+export declare class VecRecursiveSplitAndIndexTool extends Tool {
+    readonly name = "vector_recursive_split_and_index";
+    readonly description = "Split text into chunks recursively and index them into a vector database collection";
+    readonly inputSchema: {
+        type: string;
+        properties: {
+            text: {
+                type: string;
+                description: string;
+            };
+            document_id: {
+                type: string;
+                description: string;
+            };
+            chunk_size: {
+                type: string;
+                description: string;
+                default: number;
+            };
+            chunk_overlap: {
+                type: string;
+                description: string;
+                default: number;
+            };
+            separators: {
+                type: string;
+                items: {
+                    type: string;
+                };
+                description: string;
+                default: string[];
+            };
+            metadata: {
+                type: string;
+                description: string;
+                default: {};
+            };
+        };
+        required: string[];
+    };
+    private collection;
+    constructor(collection: VecCollection);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<Record<string, unknown>>;
+    userMessage(params: Record<string, unknown>): string;
+}
+export declare class VecMarkdownSplitAndIndexTool extends Tool {
+    readonly name = "vector_markdown_split_and_index";
+    readonly description = "Split markdown text into chunks based on headers and index them into a vector database collection";
+    readonly inputSchema: {
+        type: string;
+        properties: {
+            file_path: {
+                type: string;
+                description: string;
+            };
+            text: {
+                type: string;
+                description: string;
+            };
+            chunk_size: {
+                type: string;
+                description: string;
+                default: number;
+            };
+            chunk_overlap: {
+                type: string;
+                description: string;
+                default: number;
+            };
+        };
+        required: never[];
+    };
+    private collection;
+    constructor(collection: VecCollection);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<Record<string, unknown>>;
+    userMessage(params: Record<string, unknown>): string;
+}
+export declare class VecBatchIndexTool extends Tool {
+    readonly name = "vector_batch_index";
+    readonly description = "Index a batch of text chunks into a vector database collection";
+    readonly inputSchema: {
+        type: string;
+        properties: {
+            chunks: {
+                type: string;
+                items: {
+                    type: string;
+                    properties: {
+                        text: {
+                            type: string;
+                        };
+                        source_id: {
+                            type: string;
+                        };
+                        metadata: {
+                            type: string;
+                            default: {};
+                        };
+                    };
+                    required: string[];
+                };
+                description: string;
+            };
+            base_metadata: {
+                type: string;
+                description: string;
+                default: {};
+            };
+        };
+        required: string[];
+    };
+    private collection;
+    constructor(collection: VecCollection);
+    process(_context: ProcessingContext, params: Record<string, unknown>): Promise<Record<string, unknown>>;
+    userMessage(params: Record<string, unknown>): string;
+}
+//# sourceMappingURL=vector-tools.d.ts.map

package/dist/tools/vector-tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-tools.d.ts","sourceRoot":"","sources":["../../src/tools/vector-tools.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAEtC,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,MAAM,EAAE;QACZ,UAAU,EAAE,MAAM,EAAE,CAAC;QACrB,QAAQ,EAAE,MAAM,CAAC;QACjB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACxC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;KACpB,GAAG,OAAO,CAAC;QACV,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC;QAChB,SAAS,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;KAChC,CAAC,CAAC;IACH,GAAG,CAAC,MAAM,EAAE;QACV,GAAG,EAAE,MAAM,EAAE,CAAC;QACd,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,IAAI,CAAC;KAC9C,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACnB;AAUD,qBAAa,iBAAkB,SAAQ,IAAI;IACzC,QAAQ,CAAC,IAAI,wBAAwB;IACrC,QAAQ,CAAC,WAAW,mFAC8D;IAClF,QAAQ,CAAC,WAAW;;;;;;;;;;;;;;MAWlB;IAEF,OAAO,CAAC,UAAU,CAAgB;gBAEtB,UAAU,EAAE,aAAa;IAK/B,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAmBlC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAKrD;AAMD,qBAAa,YAAa,SAAQ,IAAI;IACpC,QAAQ,CAAC,IAAI,kBAAkB;IAC/B,QAAQ,CAAC,WAAW,0DAA0D;IAC9E,QAAQ,CAAC,WAAW;;;;;;;;;;;;;;;;;;MAelB;IAEF,OAAO,CAAC,UAAU,CAAgB;gBAEtB,UAAU,EAAE,aAAa;IAK/B,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAsBlC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAKrD;AAMD,qBAAa,mBAAoB,SAAQ,IAAI;IAC3C,QAAQ,CAAC,IAAI,0BAA0B;IACvC,QAAQ,CAAC,WAAW,yFACoE;IACxF,QAAQ,CAAC,WAAW;;;;;;;;;;;;;;;;;;;;;;;;MAqBlB;IAEF,OAAO,CAAC,UAAU,CAAgB;gBAEtB,UAAU,EAAE,aAAa;IAKrC,OAAO,CAAC,eAAe;IAgBjB,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAmElC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAKrD;AA4GD,qBAAa,6BAA8B,SAAQ,IAAI;IACrD,QAAQ,CAAC,IAAI,sCAAsC;IACnD,QAAQ,CAAC,WAAW,yFACoE;IACxF,QAAQ,CAAC,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;MAkClB;IAEF,OAAO,CAAC,UAAU,CAAgB;gBAEtB,UAAU,EAAE,aAAa;IAK/B,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAiDnC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAKrD;AAwBD,qBAAa,4BAA6B,SAAQ,IAAI;IACpD,QAAQ,CAAC,IAAI,qCAAqC;IAClD,QAAQ,CAAC,WAAW,uGACkF;IACtG,QAAQ,CAAC,WAAW;;;;;;;;;;;;;;;;;;;;;;;MAuBlB;IAEF,OAAO,CAAC,UAAU,CAAgB;gBAEtB,UAAU,EAAE,aAAa;IAK/B,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAqDnC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAKrD;AAMD,qBAAa,iBAAkB,SAAQ,IAAI;IACzC,QAAQ,CAAC,IAAI,wBAAwB;IACrC,QAAQ,CAAC,WAAW,oEAC+C;IACnE,QAAQ,CAAC,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;MAuBlB;IAEF,OAAO,CAAC,UAAU,CAAgB;gBAEtB,UAAU,EAAE,aAAa;IAK/B,OAAO,CACX,QAAQ,EAAE,iBAAiB,EAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IA6CnC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;CAIrD"}