@agentforge/tools 0.16.16 → 0.16.18

package/dist/index.js

@@ -4259,6 +4259,79 @@ async function initializeSQLiteConnection(connection) {
   const db = drizzle({ client });
   return { client, db };
 }
+
+// src/data/relational/connection/query-execution.ts
+async function executeQuery(context, query, logger23) {
+  logger23.debug("Executing SQL query", {
+    vendor: context.vendor
+  });
+  if (context.vendor === "sqlite") {
+    return executeSqliteQuery(
+      context.db,
+      query,
+      context.isSqliteNonQueryError
+    );
+  }
+  if (context.vendor === "mysql") {
+    return normalizeMySqlResult(
+      await context.db.execute(query)
+    );
+  }
+  return context.db.execute(query);
+}
+async function executeSqliteQuery(db, query, isSqliteNonQueryError) {
+  try {
+    return db.all(query);
+  } catch (error) {
+    if (isSqliteNonQueryError(error)) {
+      const runResult = db.run(query);
+      return { ...runResult, affectedRows: runResult.changes ?? 0 };
+    }
+    throw error;
+  }
+}
+function normalizeMySqlResult(raw) {
+  if (Array.isArray(raw) && raw.length === 2) {
+    return raw[0];
+  }
+  return raw;
+}
+
+// src/data/relational/connection/session-adapters.ts
+async function executeInDedicatedConnection(context, callback) {
+  if (context.vendor === "sqlite") {
+    return callback(
+      (query) => executeSqliteQuery(
+        context.db,
+        query,
+        context.isSqliteNonQueryError
+      )
+    );
+  }
+  if (context.vendor === "postgresql") {
+    const poolClient = await context.client.connect();
+    try {
+      const { drizzle } = await import('drizzle-orm/node-postgres');
+      const sessionDb = drizzle({ client: poolClient });
+      return await callback((query) => sessionDb.execute(query));
+    } finally {
+      poolClient.release();
+    }
+  }
+  if (context.vendor === "mysql") {
+    const mysqlConnection = await context.client.getConnection();
+    try {
+      const { drizzle } = await import('drizzle-orm/mysql2');
+      const sessionDb = drizzle({ client: mysqlConnection });
+      return await callback(
+        async (query) => normalizeMySqlResult(await sessionDb.execute(query))
+      );
+    } finally {
+      mysqlConnection.release();
+    }
+  }
+  throw new Error(`Unsupported database vendor: ${context.vendor}`);
+}
 var logger8 = createLogger("agentforge:tools:data:relational:connection");
 var ConnectionManager = class extends EventEmitter {
   vendor;
@@ -4559,28 +4632,15 @@ var ConnectionManager = class extends EventEmitter {
     if (!this.db) {
       throw new Error("Database not initialized. Call initialize() first.");
     }
-    logger8.debug("Executing SQL query", {
-      vendor: this.vendor
-    });
-    if (this.vendor === "sqlite") {
-      try {
-        return this.db.all(query);
-      } catch (error) {
-        if (this.isSqliteNonQueryError(error)) {
-          const runResult = this.db.run(query);
-          return { ...runResult, affectedRows: runResult.changes ?? 0 };
-        }
-        throw error;
-      }
-    }
-    if (this.vendor === "mysql") {
-      const raw = await this.db.execute(query);
-      if (Array.isArray(raw) && raw.length === 2) {
-        return raw[0];
-      }
-      return raw;
-    }
-    return this.db.execute(query);
+    return executeQuery(
+      {
+        vendor: this.vendor,
+        db: this.db,
+        isSqliteNonQueryError: (error) => this.isSqliteNonQueryError(error)
+      },
+      query,
+      logger8
+    );
   }
   /**
    * Execute a callback with a single dedicated database connection/session.
@@ -4592,46 +4652,15 @@ var ConnectionManager = class extends EventEmitter {
     if (!this.client || !this.db) {
       throw new Error("Database not initialized. Call connect() first.");
     }
-    if (this.vendor === "sqlite") {
-      return callback(async (query) => {
-        try {
-          return this.db.all(query);
-        } catch (error) {
-          if (this.isSqliteNonQueryError(error)) {
-            const runResult = this.db.run(query);
-            return { ...runResult, affectedRows: runResult.changes ?? 0 };
-          }
-          throw error;
-        }
-      });
-    }
-    if (this.vendor === "postgresql") {
-      const poolClient = await this.client.connect();
-      try {
-        const { drizzle } = await import('drizzle-orm/node-postgres');
-        const sessionDb = drizzle({ client: poolClient });
-        return await callback(async (query) => sessionDb.execute(query));
-      } finally {
-        poolClient.release();
-      }
-    }
-    if (this.vendor === "mysql") {
-      const mysqlConnection = await this.client.getConnection();
-      try {
-        const { drizzle } = await import('drizzle-orm/mysql2');
-        const sessionDb = drizzle({ client: mysqlConnection });
-        return await callback(async (query) => {
-          const raw = await sessionDb.execute(query);
-          if (Array.isArray(raw) && raw.length === 2) {
-            return raw[0];
-          }
-          return raw;
-        });
-      } finally {
-        mysqlConnection.release();
-      }
-    }
-    throw new Error(`Unsupported database vendor: ${this.vendor}`);
+    return executeInDedicatedConnection(
+      {
+        vendor: this.vendor,
+        client: this.client,
+        db: this.db,
+        isSqliteNonQueryError: (error) => this.isSqliteNonQueryError(error)
+      },
+      callback
+    );
   }
   /**
    * Get connection pool metrics
@@ -4850,7 +4879,7 @@ function buildParameterizedQuery(sqlString, params) {
     return sqlChunks.length > 0 ? sql.join(sqlChunks, sql.raw("")) : sql.raw(sqlString);
   }
 }
-async function executeQuery(manager, input, context) {
+async function executeQuery2(manager, input, context) {
   const startTime = Date.now();
   logger9.debug("Executing query", {
     vendor: input.vendor,
@@ -5921,11 +5950,11 @@ async function withTransaction(manager, operation, options) {
     ...resolvedOptions.isolationLevel ? { isolationLevel: resolvedOptions.isolationLevel } : {},
     ...resolvedOptions.timeoutMs !== void 0 ? { timeoutMs: resolvedOptions.timeoutMs } : {}
   });
-  return manager.executeInConnection(async (executeQuery2) => {
+  return manager.executeInConnection(async (executeQuery3) => {
     const transaction = new ManagedTransaction({
       id: transactionId,
       vendor,
-      executeQuery: executeQuery2,
+      executeQuery: executeQuery3,
       options: resolvedOptions
     });
     await transaction.begin();
@@ -6291,7 +6320,7 @@ var SchemaInspector = class _SchemaInspector {
     };
   }
   async runQueryRows(query) {
-    const result = await executeQuery(this.manager, {
+    const result = await executeQuery2(this.manager, {
       sql: query,
       vendor: this.vendor
     });
@@ -7015,7 +7044,7 @@ var relationalQuery = toolBuilder().name("relational-query").displayName("Relati
   });
   try {
     await manager.initialize();
-    const result = await executeQuery(manager, {
+    const result = await executeQuery2(manager, {
       sql: input.sql,
       params: input.params,
       vendor: input.vendor
@@ -10191,6 +10220,6 @@ function createAskHumanTool() {
 }
 var askHumanTool = createAskHumanTool();
 
-export { AskHumanInputSchema, CalculatorSchema, ConnectionManager, ConnectionState, CreditCardValidatorSchema, CurrentDateTimeSchema, DEFAULT_BATCH_SIZE, DEFAULT_CHUNK_SIZE, DEFAULT_RETRY_CONFIG2 as DEFAULT_RETRY_CONFIG, DateArithmeticSchema, DateComparisonSchema, DateDifferenceSchema, DateFormatterSchema, DuckDuckGoProvider, EmailValidatorSchema, EmbeddingManager, HttpMethod, IpValidatorSchema, MAX_BATCH_SIZE, MathFunctionsSchema, MissingPeerDependencyError, OpenAIEmbeddingProvider, PhoneValidatorSchema, RandomNumberSchema, SchemaInspector, SerperProvider, StatisticsSchema, StringCaseConverterSchema, StringJoinSchema, StringLengthSchema, StringReplaceSchema, StringSplitSchema, StringSubstringSchema, StringTrimSchema, UrlValidatorSimpleSchema, UuidValidatorSchema, archiveConfluencePage, arrayFilter, arrayFilterSchema, arrayGroupBy, arrayGroupBySchema, arrayMap, arrayMapSchema, arraySort, arraySortSchema, askHumanTool, benchmarkBatchExecution, benchmarkStreamingSelectMemory, buildDeleteQuery, buildInsertQuery, buildSelectQuery, buildUpdateQuery, calculator, checkPeerDependency, confluenceTools, createArrayFilterTool, createArrayGroupByTool, createArrayMapTool, createArraySortTool, createAskHumanTool, createCalculatorTool, createConfluencePage, createConfluenceTools, createCreditCardValidatorTool, createCsvGeneratorTool, createCsvParserTool, createCsvToJsonTool, createCsvTools, createCurrentDateTimeTool, createDateArithmeticTool, createDateComparisonTool, createDateDifferenceTool, createDateFormatterTool, createDateTimeTools, createDirectoryCreateTool, createDirectoryDeleteTool, createDirectoryListTool, createDirectoryOperationTools, createDuckDuckGoProvider, createEmailValidatorTool, createExtractImagesTool, createExtractLinksTool, createFileAppendTool, createFileDeleteTool, createFileExistsTool, createFileOperationTools, createFileReaderTool, createFileSearchTool, createFileWriterTool, createHtmlParserTool, createHtmlParserTools, createHttpTools, createIpValidatorTool, createJsonMergeTool, createJsonParserTool, createJsonQueryTool, createJsonStringifyTool, createJsonToCsvTool, createJsonToXmlTool, createJsonTools, createJsonValidatorTool, createMathFunctionsTool, createMathOperationTools, createNeo4jCreateNodeWithEmbeddingTool, createNeo4jFindNodesTool, createNeo4jGetSchemaTool, createNeo4jQueryTool, createNeo4jTools, createNeo4jTraverseTool, createNeo4jVectorSearchTool, createNeo4jVectorSearchWithEmbeddingTool, createObjectOmitTool, createObjectPickTool, createPathBasenameTool, createPathDirnameTool, createPathExtensionTool, createPathJoinTool, createPathNormalizeTool, createPathParseTool, createPathRelativeTool, createPathResolveTool, createPathUtilityTools, createPhoneValidatorTool, createRandomNumberTool, createScraperTools, createSelectReadableStream, createSerperProvider, createSlackTools, createStatisticsTool, createStringCaseConverterTool, createStringJoinTool, createStringLengthTool, createStringReplaceTool, createStringSplitTool, createStringSubstringTool, createStringTrimTool, createStringUtilityTools, createTransformerTools, createUrlBuilderTool, createUrlQueryParserTool, createUrlValidatorSimpleTool, createUrlValidatorTool, createUrlValidatorTools, createUuidValidatorTool, createValidationTools, createWebScraperTool, createXmlGeneratorTool, createXmlParserTool, createXmlToJsonTool, createXmlTools, creditCardValidator, csvGenerator, csvGeneratorSchema, csvParser, csvParserSchema, csvToJson, csvToJsonSchema, csvTools, currentDateTime, dateArithmetic, dateComparison, dateDifference, dateFormatter, dateTimeTools, diffSchemas, directoryCreate, directoryCreateSchema, directoryDelete, directoryDeleteSchema, directoryList, directoryListSchema, directoryOperationTools, emailValidator, embeddingManager, executeBatchedTask, executeQuery, executeStreamingSelect, exportSchemaToJson, extractImages, extractImagesSchema, extractLinks, extractLinksSchema, fileAppend, fileAppendSchema, fileDelete, fileDeleteSchema, fileExists, fileExistsSchema, fileOperationTools, fileReader, fileReaderSchema, fileSearch, fileSearchSchema, fileWriter, fileWriterSchema, generateBatchEmbeddings, generateEmbedding, getCohereApiKey, getConfluencePage, getEmbeddingModel, getEmbeddingProvider, getHuggingFaceApiKey, getInstallationInstructions, getOllamaBaseUrl, getOpenAIApiKey, getPeerDependencyName, getSlackChannels, getSlackMessages, getSpacePages, getVendorTypeMap, getVoyageApiKey, htmlParser, htmlParserSchema, htmlParserTools, httpClient, httpGet, httpGetSchema, httpPost, httpPostSchema, httpRequestSchema, httpTools, importSchemaFromJson, initializeEmbeddings, initializeEmbeddingsWithConfig, initializeFromEnv, initializeNeo4jTools, ipValidator, isRetryableError2 as isRetryableError, jsonMerge, jsonMergeSchema, jsonParser, jsonParserSchema, jsonQuery, jsonQuerySchema, jsonStringify, jsonStringifySchema, jsonToCsv, jsonToCsvSchema, jsonToXml, jsonToXmlSchema, jsonTools, jsonValidator, jsonValidatorSchema, listConfluenceSpaces, mapColumnType, mapSchemaTypes, mathFunctions, mathOperationTools, neo4jCoreTools, neo4jCreateNodeWithEmbedding, neo4jCreateNodeWithEmbeddingSchema, neo4jFindNodes, neo4jFindNodesSchema, neo4jGetSchema, neo4jGetSchemaSchema, neo4jPool, neo4jQuery, neo4jQuerySchema, neo4jTools, neo4jTraverse, neo4jTraverseSchema, neo4jVectorSearch, neo4jVectorSearchSchema, neo4jVectorSearchWithEmbedding, neo4jVectorSearchWithEmbeddingSchema, notifySlack, objectOmit, objectOmitSchema, objectPick, objectPickSchema, pathBasename, pathBasenameSchema, pathDirname, pathDirnameSchema, pathExtension, pathExtensionSchema, pathJoin, pathJoinSchema, pathNormalize, pathNormalizeSchema, pathParse, pathParseSchema, pathRelative, pathRelativeSchema, pathResolve, pathResolveSchema, pathUtilityTools, phoneValidator, randomNumber, relationalDelete, relationalGetSchema, relationalInsert, relationalQuery, relationalSelect, relationalUpdate, retryWithBackoff2 as retryWithBackoff, scraperTools, searchConfluence, searchResultSchema, sendSlackMessage, slackTools, statistics, streamSelectChunks, stringCaseConverter, stringJoin, stringLength, stringReplace, stringSplit, stringSubstring, stringTrim, stringUtilityTools, transformerTools, updateConfluencePage, urlBuilder, urlBuilderSchema, urlQueryParser, urlQueryParserSchema, urlValidator, urlValidatorSchema, urlValidatorSimple, urlValidatorTools, uuidValidator, validateBatch, validateColumnTypes, validateColumnsExist, validateTableExists, validateText, validationTools, webScraper, webScraperSchema, webSearch, webSearchOutputSchema, webSearchSchema, withTransaction, xmlGenerator, xmlGeneratorSchema, xmlParser, xmlParserSchema, xmlToJson, xmlToJsonSchema, xmlTools };
+export { AskHumanInputSchema, CalculatorSchema, ConnectionManager, ConnectionState, CreditCardValidatorSchema, CurrentDateTimeSchema, DEFAULT_BATCH_SIZE, DEFAULT_CHUNK_SIZE, DEFAULT_RETRY_CONFIG2 as DEFAULT_RETRY_CONFIG, DateArithmeticSchema, DateComparisonSchema, DateDifferenceSchema, DateFormatterSchema, DuckDuckGoProvider, EmailValidatorSchema, EmbeddingManager, HttpMethod, IpValidatorSchema, MAX_BATCH_SIZE, MathFunctionsSchema, MissingPeerDependencyError, OpenAIEmbeddingProvider, PhoneValidatorSchema, RandomNumberSchema, SchemaInspector, SerperProvider, StatisticsSchema, StringCaseConverterSchema, StringJoinSchema, StringLengthSchema, StringReplaceSchema, StringSplitSchema, StringSubstringSchema, StringTrimSchema, UrlValidatorSimpleSchema, UuidValidatorSchema, archiveConfluencePage, arrayFilter, arrayFilterSchema, arrayGroupBy, arrayGroupBySchema, arrayMap, arrayMapSchema, arraySort, arraySortSchema, askHumanTool, benchmarkBatchExecution, benchmarkStreamingSelectMemory, buildDeleteQuery, buildInsertQuery, buildSelectQuery, buildUpdateQuery, calculator, checkPeerDependency, confluenceTools, createArrayFilterTool, createArrayGroupByTool, createArrayMapTool, createArraySortTool, createAskHumanTool, createCalculatorTool, createConfluencePage, createConfluenceTools, createCreditCardValidatorTool, createCsvGeneratorTool, createCsvParserTool, createCsvToJsonTool, createCsvTools, createCurrentDateTimeTool, createDateArithmeticTool, createDateComparisonTool, createDateDifferenceTool, createDateFormatterTool, createDateTimeTools, createDirectoryCreateTool, createDirectoryDeleteTool, createDirectoryListTool, createDirectoryOperationTools, createDuckDuckGoProvider, createEmailValidatorTool, createExtractImagesTool, createExtractLinksTool, createFileAppendTool, createFileDeleteTool, createFileExistsTool, createFileOperationTools, createFileReaderTool, createFileSearchTool, createFileWriterTool, createHtmlParserTool, createHtmlParserTools, createHttpTools, createIpValidatorTool, createJsonMergeTool, createJsonParserTool, createJsonQueryTool, createJsonStringifyTool, createJsonToCsvTool, createJsonToXmlTool, createJsonTools, createJsonValidatorTool, createMathFunctionsTool, createMathOperationTools, createNeo4jCreateNodeWithEmbeddingTool, createNeo4jFindNodesTool, createNeo4jGetSchemaTool, createNeo4jQueryTool, createNeo4jTools, createNeo4jTraverseTool, createNeo4jVectorSearchTool, createNeo4jVectorSearchWithEmbeddingTool, createObjectOmitTool, createObjectPickTool, createPathBasenameTool, createPathDirnameTool, createPathExtensionTool, createPathJoinTool, createPathNormalizeTool, createPathParseTool, createPathRelativeTool, createPathResolveTool, createPathUtilityTools, createPhoneValidatorTool, createRandomNumberTool, createScraperTools, createSelectReadableStream, createSerperProvider, createSlackTools, createStatisticsTool, createStringCaseConverterTool, createStringJoinTool, createStringLengthTool, createStringReplaceTool, createStringSplitTool, createStringSubstringTool, createStringTrimTool, createStringUtilityTools, createTransformerTools, createUrlBuilderTool, createUrlQueryParserTool, createUrlValidatorSimpleTool, createUrlValidatorTool, createUrlValidatorTools, createUuidValidatorTool, createValidationTools, createWebScraperTool, createXmlGeneratorTool, createXmlParserTool, createXmlToJsonTool, createXmlTools, creditCardValidator, csvGenerator, csvGeneratorSchema, csvParser, csvParserSchema, csvToJson, csvToJsonSchema, csvTools, currentDateTime, dateArithmetic, dateComparison, dateDifference, dateFormatter, dateTimeTools, diffSchemas, directoryCreate, directoryCreateSchema, directoryDelete, directoryDeleteSchema, directoryList, directoryListSchema, directoryOperationTools, emailValidator, embeddingManager, executeBatchedTask, executeQuery2 as executeQuery, executeStreamingSelect, exportSchemaToJson, extractImages, extractImagesSchema, extractLinks, extractLinksSchema, fileAppend, fileAppendSchema, fileDelete, fileDeleteSchema, fileExists, fileExistsSchema, fileOperationTools, fileReader, fileReaderSchema, fileSearch, fileSearchSchema, fileWriter, fileWriterSchema, generateBatchEmbeddings, generateEmbedding, getCohereApiKey, getConfluencePage, getEmbeddingModel, getEmbeddingProvider, getHuggingFaceApiKey, getInstallationInstructions, getOllamaBaseUrl, getOpenAIApiKey, getPeerDependencyName, getSlackChannels, getSlackMessages, getSpacePages, getVendorTypeMap, getVoyageApiKey, htmlParser, htmlParserSchema, htmlParserTools, httpClient, httpGet, httpGetSchema, httpPost, httpPostSchema, httpRequestSchema, httpTools, importSchemaFromJson, initializeEmbeddings, initializeEmbeddingsWithConfig, initializeFromEnv, initializeNeo4jTools, ipValidator, isRetryableError2 as isRetryableError, jsonMerge, jsonMergeSchema, jsonParser, jsonParserSchema, jsonQuery, jsonQuerySchema, jsonStringify, jsonStringifySchema, jsonToCsv, jsonToCsvSchema, jsonToXml, jsonToXmlSchema, jsonTools, jsonValidator, jsonValidatorSchema, listConfluenceSpaces, mapColumnType, mapSchemaTypes, mathFunctions, mathOperationTools, neo4jCoreTools, neo4jCreateNodeWithEmbedding, neo4jCreateNodeWithEmbeddingSchema, neo4jFindNodes, neo4jFindNodesSchema, neo4jGetSchema, neo4jGetSchemaSchema, neo4jPool, neo4jQuery, neo4jQuerySchema, neo4jTools, neo4jTraverse, neo4jTraverseSchema, neo4jVectorSearch, neo4jVectorSearchSchema, neo4jVectorSearchWithEmbedding, neo4jVectorSearchWithEmbeddingSchema, notifySlack, objectOmit, objectOmitSchema, objectPick, objectPickSchema, pathBasename, pathBasenameSchema, pathDirname, pathDirnameSchema, pathExtension, pathExtensionSchema, pathJoin, pathJoinSchema, pathNormalize, pathNormalizeSchema, pathParse, pathParseSchema, pathRelative, pathRelativeSchema, pathResolve, pathResolveSchema, pathUtilityTools, phoneValidator, randomNumber, relationalDelete, relationalGetSchema, relationalInsert, relationalQuery, relationalSelect, relationalUpdate, retryWithBackoff2 as retryWithBackoff, scraperTools, searchConfluence, searchResultSchema, sendSlackMessage, slackTools, statistics, streamSelectChunks, stringCaseConverter, stringJoin, stringLength, stringReplace, stringSplit, stringSubstring, stringTrim, stringUtilityTools, transformerTools, updateConfluencePage, urlBuilder, urlBuilderSchema, urlQueryParser, urlQueryParserSchema, urlValidator, urlValidatorSchema, urlValidatorSimple, urlValidatorTools, uuidValidator, validateBatch, validateColumnTypes, validateColumnsExist, validateTableExists, validateText, validationTools, webScraper, webScraperSchema, webSearch, webSearchOutputSchema, webSearchSchema, withTransaction, xmlGenerator, xmlGeneratorSchema, xmlParser, xmlParserSchema, xmlToJson, xmlToJsonSchema, xmlTools };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map