cto-ai-cli 5.0.0 → 5.2.0

package/dist/cli/gateway.js

@@ -97,7 +97,8 @@ var GOOGLE_MODELS = [
   { id: "gemini-1.5-pro", contextWindow: 2e6, costPerMInput: 1.25, costPerMOutput: 5, maxOutput: 8192 }
 ];
 function parseOpenAIRequest(body) {
-  const messages = (body.messages || []).map((m) => ({
+  const rawMessages = Array.isArray(body.messages) ? body.messages : [];
+  const messages = rawMessages.map((m) => ({
     role: m.role || "user",
     content: typeof m.content === "string" ? m.content : JSON.stringify(m.content)
   }));
@@ -109,34 +110,29 @@ function parseOpenAIRequest(body) {
     temperature: body.temperature
   };
 }
-function parseOpenAIResponse(body, streaming) {
-  if (streaming) {
-    return {
-      model: body.model || "unknown",
-      inputTokens: body.usage?.prompt_tokens || 0,
-      outputTokens: body.usage?.completion_tokens || 0,
-      content: body.choices?.[0]?.message?.content || "",
-      finishReason: body.choices?.[0]?.finish_reason || "stop"
-    };
-  }
+function parseOpenAIResponse(body, _streaming) {
+  const usage = body.usage;
+  const choices = Array.isArray(body.choices) ? body.choices : [];
+  const first = choices[0];
+  const msg = first?.message;
   return {
     model: body.model || "unknown",
-    inputTokens: body.usage?.prompt_tokens || 0,
-    outputTokens: body.usage?.completion_tokens || 0,
-    content: body.choices?.[0]?.message?.content || "",
-    finishReason: body.choices?.[0]?.finish_reason || "stop"
+    inputTokens: usage?.prompt_tokens || 0,
+    outputTokens: usage?.completion_tokens || 0,
+    content: msg?.content || "",
+    finishReason: first?.finish_reason || "stop"
   };
 }
 function parseAnthropicRequest(body) {
   const messages = [];
   if (body.system) {
-    messages.push({ role: "system", content: body.system });
+    messages.push({ role: "system", content: String(body.system) });
   }
-  for (const m of body.messages || []) {
-    messages.push({
-      role: m.role || "user",
-      content: typeof m.content === "string" ? m.content : m.content?.map((b) => b.text || "").join("\n") || ""
-    });
+  const rawMessages = Array.isArray(body.messages) ? body.messages : [];
+  for (const m of rawMessages) {
+    const content = typeof m.content === "string" ? m.content : Array.isArray(m.content) ? m.content.map((b) => String(b.text ?? "")).join("\n") : "";
+    const role = m.role || "user";
+    messages.push({ role, content });
   }
   return {
     model: body.model || "unknown",
@@ -147,43 +143,53 @@ function parseAnthropicRequest(body) {
   };
 }
 function parseAnthropicResponse(body, _streaming) {
+  const usage = body.usage;
+  const contentBlocks = Array.isArray(body.content) ? body.content : [];
   return {
     model: body.model || "unknown",
-    inputTokens: body.usage?.input_tokens || 0,
-    outputTokens: body.usage?.output_tokens || 0,
-    content: body.content?.map((b) => b.text || "").join("\n") || "",
+    inputTokens: usage?.input_tokens || 0,
+    outputTokens: usage?.output_tokens || 0,
+    content: contentBlocks.map((b) => String(b.text ?? "")).join("\n"),
     finishReason: body.stop_reason || "end_turn"
   };
 }
 function parseGoogleRequest(body) {
   const messages = [];
-  if (body.systemInstruction?.parts) {
+  const sysInst = body.systemInstruction;
+  if (sysInst && Array.isArray(sysInst.parts)) {
     messages.push({
       role: "system",
-      content: body.systemInstruction.parts.map((p) => p.text || "").join("\n")
+      content: sysInst.parts.map((p) => String(p.text ?? "")).join("\n")
     });
   }
-  for (const item of body.contents || []) {
+  const contents = Array.isArray(body.contents) ? body.contents : [];
+  for (const item of contents) {
     const role = item.role === "model" ? "assistant" : "user";
-    const content = item.parts?.map((p) => p.text || "").join("\n") || "";
+    const parts = Array.isArray(item.parts) ? item.parts : [];
+    const content = parts.map((p) => String(p.text ?? "")).join("\n");
     messages.push({ role, content });
   }
   const model = body.model || body.modelId || "gemini-2.0-flash";
+  const genConfig = body.generationConfig;
   return {
     model,
     messages,
     stream: body.stream === true,
-    maxTokens: body.generationConfig?.maxOutputTokens,
-    temperature: body.generationConfig?.temperature
+    maxTokens: genConfig?.maxOutputTokens,
+    temperature: genConfig?.temperature
   };
 }
 function parseGoogleResponse(body, _streaming) {
-  const candidate = body.candidates?.[0];
+  const candidates = Array.isArray(body.candidates) ? body.candidates : [];
+  const candidate = candidates[0];
+  const usage = body.usageMetadata;
+  const candidateContent = candidate?.content;
+  const parts = Array.isArray(candidateContent?.parts) ? candidateContent.parts : [];
   return {
     model: body.modelVersion || body.model || "gemini-2.0-flash",
-    inputTokens: body.usageMetadata?.promptTokenCount || 0,
-    outputTokens: body.usageMetadata?.candidatesTokenCount || 0,
-    content: candidate?.content?.parts?.map((p) => p.text || "").join("\n") || "",
+    inputTokens: usage?.promptTokenCount || 0,
+    outputTokens: usage?.candidatesTokenCount || 0,
+    content: parts.map((p) => String(p.text ?? "")).join("\n"),
     finishReason: candidate?.finishReason || "STOP"
   };
 }
@@ -1401,7 +1407,8 @@ async function optimizeContext(messages, analysis, config) {
     );
     return { messages: optimizedMessages, injected: true, optimizeDecisions };
   } catch (err) {
-    optimizeDecisions.push(`Context optimization failed: ${err.message}`);
+    const errMsg = err instanceof Error ? err.message : String(err);
+    optimizeDecisions.push(`Context optimization failed: ${errMsg}`);
     return { messages, injected: false, optimizeDecisions };
   }
 }
@@ -2186,8 +2193,8 @@ async function analyzeProject(projectPath, config) {
     maxDepth: mergedConfig.analysis.maxDepth
   });
   const tokenMethod = mergedConfig.tokens.method;
-  const files = [];
-  for (const entry of walkEntries) {
+  const BATCH_SIZE = 50;
+  async function estimateFileTokens(entry) {
     let tokens;
     if (tokenMethod === "tiktoken") {
       try {
@@ -2199,7 +2206,7 @@ async function analyzeProject(projectPath, config) {
     } else {
       tokens = countTokensChars4(entry.size);
     }
-    files.push({
+    return {
       path: entry.path,
       relativePath: entry.relativePath,
       extension: entry.extension,
@@ -2208,16 +2215,20 @@ async function analyzeProject(projectPath, config) {
       lines: entry.lines,
       lastModified: entry.lastModified,
       kind: classifyFileKind(entry.relativePath),
-      // Graph data — populated by graph analysis
       imports: [],
       importedBy: [],
       isHub: false,
       complexity: 0,
-      // Risk data — populated by risk analysis
       riskScore: 0,
       riskFactors: [],
       exclusionImpact: "none"
-    });
+    };
+  }
+  const files = [];
+  for (let i = 0; i < walkEntries.length; i += BATCH_SIZE) {
+    const batch = walkEntries.slice(i, i + BATCH_SIZE);
+    const results = await Promise.all(batch.map(estimateFileTokens));
+    files.push(...results);
   }
   const graph = buildProjectGraph(absPath, files);
   for (const file of files) {
@@ -2363,7 +2374,8 @@ var ContextGateway = class {
       this.analysis = analysis;
       return analysis;
     } catch (err) {
-      this.emit({ type: "error", message: `Analysis failed: ${err.message}`, error: err });
+      const message = err instanceof Error ? err.message : String(err);
+      this.emit({ type: "error", message: `Analysis failed: ${message}`, error: err instanceof Error ? err : void 0 });
       throw err;
     }
   }
@@ -2400,7 +2412,8 @@ var ContextGateway = class {
     try {
       body = await readBody(req, this.config.maxBodyBytes);
     } catch (err) {
-      const status = err.message === "body-too-large" ? 413 : 400;
+      const errMsg = err instanceof Error ? err.message : String(err);
+      const status = errMsg === "body-too-large" ? 413 : 400;
       res.writeHead(status, { "Content-Type": "application/json" });
       res.end(JSON.stringify({ error: status === 413 ? `Request body too large. Max: ${Math.round(this.config.maxBodyBytes / 1024 / 1024)}MB` : "Failed to read request body" }));
       return;
@@ -2511,12 +2524,13 @@ var ContextGateway = class {
     try {
       await this.proxyRequest(targetUrl, req, res, modifiedBody, provider, parsed, interceptResult, startTime);
     } catch (err) {
+      const errMsg = err instanceof Error ? err.message : String(err);
       if (!res.headersSent) {
-        const status = err.message === "upstream-timeout" ? 504 : 502;
+        const status = errMsg === "upstream-timeout" ? 504 : 502;
         res.writeHead(status, { "Content-Type": "application/json" });
-        res.end(JSON.stringify({ error: status === 504 ? "Upstream provider timeout" : `Proxy error: ${err.message}` }));
+        res.end(JSON.stringify({ error: status === 504 ? "Upstream provider timeout" : `Proxy error: ${errMsg}` }));
       }
-      this.emit({ type: "error", message: `Proxy error: ${err.message}`, error: err });
+      this.emit({ type: "error", message: `Proxy error: ${errMsg}`, error: err instanceof Error ? err : void 0 });
     }
   }
   // ===== PROXY =====