@tetrixdev/ai-bridge 0.1.0 → 0.1.2

package/dist/cli.js

@@ -1,6 +1,7 @@
 #!/usr/bin/env node
 
 // src/cli.ts
+import { realpathSync } from "fs";
 import { fileURLToPath } from "url";
 import { Command } from "commander";
 
@@ -11,7 +12,7 @@ import WebSocket from "ws";
 
 // src/protocol/version.ts
 var PROTOCOL_VERSION = "0.1";
-var BRIDGE_VERSION = "0.1.0";
+var BRIDGE_VERSION = "0.1.2";
 
 // src/tools/manager.ts
 import fs from "fs";
@@ -276,26 +277,29 @@ var ToolManager = class {
 
 set -euo pipefail
 
-# Read arguments from stdin (the standard way AI CLIs pass tool args).
-STDIN_DATA=""
-if [ ! -t 0 ]; then
-  STDIN_DATA=$(cat)
+# Read the JSON payload from the first CLI argument when present; otherwise
+# fall back to stdin (some AI CLIs pass tool args one way, some the other).
+PAYLOAD_DATA=""
+if [ "$#" -ge 1 ] && [ -n "\${1:-}" ]; then
+  PAYLOAD_DATA="$1"
+elif [ ! -t 0 ]; then
+  PAYLOAD_DATA=$(cat)
 fi
 
 # Single Node.js invocation for input parsing, payload building, HTTP call,
 # and output extraction.
 node -e '
 const http = require("http");
-const stdinData = process.argv[1];
+const payloadData = process.argv[1];
 const toolName = process.argv[2];
 const toolCallId = process.argv[3];
 const requestId = process.argv[4];
 const secret = process.argv[5];
 
-// Parse stdin as JSON, fall back to wrapping as {input: ...}
+// Parse the payload as JSON, fall back to wrapping as {input: ...}
 let args = {};
-if (stdinData) {
-  try { args = JSON.parse(stdinData); } catch { args = { input: stdinData }; }
+if (payloadData) {
+  try { args = JSON.parse(payloadData); } catch { args = { input: payloadData }; }
 }
 
 const payload = JSON.stringify({
@@ -327,7 +331,7 @@ req.write(payload);
 req.end();
 // The $RANDOM-based ID here is discarded \u2014 the callback server overrides it
 // with a cryptographically strong UUID before use.
-' "$STDIN_DATA" "${tool.name}" "tc_\${RANDOM}\${RANDOM}" "\${AI_BRIDGE_REQUEST_ID:-}" "${secretArg}"
+' "$PAYLOAD_DATA" "${tool.name}" "tc_\${RANDOM}\${RANDOM}" "\${AI_BRIDGE_REQUEST_ID:-}" "${secretArg}"
 `;
   }
 };
@@ -1662,14 +1666,47 @@ function createFinalizer(opts) {
 var ProviderAdapter = class {
 };
 
+// src/tools/prompt.ts
+function buildToolInstructions(tools, toolScriptDir) {
+  if (tools.length === 0) return "";
+  const cmd = (name) => toolScriptDir ? `${toolScriptDir}/${name}` : name;
+  const sections = [
+    "# Available Tools",
+    "",
+    'You have access to the following tools. Each tool is an executable script. Call a tool by running it as a shell command with a single argument: a JSON object containing the parameters. The command prints its result to stdout. Run each tool using the exact path shown in its "Call it like" example. Use these tools whenever they help fulfill the request.'
+  ];
+  for (const tool of tools) {
+    sections.push("", `## ${tool.name}`, "", tool.description);
+    const parameters = tool.parameters ?? {};
+    const properties = parameters["properties"] ?? {};
+    const required = Array.isArray(parameters["required"]) ? parameters["required"].map((r) => String(r)) : [];
+    const propNames = Object.keys(properties);
+    if (propNames.length === 0) {
+      sections.push("", "Parameters: No parameters");
+      sections.push("", `Call it like: ${cmd(tool.name)} '{}'`);
+      continue;
+    }
+    sections.push("", "Parameters:");
+    for (const pname of propNames) {
+      const raw = properties[pname];
+      const prop = raw && typeof raw === "object" ? raw : {};
+      const type = typeof prop.type === "string" ? prop.type : "any";
+      const requiredLabel = required.includes(pname) ? "required" : "optional";
+      const description = typeof prop.description === "string" && prop.description.length > 0 ? prop.description : "No description";
+      sections.push(`- ${pname} (${type}, ${requiredLabel}): ${description}`);
+    }
+    const exampleParam = propNames[0];
+    sections.push(
+      "",
+      `Call it like: ${cmd(tool.name)} '{"${exampleParam}":"<value>"}'`
+    );
+  }
+  return sections.join("\n");
+}
+
 // src/providers/codex.ts
 var log7 = createLogger("CodexAdapter");
 var DEFAULT_MODEL = "gpt-5.3-codex";
-function buildToolPromptNote(tools) {
-  if (tools.length === 0) return "";
-  const lines = tools.map((t) => `- \`${t.name}\`: ${t.description}`);
-  return "\n\n---\nThe following bridge tools are available to you as shell commands. Run a tool by executing its command name in the shell (passing any arguments it documents); the command performs the action and prints the result. Use them when they help answer the request:\n" + lines.join("\n");
-}
 var CodexAdapter = class extends ProviderAdapter {
   providerName = "codex";
   async execute(context, onEvent) {
@@ -1704,14 +1741,12 @@ var CodexAdapter = class extends ProviderAdapter {
     if (hasTools) {
       args.push(
         "-s",
-        "workspace-write",
-        "-c",
-        "sandbox_workspace_write.network_access=true",
+        "danger-full-access",
         "-c",
         "approval_policy=never"
       );
     }
-    const toolNote = hasTools ? buildToolPromptNote(context.tools) : "";
+    const toolNote = hasTools ? "\n\n" + buildToolInstructions(context.tools, context.toolScriptDir) : "";
     if (!cliSessionId && request.system_prompt) {
       args.push("--", buildCombinedPrompt(request.system_prompt, userMessage) + toolNote);
     } else if (toolNote) {
@@ -1963,9 +1998,13 @@ var ClaudeAdapter = class extends ProviderAdapter {
       args.push("--max-tokens", String(request.options.max_tokens));
     }
     if (context.tools.length > 0 && context.toolScriptDir) {
-      args.push("--allowedTools=bash");
+      args.push("--permission-mode", "bypassPermissions");
+    }
+    let promptArg = userMessage;
+    if (context.tools.length > 0) {
+      promptArg += "\n\n" + buildToolInstructions(context.tools, context.toolScriptDir);
     }
-    args.push(userMessage);
+    args.push(promptArg);
     if (isDebugEnabled()) {
       log8.debug("Spawning claude", { args: args.map((a) => a.length > 50 ? a.substring(0, 50) + "..." : a) });
     }
@@ -2126,13 +2165,8 @@ var ClaudeAdapter = class extends ProviderAdapter {
           return;
         }
         if (type === "rate_limit_event") {
-          log8.warn("Claude rate limit event", { type });
-          onEvent({
-            event: "error",
-            data: {
-              code: "rate_limited",
-              message: "Claude is currently rate-limited. The request may retry automatically, or you may need to try again in a moment."
-            }
+          log8.debug("Claude rate limit event (informational)", {
+            status: parsed["rate_limit_info"]?.["status"]
           });
           return;
         }
@@ -2191,6 +2225,10 @@ var GeminiAdapter = class extends ProviderAdapter {
     if (request.system_prompt && !cliSessionId) {
       prompt = buildCombinedPrompt(request.system_prompt, userMessage);
     }
+    const hasTools = context.tools.length > 0;
+    if (hasTools) {
+      prompt += "\n\n" + buildToolInstructions(context.tools, context.toolScriptDir);
+    }
     const args = [
       "--prompt",
       prompt,
@@ -2201,6 +2239,9 @@ var GeminiAdapter = class extends ProviderAdapter {
       "--skip-trust"
       // Required for headless/non-interactive mode
     ];
+    if (hasTools) {
+      args.push("--yolo");
+    }
     if (cliSessionId) {
       args.push("--resume", cliSessionId);
       log9.debug("Resuming session", { cliSessionId });
@@ -2384,16 +2425,6 @@ var GeminiAdapter = class extends ProviderAdapter {
             });
             onEvent({ event: "done", data: {} });
             settled = true;
-          } else if (severity === "warning") {
-            const lowerMsg = (message ?? "").toLowerCase();
-            const isRateLimit = lowerMsg.includes("rate limit") || lowerMsg.includes("ratelimit") || lowerMsg.includes("quota") || lowerMsg.includes("429") || lowerMsg.includes("too many requests");
-            onEvent({
-              event: "error",
-              data: {
-                code: isRateLimit ? "rate_limited" : "provider_warning",
-                message: message ?? "Gemini warning"
-              }
-            });
           }
           return;
         }
@@ -2677,10 +2708,21 @@ program.name("ai-bridge").description("Local CLI bridge for AI web apps \u2014 c
   });
   bridge.connect();
 });
-if (process.argv[1] === fileURLToPath(import.meta.url)) {
+function isMainModule(argv1, moduleUrl) {
+  if (!argv1) return false;
+  try {
+    return realpathSync(argv1) === realpathSync(fileURLToPath(moduleUrl));
+  } catch {
+    return false;
+  }
+}
+if (isMainModule(process.argv[1], import.meta.url)) {
   program.parseAsync(process.argv).catch((err) => {
     log11.error("Fatal error", { error: err instanceof Error ? err.message : String(err) });
     process.exit(1);
   });
 }
+export {
+  isMainModule
+};
 //# sourceMappingURL=cli.js.map