@bytesbrains/pi-loki-gate 1.0.0 → 1.0.1

package/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2026 nandal
+Copyright (c) 2026 BytesBrains Pte Ltd
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

package/README.md

@@ -7,7 +7,7 @@ For raw Docker container output (console.log + everything), use [pi-docker-logs]
 ## Install
 
 ```bash
-pi install npm:pi-loki-gate
+pi install npm:@bytesbrains/pi-loki-gate
 ```
 
 ## Configuration
@@ -75,3 +75,8 @@ loki_worker_logs(container="ai-factory-orchestrator")
 ## License
 
 MIT
+
+---
+
+Built and maintained by [BytesBrains](https://bytesbrains.com) — AI automation & agents, engineered to production standards.
+*The model proposes, code guarantees.*

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytesbrains/pi-loki-gate",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Loki log gateway for pi agents — query Grafana Loki for container/job/worker logs from the wrok.in AI Factory.",
   "keywords": [
     "pi-package",
@@ -11,15 +11,14 @@
     "logging",
     "ai-factory"
   ],
-  "author": "nandal <nandal@users.noreply.github.com>",
+  "author": "BytesBrains Pte Ltd (https://bytesbrains.com)",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/nandal/pi-ext.git",
-    "directory": "loki-gate"
+    "url": "git+https://github.com/bytesbrains/pi-loki-gate.git"
   },
-  "homepage": "https://github.com/nandal/pi-ext/tree/main/loki-gate",
+  "homepage": "https://github.com/bytesbrains/pi-loki-gate#readme",
   "bugs": {
-    "url": "https://github.com/nandal/pi-ext/issues"
+    "url": "https://github.com/bytesbrains/pi-loki-gate/issues"
   },
   "license": "MIT",
   "main": "./src/index.ts",

package/src/config.ts

@@ -21,7 +21,10 @@ export function loadConfig(cwd: string): LokiConfig {
     }
     return {
       lokiUrl: (result["lokiUrl"] as string) || DEFAULT_CONFIG.lokiUrl,
-      defaultLimit: parseInt(result["defaultLimit"] as string) || DEFAULT_CONFIG.defaultLimit,
+      const defaultLimit = parseInt(result["defaultLimit"] as string);
+    return {
+      lokiUrl: (result["lokiUrl"] as string) || DEFAULT_CONFIG.lokiUrl,
+      defaultLimit: isNaN(defaultLimit) ? DEFAULT_CONFIG.defaultLimit : defaultLimit,
     };
   } catch {
     return { ...DEFAULT_CONFIG };

package/src/tools/query.ts

@@ -3,6 +3,11 @@ import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { loadConfig } from "../config";
 import { lokiQuery, formatLogOutput } from "../helpers";
 
+/** Escape backticks to prevent LogQL injection via user input. */
+function escapeLogQL(value: string): string {
+  return value.replace(/`/g, '\\`');
+}
+
 // ─── Query Logs ───────────────────────────────────────────────────────────────
 
 export const queryLogsTool = {
@@ -60,7 +65,7 @@ export const jobLogsTool = {
   async execute(_id: string, params: any, _s: any, _u: any, ctx: ExtensionContext) {
     const config = loadConfig(ctx.cwd);
     const limit = params.limit || 200;
-    const query = `{service_name=~"worker|orchestrator"} |= \`${params.jobId}\``;
+    const query = `{service_name=~"worker|orchestrator"} |= \`${escapeLogQL(params.jobId)}\``;
 
     const result = await lokiQuery(config, query, limit);
 
@@ -138,7 +143,7 @@ export const workerLogsTool = {
 
     let query = `{${serviceFilter}}`;
     if (params.search) {
-      query += ` |= \`${params.search}\``;
+      query += ` |= \`${escapeLogQL(params.search)}\``;
     }
 
     const result = await lokiQuery(config, query, limit);