triagent 0.1.0-alpha2 → 0.1.0-alpha3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "triagent",
-  "version": "0.1.0-alpha2",
+  "version": "0.1.0-alpha3",
   "description": "AI-powered Kubernetes debugging agent with terminal UI",
   "type": "module",
   "main": "src/index.ts",

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
@@ -179,6 +181,7 @@ async function handleConfigCommand(args: CliArgs): Promise<void> {
     "aiProvider",
     "aiModel",
     "apiKey",
+    "baseUrl",
     "webhookPort",
     "codebasePath",
     "kubeConfigPath",

package/src/mastra/agents/debugger.ts

@@ -163,14 +163,17 @@ 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: {
       cli: cliTool,
       git: gitTool,