comfy-qa 2.0.0 → 2.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "comfy-qa",
-  "version": "2.0.0",
+  "version": "2.1.0",
   "description": "ComfyUI QA automation CLI",
   "repository": {
     "type": "git",

package/src/agent/qa-research.ts

@@ -79,26 +79,14 @@ interface ResearchResults {
 
 const ANTHROPIC_KEY = process.env.ANTHROPIC_API_KEY_QA ?? process.env.ANTHROPIC_API_KEY ?? "";
 const OPENROUTER_KEY = process.env.OPENROUTER_API_KEY ?? "";
+const OPENROUTER_MODEL = process.env.OPENROUTER_MODEL ?? "openai/gpt-4.5";
 
 import Anthropic from "@anthropic-ai/sdk";
 
 const anthropicClient = ANTHROPIC_KEY ? new Anthropic({ apiKey: ANTHROPIC_KEY, timeout: 60_000 }) : null;
 
 async function callLLM(system: string, messages: any[]): Promise<string> {
-  if (anthropicClient) {
-    try {
-      const res = await anthropicClient.messages.create({
-        model: "claude-sonnet-4-20250514",
-        max_tokens: 8192,
-        system,
-        messages,
-      });
-      return res.content?.[0]?.type === "text" ? res.content[0].text : "";
-    } catch (err: any) {
-      console.log(`  ⚠ Anthropic SDK: ${err.message?.slice(0, 80)}`);
-    }
-  }
-
+  // Prefer OpenRouter
   if (OPENROUTER_KEY) {
     try {
       const controller = new AbortController();
@@ -108,19 +96,35 @@ async function callLLM(system: string, messages: any[]): Promise<string> {
         signal: controller.signal,
         headers: { Authorization: `Bearer ${OPENROUTER_KEY}`, "content-type": "application/json" },
         body: JSON.stringify({
-          model: "anthropic/claude-sonnet-4-20250514",
+          model: OPENROUTER_MODEL,
           messages: [{ role: "system", content: system }, ...messages],
           max_tokens: 8192,
         }),
       });
       clearTimeout(timer);
       const json = (await res.json()) as any;
-      return json.choices?.[0]?.message?.content ?? "";
+      if (json.choices?.[0]?.message?.content) return json.choices[0].message.content;
+      console.log(`  ⚠ OpenRouter: ${json.error?.message?.slice(0, 80) ?? "empty response"}`);
     } catch (err: any) {
       console.log(`  ⚠ OpenRouter: ${err.message?.slice(0, 60)}`);
     }
   }
 
+  // Fallback: Anthropic SDK
+  if (anthropicClient) {
+    try {
+      const res = await anthropicClient.messages.create({
+        model: "claude-sonnet-4-20250514",
+        max_tokens: 8192,
+        system,
+        messages,
+      });
+      return res.content?.[0]?.type === "text" ? res.content[0].text : "";
+    } catch (err: any) {
+      console.log(`  ⚠ Anthropic SDK: ${err.message?.slice(0, 80)}`);
+    }
+  }
+
   return "";
 }
 

package/src/agent/research.ts

@@ -1,8 +1,27 @@
 import { $ } from "bun";
 import type { PRInfo, IssueInfo } from "../utils/github";
 
-/** Call Claude via CLI (already authenticated) or SDK if ANTHROPIC_API_KEY is set */
+const OPENROUTER_KEY = process.env.OPENROUTER_API_KEY ?? "";
+const OPENROUTER_MODEL = process.env.OPENROUTER_MODEL ?? "openai/gpt-4.5";
+
 async function callClaude(prompt: string): Promise<string> {
+  // Prefer OpenRouter
+  if (OPENROUTER_KEY) {
+    const res = await fetch("https://openrouter.ai/api/v1/chat/completions", {
+      method: "POST",
+      headers: { Authorization: `Bearer ${OPENROUTER_KEY}`, "content-type": "application/json" },
+      body: JSON.stringify({
+        model: OPENROUTER_MODEL,
+        messages: [{ role: "user", content: prompt }],
+        max_tokens: 4096,
+      }),
+    });
+    const json = (await res.json()) as any;
+    if (json.choices?.[0]?.message?.content) return json.choices[0].message.content;
+    console.log(`  ⚠ OpenRouter: ${json.error?.message?.slice(0, 80) ?? "empty response"}`);
+  }
+
+  // Fallback: Anthropic SDK
   const apiKey = process.env.ANTHROPIC_API_KEY_QA ?? process.env.ANTHROPIC_API_KEY;
   if (apiKey) {
     const Anthropic = (await import("@anthropic-ai/sdk")).default;
@@ -14,7 +33,8 @@ async function callClaude(prompt: string): Promise<string> {
     });
     return response.content[0].type === "text" ? response.content[0].text : "";
   }
-  // Fallback: pipe through claude CLI via stdin
+
+  // Last resort: claude CLI
   const proc = Bun.spawn(["claude", "--print", "--model", "claude-opus-4-6"], {
     stdin: new TextEncoder().encode(prompt),
     stdout: "pipe",

package/src/utils/github.ts

@@ -37,46 +37,56 @@ export function parseRef(ref: string): { owner: string; repo: string; number?: n
 }
 
 export async function fetchPR(owner: string, repo: string, number: number): Promise<PRInfo> {
-  const json = await $`gh pr view ${number} --repo ${owner}/${repo} --json number,title,body,state,headRefName,baseRefName,author,labels,url,files,comments`.text();
-  const raw = JSON.parse(json);
+  const [prJson, filesJson, commentsJson] = await Promise.all([
+    $`gh api repos/${owner}/${repo}/pulls/${number}`.text(),
+    $`gh api repos/${owner}/${repo}/pulls/${number}/files --paginate`.text(),
+    $`gh api repos/${owner}/${repo}/issues/${number}/comments --paginate`.text(),
+  ]);
+  const pr = JSON.parse(prJson);
+  const files = JSON.parse(filesJson);
+  const comments = JSON.parse(commentsJson);
   return {
-    number: raw.number,
-    title: raw.title,
-    body: raw.body || "",
-    state: raw.state,
-    headRefName: raw.headRefName,
-    baseRefName: raw.baseRefName,
-    author: raw.author?.login || "unknown",
-    labels: (raw.labels || []).map((l: any) => l.name),
-    url: raw.url,
-    files: (raw.files || []).map((f: any) => ({
-      path: f.path,
+    number: pr.number,
+    title: pr.title,
+    body: pr.body || "",
+    state: pr.state,
+    headRefName: pr.head.ref,
+    baseRefName: pr.base.ref,
+    author: pr.user?.login || "unknown",
+    labels: (pr.labels || []).map((l: any) => l.name),
+    url: pr.html_url,
+    files: files.map((f: any) => ({
+      path: f.filename,
       additions: f.additions,
       deletions: f.deletions,
     })),
-    comments: (raw.comments || []).map((c: any) => ({
-      author: c.author?.login || "unknown",
+    comments: comments.map((c: any) => ({
+      author: c.user?.login || "unknown",
       body: c.body || "",
-      createdAt: c.createdAt,
+      createdAt: c.created_at,
     })),
   };
 }
 
 export async function fetchIssue(owner: string, repo: string, number: number): Promise<IssueInfo> {
-  const json = await $`gh issue view ${number} --repo ${owner}/${repo} --json number,title,body,state,author,labels,url,comments`.text();
-  const raw = JSON.parse(json);
+  const [issueJson, commentsJson] = await Promise.all([
+    $`gh api repos/${owner}/${repo}/issues/${number}`.text(),
+    $`gh api repos/${owner}/${repo}/issues/${number}/comments --paginate`.text(),
+  ]);
+  const issue = JSON.parse(issueJson);
+  const comments = JSON.parse(commentsJson);
   return {
-    number: raw.number,
-    title: raw.title,
-    body: raw.body || "",
-    state: raw.state,
-    author: raw.author?.login || "unknown",
-    labels: (raw.labels || []).map((l: any) => l.name),
-    url: raw.url,
-    comments: (raw.comments || []).map((c: any) => ({
-      author: c.author?.login || "unknown",
+    number: issue.number,
+    title: issue.title,
+    body: issue.body || "",
+    state: issue.state,
+    author: issue.user?.login || "unknown",
+    labels: (issue.labels || []).map((l: any) => l.name),
+    url: issue.html_url,
+    comments: comments.map((c: any) => ({
+      author: c.user?.login || "unknown",
       body: c.body || "",
-      createdAt: c.createdAt,
+      createdAt: c.created_at,
     })),
   };
 }