@agentforge/tools 0.15.4 → 0.15.6

package/dist/index.d.ts

@@ -3069,6 +3069,8 @@ interface Neo4jToolsConfig {
  * Manages Neo4j driver connections with pooling and health checks.
  */
 
+type Neo4jQueryResult = any;
+type Neo4jQueryParameters = Record<string, unknown>;
 /**
  * Neo4j connection pool singleton
  */
@@ -3090,15 +3092,15 @@ declare class Neo4jConnectionPool {
     /**
      * Execute a query with automatic session management
      */
-    executeQuery<T = any>(cypher: string, parameters?: Record<string, any>, database?: string): Promise<T[]>;
+    executeQuery<T = Neo4jQueryResult>(cypher: string, parameters?: Neo4jQueryParameters, database?: string): Promise<T[]>;
     /**
      * Execute a read query
      */
-    executeReadQuery<T = any>(cypher: string, parameters?: Record<string, any>, database?: string): Promise<T[]>;
+    executeReadQuery<T = Neo4jQueryResult>(cypher: string, parameters?: Neo4jQueryParameters, database?: string): Promise<T[]>;
     /**
      * Execute a write query
      */
-    executeWriteQuery<T = any>(cypher: string, parameters?: Record<string, any>, database?: string): Promise<T[]>;
+    executeWriteQuery<T = Neo4jQueryResult>(cypher: string, parameters?: Neo4jQueryParameters, database?: string): Promise<T[]>;
     /**
      * Check if driver is initialized
      */
@@ -3373,7 +3375,7 @@ declare function createNeo4jQueryTool(): _agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    data: any[];
+    data: Record<string, unknown>[];
     recordCount: number;
     summary: {
         query: string;
@@ -3464,7 +3466,7 @@ declare function createNeo4jFindNodesTool(): _agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    nodes: any[];
+    nodes: unknown[];
     count: number;
     query: {
         label: string;
@@ -3508,10 +3510,10 @@ declare function createNeo4jTraverseTool(): _agentforge_core.Tool<{
 } | {
     success: boolean;
     paths: {
-        start: any;
-        end: any;
-        relationships: any;
-        depth: any;
+        start: unknown;
+        end: unknown;
+        relationships: unknown;
+        depth: unknown;
     }[];
     count: number;
     query: {
@@ -3556,8 +3558,8 @@ declare function createNeo4jVectorSearchTool(): _agentforge_core.Tool<{
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -3603,8 +3605,8 @@ declare function createNeo4jVectorSearchWithEmbeddingTool(): _agentforge_core.To
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -3663,8 +3665,8 @@ declare function createNeo4jCreateNodeWithEmbeddingTool(): _agentforge_core.Tool
     message?: undefined;
 } | {
     success: boolean;
-    node: any;
-    nodeId: any;
+    node: unknown;
+    nodeId: unknown;
     embedding: {
         model: string;
         dimensions: number;
@@ -3694,7 +3696,7 @@ declare const neo4jQuery: _agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    data: any[];
+    data: Record<string, unknown>[];
     recordCount: number;
     summary: {
         query: string;
@@ -3773,7 +3775,7 @@ declare const neo4jFindNodes: _agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    nodes: any[];
+    nodes: unknown[];
     count: number;
     query: {
         label: string;
@@ -3811,10 +3813,10 @@ declare const neo4jTraverse: _agentforge_core.Tool<{
 } | {
     success: boolean;
     paths: {
-        start: any;
-        end: any;
-        relationships: any;
-        depth: any;
+        start: unknown;
+        end: unknown;
+        relationships: unknown;
+        depth: unknown;
     }[];
     count: number;
     query: {
@@ -3853,8 +3855,8 @@ declare const neo4jVectorSearch: _agentforge_core.Tool<{
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -3893,8 +3895,8 @@ declare const neo4jVectorSearchWithEmbedding: _agentforge_core.Tool<{
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -3946,8 +3948,8 @@ declare const neo4jCreateNodeWithEmbedding: _agentforge_core.Tool<{
     message?: undefined;
 } | {
     success: boolean;
-    node: any;
-    nodeId: any;
+    node: unknown;
+    nodeId: unknown;
     embedding: {
         model: string;
         dimensions: number;
@@ -3976,7 +3978,7 @@ declare const neo4jTools: (_agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    data: any[];
+    data: Record<string, unknown>[];
     recordCount: number;
     summary: {
         query: string;
@@ -4047,7 +4049,7 @@ declare const neo4jTools: (_agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    nodes: any[];
+    nodes: unknown[];
     count: number;
     query: {
         label: string;
@@ -4081,10 +4083,10 @@ declare const neo4jTools: (_agentforge_core.Tool<{
 } | {
     success: boolean;
     paths: {
-        start: any;
-        end: any;
-        relationships: any;
-        depth: any;
+        start: unknown;
+        end: unknown;
+        relationships: unknown;
+        depth: unknown;
     }[];
     count: number;
     query: {
@@ -4119,8 +4121,8 @@ declare const neo4jTools: (_agentforge_core.Tool<{
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -4155,8 +4157,8 @@ declare const neo4jTools: (_agentforge_core.Tool<{
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -4204,8 +4206,8 @@ declare const neo4jTools: (_agentforge_core.Tool<{
     message?: undefined;
 } | {
     success: boolean;
-    node: any;
-    nodeId: any;
+    node: unknown;
+    nodeId: unknown;
     embedding: {
         model: string;
         dimensions: number;
@@ -4234,7 +4236,7 @@ declare const neo4jCoreTools: (_agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    data: any[];
+    data: Record<string, unknown>[];
     recordCount: number;
     summary: {
         query: string;
@@ -4305,7 +4307,7 @@ declare const neo4jCoreTools: (_agentforge_core.Tool<{
     query?: undefined;
 } | {
     success: boolean;
-    nodes: any[];
+    nodes: unknown[];
     count: number;
     query: {
         label: string;
@@ -4339,10 +4341,10 @@ declare const neo4jCoreTools: (_agentforge_core.Tool<{
 } | {
     success: boolean;
     paths: {
-        start: any;
-        end: any;
-        relationships: any;
-        depth: any;
+        start: unknown;
+        end: unknown;
+        relationships: unknown;
+        depth: unknown;
     }[];
     count: number;
     query: {
@@ -4377,8 +4379,8 @@ declare const neo4jCoreTools: (_agentforge_core.Tool<{
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -4414,7 +4416,7 @@ declare function createNeo4jTools(config?: Neo4jToolsConfig, includeEmbeddingToo
     query?: undefined;
 } | {
     success: boolean;
-    data: any[];
+    data: Record<string, unknown>[];
     recordCount: number;
     summary: {
         query: string;
@@ -4485,7 +4487,7 @@ declare function createNeo4jTools(config?: Neo4jToolsConfig, includeEmbeddingToo
     query?: undefined;
 } | {
     success: boolean;
-    nodes: any[];
+    nodes: unknown[];
     count: number;
     query: {
         label: string;
@@ -4519,10 +4521,10 @@ declare function createNeo4jTools(config?: Neo4jToolsConfig, includeEmbeddingToo
 } | {
     success: boolean;
     paths: {
-        start: any;
-        end: any;
-        relationships: any;
-        depth: any;
+        start: unknown;
+        end: unknown;
+        relationships: unknown;
+        depth: unknown;
     }[];
     count: number;
     query: {
@@ -4557,8 +4559,8 @@ declare function createNeo4jTools(config?: Neo4jToolsConfig, includeEmbeddingToo
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -4593,8 +4595,8 @@ declare function createNeo4jTools(config?: Neo4jToolsConfig, includeEmbeddingToo
 } | {
     success: boolean;
     results: {
-        node: any;
-        score: any;
+        node: unknown;
+        score: unknown;
     }[];
     count: number;
     query: {
@@ -4642,8 +4644,8 @@ declare function createNeo4jTools(config?: Neo4jToolsConfig, includeEmbeddingToo
     message?: undefined;
 } | {
     success: boolean;
-    node: any;
-    nodeId: any;
+    node: unknown;
+    nodeId: unknown;
     embedding: {
         model: string;
         dimensions: number;
@@ -7867,11 +7869,11 @@ declare const StringTrimSchema: z.ZodObject<{
     characters: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
     text: string;
-    mode: "both" | "end" | "start";
+    mode: "both" | "start" | "end";
     characters?: string | undefined;
 }, {
     text: string;
-    mode?: "both" | "end" | "start" | undefined;
+    mode?: "both" | "start" | "end" | undefined;
     characters?: string | undefined;
 }>;
 /**
@@ -7986,7 +7988,7 @@ declare function createStringCaseConverterTool(): _agentforge_core.Tool<{
  */
 declare function createStringTrimTool(): _agentforge_core.Tool<{
     text: string;
-    mode?: "both" | "end" | "start" | undefined;
+    mode?: "both" | "start" | "end" | undefined;
     characters?: string | undefined;
 }, {
     original: string;
@@ -8083,7 +8085,7 @@ declare const stringCaseConverter: _agentforge_core.Tool<{
 }>;
 declare const stringTrim: _agentforge_core.Tool<{
     text: string;
-    mode?: "both" | "end" | "start" | undefined;
+    mode?: "both" | "start" | "end" | undefined;
     characters?: string | undefined;
 }, {
     original: string;
@@ -8143,7 +8145,7 @@ declare const stringUtilityTools: (_agentforge_core.Tool<{
     targetCase: "title" | "lowercase" | "uppercase" | "camel" | "snake" | "kebab" | "pascal";
 }> | _agentforge_core.Tool<{
     text: string;
-    mode?: "both" | "end" | "start" | undefined;
+    mode?: "both" | "start" | "end" | undefined;
     characters?: string | undefined;
 }, {
     original: string;
@@ -8201,7 +8203,7 @@ declare function createStringUtilityTools(config?: StringUtilitiesConfig): (_age
     targetCase: "title" | "lowercase" | "uppercase" | "camel" | "snake" | "kebab" | "pascal";
 }> | _agentforge_core.Tool<{
     text: string;
-    mode?: "both" | "end" | "start" | undefined;
+    mode?: "both" | "start" | "end" | undefined;
     characters?: string | undefined;
 }, {
     original: string;

package/dist/index.js

@@ -3053,6 +3053,21 @@ async function generateEmbedding(text, model) {
 async function generateBatchEmbeddings(texts, model) {
   return embeddingManager.generateBatchEmbeddings(texts, model);
 }
+function normalizeRecord(record) {
+  const originalKeyByString = /* @__PURE__ */ new Map();
+  const normalizedKeys = [];
+  for (const key of record.keys) {
+    const keyString = String(key);
+    if (!originalKeyByString.has(keyString)) {
+      originalKeyByString.set(keyString, key);
+      normalizedKeys.push(keyString);
+    }
+  }
+  return {
+    keys: normalizedKeys,
+    get: (key) => record.get(originalKeyByString.get(key) ?? key)
+  };
+}
 function formatInteger(value) {
   if (value.inSafeRange()) {
     return value.toNumber();
@@ -3116,7 +3131,8 @@ function formatPath(path12) {
   };
 }
 function formatResults(records) {
-  return records.map((record) => {
+  return records.map((rawRecord) => {
+    const record = normalizeRecord(rawRecord);
     const formatted = {};
     for (const key of record.keys) {
       formatted[key] = formatValue(record.get(key));