small changes to prompt and caching

package-lock.json

@@ -11,7 +11,7 @@
       "license": "ISC",
       "dependencies": {
         "@hono/node-server": "^2.0.3",
-        "@langchain/anthropic": "^1.3.29",
+        "@langchain/anthropic": "^1.4.1",
         "@langchain/core": "^1.1.45",
         "@langchain/google-genai": "^2.1.31",
         "@langchain/langgraph": "^1.3.2",
@@ -36,9 +36,9 @@
       }
     },
     "node_modules/@anthropic-ai/sdk": {
-      "version": "0.95.2",
-      "resolved": "https://registry.npmjs.org/@anthropic-ai/sdk/-/sdk-0.95.2.tgz",
-      "integrity": "sha512-Egddwo3sheo1PzUrMkZnH6VkQYwS0h/b/i8vSK8Ta9M45UQipAMeDFH57dYuDAfXMEUUGeKw6CMlremgMZgrSQ==",
+      "version": "0.103.0",
+      "resolved": "https://registry.npmjs.org/@anthropic-ai/sdk/-/sdk-0.103.0.tgz",
+      "integrity": "sha512-1uG7RNgoHTUxzOXqSCODKt0UTVlxWiHk/2Tt2/uQJiPW7XzBeKVuJyd3Aw6T3LPyvZV/jDTnPLX7SaM70WLLjA==",
       "license": "MIT",
       "dependencies": {
         "json-schema-to-ts": "^3.1.1",
@@ -199,23 +199,25 @@
       }
     },
     "node_modules/@langchain/anthropic": {
-      "version": "1.4.0",
-      "resolved": "https://registry.npmjs.org/@langchain/anthropic/-/anthropic-1.4.0.tgz",
-      "integrity": "sha512-rs1yVydrHjyiD31uChdCnKZpmDuKa0Bpz8Raiy9GvqnqmfXPMe0oOrap/2paE+NRSinDbtax8mMpP/yv8EbO1A==",
+      "version": "1.4.1",
+      "resolved": "https://registry.npmjs.org/@langchain/anthropic/-/anthropic-1.4.1.tgz",
+      "integrity": "sha512-h3b6hxThcfh0WdmpuWr+qBi74MN+0BpNI/4H681vwXxbD3hLr2qMYN6ghqcPQhCxGJjg8ufs85qu2/ldSWonYQ==",
       "license": "MIT",
       "dependencies": {
-        "@anthropic-ai/sdk": "^0.95.1",
+        "@anthropic-ai/sdk": "^0.103.0",
         "zod": "^3.25.76 || ^4"
       },
       "engines": {
         "node": ">=20"
       },
       "peerDependencies": {
-        "@langchain/core": "^1.1.47"
+        "@langchain/core": "^1.1.49"
       }
     },
     "node_modules/@langchain/core": {
-      "version": "1.1.47",
+      "version": "1.1.49",
+      "resolved": "https://registry.npmjs.org/@langchain/core/-/core-1.1.49.tgz",
+      "integrity": "sha512-7wkN3Qv/qZqsY0p3h48CNu6E6y5GMYatYxj+JrX4uVNBiqIVQm1Z528QrmayJWVW9SQTQicqRNoyTCzl+K9F8Q==",
       "license": "MIT",
       "dependencies": {
         "@cfworker/json-schema": "^4.0.2",

package.json

@@ -33,7 +33,7 @@
   },
   "dependencies": {
     "@hono/node-server": "^2.0.3",
-    "@langchain/anthropic": "^1.3.29",
+    "@langchain/anthropic": "^1.4.1",
     "@langchain/core": "^1.1.45",
     "@langchain/google-genai": "^2.1.31",
     "@langchain/langgraph": "^1.3.2",

src/agents/auditor/agent.ts

@@ -1,5 +1,4 @@
 import fs from "node:fs";
-import path from "node:path";
 
 import { HumanMessage, SystemMessage } from "@langchain/core/messages";
 import { END, type GraphNode, START, StateGraph } from "@langchain/langgraph";
@@ -7,18 +6,7 @@ import { z } from "zod";
 
 import { createLLM } from "../../config/llm.ts";
 import { logger } from "../../logger.ts";
-import {
-  DOC_BASENAMES,
-  DOC_EXTS,
-  MAX_DEPTH,
-  MAX_DOC_CHARS,
-  MAX_REFLECTIONS,
-  MAX_SOL_CHARS,
-  MIN_FILE_IMPORTANCE,
-  SKIP_DIRS,
-  SOL_EXT,
-  SOL_TEST_SUFFIXES,
-} from "./config.ts";
+import { MAX_DOC_CHARS, MAX_REFLECTIONS, MAX_SOL_CHARS, MIN_FILE_IMPORTANCE } from "./config.ts";
 import {
   FIND_VULNERABILITIES_PROMPT,
   GATHER_CONTEXT_PROMPT,
@@ -29,42 +17,12 @@ import {
 import { AuditorState, CandidateFindingSchema, FileRankingSchema, JudgeReviewSchema } from "./state.ts";
 import { buildRepoTree } from "./tools/repo-tree/tool.ts";
 import { analyzeSolidityFile } from "./tools/solidity-analyzer/tool.ts";
-import { buildReviewBlocks, matchLines } from "./utils.ts";
+import { buildReviewBlocks, matchLines, walkDirectory } from "./utils.ts";
 
 const llmHaiku = createLLM("anthropic", { model: "claude-haiku-4-5", maxTokens: 20000 });
 const llmOpus = createLLM("anthropic", { model: "claude-opus-4-8", temperature: null, maxTokens: 20000 });
 const llmSonnet = createLLM("anthropic", { model: "claude-sonnet-4-6", maxTokens: 20000 });
 
-const walkDirectory = (dir: string, depth: number, solFiles: string[], docFiles: string[]) => {
-  if (depth > MAX_DEPTH) return;
-
-  let entries: fs.Dirent[];
-  try {
-    entries = fs.readdirSync(dir, { withFileTypes: true });
-  } catch {
-    return;
-  }
-
-  for (const entry of entries) {
-    if (entry.isDirectory()) {
-      if (!SKIP_DIRS.has(entry.name)) {
-        walkDirectory(path.join(dir, entry.name), depth + 1, solFiles, docFiles);
-      }
-    } else if (entry.isFile()) {
-      const fullPath = path.join(dir, entry.name);
-      const ext = path.extname(entry.name).toLowerCase();
-      const base = path.basename(entry.name, ext).toLowerCase();
-
-      if (ext === SOL_EXT) {
-        const isTest = SOL_TEST_SUFFIXES.some((suffix) => entry.name.endsWith(suffix));
-        if (!isTest) solFiles.push(fullPath);
-      } else if (DOC_EXTS.has(ext) || DOC_BASENAMES.has(base)) {
-        docFiles.push(fullPath);
-      }
-    }
-  }
-};
-
 const defineScope: GraphNode<typeof AuditorState> = async (state) => {
   logger.info(`defineScope: walking repo at ${state.repoPath}`);
 
@@ -152,7 +110,9 @@ const gatherContext: GraphNode<typeof AuditorState> = async (state) => {
 
   const model = llmHaiku.withStructuredOutput(z.object({ context: z.string() }));
   const result = await model.invoke([
-    new SystemMessage({ content: [{ type: "text", text: GATHER_CONTEXT_PROMPT, cache_control: { type: "ephemeral" } }] }),
+    new SystemMessage({
+      content: [{ type: "text", text: GATHER_CONTEXT_PROMPT, cache_control: { type: "ephemeral" } }],
+    }),
     new HumanMessage(parts.join("\n\n")),
   ]);
 
@@ -176,46 +136,52 @@ const findVulnerabilities: GraphNode<typeof AuditorState> = async (state) => {
     `findVulnerabilities: invoking LLM for ${state.scope.length} file(s) in parallel (iteration ${state.reflectionCount + 1})`,
   );
 
-  const cachedContext = { type: "text" as const, text: `Contexto do Protocolo:\n${state.repoContext}`, cache_control: { type: "ephemeral" as const } };
-
-  const allFindings = await Promise.all(
-    state.scope.map(async (filePath) => {
-      let source: string;
-      try {
-        source = fs.readFileSync(filePath, "utf-8").slice(0, MAX_SOL_CHARS);
-      } catch {
-        return [];
-      }
-      if (!source) return [];
-
-      const fileEntries = isReflection
-        ? state.candidateFindings
-            .map((f, i) => ({ finding: f, review: state.judgeReviews[i] }))
-            .filter(({ finding }) => finding.path === filePath)
-        : [];
-
-      const isRefinement = fileEntries.length > 0;
-      const promptText = isRefinement ? REFINE_VULNERABILITIES_PROMPT : FIND_VULNERABILITIES_PROMPT;
-      const contractText = isRefinement
-        ? `Contrato (${filePath}):\n\n${source}\n\n${buildReviewBlocks(fileEntries, state.reflectionCount)}`
-        : `Contrato (${filePath}):\n\n${source}`;
-
-      logger.debug(`findVulnerabilities: processing ${filePath}`);
-
-      const result = await model.invoke([
-        new SystemMessage({ content: [{ type: "text", text: promptText, cache_control: { type: "ephemeral" } }] }),
-        new HumanMessage({ content: [cachedContext, { type: "text", text: contractText }] }),
-      ]);
-
-      return result.findings.map((finding: any) => ({
-        ...finding,
-        path: filePath,
-        location: matchLines(source, finding.codeSnippet) ?? "",
-      }));
-    }),
-  );
+  const cachedContext = {
+    type: "text" as const,
+    text: `Contexto do Protocolo:\n${state.repoContext}`,
+    cache_control: { type: "ephemeral" as const },
+  };
+
+  const processFile = async (filePath: string) => {
+    let source: string;
+    try {
+      source = fs.readFileSync(filePath, "utf-8").slice(0, MAX_SOL_CHARS);
+    } catch {
+      return [];
+    }
+    if (!source) return [];
+
+    const fileEntries = isReflection
+      ? state.candidateFindings
+          .map((f, i) => ({ finding: f, review: state.judgeReviews[i] }))
+          .filter(({ finding }) => finding.path === filePath)
+      : [];
+
+    const isRefinement = fileEntries.length > 0;
+    const promptText = isRefinement ? REFINE_VULNERABILITIES_PROMPT : FIND_VULNERABILITIES_PROMPT;
+    const contractText = isRefinement
+      ? `Contrato (${filePath}):\n\n${source}\n\n${buildReviewBlocks(fileEntries, state.reflectionCount)}`
+      : `Contrato (${filePath}):\n\n${source}`;
+
+    logger.debug(`findVulnerabilities: processing ${filePath}`);
+
+    const result = await model.invoke([
+      new SystemMessage({ content: [{ type: "text", text: promptText, cache_control: { type: "ephemeral" } }] }),
+      new HumanMessage({ content: [cachedContext, { type: "text", text: contractText }] }),
+    ]);
+
+    return result.findings.map((finding: any) => ({
+      ...finding,
+      path: filePath,
+      location: matchLines(source, finding.codeSnippet) ?? "",
+    }));
+  };
+
+  const [firstFile, ...restFiles] = state.scope;
+  const firstFindings = firstFile ? await processFile(firstFile) : [];
+  const restFindings = await Promise.all(restFiles.map(processFile));
+  const candidateFindings = [firstFindings, ...restFindings].flat();
 
-  const candidateFindings = allFindings.flat();
   logger.info(`findVulnerabilities: LLM returned ${candidateFindings.length} total candidate finding(s)`);
   logger.debug(`findVulnerabilities: findings:\n${JSON.stringify(candidateFindings, null, 2)}`);
 
@@ -236,31 +202,40 @@ const judgeFindings: GraphNode<typeof AuditorState> = async (state) => {
 
   logger.info(`judgeFindings: reviewing ${state.candidateFindings.length} candidate finding(s) in parallel`);
 
-  const cachedContext = { type: "text" as const, text: `Contexto do Protocolo:\n${state.repoContext}`, cache_control: { type: "ephemeral" as const } };
-
-  const reviews = await Promise.all(
-    state.candidateFindings.map(async (finding, i) => {
-      let source: string;
-      try {
-        source = fs.readFileSync(finding.path, "utf-8").slice(0, MAX_SOL_CHARS);
-      } catch {
-        source = "";
-      }
-
-      const findingText = `[Achado ${i + 1}] ${finding.title}\nSeveridade: ${finding.severity}\nDescrição: ${finding.description}\nLocalização: ${finding.path} linhas ${finding.location}\nCódigo:\n\`\`\`solidity\n${finding.codeSnippet}\n\`\`\``;
-
-      logger.debug(`judgeFindings: reviewing finding ${i + 1}: ${finding.title}`);
-      return model.invoke([
-        new SystemMessage({ content: [{ type: "text", text: JUDGE_FINDINGS_PROMPT, cache_control: { type: "ephemeral" } }] }),
-        new HumanMessage({
-          content: [
-            cachedContext,
-            { type: "text", text: `Contrato (${finding.path}):\n\n${source}\n\nAchado para Revisão:\n\n${findingText}` },
-          ],
-        }),
-      ]);
-    }),
-  );
+  const cachedContext = {
+    type: "text" as const,
+    text: `Contexto do Protocolo:\n${state.repoContext}`,
+    cache_control: { type: "ephemeral" as const },
+  };
+
+  const reviewFinding = async (finding: (typeof state.candidateFindings)[number], i: number) => {
+    let source: string;
+    try {
+      source = fs.readFileSync(finding.path, "utf-8").slice(0, MAX_SOL_CHARS);
+    } catch {
+      source = "";
+    }
+
+    const findingText = `[Achado ${i + 1}] ${finding.title}\nSeveridade: ${finding.severity}\nDescrição: ${finding.description}\nLocalização: ${finding.path} linhas ${finding.location}\nCódigo:\n\`\`\`solidity\n${finding.codeSnippet}\n\`\`\``;
+
+    logger.debug(`judgeFindings: reviewing finding ${i + 1}: ${finding.title}`);
+    return model.invoke([
+      new SystemMessage({
+        content: [{ type: "text", text: JUDGE_FINDINGS_PROMPT, cache_control: { type: "ephemeral" } }],
+      }),
+      new HumanMessage({
+        content: [
+          cachedContext,
+          { type: "text", text: `Contrato (${finding.path}):\n\n${source}\n\nAchado para Revisão:\n\n${findingText}` },
+        ],
+      }),
+    ]);
+  };
+
+  const [firstFinding, ...restFindings] = state.candidateFindings;
+  const firstReview = await reviewFinding(firstFinding, 0);
+  const restReviews = await Promise.all(restFindings.map((f, i) => reviewFinding(f, i + 1)));
+  const reviews = [firstReview, ...restReviews];
 
   const confirmedEntries = state.candidateFindings
     .map((finding, i) => ({ finding, review: reviews[i] }))

src/agents/auditor/prompts.ts

@@ -34,13 +34,13 @@ Apenas os caminhos críticos ponta a ponta que movem valor ou alteram estado rel
 Formato por passo: \`ação (função) → efeito colateral → variável/estado alterado\`.
 Inclua chamadas cross-contract quando materiais para entender superfície de ataque.
 
-## Invariantes e Propriedades de Segurança
+## Invariantes e Propriedades
 Condições que devem ser verdadeiras para o protocolo operar corretamente, derivadas APENAS de \`require\`/\`assert\`/\`revert\` explícitos no código, NatSpec \`@dev\`, ou comentários que as declarem literalmente. Separe em dois grupos:
-- **Contábeis**: balanços, totais, proporções (ex.: \`totalDebt == Σ userDebt[i]\`, \`reservas >= totalSupply * exchangeRate\`)
+- **Econômicas**: balanços, totais, proporções (ex.: \`totalDebt == Σ userDebt[i]\`, \`reservas >= totalSupply * exchangeRate\`)
 - **De controle**: acesso, sequência de operações, transições de estado permitidas
 
 ## Trust Assumptions
-⚠ **SOMENTE o que estiver EXPLICITAMENTE declarado** em código-fonte (require, NatSpec, comentários inline) ou na documentação. **NÃO inferir. NÃO supor. NÃO extrapolar.** Se não há declaração explícita sobre confiança em um componente externo ou comportamento esperado, ele NÃO entra nesta seção — mesmo que pareça óbvio.
+**SOMENTE o que estiver EXPLICITAMENTE declarado** em código-fonte (require, NatSpec, comentários inline) ou na documentação. **NÃO inferir. NÃO supor. NÃO extrapolar.** Se não há declaração explícita sobre confiança em um componente externo ou comportamento esperado, ele NÃO entra nesta seção — mesmo que pareça óbvio.
 Bullets curtos com referência à fonte (ex.: "owner pode pausar o contrato — \`onlyOwner\` em \`pause()\`").
 
 ## Regras de Negócio e Restrições de Segurança
@@ -64,7 +64,7 @@ Além das categorias técnicas listadas abaixo, sua análise deve cobrir:
 
 ## Categorias técnicas a verificar sistematicamente
 
-Reentrância (simples, cross-function, cross-contract, read-only), controle de acesso (funções privilegiadas desprotegidas, erros em herança de roles), overflow/underflow (Solidity <0.8 ou uso de \`unchecked\`), manipulação de oráculo (TWAP curto, preço spot, valor de reserves), ataques de flash loan (price impact, liquidações artificiais), front-running e MEV (sandwich, race condition em aprovações), replay de assinatura (nonce ausente, falta de chainId), colisões de storage (proxies, delegatecall), proxies não inicializados (initializer sem proteção), delegatecall inseguro (destino controlável pelo usuário), griefing de gas (loops ilimitados, arrays crescentes), negação de serviço (push payments, dependência de chamada externa), perda de precisão (divisão antes de multiplicação, truncamento acumulativo), lógica de negócio (violação de invariantes, casos de borda em math financeira, race conditions de estado), eficiência de gas, boas práticas.
+Reentrância (simples, cross-function, cross-contract, read-only), controle de acesso (funções privilegiadas desprotegidas, erros em herança de roles), overflow/underflow (Solidity <0.8 ou uso de \`unchecked\`), manipulação de oráculo (TWAP curto, preço spot, valor de reserves), ataques de flash loan (price impact, liquidações artificiais), front-running e MEV (sandwich, race condition em aprovações), replay de assinatura (nonce ausente, falta de chainId), colisões de storage (proxies, delegatecall), proxies não inicializados (initializer sem proteção), delegatecall inseguro (destino controlável pelo usuário), griefing de gas (loops ilimitados, arrays crescentes), negação de serviço (push payments, dependência de chamada externa), perda de precisão (divisão antes de multiplicação, truncamento acumulativo), lógica de negócio (violação de invariantes, casos de borda em math financeira, race conditions de estado).
 
 ## Formato de saída
 
@@ -114,14 +114,14 @@ Para cada achado, forneça OBRIGATORIAMENTE todos os campos abaixo:
 - **review**: Análise técnica detalhada (3 a 6 frases) explicando o veredicto. Referencie: (a) o código específico envolvido, (b) invariantes ou premissas do protocolo que confirmam ou bloqueiam o exploit, (c) pré-condições necessárias para exploração, (d) controles mitigadores existentes que o auditor pode ter ignorado. Seja preciso — cite nomes de funções, variáveis e valores.
 - **isFalsePositive**: \`true\` se o achado NÃO for explorável na prática; \`false\` se for uma vulnerabilidade real.
 - **confidence**: Inteiro de 0 a 100 refletindo sua certeza no veredicto. Use < 60 apenas quando existir ambiguidade genuína no código.
-- **exploitablePaths**: Array de strings.
+- **exploitablePaths**: Retorne uma array de strings.
   - Se verdadeiro positivo (\`isFalsePositive: false\`): forneça 1 a 3 caminhos concretos de exploit, cada um com passos numerados, entradas realistas e estado do contrato antes/depois. Ex.: ["1. Atacante chama flashLoan(500k USDC). 2. No callback, chama deposit() inflando reserves. 3. Chama withdraw() com preço manipulado. 4. Lucra 50k USDC. Estado: reserves inflado temporariamente, totalShares inalterado."].
   - Se falso positivo (\`isFalsePositive: true\`): forneça o raciocínio exato que bloqueia cada caminho de exploit tentado pelo auditor.
 
 ## Critérios para falso positivo (aplique com rigor — não seja permissivo)
 
 1. O caminho de exploit é bloqueado por controle de acesso verificável no código.
-2. A vulnerabilidade já é totalmente mitigada por outro mecanismo no código (ex.: nonReentrant, require com validação suficiente).
+2. A vulnerabilidade já é totalmente mitigada por outro mecanismo no código (ex.: nonReentrant, onlyOwner, require com validação suficiente).
 3. A condição necessária para o exploit é impossível ou economicamente inviável dado o modelo do protocolo (ex.: requer ser o próprio contrato, ou lucro < custo de gas em qualquer cenário realista).
 4. O comportamento é explicitamente documentado como intencional nas premissas de design do protocolo.
 

src/agents/auditor/tools/solidity-analyzer/utils.ts

@@ -385,13 +385,11 @@ const renderContractFull = (contract: ContractAnalysis, imports: string[], lines
     lines.push("");
   }
 
-  lines.push("## Errors");
-  if (contract.errors.length === 0) {
-    lines.push("- None");
-  } else {
+  if (contract.errors.length > 0) {
+    lines.push("## Errors");
     for (const e of contract.errors) lines.push(`- \`${e.name}(${e.params.join(", ")})\``);
+    lines.push("");
   }
-  lines.push("");
 
   if (contract.modifiers.length > 0) {
     lines.push("## Modifiers");

src/agents/auditor/utils.ts

@@ -1,3 +1,38 @@
+import fs from "node:fs";
+import path from "node:path";
+
+import { DOC_BASENAMES, DOC_EXTS, MAX_DEPTH, SKIP_DIRS, SOL_EXT, SOL_TEST_SUFFIXES } from "./config.ts";
+
+export const walkDirectory = (dir: string, depth: number, solFiles: string[], docFiles: string[]) => {
+  if (depth > MAX_DEPTH) return;
+
+  let entries: fs.Dirent[];
+  try {
+    entries = fs.readdirSync(dir, { withFileTypes: true });
+  } catch {
+    return;
+  }
+
+  for (const entry of entries) {
+    if (entry.isDirectory()) {
+      if (!SKIP_DIRS.has(entry.name)) {
+        walkDirectory(path.join(dir, entry.name), depth + 1, solFiles, docFiles);
+      }
+    } else if (entry.isFile()) {
+      const fullPath = path.join(dir, entry.name);
+      const ext = path.extname(entry.name).toLowerCase();
+      const base = path.basename(entry.name, ext).toLowerCase();
+
+      if (ext === SOL_EXT) {
+        const isTest = SOL_TEST_SUFFIXES.some((suffix) => entry.name.endsWith(suffix));
+        if (!isTest) solFiles.push(fullPath);
+      } else if (DOC_EXTS.has(ext) || DOC_BASENAMES.has(base)) {
+        docFiles.push(fullPath);
+      }
+    }
+  }
+};
+
 export const matchLines = (fileContent: string, codeSnippet: string): string | null => {
   const fileLines = fileContent.split("\n");
   const snippetLines = codeSnippet.split("\n").map((line) => line.trim());