@blogic-cz/agent-tools 0.14.45 → 0.14.46

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blogic-cz/agent-tools",
-  "version": "0.14.45",
+  "version": "0.14.46",
   "description": "CLI tools for AI coding agent workflows — GitHub, database, Kubernetes, Azure DevOps, logs, sessions, and audit",
   "keywords": [
     "agent",

package/src/db-tool/index.ts

@@ -64,16 +64,21 @@ const sqlCommand = Command.make(
       ),
     ),
     sql: Flag.string("sql").pipe(Flag.withDescription("SQL query to execute")),
+    limit: Flag.optional(Flag.integer("limit")).pipe(
+      Flag.withDescription(
+        "Max rows to return (default 50). Use 0 for no cap. Prefer a SQL LIMIT for large tables.",
+      ),
+    ),
     format: formatOption,
     profile: Flag.optional(Flag.string("profile")).pipe(
       Flag.withDescription("Database profile name from agent-tools.json5 (if multiple configured)"),
     ),
   },
-  ({ env, sql, format }) =>
+  ({ env, sql, limit, format }) =>
     Effect.gen(function* () {
       const resolvedEnv = yield* resolveEnv(env);
       const db = yield* DbService;
-      const result = yield* db.executeQuery(resolvedEnv, sql);
+      const result = yield* db.executeQuery(resolvedEnv, sql, Option.getOrUndefined(limit));
       yield* Console.log(formatOutput(result, format));
     }),
 ).pipe(Command.withDescription("Execute a SQL query"));

package/src/db-tool/service.ts

@@ -59,7 +59,11 @@ export { buildApiProbeArgs } from "#shared/k8s-probe";
 export class DbService extends Context.Service<
   DbService,
   {
-    readonly executeQuery: (env: string, sql: string) => Effect.Effect<QueryResult, DbError>;
+    readonly executeQuery: (
+      env: string,
+      sql: string,
+      limit?: number,
+    ) => Effect.Effect<QueryResult, DbError>;
     readonly executeSchemaQuery: (
       env: string,
       mode: SchemaMode,
@@ -83,7 +87,8 @@ export class DbService extends Context.Service<
               environment: env,
             });
           return {
-            executeQuery: (env: string, _sql: string) => Effect.fail(noConfigError(env)),
+            executeQuery: (env: string, _sql: string, _limit?: number) =>
+              Effect.fail(noConfigError(env)),
             executeSchemaQuery: (env: string, _mode: SchemaMode, _table?: string) =>
               Effect.fail(noConfigError(env)),
           };
@@ -468,6 +473,7 @@ export class DbService extends Context.Service<
           password: string,
           startTimeMs: number,
           applyTransform = false,
+          limit?: number,
         ) {
           const selectSql = sql.trim().replace(/;\s*$/, "");
           const wrappedSql = `SELECT json_agg(t) FROM (${selectSql}) t;`;
@@ -532,7 +538,7 @@ export class DbService extends Context.Service<
           });
 
           const transformed = applyTransform
-            ? transformQueryResult(rawData)
+            ? transformQueryResult(rawData, limit)
             : {
                 data: rawData,
                 showing: rawData.length,
@@ -720,6 +726,7 @@ export class DbService extends Context.Service<
         const executeQuery = Effect.fn("DbService.executeQuery")(function* (
           env: string,
           sql: string,
+          limit?: number,
         ) {
           const config = yield* getConfigForEnv(env);
           const startTimeMs = yield* Clock.currentTimeMillis;
@@ -753,7 +760,7 @@ export class DbService extends Context.Service<
 
           const queryEffect = mutation
             ? executeMutationQuery(resolvedConfig, sql, password, Number(startTimeMs))
-            : executeSelectQuery(resolvedConfig, sql, password, Number(startTimeMs), true);
+            : executeSelectQuery(resolvedConfig, sql, password, Number(startTimeMs), true, limit);
 
           return yield* runWithVpnPrerequisites(
             resolvedConfig.port,

package/src/db-tool/transformers.ts

@@ -22,12 +22,17 @@ function truncateValue(value: unknown): unknown {
   return `${value.slice(0, MAX_VALUE_LENGTH)}...`;
 }
 
-export function transformQueryResult(data: Record<string, unknown>[]): TransformResult {
+export function transformQueryResult(
+  data: Record<string, unknown>[],
+  // limit 0 (or negative) means "no row cap".
+  limit: number = DEFAULT_ROW_LIMIT,
+): TransformResult {
   const withoutEmptyColumns = stripEmptyColumns(data);
   const truncatedValues = withoutEmptyColumns.map((record) =>
     Object.fromEntries(Object.entries(record).map(([key, value]) => [key, truncateValue(value)])),
   );
-  const { rows, truncated, total, showing } = truncateRows(truncatedValues, DEFAULT_ROW_LIMIT);
+  const effectiveLimit = limit > 0 ? limit : Number.POSITIVE_INFINITY;
+  const { rows, truncated, total, showing } = truncateRows(truncatedValues, effectiveLimit);
 
   return {
     data: rows,