triagent 0.1.0-alpha1 → 0.1.0-alpha3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "triagent",
-  "version": "0.1.0-alpha1",
+  "version": "0.1.0-alpha3",
   "description": "AI-powered Kubernetes debugging agent with terminal UI",
   "type": "module",
   "main": "src/index.ts",
@@ -46,7 +46,9 @@
     "@opentui/core": "^0.1.72",
     "@opentui/solid": "^0.1.72",
     "hono": "^4.6.0",
+    "opentui-spinner": "^0.0.6",
     "solid-js": "^1.9.10",
+    "triagent": "^0.1.0-alpha1",
     "zod": "^3.24.0"
   },
   "devDependencies": {

package/src/cli/config.ts

@@ -7,6 +7,7 @@ export interface StoredConfig {
   aiProvider?: AIProvider;
   aiModel?: string;
   apiKey?: string;
+  baseUrl?: string;
   webhookPort?: number;
   codebasePath?: string;
   kubeConfigPath?: string;

package/src/config.ts

@@ -10,6 +10,7 @@ const ConfigSchema = z.object({
   aiProvider: AIProviderSchema,
   aiModel: z.string().min(1),
   apiKey: z.string().min(1),
+  baseUrl: z.string().url().optional(),
   webhookPort: z.number().int().positive().default(3000),
   codebasePath: z.string().min(1).default("./"),
   kubeConfigPath: z.string().min(1).default("~/.kube"),
@@ -47,6 +48,7 @@ export async function loadConfig(): Promise<Config> {
     aiProvider: provider,
     aiModel: process.env.AI_MODEL || stored.aiModel || getDefaultModel(provider),
     apiKey: getApiKey(provider, stored),
+    baseUrl: process.env.AI_BASE_URL || stored.baseUrl || undefined,
     webhookPort: parseInt(process.env.WEBHOOK_PORT || String(stored.webhookPort || 3000), 10),
     codebasePath: expandPath(process.env.CODEBASE_PATH || stored.codebasePath || "./"),
     kubeConfigPath: expandPath(process.env.KUBE_CONFIG_PATH || stored.kubeConfigPath || "~/.kube"),

package/src/index.ts

@@ -84,6 +84,7 @@ CONFIG KEYS:
   aiProvider     - AI provider (openai, anthropic, google)
   aiModel        - Model ID (e.g., gpt-4o, claude-sonnet-4-20250514)
   apiKey         - API key for the provider
+  baseUrl        - Custom API base URL (for proxies or local models)
   webhookPort    - Webhook server port (default: 3000)
   codebasePath   - Path to codebase (default: ./)
   kubeConfigPath - Kubernetes config path (default: ~/.kube)
@@ -109,6 +110,7 @@ MODES:
 ENVIRONMENT VARIABLES:
   AI_PROVIDER              - AI provider (openai, anthropic, google)
   AI_MODEL                 - Model ID (e.g., gpt-4o, claude-3-5-sonnet)
+  AI_BASE_URL              - Custom API base URL (for proxies or local models)
   OPENAI_API_KEY          - OpenAI API key
   ANTHROPIC_API_KEY       - Anthropic API key
   GOOGLE_GENERATIVE_AI_API_KEY - Google AI API key
@@ -152,7 +154,12 @@ async function runDirectIncident(description: string): Promise<void> {
         if (toolCalls && toolCalls.length > 0) {
           const toolCall = toolCalls[0];
           const toolName = "toolName" in toolCall ? toolCall.toolName : "tool";
-          console.log(`\n[Tool: ${toolName}]\n`);
+          const args = "args" in toolCall ? toolCall.args : {};
+          console.log(`\n[Tool: ${toolName}]`);
+          if (args && typeof args === "object" && "command" in args) {
+            console.log(`$ ${args.command}`);
+          }
+          console.log();
         }
       },
     });
@@ -174,6 +181,7 @@ async function handleConfigCommand(args: CliArgs): Promise<void> {
     "aiProvider",
     "aiModel",
     "apiKey",
+    "baseUrl",
     "webhookPort",
     "codebasePath",
     "kubeConfigPath",

package/src/mastra/agents/debugger.ts

@@ -1,6 +1,6 @@
 import { Agent } from "@mastra/core/agent";
 import { z } from "zod";
-import { kubectlTool } from "../tools/kubectl.js";
+import { cliTool } from "../tools/cli.js";
 import { gitTool } from "../tools/git.js";
 import { filesystemTool } from "../tools/filesystem.js";
 import type { Config } from "../../config.js";
@@ -9,12 +9,14 @@ const DEBUGGER_INSTRUCTIONS = `You are an expert Kubernetes debugging agent name
 
 ## Your Capabilities
 
-1. **Kubernetes Inspection** (kubectl tool):
-   - Get resource status (pods, deployments, services, configmaps)
-   - Describe resources for detailed information
-   - Fetch container logs
-   - Check resource usage (top)
-   - Review cluster events
+1. **CLI Access** (cli tool):
+   - Run any shell command including kubectl, grep, awk, jq, curl, etc.
+   - Pipe commands together for powerful filtering and processing
+   - Examples:
+     - \`kubectl get pods -A | grep inventory\`
+     - \`kubectl logs deploy/myapp --tail 100 | grep -i error\`
+     - \`kubectl get pods -o json | jq '.items[].metadata.name'\`
+     - \`kubectl describe pod mypod | grep -A10 Events\`
 
 2. **Code Analysis** (filesystem tool):
    - Read source code files
@@ -27,6 +29,39 @@ const DEBUGGER_INSTRUCTIONS = `You are an expert Kubernetes debugging agent name
    - Show specific commit details
    - Blame files to find who changed what
 
+## Resource Discovery Strategy
+
+When asked to find resources for a service (e.g., "inventory service"), DO NOT simply try one label like \`app=inventory\` and give up if not found. Instead, use a systematic discovery approach:
+
+1. **Search by partial name match using grep**:
+   - \`kubectl get pods -A | grep -i inventory\`
+   - \`kubectl get deploy,svc -A | grep -i inventory\`
+   - This finds resources with "inventory" anywhere in the name (e.g., \`inventory-api\`, \`svc-inventory\`)
+
+2. **If grep returns no results, list all resources to browse**:
+   - \`kubectl get pods,deploy,svc -A\` to see everything
+   - \`kubectl get pods -n <namespace>\` if namespace is known
+
+3. **Try common label patterns**:
+   - \`kubectl get pods -A -l app=inventory\`
+   - \`kubectl get pods -A -l app.kubernetes.io/name=inventory\`
+   - \`kubectl get pods -A -l component=inventory\`
+
+4. **Follow the resource chain**:
+   - Found a Service? \`kubectl describe svc <name> | grep Selector\` then find pods with that selector
+   - Found a Deployment? \`kubectl get pods -l app=<deployment-name>\`
+   - Use \`kubectl get endpoints <svc-name>\` to see which pods back a service
+
+5. **Check events for context**:
+   - \`kubectl get events -A --sort-by='.lastTimestamp' | grep -i inventory\`
+   - \`kubectl get events -A --sort-by='.lastTimestamp' | head -20\` for recent cluster activity
+
+6. **When you find a potential match**:
+   - \`kubectl describe <resource> <name>\` to confirm it's the right one
+   - Check related resources (pods for a deployment, endpoints for a service)
+
+Always report what you searched for and what you found, even if it's not an exact match. The user can confirm if you found the right resource.
+
 ## Investigation Process
 
 When given an incident, follow this systematic approach:
@@ -36,26 +71,32 @@ When given an incident, follow this systematic approach:
    - What symptoms are being observed
    - When the issue started (if known)
 
-2. **Check Cluster State**:
-   - Get pod status for affected services
-   - Check for recent events
+2. **Discover Relevant Resources**:
+   - Use the Resource Discovery Strategy above to find the affected resources
+   - Don't assume exact names or labels - search broadly first
+   - Follow the resource chain (Service → Deployment → Pods → Containers)
+
+3. **Check Cluster State**:
+   - Get pod status for discovered resources
+   - Check for recent events related to those resources
    - Look at resource usage
 
-3. **Analyze Logs**:
-   - Fetch logs from affected pods
+4. **Analyze Logs**:
+   - Fetch logs from affected pods (use \`--tail 100\` to get recent logs)
    - Look for errors, exceptions, or unusual patterns
+   - If multiple containers, check each one
 
-4. **Investigate Recent Changes**:
+5. **Investigate Recent Changes**:
    - Check git log for recent commits
    - Review diffs of suspicious changes
    - Correlate timing with when issues started
 
-5. **Examine Code**:
+6. **Examine Code**:
    - Read relevant configuration files
    - Check application code if needed
    - Look for misconfigurations
 
-6. **Synthesize Findings**:
+7. **Synthesize Findings**:
    - Identify the root cause
    - List affected resources
    - Provide actionable recommendations
@@ -122,16 +163,19 @@ export const InvestigationResultSchema = z.object({
 export type InvestigationResult = z.infer<typeof InvestigationResultSchema>;
 
 export function createDebuggerAgent(config: Config) {
-  // Construct model string based on provider
-  const modelString = `${config.aiProvider}/${config.aiModel}`;
+  // Construct model config - use OpenAICompatibleConfig if baseUrl is set
+  const modelId = `${config.aiProvider}/${config.aiModel}` as const;
+  const modelConfig = config.baseUrl
+    ? { id: modelId, url: config.baseUrl }
+    : modelId;
 
   return new Agent({
     id: "kubernetes-debugger",
     name: "Kubernetes Debugger",
     instructions: DEBUGGER_INSTRUCTIONS,
-    model: modelString as any, // Mastra handles model routing
+    model: modelConfig as any, // Mastra handles model routing
     tools: {
-      kubectl: kubectlTool,
+      cli: cliTool,
       git: gitTool,
       filesystem: filesystemTool,
     },

package/src/mastra/tools/cli.ts

@@ -0,0 +1,65 @@
+import { createTool } from "@mastra/core/tools";
+import { z } from "zod";
+import { execCommand } from "../../sandbox/bashlet.js";
+
+interface CliOutput {
+  success: boolean;
+  output: string;
+  error?: string;
+}
+
+function filterSensitiveData(output: string): string {
+  // Redact potential secrets, tokens, and passwords
+  return output
+    .replace(
+      /(password|secret|token|key|credential)[\s:=]+["']?[^\s"'\n]+["']?/gi,
+      "$1: [REDACTED]"
+    )
+    .replace(/Bearer\s+[^\s]+/gi, "Bearer [REDACTED]")
+    .replace(/-----BEGIN[^-]+-----[\s\S]*?-----END[^-]+-----/g, "[REDACTED CERTIFICATE/KEY]");
+}
+
+export const cliTool = createTool({
+  id: "cli",
+  description: `Execute shell commands in the sandbox environment.
+Use this to run any CLI commands including kubectl, grep, awk, jq, curl, etc.
+Supports pipes and command chaining.
+
+Examples:
+- List all pods: kubectl get pods -A
+- Find pods by name: kubectl get pods -A | grep inventory
+- Get logs with filtering: kubectl logs deployment/myapp -n prod --tail 100 | grep -i error
+- Check resource usage: kubectl top pods -n default
+- Describe and search: kubectl describe pod mypod | grep -A5 "Events"
+- JSON processing: kubectl get pods -o json | jq '.items[].metadata.name'`,
+
+  inputSchema: z.object({
+    command: z.string().describe("The shell command to execute"),
+  }),
+
+  execute: async ({ command }): Promise<CliOutput> => {
+    try {
+      const result = await execCommand(command);
+
+      if (result.exitCode !== 0) {
+        return {
+          success: false,
+          output: result.stdout ? filterSensitiveData(result.stdout) : "",
+          error: result.stderr || `Command failed with exit code ${result.exitCode}`,
+        };
+      }
+
+      return {
+        success: true,
+        output: filterSensitiveData(result.stdout),
+        error: result.stderr ? filterSensitiveData(result.stderr) : undefined,
+      };
+    } catch (error) {
+      return {
+        success: false,
+        output: "",
+        error: error instanceof Error ? error.message : String(error),
+      };
+    }
+  },
+});

package/src/server/webhook.ts

@@ -149,7 +149,11 @@ async function runInvestigation(id: string): Promise<void> {
         if (toolCalls && toolCalls.length > 0) {
           const toolCall = toolCalls[0];
           const toolName = "toolName" in toolCall ? toolCall.toolName : "tool";
+          const args = "args" in toolCall ? toolCall.args : {};
           console.log(`[Investigation ${id}] Tool: ${toolName}`);
+          if (args && typeof args === "object" && "command" in args) {
+            console.log(`[Investigation ${id}] $ ${args.command}`);
+          }
         }
       },
     });

package/src/tui/app.tsx

@@ -1,6 +1,7 @@
 import { render } from "@opentui/solid";
 import { createSignal, For, Show, onMount } from "solid-js";
 import { createTextAttributes } from "@opentui/core";
+import "opentui-spinner/solid";
 import { getDebuggerAgent, buildIncidentPrompt } from "../mastra/index.js";
 import type { IncidentInput } from "../mastra/agents/debugger.js";
 
@@ -10,6 +11,25 @@ interface Message {
   content: string;
   timestamp: Date;
   toolName?: string;
+  command?: string;
+}
+
+// Conversation history for multi-turn debugging
+interface ConversationMessage {
+  role: "user" | "assistant";
+  content: string;
+}
+
+function formatHistoryAsPrompt(history: ConversationMessage[], newMessage: string): string {
+  if (history.length === 0) {
+    return newMessage;
+  }
+
+  const historyText = history
+    .map((msg) => `${msg.role === "user" ? "User" : "Assistant"}: ${msg.content}`)
+    .join("\n\n");
+
+  return `Previous conversation:\n${historyText}\n\nUser: ${newMessage}`;
 }
 
 type AppStatus = "idle" | "investigating" | "complete" | "error";
@@ -17,8 +37,55 @@ type AppStatus = "idle" | "investigating" | "complete" | "error";
 const ATTR_DIM = createTextAttributes({ dim: true });
 const ATTR_BOLD = createTextAttributes({ bold: true });
 
+function buildDisplayCommand(toolName: string, args: unknown): string | undefined {
+  if (!args || typeof args !== "object") return undefined;
+
+  const a = args as Record<string, unknown>;
+
+  switch (toolName) {
+    case "cli":
+      // CLI tool has direct command
+      return "command" in a ? String(a.command) : undefined;
+
+    case "git": {
+      // Build git command: git <command> [args...] [path]
+      if (!("command" in a)) return undefined;
+      const parts = ["git", String(a.command)];
+      if ("args" in a && Array.isArray(a.args)) {
+        parts.push(...a.args.map(String));
+      }
+      if ("path" in a && a.path) {
+        parts.push(String(a.path));
+      }
+      return parts.join(" ");
+    }
+
+    case "filesystem": {
+      // Build filesystem display: <operation> <path> [pattern]
+      if (!("operation" in a)) return undefined;
+      const op = String(a.operation);
+      const path = "path" in a ? String(a.path) : "";
+      if (op === "search" && "pattern" in a) {
+        return `grep "${a.pattern}" ${path}`;
+      }
+      if (op === "read") {
+        return `cat ${path}`;
+      }
+      if (op === "list") {
+        return `ls ${path}`;
+      }
+      return `${op} ${path}`;
+    }
+
+    default:
+      // Fallback: try to use command if it exists
+      return "command" in a ? String(a.command) : undefined;
+  }
+}
+
 function App() {
   const [messages, setMessages] = createSignal<Message[]>([]);
+  const [conversationHistory, setConversationHistory] = createSignal<ConversationMessage[]>([]);
   const [status, setStatus] = createSignal<AppStatus>("idle");
   const [currentTool, setCurrentTool] = createSignal<string | null>(null);
   const [inputValue, setInputValue] = createSignal("");
@@ -40,29 +107,51 @@ function App() {
     setError(null);
     setCurrentTool(null);
 
+    // Add user message to UI
     addMessage({
       role: "user",
       content: incident.description,
     });
 
+    // Build prompt: use full incident prompt for first message, include history for follow-ups
+    const isFirstMessage = conversationHistory().length === 0;
+    const userContent = isFirstMessage
+      ? buildIncidentPrompt(incident)
+      : incident.description;
+
+    // Format prompt with conversation history
+    const prompt = formatHistoryAsPrompt(conversationHistory(), userContent);
+
+    // Add user message to conversation history
+    setConversationHistory((prev) => [
+      ...prev,
+      { role: "user", content: userContent },
+    ]);
+
     try {
       const agent = getDebuggerAgent();
-      const prompt = buildIncidentPrompt(incident);
 
       let assistantContent = "";
 
+      // Send the formatted prompt to the agent
       const stream = await agent.stream(prompt, {
         maxSteps: 20,
         onStepFinish: ({ toolCalls }) => {
           if (toolCalls && toolCalls.length > 0) {
-            const toolCall = toolCalls[0];
-            const toolName =
-              "toolName" in toolCall ? String(toolCall.toolName) : "tool";
+            const toolCall = toolCalls[0] as { payload?: { toolName?: string; args?: unknown } };
+            const payload = toolCall.payload;
+            const toolName = payload?.toolName ?? "tool";
+            const args = payload?.args ?? {};
+
+            // Build display command based on tool type
+            const command = buildDisplayCommand(toolName, args);
+
             setCurrentTool(toolName);
             addMessage({
               role: "tool",
-              content: `Executing ${toolName}...`,
+              content: command ? `$ ${command}` : `Executing ${toolName}...`,
               toolName,
+              command,
             });
           }
         },
@@ -72,11 +161,18 @@ function App() {
         assistantContent += chunk;
       }
 
+      // Add assistant response to UI
       addMessage({
         role: "assistant",
         content: assistantContent,
       });
 
+      // Add assistant response to conversation history
+      setConversationHistory((prev) => [
+        ...prev,
+        { role: "assistant", content: assistantContent },
+      ]);
+
       setStatus("complete");
       setCurrentTool(null);
     } catch (err) {
@@ -185,7 +281,13 @@ function App() {
                     </box>
                   </Show>
                   <Show when={msg.role === "tool"}>
-                    <box flexDirection="row" gap={1}>
+                    <box flexDirection="row" gap={1} alignItems="center">
+                      <Show
+                        when={status() === "investigating" && msg.id === messages().filter(m => m.role === "tool").at(-1)?.id}
+                        fallback={<text fg="green">✓</text>}
+                      >
+                        <spinner name="dots" color="blue" />
+                      </Show>
                       <text fg="blue" attributes={ATTR_DIM}>
                         [{msg.toolName}]
                       </text>

package/src/mastra/tools/kubectl.ts

@@ -1,107 +0,0 @@
-import { createTool } from "@mastra/core/tools";
-import { z } from "zod";
-import { execCommand } from "../../sandbox/bashlet.js";
-
-const ALLOWED_COMMANDS = ["get", "describe", "logs", "top", "events"] as const;
-
-const KubectlInputSchema = z.object({
-  command: z.enum(ALLOWED_COMMANDS).describe("The kubectl command to run"),
-  resource: z
-    .string()
-    .optional()
-    .describe(
-      "Resource type (e.g., pods, deployments, services, configmaps, secrets)"
-    ),
-  name: z.string().optional().describe("Specific resource name"),
-  namespace: z
-    .string()
-    .optional()
-    .describe("Kubernetes namespace (defaults to current context namespace)"),
-  flags: z
-    .array(z.string())
-    .optional()
-    .describe(
-      "Additional flags (e.g., ['-o', 'yaml'], ['--tail', '100'], ['-l', 'app=myapp'])"
-    ),
-});
-
-// Output type (no schema validation to allow error returns)
-interface KubectlOutput {
-  success: boolean;
-  output: string;
-  error?: string;
-}
-
-function filterSensitiveData(output: string): string {
-  // Redact potential secrets, tokens, and passwords
-  return output
-    .replace(
-      /(password|secret|token|key|credential)[\s:=]+["']?[^\s"'\n]+["']?/gi,
-      "$1: [REDACTED]"
-    )
-    .replace(/Bearer\s+[^\s]+/gi, "Bearer [REDACTED]")
-    .replace(/-----BEGIN[^-]+-----[\s\S]*?-----END[^-]+-----/g, "[REDACTED CERTIFICATE/KEY]");
-}
-
-export const kubectlTool = createTool({
-  id: "kubectl",
-  description: `Execute kubectl commands to inspect Kubernetes resources.
-Available commands: ${ALLOWED_COMMANDS.join(", ")}.
-Use this to get information about pods, deployments, services, logs, and cluster events.
-Examples:
-- Get all pods: { command: "get", resource: "pods", flags: ["-A"] }
-- Get pod logs: { command: "logs", name: "my-pod", namespace: "default", flags: ["--tail", "100"] }
-- Describe deployment: { command: "describe", resource: "deployment", name: "my-app" }
-- Get events: { command: "events", namespace: "production", flags: ["--sort-by", ".lastTimestamp"] }`,
-
-  inputSchema: KubectlInputSchema,
-
-  execute: async (inputData): Promise<KubectlOutput> => {
-    const { command, resource, name, namespace, flags } = inputData;
-
-    // Build kubectl command
-    const parts = ["kubectl", command];
-
-    if (resource) {
-      parts.push(resource);
-    }
-
-    if (name) {
-      parts.push(name);
-    }
-
-    if (namespace) {
-      parts.push("-n", namespace);
-    }
-
-    if (flags && flags.length > 0) {
-      parts.push(...flags);
-    }
-
-    const fullCommand = parts.join(" ");
-
-    try {
-      const result = await execCommand(fullCommand);
-
-      if (result.exitCode !== 0) {
-        return {
-          success: false,
-          output: "",
-          error: result.stderr || `Command failed with exit code ${result.exitCode}`,
-        };
-      }
-
-      return {
-        success: true,
-        output: filterSensitiveData(result.stdout),
-        error: result.stderr ? filterSensitiveData(result.stderr) : undefined,
-      };
-    } catch (error) {
-      return {
-        success: false,
-        output: "",
-        error: error instanceof Error ? error.message : String(error),
-      };
-    }
-  },
-});