@vedangiitb/qwintly-core 1.4.9 → 1.4.11

package/dist/ai/generate/gemini.client.d.ts

@@ -1,5 +1,6 @@
 import { FunctionCallingConfigMode, GoogleGenAI, Tool } from "@google/genai";
 import type { ZodSchema } from "zod";
+export declare const DEFAULT_MODEL = "gemini-3.1-flash-lite";
 type AIResponseOptions = {
     tools?: Tool[];
     schema?: ZodSchema;

package/dist/ai/generate/gemini.client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.client.d.ts","sourceRoot":"","sources":["../../../src/ai/generate/gemini.client.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EAEzB,WAAW,EACX,IAAI,EACL,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAKrC,KAAK,iBAAiB,GAAG;IACvB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,eAAe,CAAC,EAAE,yBAAyB,CAAC;CAC7C,CAAC;AAEF,qBAAa,qBAAqB;IACzB,MAAM,EAAE,WAAW,CAAC;IACpB,KAAK,EAAE,MAAM,CAAiB;gBAEzB,YAAY,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM;IAOnC,UAAU,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,GAAE,iBAAsB;CAqC1E"}
+{"version":3,"file":"gemini.client.d.ts","sourceRoot":"","sources":["../../../src/ai/generate/gemini.client.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EAEzB,WAAW,EACX,IAAI,EACL,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAGrC,eAAO,MAAM,aAAa,0BAA0B,CAAC;AAErD,KAAK,iBAAiB,GAAG;IACvB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,eAAe,CAAC,EAAE,yBAAyB,CAAC;CAC7C,CAAC;AAEF,qBAAa,qBAAqB;IACzB,MAAM,EAAE,WAAW,CAAC;IACpB,KAAK,EAAE,MAAM,CAAiB;gBAEzB,YAAY,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM;IAOnC,UAAU,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,GAAE,iBAAsB;CAqC1E"}

package/dist/ai/generate/gemini.client.js

@@ -1,6 +1,6 @@
 import { FunctionCallingConfigMode, GoogleGenAI, } from "@google/genai";
 import { zodToJsonSchema } from "zod-to-json-schema";
-const DEFAULT_MODEL = "gemini-2.5-flash-lite";
+export const DEFAULT_MODEL = "gemini-3.1-flash-lite";
 export class GenerateGeminiReponse {
     constructor(geminiApiKey, model) {
         this.model = DEFAULT_MODEL;

package/dist/ai/generate/gemini.client.js.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.client.js","sourceRoot":"","sources":["../../../src/ai/generate/gemini.client.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EAEzB,WAAW,GAEZ,MAAM,eAAe,CAAC;AAEvB,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,MAAM,aAAa,GAAG,uBAAuB,CAAC;AAQ9C,MAAM,OAAO,qBAAqB;IAIhC,YAAY,YAAoB,EAAE,KAAc;QAFzC,UAAK,GAAW,aAAa,CAAC;QAGnC,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QACxD,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,OAAgB,EAAE,UAA6B,EAAE;QACvE,MAAM,EACJ,KAAK,EACL,MAAM,EACN,eAAe,GAAG,yBAAyB,CAAC,IAAI,GACjD,GAAG,OAAO,CAAC;QAEZ,MAAM,MAAM,GAA0B,EAAE,CAAC;QAEzC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;YACrB,MAAM,CAAC,UAAU,GAAG;gBAClB,qBAAqB,EAAE;oBACrB,IAAI,EAAE,eAAe;iBACtB;aACF,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,gBAAgB,GAAG,kBAAkB,CAAC;YAC7C,MAAM,CAAC,kBAAkB,GAAG,eAAe,CAAC,MAAa,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;gBAC9C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,OAAc;gBACxB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;aAClD,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,OAAO,GAAG,IAAI,KAAK,CACvB,yBAAyB,GAAG,EAAE,OAAO,IAAI,eAAe,EAAE,CAC3D,CAAC;YACD,OAAe,CAAC,KAAK,GAAG,GAAG,CAAC;YAC7B,MAAM,OAAO,CAAC;QAChB,CAAC;IACH,CAAC;CACF","sourcesContent":["import {\n  FunctionCallingConfigMode,\n  GenerateContentConfig,\n  GoogleGenAI,\n  Tool,\n} from \"@google/genai\";\nimport type { ZodSchema } from \"zod\";\nimport { zodToJsonSchema } from \"zod-to-json-schema\";\n\nconst DEFAULT_MODEL = \"gemini-2.5-flash-lite\";\n\ntype AIResponseOptions = {\n  tools?: Tool[];\n  schema?: ZodSchema;\n  toolCallingMode?: FunctionCallingConfigMode;\n};\n\nexport class GenerateGeminiReponse {\n  public gemini: GoogleGenAI;\n  public model: string = DEFAULT_MODEL;\n\n  constructor(geminiApiKey: string, model?: string) {\n    this.gemini = new GoogleGenAI({ apiKey: geminiApiKey });\n    if (model) {\n      this.model = model;\n    }\n  }\n\n  public async aiResponse(request: unknown, options: AIResponseOptions = {}) {\n    const {\n      tools,\n      schema,\n      toolCallingMode = FunctionCallingConfigMode.AUTO,\n    } = options;\n\n    const config: GenerateContentConfig = {};\n\n    if (tools && tools.length > 0) {\n      config.tools = tools;\n      config.toolConfig = {\n        functionCallingConfig: {\n          mode: toolCallingMode,\n        },\n      };\n    }\n\n    if (schema) {\n      config.responseMimeType = \"application/json\";\n      config.responseJsonSchema = zodToJsonSchema(schema as any);\n    }\n\n    try {\n      return await this.gemini.models.generateContent({\n        model: this.model,\n        contents: request as any,\n        ...(Object.keys(config).length > 0 && { config }),\n      });\n    } catch (err: any) {\n      const wrapped = new Error(\n        `AI generation failed: ${err?.message || \"Unknown error\"}`,\n      );\n      (wrapped as any).cause = err;\n      throw wrapped;\n    }\n  }\n}\n"]}
+{"version":3,"file":"gemini.client.js","sourceRoot":"","sources":["../../../src/ai/generate/gemini.client.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EAEzB,WAAW,GAEZ,MAAM,eAAe,CAAC;AAEvB,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,MAAM,CAAC,MAAM,aAAa,GAAG,uBAAuB,CAAC;AAQrD,MAAM,OAAO,qBAAqB;IAIhC,YAAY,YAAoB,EAAE,KAAc;QAFzC,UAAK,GAAW,aAAa,CAAC;QAGnC,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;QACxD,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,OAAgB,EAAE,UAA6B,EAAE;QACvE,MAAM,EACJ,KAAK,EACL,MAAM,EACN,eAAe,GAAG,yBAAyB,CAAC,IAAI,GACjD,GAAG,OAAO,CAAC;QAEZ,MAAM,MAAM,GAA0B,EAAE,CAAC;QAEzC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;YACrB,MAAM,CAAC,UAAU,GAAG;gBAClB,qBAAqB,EAAE;oBACrB,IAAI,EAAE,eAAe;iBACtB;aACF,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,gBAAgB,GAAG,kBAAkB,CAAC;YAC7C,MAAM,CAAC,kBAAkB,GAAG,eAAe,CAAC,MAAa,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;gBAC9C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,OAAc;gBACxB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;aAClD,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,OAAO,GAAG,IAAI,KAAK,CACvB,yBAAyB,GAAG,EAAE,OAAO,IAAI,eAAe,EAAE,CAC3D,CAAC;YACD,OAAe,CAAC,KAAK,GAAG,GAAG,CAAC;YAC7B,MAAM,OAAO,CAAC;QAChB,CAAC;IACH,CAAC;CACF","sourcesContent":["import {\n  FunctionCallingConfigMode,\n  GenerateContentConfig,\n  GoogleGenAI,\n  Tool,\n} from \"@google/genai\";\nimport type { ZodSchema } from \"zod\";\nimport { zodToJsonSchema } from \"zod-to-json-schema\";\n\nexport const DEFAULT_MODEL = \"gemini-3.1-flash-lite\";\n\ntype AIResponseOptions = {\n  tools?: Tool[];\n  schema?: ZodSchema;\n  toolCallingMode?: FunctionCallingConfigMode;\n};\n\nexport class GenerateGeminiReponse {\n  public gemini: GoogleGenAI;\n  public model: string = DEFAULT_MODEL;\n\n  constructor(geminiApiKey: string, model?: string) {\n    this.gemini = new GoogleGenAI({ apiKey: geminiApiKey });\n    if (model) {\n      this.model = model;\n    }\n  }\n\n  public async aiResponse(request: unknown, options: AIResponseOptions = {}) {\n    const {\n      tools,\n      schema,\n      toolCallingMode = FunctionCallingConfigMode.AUTO,\n    } = options;\n\n    const config: GenerateContentConfig = {};\n\n    if (tools && tools.length > 0) {\n      config.tools = tools;\n      config.toolConfig = {\n        functionCallingConfig: {\n          mode: toolCallingMode,\n        },\n      };\n    }\n\n    if (schema) {\n      config.responseMimeType = \"application/json\";\n      config.responseJsonSchema = zodToJsonSchema(schema as any);\n    }\n\n    try {\n      return await this.gemini.models.generateContent({\n        model: this.model,\n        contents: request as any,\n        ...(Object.keys(config).length > 0 && { config }),\n      });\n    } catch (err: any) {\n      const wrapped = new Error(\n        `AI generation failed: ${err?.message || \"Unknown error\"}`,\n      );\n      (wrapped as any).cause = err;\n      throw wrapped;\n    }\n  }\n}\n"]}

package/dist/ai/toolLoop/toolLoopRunner.d.ts

@@ -1,4 +1,5 @@
 import { FunctionCallingConfigMode, Tool } from "@google/genai";
+import type { GenTokensRepository } from "../../repository/genTokens.repository.js";
 import { EventType } from "../../types/events.js";
 import { ToolLoopContextPolicy } from "./toolLoopContext.js";
 export type ToolLoopResult = {
@@ -38,6 +39,11 @@ export type RunToolLoopOptions = {
     aiCallAutoRetryBaseMs?: number;
     aiCallAutoRetryMaxMs?: number;
     persistResponse?: (modelInput: any, modelOutput: any) => Promise<void>;
+    tokenPersistence?: {
+        repository: Pick<GenTokensRepository, "persistGenTokens">;
+        sessionId: string;
+        model: string;
+    };
 };
 export declare function runToolLoop(options: RunToolLoopOptions): Promise<ToolLoopResult>;
 //# sourceMappingURL=toolLoopRunner.d.ts.map

package/dist/ai/toolLoop/toolLoopRunner.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"toolLoopRunner.d.ts","sourceRoot":"","sources":["../../../src/ai/toolLoop/toolLoopRunner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAGhE,OAAO,EAAe,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAM/D,OAAO,EAML,qBAAqB,EACtB,MAAM,sBAAsB,CAAC;AAQ9B,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,GAAG,EAAE,CAAC;IAChB,aAAa,EAAE,GAAG,EAAE,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9B,QAAQ,EAAE,OAAO,CAAC;KACnB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,MAAM,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AAE9E,MAAM,MAAM,cAAc,GAAG;IAC3B,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,QAAQ,GAAG,CACrB,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE;IACP,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,yBAAyB,CAAC;CAC7C,KACE,OAAO,CAAC,cAAc,CAAC,CAAC;AAE7B,MAAM,MAAM,kBAAkB,GAAG;IAC/B,eAAe,EAAE,GAAG,EAAE,CAAC;IACvB,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,yBAAyB,CAAC;IAC5C,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,aAAa,CAAC,EAAE,qBAAqB,CAAC;IACtC,MAAM,EAAE,QAAQ,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,eAAe,CAAC,EAAE,CAAC,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACxE,CAAC;AAEF,wBAAsB,WAAW,CAC/B,OAAO,EAAE,kBAAkB,GAC1B,OAAO,CAAC,cAAc,CAAC,CA6kBzB"}
+{"version":3,"file":"toolLoopRunner.d.ts","sourceRoot":"","sources":["../../../src/ai/toolLoop/toolLoopRunner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAEhE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,0CAA0C,CAAC;AAEpF,OAAO,EAAe,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAM/D,OAAO,EAML,qBAAqB,EACtB,MAAM,sBAAsB,CAAC;AAQ9B,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,GAAG,EAAE,CAAC;IAChB,aAAa,EAAE,GAAG,EAAE,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9B,QAAQ,EAAE,OAAO,CAAC;KACnB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,MAAM,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AAE9E,MAAM,MAAM,cAAc,GAAG;IAC3B,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,QAAQ,GAAG,CACrB,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE;IACP,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,yBAAyB,CAAC;CAC7C,KACE,OAAO,CAAC,cAAc,CAAC,CAAC;AAE7B,MAAM,MAAM,kBAAkB,GAAG;IAC/B,eAAe,EAAE,GAAG,EAAE,CAAC;IACvB,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,yBAAyB,CAAC;IAC5C,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,aAAa,CAAC,EAAE,qBAAqB,CAAC;IACtC,MAAM,EAAE,QAAQ,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,eAAe,CAAC,EAAE,CAAC,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACvE,gBAAgB,CAAC,EAAE;QACjB,UAAU,EAAE,IAAI,CAAC,mBAAmB,EAAE,kBAAkB,CAAC,CAAC;QAC1D,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;CACH,CAAC;AAEF,wBAAsB,WAAW,CAC/B,OAAO,EAAE,kBAAkB,GAC1B,OAAO,CAAC,cAAc,CAAC,CA+nBzB"}

package/dist/ai/toolLoop/toolLoopRunner.js

@@ -3,16 +3,43 @@ import fs from "node:fs/promises";
 import { persistToolCall } from "../../services/toolcallPersist.service.js";
 import { EVENT_TYPES } from "../../types/events.js";
 import { STYLE_TOKEN_KEYS } from "../../types/styleConfig.js";
+import { getAvailableRoutes } from "../tools/helpers/pageConfigJson.helpers.js";
 import { createWorkspaceToolImpls } from "../tools/implementations/factories.js";
 import { parsePlannerTasksUnknown } from "./plannerTaskParser.js";
-import { getAvailableRoutes } from "../tools/helpers/pageConfigJson.helpers.js";
 import { compactForModel, DEFAULT_CONTEXT_POLICY, normalizeReadFileArgs, redactFunctionCallArgs, } from "./toolLoopContext.js";
 import { aiCallWithRetry, buildToolStatusMessage, recordToolEvent, serializeError, } from "./toolLoopRunnerUtils.js";
 export async function runToolLoop(options) {
     const isPlainObject = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
     const styleTokenKeySet = new Set(STYLE_TOKEN_KEYS);
     const { initialContents, tools, workspaceRoot, maxSteps = 30, toolCallingMode = FunctionCallingConfigMode.ANY, terminalToolNames = [], keepFullTrace = true, contextPolicy, aiCall, logger, applyPatchAutoRetryMax = 2, aiCallAutoRetryMax = 3, // must have it to try 3 times as gemini errors a lot due to high demand sometimes
-    aiCallAutoRetryBaseMs = 400, aiCallAutoRetryMaxMs = 10000, persistResponse, } = options;
+    aiCallAutoRetryBaseMs = 400, aiCallAutoRetryMaxMs = 10000, persistResponse, tokenPersistence, } = options;
+    const extractUsageTokenCounts = (rawResponse) => {
+        const usage = rawResponse?.usageMetadata;
+        const promptTokenCount = usage?.promptTokenCount;
+        const candidatesTokenCount = usage?.candidatesTokenCount;
+        const inputTokens = Number(promptTokenCount);
+        const outputTokens = Number(candidatesTokenCount);
+        if (!Number.isFinite(inputTokens) || inputTokens < 0)
+            return null;
+        if (!Number.isFinite(outputTokens) || outputTokens < 0)
+            return null;
+        return { inputTokens, outputTokens };
+    };
+    let totalInputTokens = 0;
+    let totalOutputTokens = 0;
+    let sawAnyTokenUsage = false;
+    const persistTokensOnce = async () => {
+        if (!tokenPersistence)
+            return;
+        if (!sawAnyTokenUsage)
+            return;
+        try {
+            await tokenPersistence.repository.persistGenTokens(tokenPersistence.sessionId, totalInputTokens, totalOutputTokens, tokenPersistence.model);
+        }
+        catch (err) {
+            console.error("Tool loop: failed to persist gen tokens", err);
+        }
+    };
     const nodeFs = {
         readFile: async (absolutePath) => fs.readFile(absolutePath, "utf-8"),
         writeFile: async (absolutePath, content) => fs.writeFile(absolutePath, content ?? "", "utf-8"),
@@ -64,7 +91,10 @@ export async function runToolLoop(options) {
         insert_element: async (args) => {
             const result = await impls.insertElementImpl(args);
             if (!result.success) {
-                const available = await getAvailableRoutes({ workspaceRoot, fs: nodeFs });
+                const available = await getAvailableRoutes({
+                    workspaceRoot,
+                    fs: nodeFs,
+                });
                 return {
                     success: false,
                     error: `insert_element failed: ${result.error}. Available routes are: ${JSON.stringify(available)}. If you intend to create a new route, create it using the 'create_new_route' tool.`,
@@ -196,8 +226,17 @@ export async function runToolLoop(options) {
                 });
             }
         }
+        {
+            const usage = extractUsageTokenCounts(response);
+            if (usage) {
+                sawAnyTokenUsage = true;
+                totalInputTokens += usage.inputTokens;
+                totalOutputTokens += usage.outputTokens;
+            }
+        }
         const functionCalls = response.functionCalls ?? [];
         if (functionCalls.length === 0) {
+            await persistTokensOnce();
             return {
                 contents: keepFullTrace ? fullTraceContents : modelContents,
                 modelContents,
@@ -484,6 +523,7 @@ export async function runToolLoop(options) {
                 toolResultRaw,
             });
             if (terminalToolNames.includes(name)) {
+                await persistTokensOnce();
                 return {
                     contents: keepFullTrace ? fullTraceContents : modelContents,
                     modelContents,
@@ -494,6 +534,7 @@ export async function runToolLoop(options) {
             }
         }
     }
+    await persistTokensOnce();
     return {
         contents: keepFullTrace ? fullTraceContents : modelContents,
         modelContents,

package/dist/ai/toolLoop/toolLoopRunner.js.map

@@ -1 +1 @@
-{"version":3,"file":"toolLoopRunner.js","sourceRoot":"","sources":["../../../src/ai/toolLoop/toolLoopRunner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAQ,MAAM,eAAe,CAAC;AAChE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,eAAe,EAAE,MAAM,2CAA2C,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAa,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AAEjF,OAAO,EAAE,wBAAwB,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,4CAA4C,CAAC;AAChF,OAAO,EACL,eAAe,EACf,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,GAGvB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,eAAe,EACf,sBAAsB,EACtB,eAAe,EACf,cAAc,GACf,MAAM,0BAA0B,CAAC;AAgDlC,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,OAA2B;IAE3B,MAAM,aAAa,GAAG,CAAC,KAAc,EAAoC,EAAE,CACzE,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAEvE,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAC9B,gBAAuC,CACxC,CAAC;IAEF,MAAM,EACJ,eAAe,EACf,KAAK,EACL,aAAa,EACb,QAAQ,GAAG,EAAE,EACb,eAAe,GAAG,yBAAyB,CAAC,GAAG,EAC/C,iBAAiB,GAAG,EAAE,EACtB,aAAa,GAAG,IAAI,EACpB,aAAa,EACb,MAAM,EACN,MAAM,EACN,sBAAsB,GAAG,CAAC,EAC1B,kBAAkB,GAAG,CAAC,EAAE,kFAAkF;IAC1G,qBAAqB,GAAG,GAAG,EAC3B,oBAAoB,GAAG,KAAM,EAC7B,eAAe,GAChB,GAAG,OAAO,CAAC;IAEZ,MAAM,MAAM,GAAW;QACrB,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC;QACpE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,EAAE,CACzC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,OAAO,IAAI,EAAE,EAAE,OAAO,CAAC;QACpD,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE;YAC5B,MAAM,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACpE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC;QACnD,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,CACjC,EAAE,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;KACnD,CAAC;IAEF,MAAM,KAAK,GAAG,wBAAwB,CAAC;QACrC,aAAa;QACb,EAAE,EAAE,MAAM;KACX,CAAC,CAAC;IAEH,MAAM,YAAY,GAAgD;QAChE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACxB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;YACrC,MAAM,SAAS,GACb,IAAI,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACtE,MAAM,OAAO,GACX,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClE,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YACnE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QAC3B,CAAC;QACD,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE,CACnB,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAC1E,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,WAAW,CACrC,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,EACvB,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,CACxB,CAAC;YACF,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACrB,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,CAAC;YACxE,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QACD,WAAW,EAAE,CAAC,IAAI,EAAE,EAAE,CACpB,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;QACvD,oBAAoB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACnC,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC;YACxD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,gBAAgB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC/B,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YACtE,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAChE,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC7B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,SAAS,GAAG,MAAM,kBAAkB,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC1E,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,0BAA0B,MAAM,CAAC,KAAK,2BAA2B,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,qFAAqF;oBACtL,gBAAgB,EAAE,SAAS;iBAC5B,CAAC;YACJ,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,KAAK,GAAQ,IAAI,CAAC,KAAK,CAAC;YAC9B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC;gBACzC,KAAK;gBACL,UAAU;gBACV,GAAG,KAAK;aACT,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,gBAAgB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAC5C,KAAK,EACL,UAAU,EACV,UAAU,CACX,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,oBAAoB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACnC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YACvE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACnC,CAAC;QACD,mBAAmB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;SAC3C,CAAC;QACF,oBAAoB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACnC,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC3D,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;QAChD,CAAC;KACF,CAAC;IAEF,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,MAAM,GAAoC;QAC9C,GAAG,sBAAsB;QACzB,GAAG,CAAC,aAAa,IAAI,EAAE,CAAC;KACzB,CAAC;IAEF,MAAM,UAAU,GAAgB,EAAE,CAAC;IACnC,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAE7B,MAAM,sBAAsB,GAAG,8BAA8B,CAAC;IAC9D,MAAM,yBAAyB,GAAG;QAChC,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE;YACL;gBACE,IAAI,EACF,GAAG,sBAAsB,IAAI;oBAC7B,4BAA4B,QAAQ,wCAAwC;oBAC5E,uDAAuD;oBACvD,sGAAsG;aACzG;SACF;KACF,CAAC;IAEF,MAAM,iBAAiB,GAAU,aAAa;QAC5C,CAAC,CAAC,CAAC,GAAG,eAAe,EAAE,yBAAyB,CAAC;QACjD,CAAC,CAAC,EAAE,CAAC;IACP,IAAI,aAAa,GAAU,CAAC,GAAG,eAAe,EAAE,yBAAyB,CAAC,CAAC;IAC3E,MAAM,kBAAkB,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,CAAC,QAAa,EAAE,SAAc,EAAE,EAAE;QACjD,IAAI,aAAa;YAAE,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpD,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC,CAAC;IACF,MAAM,aAAa,GAAG,CAAC,SAAc,EAAE,EAAE;QACvC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC,CAAC;IAEF,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC;QAC3C,aAAa,GAAG,eAAe,CAAC;YAC9B,YAAY,EAAE,kBAAkB;YAChC,aAAa;YACb,UAAU;YACV,MAAM;SACP,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;YACzD,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE;gBAC3C,WAAW;gBACX,IAAI,EAAE,IAAI,GAAG,CAAC;aACf,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAuC,CAAC;QAC5C,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,eAAe,CAAC;gBAC/B,MAAM;gBACN,OAAO,EAAE,aAAa;gBACtB,OAAO,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE;gBACnC,QAAQ,EAAE,kBAAkB;gBAC5B,WAAW,EAAE,qBAAqB;gBAClC,UAAU,EAAE,oBAAoB;gBAChC,IAAI,EAAE,IAAI,GAAG,CAAC;gBACd,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CACJ,iEAAiE,EACjE,WAAW,CAAC,UAAU,CACvB,CAAC;YACF,OAAO,CAAC,KAAK,CAAC,iDAAiD,EAAE,GAAG,EAAE;gBACpE,IAAI,EAAE,IAAI,GAAG,CAAC;gBACd,KAAK,EAAE,cAAc,CAAC,GAAG,CAAC;aAC3B,CAAC,CAAC;YAEH,MAAM,OAAO,GACX,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;YACnE,MAAM,wBAAwB,GAAG;gBAC/B,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL;wBACE,IAAI,EACF,oHAAoH;4BACpH,UAAU,OAAO,IAAI;4BACrB,iHAAiH;qBACpH;iBACF;aACF,CAAC;YACF,IAAI,aAAa;gBAAE,iBAAiB,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACpE,SAAS;QACX,CAAC;QAED,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,eAAe,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,EAAE;oBAC1D,IAAI,EAAE,IAAI,GAAG,CAAC;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,CAAC,aAAa,IAAI,EAAE,CAAC;QACnD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa;gBAC3D,aAAa;gBACb,SAAS,EAAE,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;gBACvC,KAAK,EAAE,IAAI,GAAG,CAAC;aAChB,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE;YAC1B,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAE,QAAgB,EAAE,UAAU,CAAC;oBAC7D,CAAC,CAAG,QAAgB,CAAC,UAAoB;oBACzC,CAAC,CAAC,EAAE,CAAC;gBACP,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC;gBAC9C,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAE,KAAe,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;gBACtC,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;oBACpB,MAAM,EAAE,GAAG,CAAC,EAAE,YAAY,CAAC;oBAC3B,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC;oBAClB,MAAM,GAAG,GAAG,CAAC,EAAE,gBAAgB,IAAI,CAAC,EAAE,iBAAiB,CAAC;oBACxD,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,EAAE,CAAC;wBAC7D,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;oBACnB,CAAC;gBACH,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,IAAI,GAAG,EAAkB,CAAC;YACnC,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YACtE,MAAM,IAAI,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YACzC,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAA4B,CAAC;YAC1D,MAAM,gBAAgB,GAAuB,CAAC,GAAG,EAAE;gBACjD,MAAM,MAAM,GACT,IAAY,EAAE,iBAAiB,IAAK,IAAY,EAAE,gBAAgB,CAAC;gBACtE,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM;oBAAE,OAAO,MAAM,CAAC;gBACxD,MAAM,EAAE,GAAI,IAAY,EAAE,EAAE,CAAC;gBAC7B,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACpD,OAAO,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/B,CAAC;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC,EAAE,CAAC;YAEL,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,CACJ,kFAAkF,EAClF,WAAW,CAAC,UAAU,CACvB,CAAC;gBACF,MAAM,oBAAoB,GAAG;oBAC3B,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE;wBACL;4BACE,IAAI,EACF,0FAA0F;gCAC1F,wEAAwE;gCACxE,aAAa,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE;yBAC7D;qBACF;iBACF,CAAC;gBACF,IAAI,aAAa;oBAAE,iBAAiB,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;gBAChE,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;gBACzC,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,oBAAoB,GAAG,CAAC,OAAO;gBACnC,CAAC,CAAC;oBACE,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,8BAA8B,IAAI,IAAI;oBAC7C,YAAY,EAAE;wBACZ,IAAI,EAAE,yBAAyB;wBAC/B,OAAO,EAAE,8BAA8B,IAAI,IAAI;qBAChD;iBACF;gBACH,CAAC,CAAC,IAAI,CAAC;YAET,IAAI,aAAa,GAA4B,IAAI,CAAC;YAClD,IAAI,YAAY,GAIL,IAAI,CAAC;YAChB,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;gBACzB,MAAM,UAAU,GAAG,qBAAqB,CACtC,aAAa,EACb,MAAM,CAAC,uBAAuB,CAC/B,CAAC;gBACF,aAAa,GAAG,UAAU,CAAC,aAAa,CAAC;gBACzC,YAAY,GAAG;oBACb,KAAK,EAAE,UAAU,CAAC,KAAK;oBACvB,GAAG,EAAE,UAAU,CAAC,GAAG;oBACnB,SAAS,EAAE,UAAU,CAAC,SAAS;iBAChC,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,KAAK,sBAAsB,EAAE,CAAC;gBACpC,8EAA8E;gBAC9E,gFAAgF;gBAChF,MAAM,WAAW,GAAI,aAAqB,EAAE,MAAM,CAAC;gBACnD,MAAM,UAAU,GAA4B,EAAE,CAAC;gBAE/C,IAAI,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC/B,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;wBACjD,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;4BAAE,SAAS;wBACvC,IAAI,OAAO,CAAC,KAAK,QAAQ;4BAAE,SAAS;wBACpC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBACpB,CAAC;gBACH,CAAC;gBAED,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,CAAC;oBACzD,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;wBAAE,SAAS;oBACvC,IAAI,OAAO,CAAC,KAAK,QAAQ;wBAAE,SAAS;oBACpC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACpB,CAAC;gBAED,aAAa,GAAG,UAAU,CAAC;YAC7B,CAAC;YAED,MAAM,CACJ,sBAAsB,CAAC,IAAI,EAAE,aAAa,EAAE,YAAY,CAAC,EACzD,WAAW,CAAC,YAAY,CACzB,CAAC;YAEF,MAAM,SAAS,GAAG,sBAAsB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAE9D,MAAM,gBAAgB,GAAG;gBACvB,YAAY,EAAE;oBACZ,IAAI;oBACJ,IAAI,EAAE,aAAa;iBACpB;gBACD,GAAG,CAAC,gBAAgB;oBAClB,CAAC,CAAC;wBACE,gBAAgB,EAAE,gBAAgB;wBAClC,iBAAiB,EAAE,gBAAgB;qBACpC;oBACH,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;YAEF,MAAM,qBAAqB,GAAG;gBAC5B,YAAY,EAAE;oBACZ,IAAI;oBACJ,IAAI,EAAE,SAAS;iBAChB;gBACD,GAAG,CAAC,gBAAgB;oBAClB,CAAC,CAAC;wBACE,gBAAgB,EAAE,gBAAgB;wBAClC,iBAAiB,EAAE,gBAAgB;qBACpC;oBACH,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,GAAG,gBAAgB;qBACpB;iBACF;aACF,CAAC;YAEF,MAAM,cAAc,GAAG;gBACrB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,GAAG,qBAAqB;qBACzB;iBACF;aACF,CAAC;YAEF,IAAI,aAAa,EAAE,CAAC;gBAClB,QAAQ,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,cAAc,CAAC,CAAC;YAChC,CAAC;YAED,IAAI,aAAsB,CAAC;YAC3B,IAAI,oBAAoB,EAAE,CAAC;gBACzB,aAAa,GAAG,oBAAoB,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC;oBACH,IAAI,IAAI,KAAK,sBAAsB,EAAE,CAAC;wBACpC,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAC7D,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CACxB,CAAC;wBACF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;4BAC1B,aAAa,GAAG;gCACd,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,2CAA2C;gCAClD,YAAY,EAAE;oCACZ,IAAI,EAAE,2BAA2B;oCACjC,OAAO,EACL,0FAA0F;iCAC7F;gCACD,IAAI,EAAE,wFAAwF;6BAC/F,CAAC;wBACJ,CAAC;6BAAM,CAAC;4BACN,aAAa,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,CAAC;wBAC/C,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,aAAa,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,YAAY,IAAI,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;oBAC1D,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,GAAG,EAAE;wBAC7C,IAAI,EAAE,IAAI;wBACV,IAAI,EAAE,IAAI,GAAG,CAAC;qBACf,CAAC,CAAC;oBACH,aAAa,GAAG;wBACd,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;wBACvD,YAAY,EAAE,cAAc,CAAC,GAAG,CAAC;wBACjC,IAAI,EAAE,iGAAiG;qBACxG,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,IAAI,UAAU,GAAY,aAAa,CAAC;YAExC,IAAI,IAAI,KAAK,WAAW,IAAI,YAAY,EAAE,CAAC;gBACzC,MAAM,IAAI,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBAE9C,MAAM,WAAW,GACd,aAAqB,EAAE,IAAI,KAAK,MAAM;oBACrC,CAAC,CAAE,aAAqB,EAAE,IAAI;oBAC9B,CAAC,CAAC,SAAS,CAAC;gBAChB,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;oBAC9B,4EAA4E;oBAC5E,UAAU,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,MAAM,UAAU,GACd,OAAQ,aAAqB,EAAE,OAAO,KAAK,QAAQ;wBACjD,CAAC,CAAC,MAAM,CAAE,aAAqB,CAAC,OAAO,CAAC;wBACxC,CAAC,CAAC,OAAO,aAAa,KAAK,QAAQ;4BACjC,CAAC,CAAC,aAAa;4BACf,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,IAAI,IAAI,CAAC,CAAC;oBAE9C,UAAU,GAAG;wBACX,IAAI;wBACJ,UAAU,EAAE,YAAY,CAAC,KAAK;wBAC9B,QAAQ,EAAE,YAAY,CAAC,GAAG;wBAC1B,SAAS,EAAE,YAAY,CAAC,SAAS;wBACjC,OAAO,EAAE,UAAU;wBACnB,IAAI,EAAE,YAAY,CAAC,SAAS;4BAC1B,CAAC,CAAC,aAAa,MAAM,CAAC,uBAAuB,gDAAgD;4BAC7F,CAAC,CAAC,SAAS;qBACd,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,eAAe,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;YACrD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,GAAG,EAAE;oBAC3D,IAAI,EAAE,IAAI;oBACV,IAAI,EAAE,IAAI,GAAG,CAAC;iBACf,CAAC,CAAC;YACL,CAAC;YAED,MAAM,YAAY,GAAG;gBACnB,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL;wBACE,gBAAgB,EAAE;4BAChB,IAAI;4BACJ,QAAQ,EAAE,UAAU;yBACrB;qBACF;iBACF;aACF,CAAC;YAEF,IAAI,aAAa,EAAE,CAAC;gBAClB,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACvC,CAAC;YACD,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAEjC,IACE,IAAI,KAAK,aAAa;gBACrB,UAAkB,EAAE,OAAO,KAAK,KAAK;gBACtC,sBAAsB,GAAG,CAAC;gBAC1B,oBAAoB,GAAG,sBAAsB,EAC7C,CAAC;gBACD,oBAAoB,IAAI,CAAC,CAAC;gBAE1B,MAAM,KAAK,GAAG,MAAM,CAAE,UAAkB,EAAE,KAAK,IAAI,eAAe,CAAC,CAAC;gBACpE,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAE,UAAkB,EAAE,KAAK,EAAE,KAAK,CAAC;oBACjE,CAAC,CAAG,UAAkB,CAAC,KAAK,CAAC,KAGxB;oBACL,CAAC,CAAC,EAAE,CAAC;gBAEP,MAAM,SAAS,GACb,UAAU,CAAC,MAAM,GAAG,CAAC;oBACnB,CAAC,CAAC,qDAAqD,UAAU;yBAC5D,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;yBACX,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACJ,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,SAAS,MAAM,CACxC,CAAC,CAAC,IAAI,IAAI,EAAE,CACb,eAAe,CACnB;yBACA,IAAI,CAAC,MAAM,CAAC,EAAE;oBACnB,CAAC,CAAC,EAAE,CAAC;gBAET,MAAM,gBAAgB,GAAG;oBACvB,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE;wBACL;4BACE,IAAI,EACF,+BAA+B,oBAAoB,IAAI,sBAAsB,MAAM,KAAK,IAAI;gCAC5F,6DAA6D;gCAC7D,uFAAuF;gCACvF,SAAS;yBACZ;qBACF;iBACF,CAAC;gBAEF,IAAI,aAAa;oBAAE,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBAC5D,aAAa,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACvC,CAAC;YAED,eAAe,CAAC;gBACd,UAAU;gBACV,IAAI;gBACJ,aAAa;gBACb,SAAS;gBACT,YAAY;gBACZ,UAAU;gBACV,aAAa;aACd,CAAC,CAAC;YAEH,IAAI,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,OAAO;oBACL,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa;oBAC3D,aAAa;oBACb,SAAS,EAAE,EAAE;oBACb,KAAK,EAAE,IAAI,GAAG,CAAC;oBACf,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,aAAa,EAAE;iBACrE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa;QAC3D,aAAa;QACb,SAAS,EAAE,+BAA+B,QAAQ,IAAI;QACtD,KAAK,EAAE,QAAQ;KAChB,CAAC;AACJ,CAAC","sourcesContent":["import { FunctionCallingConfigMode, Tool } from \"@google/genai\";\nimport fs from \"node:fs/promises\";\nimport { persistToolCall } from \"../../services/toolcallPersist.service.js\";\nimport { EVENT_TYPES, EventType } from \"../../types/events.js\";\nimport { STYLE_TOKEN_KEYS } from \"../../types/styleConfig.js\";\nimport { createWorkspaceToolImpls } from \"../tools/implementations/factories.js\";\nimport { CoreFs } from \"../tools/implementations/workspaceDeps.js\";\nimport { parsePlannerTasksUnknown } from \"./plannerTaskParser.js\";\nimport { getAvailableRoutes } from \"../tools/helpers/pageConfigJson.helpers.js\";\nimport {\n  compactForModel,\n  DEFAULT_CONTEXT_POLICY,\n  normalizeReadFileArgs,\n  redactFunctionCallArgs,\n  ToolEvent,\n  ToolLoopContextPolicy,\n} from \"./toolLoopContext.js\";\nimport {\n  aiCallWithRetry,\n  buildToolStatusMessage,\n  recordToolEvent,\n  serializeError,\n} from \"./toolLoopRunnerUtils.js\";\n\nexport type ToolLoopResult = {\n  contents: any[];\n  modelContents: any[];\n  finalText: string;\n  steps: number;\n  terminalCall?: {\n    name: string;\n    args: Record<string, unknown>;\n    response: unknown;\n  };\n};\n\nexport type Logger = (message: string, eventType: EventType) => Promise<void>;\n\nexport type AiCallResponse = {\n  functionCalls?: any[];\n  text?: string;\n};\n\nexport type AiCallFn = (\n  request: unknown,\n  options: {\n    tools?: Tool[];\n    model?: string;\n    toolCallingMode?: FunctionCallingConfigMode;\n  },\n) => Promise<AiCallResponse>;\n\nexport type RunToolLoopOptions = {\n  initialContents: any[];\n  tools: Tool[];\n  workspaceRoot: string;\n  maxSteps?: number;\n  toolCallingMode?: FunctionCallingConfigMode;\n  terminalToolNames?: string[];\n  keepFullTrace?: boolean;\n  contextPolicy?: ToolLoopContextPolicy;\n  aiCall: AiCallFn;\n  logger: Logger;\n  applyPatchAutoRetryMax?: number;\n  aiCallAutoRetryMax?: number;\n  aiCallAutoRetryBaseMs?: number;\n  aiCallAutoRetryMaxMs?: number;\n  persistResponse?: (modelInput: any, modelOutput: any) => Promise<void>;\n};\n\nexport async function runToolLoop(\n  options: RunToolLoopOptions,\n): Promise<ToolLoopResult> {\n  const isPlainObject = (value: unknown): value is Record<string, unknown> =>\n    typeof value === \"object\" && value !== null && !Array.isArray(value);\n\n  const styleTokenKeySet = new Set<string>(\n    STYLE_TOKEN_KEYS as unknown as string[],\n  );\n\n  const {\n    initialContents,\n    tools,\n    workspaceRoot,\n    maxSteps = 30,\n    toolCallingMode = FunctionCallingConfigMode.ANY,\n    terminalToolNames = [],\n    keepFullTrace = true,\n    contextPolicy,\n    aiCall,\n    logger,\n    applyPatchAutoRetryMax = 2,\n    aiCallAutoRetryMax = 3, // must have it to try 3 times as gemini errors a lot due to high demand sometimes\n    aiCallAutoRetryBaseMs = 400,\n    aiCallAutoRetryMaxMs = 10_000,\n    persistResponse,\n  } = options;\n\n  const nodeFs: CoreFs = {\n    readFile: async (absolutePath) => fs.readFile(absolutePath, \"utf-8\"),\n    writeFile: async (absolutePath, content) =>\n      fs.writeFile(absolutePath, content ?? \"\", \"utf-8\"),\n    mkdirp: async (absoluteDir) => {\n      await fs.mkdir(absoluteDir, { recursive: true });\n    },\n    rmFile: async (absolutePath) => fs.rm(absolutePath, { force: true }),\n    stat: async (absolutePath) => fs.stat(absolutePath),\n    safeReadDir: async (absoluteDir) =>\n      fs.readdir(absoluteDir, { withFileTypes: true }),\n  };\n\n  const impls = createWorkspaceToolImpls({\n    workspaceRoot,\n    fs: nodeFs,\n  });\n\n  const toolHandlers: Record<string, (args: any) => Promise<any>> = {\n    read_file: async (args) => {\n      const path = String(args.path ?? \"\");\n      const startLine =\n        args.start_line === undefined ? undefined : Number(args.start_line);\n      const endLine =\n        args.end_line === undefined ? undefined : Number(args.end_line);\n      const content = await impls.readFileImpl(path, startLine, endLine);\n      return { path, content };\n    },\n    write_file: (args) =>\n      impls.writeFileImpl(String(args.path ?? \"\"), String(args.content ?? \"\")),\n    list_dir: async (args) => {\n      const content = await impls.listDirImpl(\n        String(args.path ?? \"\"),\n        Number(args.depth ?? 1),\n      );\n      return { content };\n    },\n    search: async (args) => {\n      const results = await impls.searchImpl(String(args.search_query ?? \"\"));\n      return { results };\n    },\n    apply_patch: (args) =>\n      impls.applyPatchImpl(String(args.patch_string ?? \"\")),\n    update_global_styles: async (args) => {\n      const result = await impls.updateGlobalStylesImpl(args);\n      return result;\n    },\n    create_new_route: async (args) => {\n      const parentRoute = String(args.parent_route ?? \"\");\n      const routeName = String(args.route_name ?? \"\");\n      const result = await impls.createNewRouteImpl(parentRoute, routeName);\n      return result;\n    },\n    delete_element: async (args) => {\n      const route = String(args.route ?? \"\");\n      const element_id = String(args.element_id ?? \"\");\n      const result = await impls.deleteElementImpl(route, element_id);\n      return result;\n    },\n    insert_element: async (args) => {\n      const result = await impls.insertElementImpl(args);\n      if (!result.success) {\n        const available = await getAvailableRoutes({ workspaceRoot, fs: nodeFs });\n        return {\n          success: false,\n          error: `insert_element failed: ${result.error}. Available routes are: ${JSON.stringify(available)}. If you intend to create a new route, create it using the 'create_new_route' tool.`,\n          available_routes: available,\n        };\n      }\n      return result;\n    },\n    update_props: async (args) => {\n      const route = String(args.route ?? \"\");\n      const element_id = String(args.element_id ?? \"\");\n      const props: any = args.props;\n      const result = await impls.updatePropsImpl({\n        route,\n        element_id,\n        ...props,\n      });\n      return result;\n    },\n    update_classname: async (args) => {\n      const route = String(args.route ?? \"\");\n      const element_id = String(args.element_id ?? \"\");\n      const class_name = String(args.class_name ?? \"\");\n      const result = await impls.updateClassNameImpl(\n        route,\n        element_id,\n        class_name,\n      );\n      return result;\n    },\n    get_available_routes: async (args) => {\n      const routes = await getAvailableRoutes({ workspaceRoot, fs: nodeFs });\n      return { success: true, routes };\n    },\n    submit_codegen_done: async (args) => ({\n      success: true,\n      summary: String(args.summary ?? \"\").trim(),\n    }),\n    submit_planner_tasks: async (args) => {\n      const tasks = parsePlannerTasksUnknown(args.planner_tasks);\n      return { success: true, count: tasks.length };\n    },\n  };\n\n  if (typeof aiCall !== \"function\") {\n    throw new Error(\"Tool loop: aiCall is required.\");\n  }\n\n  const policy: Required<ToolLoopContextPolicy> = {\n    ...DEFAULT_CONTEXT_POLICY,\n    ...(contextPolicy ?? {}),\n  };\n\n  const toolEvents: ToolEvent[] = [];\n  let applyPatchRetryCount = 0;\n\n  const EXECUTION_GUIDE_MARKER = \"TOOL_LOOP_EXECUTION_GUIDE_V1\";\n  const executionGuideInstruction = {\n    role: \"user\",\n    parts: [\n      {\n        text:\n          `${EXECUTION_GUIDE_MARKER}\\n` +\n          `Execution limit: At most ${maxSteps} assistant turn(s) in this tool loop. ` +\n          `One turn = one assistant response in the tool loop.\\n` +\n          `Complete the task in as few turns as possible and avoid unnecessary actions. Prioritize correctness.`,\n      },\n    ],\n  };\n\n  const fullTraceContents: any[] = keepFullTrace\n    ? [...initialContents, executionGuideInstruction]\n    : [];\n  let modelContents: any[] = [...initialContents, executionGuideInstruction];\n  const pinnedInitialCount = initialContents.length + 1;\n  const pushBoth = (fullItem: any, modelItem: any) => {\n    if (keepFullTrace) fullTraceContents.push(fullItem);\n    modelContents.push(modelItem);\n  };\n  const pushModelOnly = (modelItem: any) => {\n    modelContents.push(modelItem);\n  };\n\n  for (let step = 0; step < maxSteps; step++) {\n    modelContents = compactForModel({\n      initialCount: pinnedInitialCount,\n      modelContents,\n      toolEvents,\n      policy,\n    });\n\n    if (policy.logApproxModelChars) {\n      const approxChars = JSON.stringify(modelContents).length;\n      console.log(\"Tool loop: approx model chars\", {\n        approxChars,\n        step: step + 1,\n      });\n    }\n\n    let response: Awaited<ReturnType<AiCallFn>>;\n    try {\n      response = await aiCallWithRetry({\n        aiCall,\n        request: modelContents,\n        options: { tools, toolCallingMode },\n        retryMax: aiCallAutoRetryMax,\n        retryBaseMs: aiCallAutoRetryBaseMs,\n        retryMaxMs: aiCallAutoRetryMaxMs,\n        step: step + 1,\n        logger,\n      });\n    } catch (err) {\n      logger(\n        \"Tool loop: AI provider error; preserving context and continuing\",\n        EVENT_TYPES.STEP_ERROR,\n      );\n      console.error(\"Tool loop: aiCall failed (provider/server side)\", err, {\n        step: step + 1,\n        error: serializeError(err),\n      });\n\n      const message =\n        err instanceof Error ? err.message : JSON.stringify(err ?? null);\n      const providerErrorInstruction = {\n        role: \"user\",\n        parts: [\n          {\n            text:\n              `AI provider error (server-side). Do NOT clear or restart context; continue from the existing conversation state.\\n` +\n              `Error: ${message}\\n` +\n              `Next: retry the last request using the same context. If you were about to call tools, resend a valid tool call.`,\n          },\n        ],\n      };\n      if (keepFullTrace) fullTraceContents.push(providerErrorInstruction);\n      continue;\n    }\n\n    if (persistResponse) {\n      try {\n        await persistResponse(modelContents, response);\n      } catch (err) {\n        console.error(\"Tool loop: failed to persist response\", err, {\n          step: step + 1,\n        });\n      }\n    }\n\n    const functionCalls = response.functionCalls ?? [];\n    if (functionCalls.length === 0) {\n      return {\n        contents: keepFullTrace ? fullTraceContents : modelContents,\n        modelContents,\n        finalText: (response.text ?? \"\").trim(),\n        steps: step + 1,\n      };\n    }\n\n    const signatureById = (() => {\n      try {\n        const candidates = Array.isArray((response as any)?.candidates)\n          ? ((response as any).candidates as any[])\n          : [];\n        const parts = candidates?.[0]?.content?.parts;\n        const arr = Array.isArray(parts) ? (parts as any[]) : [];\n        const map = new Map<string, string>();\n        for (const p of arr) {\n          const fc = p?.functionCall;\n          const id = fc?.id;\n          const sig = p?.thoughtSignature ?? p?.thought_signature;\n          if (typeof id === \"string\" && typeof sig === \"string\" && sig) {\n            map.set(id, sig);\n          }\n        }\n        return map;\n      } catch {\n        return new Map<string, string>();\n      }\n    })();\n\n    for (let callIndex = 0; callIndex < functionCalls.length; callIndex++) {\n      const call = functionCalls[callIndex];\n      const name = call.name?.toString() ?? \"\";\n      const args = (call.args ?? {}) as Record<string, unknown>;\n      const thoughtSignature: string | undefined = (() => {\n        const direct =\n          (call as any)?.thought_signature ?? (call as any)?.thoughtSignature;\n        if (typeof direct === \"string\" && direct) return direct;\n        const id = (call as any)?.id;\n        if (typeof id === \"string\" && signatureById.has(id)) {\n          return signatureById.get(id);\n        }\n        return undefined;\n      })();\n\n      if (!name) {\n        logger(\n          \"Tool loop: malformed function call from model; preserving context and continuing\",\n          EVENT_TYPES.STEP_ERROR,\n        );\n        const malformedInstruction = {\n          role: \"user\",\n          parts: [\n            {\n              text:\n                `Malformed function call received (missing tool name). Do NOT clear or restart context.\\n` +\n                `Resend a single valid tool call with a non-empty name and JSON args.\\n` +\n                `Bad call: ${JSON.stringify(call ?? null).slice(0, 1500)}`,\n            },\n          ],\n        };\n        if (keepFullTrace) fullTraceContents.push(malformedInstruction);\n        modelContents.push(malformedInstruction);\n        continue;\n      }\n\n      const handler = toolHandlers[name];\n      const handlerMissingResult = !handler\n        ? {\n            success: false,\n            error: `No handler registered for \"${name}\".`,\n            error_detail: {\n              name: \"MissingToolHandlerError\",\n              message: `No handler registered for \"${name}\".`,\n            },\n          }\n        : null;\n\n      let effectiveArgs: Record<string, unknown> = args;\n      let readFileMeta: {\n        start: number;\n        end: number;\n        wasCapped: boolean;\n      } | null = null;\n      if (name === \"read_file\") {\n        const normalized = normalizeReadFileArgs(\n          effectiveArgs,\n          policy.readFileDefaultMaxLines,\n        );\n        effectiveArgs = normalized.effectiveArgs;\n        readFileMeta = {\n          start: normalized.start,\n          end: normalized.end,\n          wasCapped: normalized.wasCapped,\n        };\n      }\n\n      if (name === \"update_global_styles\") {\n        // Be forgiving: models sometimes include a legacy \"tokens\" key or other junk.\n        // We accept the call as long as at least one valid token key/value is provided.\n        const tokensMaybe = (effectiveArgs as any)?.tokens;\n        const normalized: Record<string, unknown> = {};\n\n        if (isPlainObject(tokensMaybe)) {\n          for (const [k, v] of Object.entries(tokensMaybe)) {\n            if (!styleTokenKeySet.has(k)) continue;\n            if (typeof v !== \"string\") continue;\n            normalized[k] = v;\n          }\n        }\n\n        for (const [k, v] of Object.entries(effectiveArgs ?? {})) {\n          if (!styleTokenKeySet.has(k)) continue;\n          if (typeof v !== \"string\") continue;\n          normalized[k] = v;\n        }\n\n        effectiveArgs = normalized;\n      }\n\n      logger(\n        buildToolStatusMessage(name, effectiveArgs, readFileMeta),\n        EVENT_TYPES.STEP_STARTED,\n      );\n\n      const modelArgs = redactFunctionCallArgs(name, effectiveArgs);\n\n      const functionCallPart = {\n        functionCall: {\n          name,\n          args: effectiveArgs,\n        },\n        ...(thoughtSignature\n          ? {\n              thoughtSignature: thoughtSignature,\n              thought_signature: thoughtSignature,\n            }\n          : {}),\n      };\n\n      const functionCallPartModel = {\n        functionCall: {\n          name,\n          args: modelArgs,\n        },\n        ...(thoughtSignature\n          ? {\n              thoughtSignature: thoughtSignature,\n              thought_signature: thoughtSignature,\n            }\n          : {}),\n      };\n\n      const assistantFull = {\n        role: \"model\",\n        parts: [\n          {\n            ...functionCallPart,\n          },\n        ],\n      };\n\n      const assistantModel = {\n        role: \"model\",\n        parts: [\n          {\n            ...functionCallPartModel,\n          },\n        ],\n      };\n\n      if (keepFullTrace) {\n        pushBoth(assistantFull, assistantModel);\n      } else {\n        pushModelOnly(assistantModel);\n      }\n\n      let toolResultRaw: unknown;\n      if (handlerMissingResult) {\n        toolResultRaw = handlerMissingResult;\n      } else {\n        try {\n          if (name === \"update_global_styles\") {\n            const flatKeys = Object.keys(effectiveArgs ?? {}).filter((k) =>\n              styleTokenKeySet.has(k),\n            );\n            if (flatKeys.length === 0) {\n              toolResultRaw = {\n                success: false,\n                error: \"must include at least one token key/value\",\n                error_detail: {\n                  name: \"InvalidToolArgumentsError\",\n                  message:\n                    'update_global_styles requires at least one token key/value (e.g. { radius: \"0.75rem\" }).',\n                },\n                note: \"Resend update_global_styles with at least one token key/value, or skip this tool call.\",\n              };\n            } else {\n              toolResultRaw = await handler(effectiveArgs);\n            }\n          } else {\n            toolResultRaw = await handler(effectiveArgs);\n          }\n        } catch (err) {\n          logger(`AI tool: ${name} failed`, EVENT_TYPES.STEP_ERROR);\n          console.error(\"Tool loop: handler threw\", err, {\n            tool: name,\n            step: step + 1,\n          });\n          toolResultRaw = {\n            success: false,\n            error: err instanceof Error ? err.message : String(err),\n            error_detail: serializeError(err),\n            note: \"Tool handler threw. Inspect error_detail and retry with corrected args or a different approach.\",\n          };\n        }\n      }\n      let toolResult: unknown = toolResultRaw;\n\n      if (name === \"read_file\" && readFileMeta) {\n        const path = String(effectiveArgs.path ?? \"\");\n\n        const jsonPayload =\n          (toolResultRaw as any)?.kind === \"json\"\n            ? (toolResultRaw as any)?.json\n            : undefined;\n        if (jsonPayload !== undefined) {\n          // Token-efficient: return JSON as structured data (no double-stringifying).\n          toolResult = { path, json: jsonPayload };\n        } else {\n          const rawContent =\n            typeof (toolResultRaw as any)?.content === \"string\"\n              ? String((toolResultRaw as any).content)\n              : typeof toolResultRaw === \"string\"\n                ? toolResultRaw\n                : JSON.stringify(toolResultRaw ?? null);\n\n          toolResult = {\n            path,\n            start_line: readFileMeta.start,\n            end_line: readFileMeta.end,\n            truncated: readFileMeta.wasCapped,\n            content: rawContent,\n            note: readFileMeta.wasCapped\n              ? `Capped to ${policy.readFileDefaultMaxLines} lines. Request more with start_line/end_line.`\n              : undefined,\n          };\n        }\n      }\n\n      try {\n        await persistToolCall(name, modelArgs, toolResult);\n      } catch (err) {\n        console.error(\"Tool loop: failed to persist tool call\", err, {\n          tool: name,\n          step: step + 1,\n        });\n      }\n\n      const responseFull = {\n        role: \"user\",\n        parts: [\n          {\n            functionResponse: {\n              name,\n              response: toolResult,\n            },\n          },\n        ],\n      };\n\n      if (keepFullTrace) {\n        fullTraceContents.push(responseFull);\n      }\n      modelContents.push(responseFull);\n\n      if (\n        name === \"apply_patch\" &&\n        (toolResult as any)?.success === false &&\n        applyPatchAutoRetryMax > 0 &&\n        applyPatchRetryCount < applyPatchAutoRetryMax\n      ) {\n        applyPatchRetryCount += 1;\n\n        const error = String((toolResult as any)?.error ?? \"unknown error\");\n        const debugFiles = Array.isArray((toolResult as any)?.debug?.files)\n          ? ((toolResult as any).debug.files as Array<{\n              path?: string;\n              head?: string;\n            }>)\n          : [];\n\n        const debugText =\n          debugFiles.length > 0\n            ? `\\n\\nFILE SNAPSHOTS (for regenerating the patch):\\n${debugFiles\n                .slice(0, 3)\n                .map(\n                  (f) =>\n                    `--- ${String(f.path ?? \"\")} ---\\n${String(\n                      f.head ?? \"\",\n                    )}\\n--- end ---`,\n                )\n                .join(\"\\n\\n\")}`\n            : \"\";\n\n        const retryInstruction = {\n          role: \"user\",\n          parts: [\n            {\n              text:\n                `apply_patch failed (attempt ${applyPatchRetryCount}/${applyPatchAutoRetryMax}): ${error}\\n` +\n                `Regenerate a patch that matches the current file contents. ` +\n                `For large rewrites, prefer write_file(path, content) or Delete+Add instead of Update.` +\n                debugText,\n            },\n          ],\n        };\n\n        if (keepFullTrace) fullTraceContents.push(retryInstruction);\n        modelContents.push(retryInstruction);\n      }\n\n      recordToolEvent({\n        toolEvents,\n        name,\n        effectiveArgs,\n        modelArgs,\n        readFileMeta,\n        toolResult,\n        toolResultRaw,\n      });\n\n      if (terminalToolNames.includes(name)) {\n        return {\n          contents: keepFullTrace ? fullTraceContents : modelContents,\n          modelContents,\n          finalText: \"\",\n          steps: step + 1,\n          terminalCall: { name, args: effectiveArgs, response: toolResultRaw },\n        };\n      }\n    }\n  }\n\n  return {\n    contents: keepFullTrace ? fullTraceContents : modelContents,\n    modelContents,\n    finalText: `Stopped: max steps reached (${maxSteps}).`,\n    steps: maxSteps,\n  };\n}\n"]}
+{"version":3,"file":"toolLoopRunner.js","sourceRoot":"","sources":["../../../src/ai/toolLoop/toolLoopRunner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAQ,MAAM,eAAe,CAAC;AAChE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAElC,OAAO,EAAE,eAAe,EAAE,MAAM,2CAA2C,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAa,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,4CAA4C,CAAC;AAChF,OAAO,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AAEjF,OAAO,EAAE,wBAAwB,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EACL,eAAe,EACf,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,GAGvB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACL,eAAe,EACf,sBAAsB,EACtB,eAAe,EACf,cAAc,GACf,MAAM,0BAA0B,CAAC;AAqDlC,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,OAA2B;IAE3B,MAAM,aAAa,GAAG,CAAC,KAAc,EAAoC,EAAE,CACzE,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAEvE,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAC9B,gBAAuC,CACxC,CAAC;IAEF,MAAM,EACJ,eAAe,EACf,KAAK,EACL,aAAa,EACb,QAAQ,GAAG,EAAE,EACb,eAAe,GAAG,yBAAyB,CAAC,GAAG,EAC/C,iBAAiB,GAAG,EAAE,EACtB,aAAa,GAAG,IAAI,EACpB,aAAa,EACb,MAAM,EACN,MAAM,EACN,sBAAsB,GAAG,CAAC,EAC1B,kBAAkB,GAAG,CAAC,EAAE,kFAAkF;IAC1G,qBAAqB,GAAG,GAAG,EAC3B,oBAAoB,GAAG,KAAM,EAC7B,eAAe,EACf,gBAAgB,GACjB,GAAG,OAAO,CAAC;IAEZ,MAAM,uBAAuB,GAAG,CAAC,WAAoB,EAAE,EAAE;QACvD,MAAM,KAAK,GAAI,WAAmB,EAAE,aAAa,CAAC;QAClD,MAAM,gBAAgB,GAAG,KAAK,EAAE,gBAAgB,CAAC;QACjD,MAAM,oBAAoB,GAAG,KAAK,EAAE,oBAAoB,CAAC;QAEzD,MAAM,WAAW,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAElD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC;QAClE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC;QAEpE,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;IACvC,CAAC,CAAC;IAEF,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,IAAI,gBAAgB,GAAG,KAAK,CAAC;IAE7B,MAAM,iBAAiB,GAAG,KAAK,IAAI,EAAE;QACnC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAC9B,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAE9B,IAAI,CAAC;YACH,MAAM,gBAAgB,CAAC,UAAU,CAAC,gBAAgB,CAChD,gBAAgB,CAAC,SAAS,EAC1B,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,CAAC,KAAK,CACvB,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,yCAAyC,EAAE,GAAG,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,MAAM,GAAW;QACrB,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC;QACpE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,EAAE,CACzC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,OAAO,IAAI,EAAE,EAAE,OAAO,CAAC;QACpD,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE;YAC5B,MAAM,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACpE,IAAI,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC;QACnD,WAAW,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,CACjC,EAAE,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;KACnD,CAAC;IAEF,MAAM,KAAK,GAAG,wBAAwB,CAAC;QACrC,aAAa;QACb,EAAE,EAAE,MAAM;KACX,CAAC,CAAC;IAEH,MAAM,YAAY,GAAgD;QAChE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACxB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;YACrC,MAAM,SAAS,GACb,IAAI,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACtE,MAAM,OAAO,GACX,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClE,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YACnE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QAC3B,CAAC;QACD,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE,CACnB,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAC1E,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,WAAW,CACrC,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,EACvB,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,CACxB,CAAC;YACF,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACrB,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,CAAC;YACxE,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QACD,WAAW,EAAE,CAAC,IAAI,EAAE,EAAE,CACpB,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;QACvD,oBAAoB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACnC,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC;YACxD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,gBAAgB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC/B,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;YACpD,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YACtE,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAChE,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC7B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,SAAS,GAAG,MAAM,kBAAkB,CAAC;oBACzC,aAAa;oBACb,EAAE,EAAE,MAAM;iBACX,CAAC,CAAC;gBACH,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,0BAA0B,MAAM,CAAC,KAAK,2BAA2B,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,qFAAqF;oBACtL,gBAAgB,EAAE,SAAS;iBAC5B,CAAC;YACJ,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,KAAK,GAAQ,IAAI,CAAC,KAAK,CAAC;YAC9B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC;gBACzC,KAAK;gBACL,UAAU;gBACV,GAAG,KAAK;aACT,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,gBAAgB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACvC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAC5C,KAAK,EACL,UAAU,EACV,UAAU,CACX,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,oBAAoB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACnC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YACvE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACnC,CAAC;QACD,mBAAmB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;SAC3C,CAAC;QACF,oBAAoB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACnC,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC3D,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;QAChD,CAAC;KACF,CAAC;IAEF,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,MAAM,GAAoC;QAC9C,GAAG,sBAAsB;QACzB,GAAG,CAAC,aAAa,IAAI,EAAE,CAAC;KACzB,CAAC;IAEF,MAAM,UAAU,GAAgB,EAAE,CAAC;IACnC,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAE7B,MAAM,sBAAsB,GAAG,8BAA8B,CAAC;IAC9D,MAAM,yBAAyB,GAAG;QAChC,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE;YACL;gBACE,IAAI,EACF,GAAG,sBAAsB,IAAI;oBAC7B,4BAA4B,QAAQ,wCAAwC;oBAC5E,uDAAuD;oBACvD,sGAAsG;aACzG;SACF;KACF,CAAC;IAEF,MAAM,iBAAiB,GAAU,aAAa;QAC5C,CAAC,CAAC,CAAC,GAAG,eAAe,EAAE,yBAAyB,CAAC;QACjD,CAAC,CAAC,EAAE,CAAC;IACP,IAAI,aAAa,GAAU,CAAC,GAAG,eAAe,EAAE,yBAAyB,CAAC,CAAC;IAC3E,MAAM,kBAAkB,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,CAAC,QAAa,EAAE,SAAc,EAAE,EAAE;QACjD,IAAI,aAAa;YAAE,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpD,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC,CAAC;IACF,MAAM,aAAa,GAAG,CAAC,SAAc,EAAE,EAAE;QACvC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC,CAAC;IAEF,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC;QAC3C,aAAa,GAAG,eAAe,CAAC;YAC9B,YAAY,EAAE,kBAAkB;YAChC,aAAa;YACb,UAAU;YACV,MAAM;SACP,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;YACzD,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE;gBAC3C,WAAW;gBACX,IAAI,EAAE,IAAI,GAAG,CAAC;aACf,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAuC,CAAC;QAC5C,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,eAAe,CAAC;gBAC/B,MAAM;gBACN,OAAO,EAAE,aAAa;gBACtB,OAAO,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE;gBACnC,QAAQ,EAAE,kBAAkB;gBAC5B,WAAW,EAAE,qBAAqB;gBAClC,UAAU,EAAE,oBAAoB;gBAChC,IAAI,EAAE,IAAI,GAAG,CAAC;gBACd,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CACJ,iEAAiE,EACjE,WAAW,CAAC,UAAU,CACvB,CAAC;YACF,OAAO,CAAC,KAAK,CAAC,iDAAiD,EAAE,GAAG,EAAE;gBACpE,IAAI,EAAE,IAAI,GAAG,CAAC;gBACd,KAAK,EAAE,cAAc,CAAC,GAAG,CAAC;aAC3B,CAAC,CAAC;YAEH,MAAM,OAAO,GACX,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;YACnE,MAAM,wBAAwB,GAAG;gBAC/B,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL;wBACE,IAAI,EACF,oHAAoH;4BACpH,UAAU,OAAO,IAAI;4BACrB,iHAAiH;qBACpH;iBACF;aACF,CAAC;YACF,IAAI,aAAa;gBAAE,iBAAiB,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACpE,SAAS;QACX,CAAC;QAED,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,eAAe,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,GAAG,EAAE;oBAC1D,IAAI,EAAE,IAAI,GAAG,CAAC;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,CAAC;YACC,MAAM,KAAK,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;YAChD,IAAI,KAAK,EAAE,CAAC;gBACV,gBAAgB,GAAG,IAAI,CAAC;gBACxB,gBAAgB,IAAI,KAAK,CAAC,WAAW,CAAC;gBACtC,iBAAiB,IAAI,KAAK,CAAC,YAAY,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,CAAC,aAAa,IAAI,EAAE,CAAC;QACnD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,MAAM,iBAAiB,EAAE,CAAC;YAC1B,OAAO;gBACL,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa;gBAC3D,aAAa;gBACb,SAAS,EAAE,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;gBACvC,KAAK,EAAE,IAAI,GAAG,CAAC;aAChB,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE;YAC1B,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAE,QAAgB,EAAE,UAAU,CAAC;oBAC7D,CAAC,CAAG,QAAgB,CAAC,UAAoB;oBACzC,CAAC,CAAC,EAAE,CAAC;gBACP,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC;gBAC9C,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAE,KAAe,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzD,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;gBACtC,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;oBACpB,MAAM,EAAE,GAAG,CAAC,EAAE,YAAY,CAAC;oBAC3B,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC;oBAClB,MAAM,GAAG,GAAG,CAAC,EAAE,gBAAgB,IAAI,CAAC,EAAE,iBAAiB,CAAC;oBACxD,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,EAAE,CAAC;wBAC7D,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;oBACnB,CAAC;gBACH,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,IAAI,GAAG,EAAkB,CAAC;YACnC,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YACtE,MAAM,IAAI,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YACzC,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAA4B,CAAC;YAC1D,MAAM,gBAAgB,GAAuB,CAAC,GAAG,EAAE;gBACjD,MAAM,MAAM,GACT,IAAY,EAAE,iBAAiB,IAAK,IAAY,EAAE,gBAAgB,CAAC;gBACtE,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM;oBAAE,OAAO,MAAM,CAAC;gBACxD,MAAM,EAAE,GAAI,IAAY,EAAE,EAAE,CAAC;gBAC7B,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACpD,OAAO,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/B,CAAC;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC,EAAE,CAAC;YAEL,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,CACJ,kFAAkF,EAClF,WAAW,CAAC,UAAU,CACvB,CAAC;gBACF,MAAM,oBAAoB,GAAG;oBAC3B,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE;wBACL;4BACE,IAAI,EACF,0FAA0F;gCAC1F,wEAAwE;gCACxE,aAAa,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE;yBAC7D;qBACF;iBACF,CAAC;gBACF,IAAI,aAAa;oBAAE,iBAAiB,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;gBAChE,aAAa,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;gBACzC,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,oBAAoB,GAAG,CAAC,OAAO;gBACnC,CAAC,CAAC;oBACE,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,8BAA8B,IAAI,IAAI;oBAC7C,YAAY,EAAE;wBACZ,IAAI,EAAE,yBAAyB;wBAC/B,OAAO,EAAE,8BAA8B,IAAI,IAAI;qBAChD;iBACF;gBACH,CAAC,CAAC,IAAI,CAAC;YAET,IAAI,aAAa,GAA4B,IAAI,CAAC;YAClD,IAAI,YAAY,GAIL,IAAI,CAAC;YAChB,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;gBACzB,MAAM,UAAU,GAAG,qBAAqB,CACtC,aAAa,EACb,MAAM,CAAC,uBAAuB,CAC/B,CAAC;gBACF,aAAa,GAAG,UAAU,CAAC,aAAa,CAAC;gBACzC,YAAY,GAAG;oBACb,KAAK,EAAE,UAAU,CAAC,KAAK;oBACvB,GAAG,EAAE,UAAU,CAAC,GAAG;oBACnB,SAAS,EAAE,UAAU,CAAC,SAAS;iBAChC,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,KAAK,sBAAsB,EAAE,CAAC;gBACpC,8EAA8E;gBAC9E,gFAAgF;gBAChF,MAAM,WAAW,GAAI,aAAqB,EAAE,MAAM,CAAC;gBACnD,MAAM,UAAU,GAA4B,EAAE,CAAC;gBAE/C,IAAI,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC/B,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;wBACjD,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;4BAAE,SAAS;wBACvC,IAAI,OAAO,CAAC,KAAK,QAAQ;4BAAE,SAAS;wBACpC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBACpB,CAAC;gBACH,CAAC;gBAED,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,CAAC;oBACzD,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;wBAAE,SAAS;oBACvC,IAAI,OAAO,CAAC,KAAK,QAAQ;wBAAE,SAAS;oBACpC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBACpB,CAAC;gBAED,aAAa,GAAG,UAAU,CAAC;YAC7B,CAAC;YAED,MAAM,CACJ,sBAAsB,CAAC,IAAI,EAAE,aAAa,EAAE,YAAY,CAAC,EACzD,WAAW,CAAC,YAAY,CACzB,CAAC;YAEF,MAAM,SAAS,GAAG,sBAAsB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAE9D,MAAM,gBAAgB,GAAG;gBACvB,YAAY,EAAE;oBACZ,IAAI;oBACJ,IAAI,EAAE,aAAa;iBACpB;gBACD,GAAG,CAAC,gBAAgB;oBAClB,CAAC,CAAC;wBACE,gBAAgB,EAAE,gBAAgB;wBAClC,iBAAiB,EAAE,gBAAgB;qBACpC;oBACH,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;YAEF,MAAM,qBAAqB,GAAG;gBAC5B,YAAY,EAAE;oBACZ,IAAI;oBACJ,IAAI,EAAE,SAAS;iBAChB;gBACD,GAAG,CAAC,gBAAgB;oBAClB,CAAC,CAAC;wBACE,gBAAgB,EAAE,gBAAgB;wBAClC,iBAAiB,EAAE,gBAAgB;qBACpC;oBACH,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,GAAG,gBAAgB;qBACpB;iBACF;aACF,CAAC;YAEF,MAAM,cAAc,GAAG;gBACrB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,GAAG,qBAAqB;qBACzB;iBACF;aACF,CAAC;YAEF,IAAI,aAAa,EAAE,CAAC;gBAClB,QAAQ,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,cAAc,CAAC,CAAC;YAChC,CAAC;YAED,IAAI,aAAsB,CAAC;YAC3B,IAAI,oBAAoB,EAAE,CAAC;gBACzB,aAAa,GAAG,oBAAoB,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC;oBACH,IAAI,IAAI,KAAK,sBAAsB,EAAE,CAAC;wBACpC,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAC7D,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CACxB,CAAC;wBACF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;4BAC1B,aAAa,GAAG;gCACd,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,2CAA2C;gCAClD,YAAY,EAAE;oCACZ,IAAI,EAAE,2BAA2B;oCACjC,OAAO,EACL,0FAA0F;iCAC7F;gCACD,IAAI,EAAE,wFAAwF;6BAC/F,CAAC;wBACJ,CAAC;6BAAM,CAAC;4BACN,aAAa,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,CAAC;wBAC/C,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,aAAa,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,YAAY,IAAI,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;oBAC1D,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,GAAG,EAAE;wBAC7C,IAAI,EAAE,IAAI;wBACV,IAAI,EAAE,IAAI,GAAG,CAAC;qBACf,CAAC,CAAC;oBACH,aAAa,GAAG;wBACd,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;wBACvD,YAAY,EAAE,cAAc,CAAC,GAAG,CAAC;wBACjC,IAAI,EAAE,iGAAiG;qBACxG,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,IAAI,UAAU,GAAY,aAAa,CAAC;YAExC,IAAI,IAAI,KAAK,WAAW,IAAI,YAAY,EAAE,CAAC;gBACzC,MAAM,IAAI,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBAE9C,MAAM,WAAW,GACd,aAAqB,EAAE,IAAI,KAAK,MAAM;oBACrC,CAAC,CAAE,aAAqB,EAAE,IAAI;oBAC9B,CAAC,CAAC,SAAS,CAAC;gBAChB,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;oBAC9B,4EAA4E;oBAC5E,UAAU,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,MAAM,UAAU,GACd,OAAQ,aAAqB,EAAE,OAAO,KAAK,QAAQ;wBACjD,CAAC,CAAC,MAAM,CAAE,aAAqB,CAAC,OAAO,CAAC;wBACxC,CAAC,CAAC,OAAO,aAAa,KAAK,QAAQ;4BACjC,CAAC,CAAC,aAAa;4BACf,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,IAAI,IAAI,CAAC,CAAC;oBAE9C,UAAU,GAAG;wBACX,IAAI;wBACJ,UAAU,EAAE,YAAY,CAAC,KAAK;wBAC9B,QAAQ,EAAE,YAAY,CAAC,GAAG;wBAC1B,SAAS,EAAE,YAAY,CAAC,SAAS;wBACjC,OAAO,EAAE,UAAU;wBACnB,IAAI,EAAE,YAAY,CAAC,SAAS;4BAC1B,CAAC,CAAC,aAAa,MAAM,CAAC,uBAAuB,gDAAgD;4BAC7F,CAAC,CAAC,SAAS;qBACd,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,eAAe,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;YACrD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,GAAG,EAAE;oBAC3D,IAAI,EAAE,IAAI;oBACV,IAAI,EAAE,IAAI,GAAG,CAAC;iBACf,CAAC,CAAC;YACL,CAAC;YAED,MAAM,YAAY,GAAG;gBACnB,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL;wBACE,gBAAgB,EAAE;4BAChB,IAAI;4BACJ,QAAQ,EAAE,UAAU;yBACrB;qBACF;iBACF;aACF,CAAC;YAEF,IAAI,aAAa,EAAE,CAAC;gBAClB,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACvC,CAAC;YACD,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAEjC,IACE,IAAI,KAAK,aAAa;gBACrB,UAAkB,EAAE,OAAO,KAAK,KAAK;gBACtC,sBAAsB,GAAG,CAAC;gBAC1B,oBAAoB,GAAG,sBAAsB,EAC7C,CAAC;gBACD,oBAAoB,IAAI,CAAC,CAAC;gBAE1B,MAAM,KAAK,GAAG,MAAM,CAAE,UAAkB,EAAE,KAAK,IAAI,eAAe,CAAC,CAAC;gBACpE,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAE,UAAkB,EAAE,KAAK,EAAE,KAAK,CAAC;oBACjE,CAAC,CAAG,UAAkB,CAAC,KAAK,CAAC,KAGxB;oBACL,CAAC,CAAC,EAAE,CAAC;gBAEP,MAAM,SAAS,GACb,UAAU,CAAC,MAAM,GAAG,CAAC;oBACnB,CAAC,CAAC,qDAAqD,UAAU;yBAC5D,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;yBACX,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACJ,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,SAAS,MAAM,CACxC,CAAC,CAAC,IAAI,IAAI,EAAE,CACb,eAAe,CACnB;yBACA,IAAI,CAAC,MAAM,CAAC,EAAE;oBACnB,CAAC,CAAC,EAAE,CAAC;gBAET,MAAM,gBAAgB,GAAG;oBACvB,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE;wBACL;4BACE,IAAI,EACF,+BAA+B,oBAAoB,IAAI,sBAAsB,MAAM,KAAK,IAAI;gCAC5F,6DAA6D;gCAC7D,uFAAuF;gCACvF,SAAS;yBACZ;qBACF;iBACF,CAAC;gBAEF,IAAI,aAAa;oBAAE,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBAC5D,aAAa,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACvC,CAAC;YAED,eAAe,CAAC;gBACd,UAAU;gBACV,IAAI;gBACJ,aAAa;gBACb,SAAS;gBACT,YAAY;gBACZ,UAAU;gBACV,aAAa;aACd,CAAC,CAAC;YAEH,IAAI,iBAAiB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,MAAM,iBAAiB,EAAE,CAAC;gBAC1B,OAAO;oBACL,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa;oBAC3D,aAAa;oBACb,SAAS,EAAE,EAAE;oBACb,KAAK,EAAE,IAAI,GAAG,CAAC;oBACf,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,aAAa,EAAE;iBACrE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,iBAAiB,EAAE,CAAC;IAC1B,OAAO;QACL,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,aAAa;QAC3D,aAAa;QACb,SAAS,EAAE,+BAA+B,QAAQ,IAAI;QACtD,KAAK,EAAE,QAAQ;KAChB,CAAC;AACJ,CAAC","sourcesContent":["import { FunctionCallingConfigMode, Tool } from \"@google/genai\";\nimport fs from \"node:fs/promises\";\nimport type { GenTokensRepository } from \"../../repository/genTokens.repository.js\";\nimport { persistToolCall } from \"../../services/toolcallPersist.service.js\";\nimport { EVENT_TYPES, EventType } from \"../../types/events.js\";\nimport { STYLE_TOKEN_KEYS } from \"../../types/styleConfig.js\";\nimport { getAvailableRoutes } from \"../tools/helpers/pageConfigJson.helpers.js\";\nimport { createWorkspaceToolImpls } from \"../tools/implementations/factories.js\";\nimport { CoreFs } from \"../tools/implementations/workspaceDeps.js\";\nimport { parsePlannerTasksUnknown } from \"./plannerTaskParser.js\";\nimport {\n  compactForModel,\n  DEFAULT_CONTEXT_POLICY,\n  normalizeReadFileArgs,\n  redactFunctionCallArgs,\n  ToolEvent,\n  ToolLoopContextPolicy,\n} from \"./toolLoopContext.js\";\nimport {\n  aiCallWithRetry,\n  buildToolStatusMessage,\n  recordToolEvent,\n  serializeError,\n} from \"./toolLoopRunnerUtils.js\";\n\nexport type ToolLoopResult = {\n  contents: any[];\n  modelContents: any[];\n  finalText: string;\n  steps: number;\n  terminalCall?: {\n    name: string;\n    args: Record<string, unknown>;\n    response: unknown;\n  };\n};\n\nexport type Logger = (message: string, eventType: EventType) => Promise<void>;\n\nexport type AiCallResponse = {\n  functionCalls?: any[];\n  text?: string;\n};\n\nexport type AiCallFn = (\n  request: unknown,\n  options: {\n    tools?: Tool[];\n    model?: string;\n    toolCallingMode?: FunctionCallingConfigMode;\n  },\n) => Promise<AiCallResponse>;\n\nexport type RunToolLoopOptions = {\n  initialContents: any[];\n  tools: Tool[];\n  workspaceRoot: string;\n  maxSteps?: number;\n  toolCallingMode?: FunctionCallingConfigMode;\n  terminalToolNames?: string[];\n  keepFullTrace?: boolean;\n  contextPolicy?: ToolLoopContextPolicy;\n  aiCall: AiCallFn;\n  logger: Logger;\n  applyPatchAutoRetryMax?: number;\n  aiCallAutoRetryMax?: number;\n  aiCallAutoRetryBaseMs?: number;\n  aiCallAutoRetryMaxMs?: number;\n  persistResponse?: (modelInput: any, modelOutput: any) => Promise<void>;\n  tokenPersistence?: {\n    repository: Pick<GenTokensRepository, \"persistGenTokens\">;\n    sessionId: string;\n    model: string;\n  };\n};\n\nexport async function runToolLoop(\n  options: RunToolLoopOptions,\n): Promise<ToolLoopResult> {\n  const isPlainObject = (value: unknown): value is Record<string, unknown> =>\n    typeof value === \"object\" && value !== null && !Array.isArray(value);\n\n  const styleTokenKeySet = new Set<string>(\n    STYLE_TOKEN_KEYS as unknown as string[],\n  );\n\n  const {\n    initialContents,\n    tools,\n    workspaceRoot,\n    maxSteps = 30,\n    toolCallingMode = FunctionCallingConfigMode.ANY,\n    terminalToolNames = [],\n    keepFullTrace = true,\n    contextPolicy,\n    aiCall,\n    logger,\n    applyPatchAutoRetryMax = 2,\n    aiCallAutoRetryMax = 3, // must have it to try 3 times as gemini errors a lot due to high demand sometimes\n    aiCallAutoRetryBaseMs = 400,\n    aiCallAutoRetryMaxMs = 10_000,\n    persistResponse,\n    tokenPersistence,\n  } = options;\n\n  const extractUsageTokenCounts = (rawResponse: unknown) => {\n    const usage = (rawResponse as any)?.usageMetadata;\n    const promptTokenCount = usage?.promptTokenCount;\n    const candidatesTokenCount = usage?.candidatesTokenCount;\n\n    const inputTokens = Number(promptTokenCount);\n    const outputTokens = Number(candidatesTokenCount);\n\n    if (!Number.isFinite(inputTokens) || inputTokens < 0) return null;\n    if (!Number.isFinite(outputTokens) || outputTokens < 0) return null;\n\n    return { inputTokens, outputTokens };\n  };\n\n  let totalInputTokens = 0;\n  let totalOutputTokens = 0;\n  let sawAnyTokenUsage = false;\n\n  const persistTokensOnce = async () => {\n    if (!tokenPersistence) return;\n    if (!sawAnyTokenUsage) return;\n\n    try {\n      await tokenPersistence.repository.persistGenTokens(\n        tokenPersistence.sessionId,\n        totalInputTokens,\n        totalOutputTokens,\n        tokenPersistence.model,\n      );\n    } catch (err) {\n      console.error(\"Tool loop: failed to persist gen tokens\", err);\n    }\n  };\n\n  const nodeFs: CoreFs = {\n    readFile: async (absolutePath) => fs.readFile(absolutePath, \"utf-8\"),\n    writeFile: async (absolutePath, content) =>\n      fs.writeFile(absolutePath, content ?? \"\", \"utf-8\"),\n    mkdirp: async (absoluteDir) => {\n      await fs.mkdir(absoluteDir, { recursive: true });\n    },\n    rmFile: async (absolutePath) => fs.rm(absolutePath, { force: true }),\n    stat: async (absolutePath) => fs.stat(absolutePath),\n    safeReadDir: async (absoluteDir) =>\n      fs.readdir(absoluteDir, { withFileTypes: true }),\n  };\n\n  const impls = createWorkspaceToolImpls({\n    workspaceRoot,\n    fs: nodeFs,\n  });\n\n  const toolHandlers: Record<string, (args: any) => Promise<any>> = {\n    read_file: async (args) => {\n      const path = String(args.path ?? \"\");\n      const startLine =\n        args.start_line === undefined ? undefined : Number(args.start_line);\n      const endLine =\n        args.end_line === undefined ? undefined : Number(args.end_line);\n      const content = await impls.readFileImpl(path, startLine, endLine);\n      return { path, content };\n    },\n    write_file: (args) =>\n      impls.writeFileImpl(String(args.path ?? \"\"), String(args.content ?? \"\")),\n    list_dir: async (args) => {\n      const content = await impls.listDirImpl(\n        String(args.path ?? \"\"),\n        Number(args.depth ?? 1),\n      );\n      return { content };\n    },\n    search: async (args) => {\n      const results = await impls.searchImpl(String(args.search_query ?? \"\"));\n      return { results };\n    },\n    apply_patch: (args) =>\n      impls.applyPatchImpl(String(args.patch_string ?? \"\")),\n    update_global_styles: async (args) => {\n      const result = await impls.updateGlobalStylesImpl(args);\n      return result;\n    },\n    create_new_route: async (args) => {\n      const parentRoute = String(args.parent_route ?? \"\");\n      const routeName = String(args.route_name ?? \"\");\n      const result = await impls.createNewRouteImpl(parentRoute, routeName);\n      return result;\n    },\n    delete_element: async (args) => {\n      const route = String(args.route ?? \"\");\n      const element_id = String(args.element_id ?? \"\");\n      const result = await impls.deleteElementImpl(route, element_id);\n      return result;\n    },\n    insert_element: async (args) => {\n      const result = await impls.insertElementImpl(args);\n      if (!result.success) {\n        const available = await getAvailableRoutes({\n          workspaceRoot,\n          fs: nodeFs,\n        });\n        return {\n          success: false,\n          error: `insert_element failed: ${result.error}. Available routes are: ${JSON.stringify(available)}. If you intend to create a new route, create it using the 'create_new_route' tool.`,\n          available_routes: available,\n        };\n      }\n      return result;\n    },\n    update_props: async (args) => {\n      const route = String(args.route ?? \"\");\n      const element_id = String(args.element_id ?? \"\");\n      const props: any = args.props;\n      const result = await impls.updatePropsImpl({\n        route,\n        element_id,\n        ...props,\n      });\n      return result;\n    },\n    update_classname: async (args) => {\n      const route = String(args.route ?? \"\");\n      const element_id = String(args.element_id ?? \"\");\n      const class_name = String(args.class_name ?? \"\");\n      const result = await impls.updateClassNameImpl(\n        route,\n        element_id,\n        class_name,\n      );\n      return result;\n    },\n    get_available_routes: async (args) => {\n      const routes = await getAvailableRoutes({ workspaceRoot, fs: nodeFs });\n      return { success: true, routes };\n    },\n    submit_codegen_done: async (args) => ({\n      success: true,\n      summary: String(args.summary ?? \"\").trim(),\n    }),\n    submit_planner_tasks: async (args) => {\n      const tasks = parsePlannerTasksUnknown(args.planner_tasks);\n      return { success: true, count: tasks.length };\n    },\n  };\n\n  if (typeof aiCall !== \"function\") {\n    throw new Error(\"Tool loop: aiCall is required.\");\n  }\n\n  const policy: Required<ToolLoopContextPolicy> = {\n    ...DEFAULT_CONTEXT_POLICY,\n    ...(contextPolicy ?? {}),\n  };\n\n  const toolEvents: ToolEvent[] = [];\n  let applyPatchRetryCount = 0;\n\n  const EXECUTION_GUIDE_MARKER = \"TOOL_LOOP_EXECUTION_GUIDE_V1\";\n  const executionGuideInstruction = {\n    role: \"user\",\n    parts: [\n      {\n        text:\n          `${EXECUTION_GUIDE_MARKER}\\n` +\n          `Execution limit: At most ${maxSteps} assistant turn(s) in this tool loop. ` +\n          `One turn = one assistant response in the tool loop.\\n` +\n          `Complete the task in as few turns as possible and avoid unnecessary actions. Prioritize correctness.`,\n      },\n    ],\n  };\n\n  const fullTraceContents: any[] = keepFullTrace\n    ? [...initialContents, executionGuideInstruction]\n    : [];\n  let modelContents: any[] = [...initialContents, executionGuideInstruction];\n  const pinnedInitialCount = initialContents.length + 1;\n  const pushBoth = (fullItem: any, modelItem: any) => {\n    if (keepFullTrace) fullTraceContents.push(fullItem);\n    modelContents.push(modelItem);\n  };\n  const pushModelOnly = (modelItem: any) => {\n    modelContents.push(modelItem);\n  };\n\n  for (let step = 0; step < maxSteps; step++) {\n    modelContents = compactForModel({\n      initialCount: pinnedInitialCount,\n      modelContents,\n      toolEvents,\n      policy,\n    });\n\n    if (policy.logApproxModelChars) {\n      const approxChars = JSON.stringify(modelContents).length;\n      console.log(\"Tool loop: approx model chars\", {\n        approxChars,\n        step: step + 1,\n      });\n    }\n\n    let response: Awaited<ReturnType<AiCallFn>>;\n    try {\n      response = await aiCallWithRetry({\n        aiCall,\n        request: modelContents,\n        options: { tools, toolCallingMode },\n        retryMax: aiCallAutoRetryMax,\n        retryBaseMs: aiCallAutoRetryBaseMs,\n        retryMaxMs: aiCallAutoRetryMaxMs,\n        step: step + 1,\n        logger,\n      });\n    } catch (err) {\n      logger(\n        \"Tool loop: AI provider error; preserving context and continuing\",\n        EVENT_TYPES.STEP_ERROR,\n      );\n      console.error(\"Tool loop: aiCall failed (provider/server side)\", err, {\n        step: step + 1,\n        error: serializeError(err),\n      });\n\n      const message =\n        err instanceof Error ? err.message : JSON.stringify(err ?? null);\n      const providerErrorInstruction = {\n        role: \"user\",\n        parts: [\n          {\n            text:\n              `AI provider error (server-side). Do NOT clear or restart context; continue from the existing conversation state.\\n` +\n              `Error: ${message}\\n` +\n              `Next: retry the last request using the same context. If you were about to call tools, resend a valid tool call.`,\n          },\n        ],\n      };\n      if (keepFullTrace) fullTraceContents.push(providerErrorInstruction);\n      continue;\n    }\n\n    if (persistResponse) {\n      try {\n        await persistResponse(modelContents, response);\n      } catch (err) {\n        console.error(\"Tool loop: failed to persist response\", err, {\n          step: step + 1,\n        });\n      }\n    }\n\n    {\n      const usage = extractUsageTokenCounts(response);\n      if (usage) {\n        sawAnyTokenUsage = true;\n        totalInputTokens += usage.inputTokens;\n        totalOutputTokens += usage.outputTokens;\n      }\n    }\n\n    const functionCalls = response.functionCalls ?? [];\n    if (functionCalls.length === 0) {\n      await persistTokensOnce();\n      return {\n        contents: keepFullTrace ? fullTraceContents : modelContents,\n        modelContents,\n        finalText: (response.text ?? \"\").trim(),\n        steps: step + 1,\n      };\n    }\n\n    const signatureById = (() => {\n      try {\n        const candidates = Array.isArray((response as any)?.candidates)\n          ? ((response as any).candidates as any[])\n          : [];\n        const parts = candidates?.[0]?.content?.parts;\n        const arr = Array.isArray(parts) ? (parts as any[]) : [];\n        const map = new Map<string, string>();\n        for (const p of arr) {\n          const fc = p?.functionCall;\n          const id = fc?.id;\n          const sig = p?.thoughtSignature ?? p?.thought_signature;\n          if (typeof id === \"string\" && typeof sig === \"string\" && sig) {\n            map.set(id, sig);\n          }\n        }\n        return map;\n      } catch {\n        return new Map<string, string>();\n      }\n    })();\n\n    for (let callIndex = 0; callIndex < functionCalls.length; callIndex++) {\n      const call = functionCalls[callIndex];\n      const name = call.name?.toString() ?? \"\";\n      const args = (call.args ?? {}) as Record<string, unknown>;\n      const thoughtSignature: string | undefined = (() => {\n        const direct =\n          (call as any)?.thought_signature ?? (call as any)?.thoughtSignature;\n        if (typeof direct === \"string\" && direct) return direct;\n        const id = (call as any)?.id;\n        if (typeof id === \"string\" && signatureById.has(id)) {\n          return signatureById.get(id);\n        }\n        return undefined;\n      })();\n\n      if (!name) {\n        logger(\n          \"Tool loop: malformed function call from model; preserving context and continuing\",\n          EVENT_TYPES.STEP_ERROR,\n        );\n        const malformedInstruction = {\n          role: \"user\",\n          parts: [\n            {\n              text:\n                `Malformed function call received (missing tool name). Do NOT clear or restart context.\\n` +\n                `Resend a single valid tool call with a non-empty name and JSON args.\\n` +\n                `Bad call: ${JSON.stringify(call ?? null).slice(0, 1500)}`,\n            },\n          ],\n        };\n        if (keepFullTrace) fullTraceContents.push(malformedInstruction);\n        modelContents.push(malformedInstruction);\n        continue;\n      }\n\n      const handler = toolHandlers[name];\n      const handlerMissingResult = !handler\n        ? {\n            success: false,\n            error: `No handler registered for \"${name}\".`,\n            error_detail: {\n              name: \"MissingToolHandlerError\",\n              message: `No handler registered for \"${name}\".`,\n            },\n          }\n        : null;\n\n      let effectiveArgs: Record<string, unknown> = args;\n      let readFileMeta: {\n        start: number;\n        end: number;\n        wasCapped: boolean;\n      } | null = null;\n      if (name === \"read_file\") {\n        const normalized = normalizeReadFileArgs(\n          effectiveArgs,\n          policy.readFileDefaultMaxLines,\n        );\n        effectiveArgs = normalized.effectiveArgs;\n        readFileMeta = {\n          start: normalized.start,\n          end: normalized.end,\n          wasCapped: normalized.wasCapped,\n        };\n      }\n\n      if (name === \"update_global_styles\") {\n        // Be forgiving: models sometimes include a legacy \"tokens\" key or other junk.\n        // We accept the call as long as at least one valid token key/value is provided.\n        const tokensMaybe = (effectiveArgs as any)?.tokens;\n        const normalized: Record<string, unknown> = {};\n\n        if (isPlainObject(tokensMaybe)) {\n          for (const [k, v] of Object.entries(tokensMaybe)) {\n            if (!styleTokenKeySet.has(k)) continue;\n            if (typeof v !== \"string\") continue;\n            normalized[k] = v;\n          }\n        }\n\n        for (const [k, v] of Object.entries(effectiveArgs ?? {})) {\n          if (!styleTokenKeySet.has(k)) continue;\n          if (typeof v !== \"string\") continue;\n          normalized[k] = v;\n        }\n\n        effectiveArgs = normalized;\n      }\n\n      logger(\n        buildToolStatusMessage(name, effectiveArgs, readFileMeta),\n        EVENT_TYPES.STEP_STARTED,\n      );\n\n      const modelArgs = redactFunctionCallArgs(name, effectiveArgs);\n\n      const functionCallPart = {\n        functionCall: {\n          name,\n          args: effectiveArgs,\n        },\n        ...(thoughtSignature\n          ? {\n              thoughtSignature: thoughtSignature,\n              thought_signature: thoughtSignature,\n            }\n          : {}),\n      };\n\n      const functionCallPartModel = {\n        functionCall: {\n          name,\n          args: modelArgs,\n        },\n        ...(thoughtSignature\n          ? {\n              thoughtSignature: thoughtSignature,\n              thought_signature: thoughtSignature,\n            }\n          : {}),\n      };\n\n      const assistantFull = {\n        role: \"model\",\n        parts: [\n          {\n            ...functionCallPart,\n          },\n        ],\n      };\n\n      const assistantModel = {\n        role: \"model\",\n        parts: [\n          {\n            ...functionCallPartModel,\n          },\n        ],\n      };\n\n      if (keepFullTrace) {\n        pushBoth(assistantFull, assistantModel);\n      } else {\n        pushModelOnly(assistantModel);\n      }\n\n      let toolResultRaw: unknown;\n      if (handlerMissingResult) {\n        toolResultRaw = handlerMissingResult;\n      } else {\n        try {\n          if (name === \"update_global_styles\") {\n            const flatKeys = Object.keys(effectiveArgs ?? {}).filter((k) =>\n              styleTokenKeySet.has(k),\n            );\n            if (flatKeys.length === 0) {\n              toolResultRaw = {\n                success: false,\n                error: \"must include at least one token key/value\",\n                error_detail: {\n                  name: \"InvalidToolArgumentsError\",\n                  message:\n                    'update_global_styles requires at least one token key/value (e.g. { radius: \"0.75rem\" }).',\n                },\n                note: \"Resend update_global_styles with at least one token key/value, or skip this tool call.\",\n              };\n            } else {\n              toolResultRaw = await handler(effectiveArgs);\n            }\n          } else {\n            toolResultRaw = await handler(effectiveArgs);\n          }\n        } catch (err) {\n          logger(`AI tool: ${name} failed`, EVENT_TYPES.STEP_ERROR);\n          console.error(\"Tool loop: handler threw\", err, {\n            tool: name,\n            step: step + 1,\n          });\n          toolResultRaw = {\n            success: false,\n            error: err instanceof Error ? err.message : String(err),\n            error_detail: serializeError(err),\n            note: \"Tool handler threw. Inspect error_detail and retry with corrected args or a different approach.\",\n          };\n        }\n      }\n      let toolResult: unknown = toolResultRaw;\n\n      if (name === \"read_file\" && readFileMeta) {\n        const path = String(effectiveArgs.path ?? \"\");\n\n        const jsonPayload =\n          (toolResultRaw as any)?.kind === \"json\"\n            ? (toolResultRaw as any)?.json\n            : undefined;\n        if (jsonPayload !== undefined) {\n          // Token-efficient: return JSON as structured data (no double-stringifying).\n          toolResult = { path, json: jsonPayload };\n        } else {\n          const rawContent =\n            typeof (toolResultRaw as any)?.content === \"string\"\n              ? String((toolResultRaw as any).content)\n              : typeof toolResultRaw === \"string\"\n                ? toolResultRaw\n                : JSON.stringify(toolResultRaw ?? null);\n\n          toolResult = {\n            path,\n            start_line: readFileMeta.start,\n            end_line: readFileMeta.end,\n            truncated: readFileMeta.wasCapped,\n            content: rawContent,\n            note: readFileMeta.wasCapped\n              ? `Capped to ${policy.readFileDefaultMaxLines} lines. Request more with start_line/end_line.`\n              : undefined,\n          };\n        }\n      }\n\n      try {\n        await persistToolCall(name, modelArgs, toolResult);\n      } catch (err) {\n        console.error(\"Tool loop: failed to persist tool call\", err, {\n          tool: name,\n          step: step + 1,\n        });\n      }\n\n      const responseFull = {\n        role: \"user\",\n        parts: [\n          {\n            functionResponse: {\n              name,\n              response: toolResult,\n            },\n          },\n        ],\n      };\n\n      if (keepFullTrace) {\n        fullTraceContents.push(responseFull);\n      }\n      modelContents.push(responseFull);\n\n      if (\n        name === \"apply_patch\" &&\n        (toolResult as any)?.success === false &&\n        applyPatchAutoRetryMax > 0 &&\n        applyPatchRetryCount < applyPatchAutoRetryMax\n      ) {\n        applyPatchRetryCount += 1;\n\n        const error = String((toolResult as any)?.error ?? \"unknown error\");\n        const debugFiles = Array.isArray((toolResult as any)?.debug?.files)\n          ? ((toolResult as any).debug.files as Array<{\n              path?: string;\n              head?: string;\n            }>)\n          : [];\n\n        const debugText =\n          debugFiles.length > 0\n            ? `\\n\\nFILE SNAPSHOTS (for regenerating the patch):\\n${debugFiles\n                .slice(0, 3)\n                .map(\n                  (f) =>\n                    `--- ${String(f.path ?? \"\")} ---\\n${String(\n                      f.head ?? \"\",\n                    )}\\n--- end ---`,\n                )\n                .join(\"\\n\\n\")}`\n            : \"\";\n\n        const retryInstruction = {\n          role: \"user\",\n          parts: [\n            {\n              text:\n                `apply_patch failed (attempt ${applyPatchRetryCount}/${applyPatchAutoRetryMax}): ${error}\\n` +\n                `Regenerate a patch that matches the current file contents. ` +\n                `For large rewrites, prefer write_file(path, content) or Delete+Add instead of Update.` +\n                debugText,\n            },\n          ],\n        };\n\n        if (keepFullTrace) fullTraceContents.push(retryInstruction);\n        modelContents.push(retryInstruction);\n      }\n\n      recordToolEvent({\n        toolEvents,\n        name,\n        effectiveArgs,\n        modelArgs,\n        readFileMeta,\n        toolResult,\n        toolResultRaw,\n      });\n\n      if (terminalToolNames.includes(name)) {\n        await persistTokensOnce();\n        return {\n          contents: keepFullTrace ? fullTraceContents : modelContents,\n          modelContents,\n          finalText: \"\",\n          steps: step + 1,\n          terminalCall: { name, args: effectiveArgs, response: toolResultRaw },\n        };\n      }\n    }\n  }\n\n  await persistTokensOnce();\n  return {\n    contents: keepFullTrace ? fullTraceContents : modelContents,\n    modelContents,\n    finalText: `Stopped: max steps reached (${maxSteps}).`,\n    steps: maxSteps,\n  };\n}\n"]}

package/dist/core.d.ts

@@ -32,11 +32,13 @@ export declare class QwintlyCore {
     readonly workspacePath: string;
     readonly source: string;
     readonly step: GenStep;
+    readonly geminiModel: string;
     private readonly aiClient?;
     private readonly statusRepo;
     private readonly ctxRepo;
     private readonly redisStatusPublisher;
     private readonly projectOpsRepo;
+    private readonly genTokensRepo;
     constructor(options: QwintlyCoreOptions);
     runAiFlow(initialContents: any[], tools: Tool[], maxSteps: number, terminalToolNames: string[], persistResponse?: (modelInput: any, modelOutput: any) => Promise<void>): Promise<ToolLoopResult>;
     streamLog(message: string, eventType: EventType, displayedSummary?: boolean): Promise<void>;

package/dist/core.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"core.d.ts","sourceRoot":"","sources":["../src/core.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,IAAI,EAAE,MAAM,eAAe,CAAC;AAEhE,OAAO,EAGL,cAAc,EACf,MAAM,iCAAiC,CAAC;AAazC,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,cAAc,EACf,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAGhE,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,OAAO,CAAC;IACd,QAAQ,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IAC/C,OAAO,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,QAAQ,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAC7C,MAAM,CAAC,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CAC7C,CAAC;AAiBF,qBAAa,WAAW;IACtB,SAAgB,MAAM,EAAE,MAAM,CAAC;IAC/B,SAAgB,SAAS,EAAE,MAAM,CAAC;IAClC,SAAgB,aAAa,EAAE,MAAM,CAAC;IACtC,SAAgB,MAAM,EAAE,MAAM,CAAC;IAC/B,SAAgB,IAAI,EAAE,OAAO,CAAC;IAE9B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAW;IACrC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAsB;IACjD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoB;IAC5C,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAoB;IACzD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAuB;gBAE1C,OAAO,EAAE,kBAAkB;IA2D1B,SAAS,CACpB,eAAe,EAAE,GAAG,EAAE,EACtB,KAAK,EAAE,IAAI,EAAE,EACb,QAAQ,EAAE,MAAM,EAChB,iBAAiB,EAAE,MAAM,EAAE,EAC3B,eAAe,CAAC,EAAE,CAAC,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,GACrE,OAAO,CAAC,cAAc,CAAC;IAyBb,SAAS,CACpB,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,SAAS,EACpB,gBAAgB,GAAE,OAAe,GAChC,OAAO,CAAC,IAAI,CAAC;IA2BH,mBAAmB,IAAI,OAAO,CAAC,WAAW,CAAC;YAQ1C,UAAU;IAMX,eAAe,IAAI,OAAO,CAAC,YAAY,CAAC;IAIxC,eAAe,IAAI,OAAO,CAAC,YAAY,CAAC;IAIxC,iBAAiB,IAAI,OAAO,CAAC,cAAc,CAAC;IAI5C,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAY7E"}
+{"version":3,"file":"core.d.ts","sourceRoot":"","sources":["../src/core.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,IAAI,EAAE,MAAM,eAAe,CAAC;AAEhE,OAAO,EAGL,cAAc,EACf,MAAM,iCAAiC,CAAC;AAczC,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,cAAc,EACf,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAIhE,MAAM,MAAM,kBAAkB,GAAG;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,OAAO,CAAC;IACd,QAAQ,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IAC/C,OAAO,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,QAAQ,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAC7C,MAAM,CAAC,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CAC7C,CAAC;AAiBF,qBAAa,WAAW;IACtB,SAAgB,MAAM,EAAE,MAAM,CAAC;IAC/B,SAAgB,SAAS,EAAE,MAAM,CAAC;IAClC,SAAgB,aAAa,EAAE,MAAM,CAAC;IACtC,SAAgB,MAAM,EAAE,MAAM,CAAC;IAC/B,SAAgB,IAAI,EAAE,OAAO,CAAC;IAC9B,SAAgB,WAAW,EAAE,MAAM,CAAC;IAEpC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAW;IACrC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAsB;IACjD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoB;IAC5C,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAoB;IACzD,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAuB;IACtD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB;gBAExC,OAAO,EAAE,kBAAkB;IAiE1B,SAAS,CACpB,eAAe,EAAE,GAAG,EAAE,EACtB,KAAK,EAAE,IAAI,EAAE,EACb,QAAQ,EAAE,MAAM,EAChB,iBAAiB,EAAE,MAAM,EAAE,EAC3B,eAAe,CAAC,EAAE,CAAC,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,GACrE,OAAO,CAAC,cAAc,CAAC;IA8Bb,SAAS,CACpB,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,SAAS,EACpB,gBAAgB,GAAE,OAAe,GAChC,OAAO,CAAC,IAAI,CAAC;IA2BH,mBAAmB,IAAI,OAAO,CAAC,WAAW,CAAC;YAQ1C,UAAU;IAMX,eAAe,IAAI,OAAO,CAAC,YAAY,CAAC;IAIxC,eAAe,IAAI,OAAO,CAAC,YAAY,CAAC;IAIxC,iBAAiB,IAAI,OAAO,CAAC,cAAc,CAAC;IAI5C,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAY7E"}

package/dist/core.js

@@ -9,10 +9,12 @@ import { statusService } from "./logging/genStatus.service.js";
 import { SendStatusToRedis } from "./logging/redis.service.js";
 import { ContextRepository } from "./repository/context.repository.js";
 import { GenStatusRepository } from "./repository/genStatus.repository.js";
+import { GenTokensRepository } from "./repository/genTokens.repository.js";
 import { ProjectOpsRepository } from "./repository/projectOperations.repository.js";
 import { applyOperations } from "./services/syncEditOps.service.js";
 import { initializeToolCallsRepository } from "./services/toolcallPersist.service.js";
 import { assertNonEmptyString } from "./utils/utils.js";
+import { DEFAULT_MODEL } from "./ai/generate/gemini.client.js";
 export class QwintlyCore {
     constructor(options) {
         assertNonEmptyString(options.chatId, "chatId");
@@ -38,9 +40,11 @@ export class QwintlyCore {
         if (options.gemini?.apiKey) {
             this.aiClient = getClient("gemini", options.gemini.apiKey, options.gemini.model);
         }
+        this.geminiModel = options.gemini?.model ?? DEFAULT_MODEL;
         this.statusRepo = new GenStatusRepository(options.supabase.endpoint, options.supabase.secret);
         this.ctxRepo = new ContextRepository(options.supabase.endpoint, options.supabase.secret);
         this.projectOpsRepo = new ProjectOpsRepository(options.supabase.endpoint, options.supabase.secret);
+        this.genTokensRepo = new GenTokensRepository(options.supabase.endpoint, options.supabase.secret);
         this.redisStatusPublisher = new SendStatusToRedis(options.upstash.url, options.upstash.token);
         initializeToolCallsRepository(options.supabase.endpoint, options.supabase.secret, options.sessionId);
         console.log(`QwintlyCore initialized (chatId=${this.chatId}, sessionId=${this.sessionId})`);
@@ -61,6 +65,11 @@ export class QwintlyCore {
             }),
             logger: this.streamLog.bind(this),
             persistResponse,
+            tokenPersistence: {
+                repository: this.genTokensRepo,
+                sessionId: this.sessionId,
+                model: this.geminiModel,
+            },
         };
         const result = await runToolLoop(toolLoopOptions);
         return result;

package/dist/core.js.map

@@ -1 +1 @@
-{"version":3,"file":"core.js","sourceRoot":"","sources":["../src/core.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAC5D,OAAO,EACL,WAAW,GAGZ,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,OAAO,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAC;AACpF,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AAQtF,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AA6BxD,MAAM,OAAO,WAAW;IAatB,YAAY,OAA2B;QACrC,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC/C,oBAAoB,CAAC,OAAO,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QACrD,oBAAoB,CAAC,OAAO,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAC7D,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC/C,oBAAoB,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC3C,oBAAoB,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,EAAE,mBAAmB,CAAC,CAAC;QACtE,oBAAoB,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC;QAClE,oBAAoB,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,EAAE,aAAa,CAAC,CAAC;QAC1D,oBAAoB,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;QAE9D,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QAC3C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAEzB,IAAI,OAAO,CAAC,QAAQ,EAAE,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;YACzD,YAAY,CAAC;gBACX,GAAG,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG;gBACzB,SAAS,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS;aACtC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,QAAQ,GAAG,SAAS,CACvB,QAAQ,EACR,OAAO,CAAC,MAAM,CAAC,MAAM,EACrB,OAAO,CAAC,MAAM,CAAC,KAAK,CACT,CAAC;QAChB,CAAC;QAED,IAAI,CAAC,UAAU,GAAG,IAAI,mBAAmB,CACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CAAC;QACF,IAAI,CAAC,OAAO,GAAG,IAAI,iBAAiB,CAClC,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,IAAI,oBAAoB,CAC5C,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CAAC;QACF,IAAI,CAAC,oBAAoB,GAAG,IAAI,iBAAiB,CAC/C,OAAO,CAAC,OAAO,CAAC,GAAG,EACnB,OAAO,CAAC,OAAO,CAAC,KAAK,CACtB,CAAC;QACF,6BAA6B,CAC3B,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,EACvB,OAAO,CAAC,SAAS,CAClB,CAAC;QAEF,OAAO,CAAC,GAAG,CACT,mCAAmC,IAAI,CAAC,MAAM,eAAe,IAAI,CAAC,SAAS,GAAG,CAC/E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,SAAS,CACpB,eAAsB,EACtB,KAAa,EACb,QAAgB,EAChB,iBAA2B,EAC3B,eAAsE;QAEtE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CACd,6EAA6E,CAC9E,CAAC;QACJ,CAAC;QACD,MAAM,eAAe,GAAuB;YAC1C,eAAe,EAAE,eAAe;YAChC,KAAK,EAAE,KAAK;YACZ,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,QAAQ,EAAE,QAAQ;YAClB,iBAAiB,EAAE,iBAAiB;YACpC,MAAM,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,CAC3B,IAAI,CAAC,QAAS,CAAC,UAAU,CAAC,OAAO,EAAE;gBACjC,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,eAAe,EAAE,OAAO,CAAC,eAAe;aACzC,CAAC;YACJ,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;YACjC,eAAe;SAChB,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,eAAe,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,SAAS,CACpB,OAAe,EACf,SAAoB,EACpB,mBAA4B,KAAK;QAEjC,IAAI,CAAC;YACH,oBAAoB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACzC,MAAM,aAAa,CACjB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,SAAS,EACd,SAAS,EACT,IAAI,CAAC,IAAI,EACT,OAAO,EACP,IAAI,CAAC,MAAM,EACX;gBACE,UAAU,EAAE;oBACV,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;iBACpE;gBACD,SAAS,EAAE;oBACT,OAAO,EAAE,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,IAAI,CACvD,IAAI,CAAC,oBAAoB,CAC1B;iBACF;aACF,EACD,gBAAgB,CACjB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,mBAAmB;QAC9B,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAEjE,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAE/D,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,OAAqC;QAErC,OAAO,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACrC,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,OAAO,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,OAAO,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,iBAAiB;QAC5B,OAAO,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;IAC9C,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,SAAiB;QACvD,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QACzB,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAC3C,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAC3C,MAAM,UAAU,GAAG,MAAM,eAAe,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACzD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;QACzC,OAAO,UAAU,CAAC;IACpB,CAAC;CACF","sourcesContent":["import { FunctionCallingConfigMode, Tool } from \"@google/genai\";\nimport { getClient } from \"./ai/generate/generateClient.js\";\nimport {\n  runToolLoop,\n  RunToolLoopOptions,\n  ToolLoopResult,\n} from \"./ai/toolLoop/toolLoopRunner.js\";\nimport { initUnsplash } from \"./image/unsplash.service.js\";\nimport { buildCodegenIndex } from \"./indexer/codegenIndex.js\";\nimport { buildPlannerIndex } from \"./indexer/plannerIndex.js\";\nimport { computeProjectInfo } from \"./indexer/projectInfoIndex.js\";\nimport { buildValidatorIndex } from \"./indexer/validatorIndex.js\";\nimport { statusService } from \"./logging/genStatus.service.js\";\nimport { SendStatusToRedis } from \"./logging/redis.service.js\";\nimport { ContextRepository } from \"./repository/context.repository.js\";\nimport { GenStatusRepository } from \"./repository/genStatus.repository.js\";\nimport { ProjectOpsRepository } from \"./repository/projectOperations.repository.js\";\nimport { applyOperations } from \"./services/syncEditOps.service.js\";\nimport { initializeToolCallsRepository } from \"./services/toolcallPersist.service.js\";\nimport { EventType, GenStep } from \"./types/events.js\";\nimport {\n  CodegenIndex,\n  PlannerIndex,\n  ValidatorIndex,\n} from \"./types/index/index.types.js\";\nimport type { ProjectInfo } from \"./types/projectInfo.types.js\";\nimport { assertNonEmptyString } from \"./utils/utils.js\";\n\nexport type QwintlyCoreOptions = {\n  chatId: string;\n  sessionId: string;\n  workspacePath: string;\n  source: string;\n  step: GenStep;\n  supabase: { endpoint: string; secret: string };\n  upstash: { url: string; token: string };\n  unsplash: { url: string; accessKey: string };\n  gemini?: { apiKey: string; model?: string };\n};\n\ntype AiResponseOptions = {\n  tools?: Tool[];\n  toolCallingMode?: FunctionCallingConfigMode;\n};\n\ntype AiClient = {\n  aiResponse: (\n    request: unknown,\n    options?: AiResponseOptions,\n  ) => Promise<{\n    functionCalls?: any[];\n    text?: string;\n  }>;\n};\n\nexport class QwintlyCore {\n  public readonly chatId: string;\n  public readonly sessionId: string;\n  public readonly workspacePath: string;\n  public readonly source: string;\n  public readonly step: GenStep;\n\n  private readonly aiClient?: AiClient;\n  private readonly statusRepo: GenStatusRepository;\n  private readonly ctxRepo: ContextRepository;\n  private readonly redisStatusPublisher: SendStatusToRedis;\n  private readonly projectOpsRepo: ProjectOpsRepository;\n\n  constructor(options: QwintlyCoreOptions) {\n    assertNonEmptyString(options.chatId, \"chatId\");\n    assertNonEmptyString(options.sessionId, \"sessionId\");\n    assertNonEmptyString(options.workspacePath, \"workspacePath\");\n    assertNonEmptyString(options.source, \"source\");\n    assertNonEmptyString(options.step, \"step\");\n    assertNonEmptyString(options.supabase?.endpoint, \"supabase.endpoint\");\n    assertNonEmptyString(options.supabase?.secret, \"supabase.secret\");\n    assertNonEmptyString(options.upstash?.url, \"upstash.url\");\n    assertNonEmptyString(options.upstash?.token, \"upstash.token\");\n\n    this.chatId = options.chatId;\n    this.sessionId = options.sessionId;\n    this.workspacePath = options.workspacePath;\n    this.source = options.source;\n    this.step = options.step;\n\n    if (options.unsplash?.url && options.unsplash?.accessKey) {\n      initUnsplash({\n        url: options.unsplash.url,\n        accessKey: options.unsplash.accessKey,\n      });\n    }\n\n    if (options.gemini?.apiKey) {\n      this.aiClient = getClient(\n        \"gemini\",\n        options.gemini.apiKey,\n        options.gemini.model,\n      ) as AiClient;\n    }\n\n    this.statusRepo = new GenStatusRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n    );\n    this.ctxRepo = new ContextRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n    );\n    this.projectOpsRepo = new ProjectOpsRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n    );\n    this.redisStatusPublisher = new SendStatusToRedis(\n      options.upstash.url,\n      options.upstash.token,\n    );\n    initializeToolCallsRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n      options.sessionId,\n    );\n\n    console.log(\n      `QwintlyCore initialized (chatId=${this.chatId}, sessionId=${this.sessionId})`,\n    );\n  }\n\n  public async runAiFlow(\n    initialContents: any[],\n    tools: Tool[],\n    maxSteps: number,\n    terminalToolNames: string[],\n    persistResponse?: (modelInput: any, modelOutput: any) => Promise<void>,\n  ): Promise<ToolLoopResult> {\n    if (!this.aiClient) {\n       throw new Error(\n        \"AI client not initialized. Please provide 'gemini' config to use runAiFlow.\",\n      );\n    }\n    const toolLoopOptions: RunToolLoopOptions = {\n      initialContents: initialContents,\n      tools: tools,\n      workspaceRoot: this.workspacePath,\n      maxSteps: maxSteps,\n      terminalToolNames: terminalToolNames,\n      aiCall: (request, options) =>\n        this.aiClient!.aiResponse(request, {\n          tools: options.tools,\n          toolCallingMode: options.toolCallingMode,\n        }),\n      logger: this.streamLog.bind(this),\n      persistResponse,\n    };\n\n    const result = await runToolLoop(toolLoopOptions);\n    return result;\n  }\n\n  public async streamLog(\n    message: string,\n    eventType: EventType,\n    displayedSummary: boolean = false,\n  ): Promise<void> {\n    try {\n      assertNonEmptyString(message, \"message\");\n      await statusService(\n        this.chatId,\n        this.sessionId,\n        eventType,\n        this.step,\n        message,\n        this.source,\n        {\n          repository: {\n            persist: this.statusRepo.persistStatusMessage.bind(this.statusRepo),\n          },\n          publisher: {\n            publish: this.redisStatusPublisher.sendStatusToRedis.bind(\n              this.redisStatusPublisher,\n            ),\n          },\n        },\n        displayedSummary,\n      );\n    } catch (error) {\n      console.error(error);\n    }\n  }\n\n  public async buildProjectInfoIdx(): Promise<ProjectInfo> {\n    const projectInfo = await computeProjectInfo(this.workspacePath);\n\n    await this.ctxRepo.updateProjectInfo(this.chatId, projectInfo);\n\n    return projectInfo;\n  }\n\n  private async buildIndex<T>(\n    builder: (path: string) => Promise<T>,\n  ): Promise<T> {\n    return builder(this.workspacePath);\n  }\n\n  public async buildPlannerIdx(): Promise<PlannerIndex> {\n    return this.buildIndex(buildPlannerIndex);\n  }\n\n  public async buildCodegenIdx(): Promise<CodegenIndex> {\n    return this.buildIndex(buildCodegenIndex);\n  }\n\n  public async buildValidatorIdx(): Promise<ValidatorIndex> {\n    return this.buildIndex(buildValidatorIndex);\n  }\n\n  public async syncEditOps(genId: string, workspace: string): Promise<boolean> {\n    if (!genId) return false;\n    const projectOpsRepo = this.projectOpsRepo;\n    const ops = await projectOpsRepo.fetchProjectOperations(genId);\n    if (!ops || ops.length === 0) return false;\n    const appliedIds = await applyOperations(ops, workspace);\n    if (appliedIds.length > 0) {\n      await projectOpsRepo.markOperationsApplied(appliedIds);\n    }\n    const isModified = appliedIds.length > 0;\n    return isModified;\n  }\n}\n"]}
+{"version":3,"file":"core.js","sourceRoot":"","sources":["../src/core.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAC5D,OAAO,EACL,WAAW,GAGZ,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,OAAO,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAC;AACpF,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AAQtF,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AA6B/D,MAAM,OAAO,WAAW;IAetB,YAAY,OAA2B;QACrC,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC/C,oBAAoB,CAAC,OAAO,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QACrD,oBAAoB,CAAC,OAAO,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAC7D,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC/C,oBAAoB,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC3C,oBAAoB,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,EAAE,mBAAmB,CAAC,CAAC;QACtE,oBAAoB,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC;QAClE,oBAAoB,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,EAAE,aAAa,CAAC,CAAC;QAC1D,oBAAoB,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC;QAE9D,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QAC3C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAEzB,IAAI,OAAO,CAAC,QAAQ,EAAE,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;YACzD,YAAY,CAAC;gBACX,GAAG,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG;gBACzB,SAAS,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS;aACtC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,QAAQ,GAAG,SAAS,CACvB,QAAQ,EACR,OAAO,CAAC,MAAM,CAAC,MAAM,EACrB,OAAO,CAAC,MAAM,CAAC,KAAK,CACT,CAAC;QAChB,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,MAAM,EAAE,KAAK,IAAI,aAAa,CAAC;QAE1D,IAAI,CAAC,UAAU,GAAG,IAAI,mBAAmB,CACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CAAC;QACF,IAAI,CAAC,OAAO,GAAG,IAAI,iBAAiB,CAClC,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CAAC;QACF,IAAI,CAAC,cAAc,GAAG,IAAI,oBAAoB,CAC5C,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CAAC;QACF,IAAI,CAAC,aAAa,GAAG,IAAI,mBAAmB,CAC1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CACxB,CAAC;QACF,IAAI,CAAC,oBAAoB,GAAG,IAAI,iBAAiB,CAC/C,OAAO,CAAC,OAAO,CAAC,GAAG,EACnB,OAAO,CAAC,OAAO,CAAC,KAAK,CACtB,CAAC;QACF,6BAA6B,CAC3B,OAAO,CAAC,QAAQ,CAAC,QAAQ,EACzB,OAAO,CAAC,QAAQ,CAAC,MAAM,EACvB,OAAO,CAAC,SAAS,CAClB,CAAC;QAEF,OAAO,CAAC,GAAG,CACT,mCAAmC,IAAI,CAAC,MAAM,eAAe,IAAI,CAAC,SAAS,GAAG,CAC/E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,SAAS,CACpB,eAAsB,EACtB,KAAa,EACb,QAAgB,EAChB,iBAA2B,EAC3B,eAAsE;QAEtE,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CACb,6EAA6E,CAC9E,CAAC;QACJ,CAAC;QACD,MAAM,eAAe,GAAuB;YAC1C,eAAe,EAAE,eAAe;YAChC,KAAK,EAAE,KAAK;YACZ,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,QAAQ,EAAE,QAAQ;YAClB,iBAAiB,EAAE,iBAAiB;YACpC,MAAM,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,CAC3B,IAAI,CAAC,QAAS,CAAC,UAAU,CAAC,OAAO,EAAE;gBACjC,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,eAAe,EAAE,OAAO,CAAC,eAAe;aACzC,CAAC;YACJ,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;YACjC,eAAe;YACf,gBAAgB,EAAE;gBAChB,UAAU,EAAE,IAAI,CAAC,aAAa;gBAC9B,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,KAAK,EAAE,IAAI,CAAC,WAAW;aACxB;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,eAAe,CAAC,CAAC;QAClD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,SAAS,CACpB,OAAe,EACf,SAAoB,EACpB,mBAA4B,KAAK;QAEjC,IAAI,CAAC;YACH,oBAAoB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACzC,MAAM,aAAa,CACjB,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,SAAS,EACd,SAAS,EACT,IAAI,CAAC,IAAI,EACT,OAAO,EACP,IAAI,CAAC,MAAM,EACX;gBACE,UAAU,EAAE;oBACV,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;iBACpE;gBACD,SAAS,EAAE;oBACT,OAAO,EAAE,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,IAAI,CACvD,IAAI,CAAC,oBAAoB,CAC1B;iBACF;aACF,EACD,gBAAgB,CACjB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,mBAAmB;QAC9B,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAEjE,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAE/D,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,OAAqC;QAErC,OAAO,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACrC,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,OAAO,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,OAAO,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,iBAAiB;QAC5B,OAAO,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;IAC9C,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,SAAiB;QACvD,IAAI,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QACzB,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;QAC3C,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAC3C,MAAM,UAAU,GAAG,MAAM,eAAe,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACzD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,cAAc,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;QACzC,OAAO,UAAU,CAAC;IACpB,CAAC;CACF","sourcesContent":["import { FunctionCallingConfigMode, Tool } from \"@google/genai\";\nimport { getClient } from \"./ai/generate/generateClient.js\";\nimport {\n  runToolLoop,\n  RunToolLoopOptions,\n  ToolLoopResult,\n} from \"./ai/toolLoop/toolLoopRunner.js\";\nimport { initUnsplash } from \"./image/unsplash.service.js\";\nimport { buildCodegenIndex } from \"./indexer/codegenIndex.js\";\nimport { buildPlannerIndex } from \"./indexer/plannerIndex.js\";\nimport { computeProjectInfo } from \"./indexer/projectInfoIndex.js\";\nimport { buildValidatorIndex } from \"./indexer/validatorIndex.js\";\nimport { statusService } from \"./logging/genStatus.service.js\";\nimport { SendStatusToRedis } from \"./logging/redis.service.js\";\nimport { ContextRepository } from \"./repository/context.repository.js\";\nimport { GenStatusRepository } from \"./repository/genStatus.repository.js\";\nimport { GenTokensRepository } from \"./repository/genTokens.repository.js\";\nimport { ProjectOpsRepository } from \"./repository/projectOperations.repository.js\";\nimport { applyOperations } from \"./services/syncEditOps.service.js\";\nimport { initializeToolCallsRepository } from \"./services/toolcallPersist.service.js\";\nimport { EventType, GenStep } from \"./types/events.js\";\nimport {\n  CodegenIndex,\n  PlannerIndex,\n  ValidatorIndex,\n} from \"./types/index/index.types.js\";\nimport type { ProjectInfo } from \"./types/projectInfo.types.js\";\nimport { assertNonEmptyString } from \"./utils/utils.js\";\nimport { DEFAULT_MODEL } from \"./ai/generate/gemini.client.js\";\n\nexport type QwintlyCoreOptions = {\n  chatId: string;\n  sessionId: string;\n  workspacePath: string;\n  source: string;\n  step: GenStep;\n  supabase: { endpoint: string; secret: string };\n  upstash: { url: string; token: string };\n  unsplash: { url: string; accessKey: string };\n  gemini?: { apiKey: string; model?: string };\n};\n\ntype AiResponseOptions = {\n  tools?: Tool[];\n  toolCallingMode?: FunctionCallingConfigMode;\n};\n\ntype AiClient = {\n  aiResponse: (\n    request: unknown,\n    options?: AiResponseOptions,\n  ) => Promise<{\n    functionCalls?: any[];\n    text?: string;\n  }>;\n};\n\nexport class QwintlyCore {\n  public readonly chatId: string;\n  public readonly sessionId: string;\n  public readonly workspacePath: string;\n  public readonly source: string;\n  public readonly step: GenStep;\n  public readonly geminiModel: string;\n\n  private readonly aiClient?: AiClient;\n  private readonly statusRepo: GenStatusRepository;\n  private readonly ctxRepo: ContextRepository;\n  private readonly redisStatusPublisher: SendStatusToRedis;\n  private readonly projectOpsRepo: ProjectOpsRepository;\n  private readonly genTokensRepo: GenTokensRepository;\n\n  constructor(options: QwintlyCoreOptions) {\n    assertNonEmptyString(options.chatId, \"chatId\");\n    assertNonEmptyString(options.sessionId, \"sessionId\");\n    assertNonEmptyString(options.workspacePath, \"workspacePath\");\n    assertNonEmptyString(options.source, \"source\");\n    assertNonEmptyString(options.step, \"step\");\n    assertNonEmptyString(options.supabase?.endpoint, \"supabase.endpoint\");\n    assertNonEmptyString(options.supabase?.secret, \"supabase.secret\");\n    assertNonEmptyString(options.upstash?.url, \"upstash.url\");\n    assertNonEmptyString(options.upstash?.token, \"upstash.token\");\n\n    this.chatId = options.chatId;\n    this.sessionId = options.sessionId;\n    this.workspacePath = options.workspacePath;\n    this.source = options.source;\n    this.step = options.step;\n\n    if (options.unsplash?.url && options.unsplash?.accessKey) {\n      initUnsplash({\n        url: options.unsplash.url,\n        accessKey: options.unsplash.accessKey,\n      });\n    }\n\n    if (options.gemini?.apiKey) {\n      this.aiClient = getClient(\n        \"gemini\",\n        options.gemini.apiKey,\n        options.gemini.model,\n      ) as AiClient;\n    }\n\n    this.geminiModel = options.gemini?.model ?? DEFAULT_MODEL;\n\n    this.statusRepo = new GenStatusRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n    );\n    this.ctxRepo = new ContextRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n    );\n    this.projectOpsRepo = new ProjectOpsRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n    );\n    this.genTokensRepo = new GenTokensRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n    );\n    this.redisStatusPublisher = new SendStatusToRedis(\n      options.upstash.url,\n      options.upstash.token,\n    );\n    initializeToolCallsRepository(\n      options.supabase.endpoint,\n      options.supabase.secret,\n      options.sessionId,\n    );\n\n    console.log(\n      `QwintlyCore initialized (chatId=${this.chatId}, sessionId=${this.sessionId})`,\n    );\n  }\n\n  public async runAiFlow(\n    initialContents: any[],\n    tools: Tool[],\n    maxSteps: number,\n    terminalToolNames: string[],\n    persistResponse?: (modelInput: any, modelOutput: any) => Promise<void>,\n  ): Promise<ToolLoopResult> {\n    if (!this.aiClient) {\n      throw new Error(\n        \"AI client not initialized. Please provide 'gemini' config to use runAiFlow.\",\n      );\n    }\n    const toolLoopOptions: RunToolLoopOptions = {\n      initialContents: initialContents,\n      tools: tools,\n      workspaceRoot: this.workspacePath,\n      maxSteps: maxSteps,\n      terminalToolNames: terminalToolNames,\n      aiCall: (request, options) =>\n        this.aiClient!.aiResponse(request, {\n          tools: options.tools,\n          toolCallingMode: options.toolCallingMode,\n        }),\n      logger: this.streamLog.bind(this),\n      persistResponse,\n      tokenPersistence: {\n        repository: this.genTokensRepo,\n        sessionId: this.sessionId,\n        model: this.geminiModel,\n      },\n    };\n\n    const result = await runToolLoop(toolLoopOptions);\n    return result;\n  }\n\n  public async streamLog(\n    message: string,\n    eventType: EventType,\n    displayedSummary: boolean = false,\n  ): Promise<void> {\n    try {\n      assertNonEmptyString(message, \"message\");\n      await statusService(\n        this.chatId,\n        this.sessionId,\n        eventType,\n        this.step,\n        message,\n        this.source,\n        {\n          repository: {\n            persist: this.statusRepo.persistStatusMessage.bind(this.statusRepo),\n          },\n          publisher: {\n            publish: this.redisStatusPublisher.sendStatusToRedis.bind(\n              this.redisStatusPublisher,\n            ),\n          },\n        },\n        displayedSummary,\n      );\n    } catch (error) {\n      console.error(error);\n    }\n  }\n\n  public async buildProjectInfoIdx(): Promise<ProjectInfo> {\n    const projectInfo = await computeProjectInfo(this.workspacePath);\n\n    await this.ctxRepo.updateProjectInfo(this.chatId, projectInfo);\n\n    return projectInfo;\n  }\n\n  private async buildIndex<T>(\n    builder: (path: string) => Promise<T>,\n  ): Promise<T> {\n    return builder(this.workspacePath);\n  }\n\n  public async buildPlannerIdx(): Promise<PlannerIndex> {\n    return this.buildIndex(buildPlannerIndex);\n  }\n\n  public async buildCodegenIdx(): Promise<CodegenIndex> {\n    return this.buildIndex(buildCodegenIndex);\n  }\n\n  public async buildValidatorIdx(): Promise<ValidatorIndex> {\n    return this.buildIndex(buildValidatorIndex);\n  }\n\n  public async syncEditOps(genId: string, workspace: string): Promise<boolean> {\n    if (!genId) return false;\n    const projectOpsRepo = this.projectOpsRepo;\n    const ops = await projectOpsRepo.fetchProjectOperations(genId);\n    if (!ops || ops.length === 0) return false;\n    const appliedIds = await applyOperations(ops, workspace);\n    if (appliedIds.length > 0) {\n      await projectOpsRepo.markOperationsApplied(appliedIds);\n    }\n    const isModified = appliedIds.length > 0;\n    return isModified;\n  }\n}\n"]}

package/dist/repository/genTokens.repository.d.ts

@@ -0,0 +1,5 @@
+import { DBRepository } from "./repository.js";
+export declare class GenTokensRepository extends DBRepository {
+    persistGenTokens(sessionId: string, input_tokens: number, output_tokens: number, model: string): Promise<void>;
+}
+//# sourceMappingURL=genTokens.repository.d.ts.map

package/dist/repository/genTokens.repository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"genTokens.repository.d.ts","sourceRoot":"","sources":["../../src/repository/genTokens.repository.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,qBAAa,mBAAoB,SAAQ,YAAY;IACtC,gBAAgB,CAC3B,SAAS,EAAE,MAAM,EACjB,YAAY,EAAE,MAAM,EACpB,aAAa,EAAE,MAAM,EACrB,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC;CAYjB"}

package/dist/repository/genTokens.repository.js

@@ -0,0 +1,15 @@
+import { DBRepository } from "./repository.js";
+export class GenTokensRepository extends DBRepository {
+    async persistGenTokens(sessionId, input_tokens, output_tokens, model) {
+        const { error } = await this.client.rpc("persist_gen_tokens", {
+            p_gen_id: sessionId,
+            p_input_tokens: input_tokens,
+            p_output_tokens: output_tokens,
+            p_model: model,
+        });
+        if (error) {
+            console.error(`Failed calling persist_gen_tokens RPC: ${error.message}`);
+        }
+    }
+}
+//# sourceMappingURL=genTokens.repository.js.map

package/dist/repository/genTokens.repository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"genTokens.repository.js","sourceRoot":"","sources":["../../src/repository/genTokens.repository.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,OAAO,mBAAoB,SAAQ,YAAY;IAC5C,KAAK,CAAC,gBAAgB,CAC3B,SAAiB,EACjB,YAAoB,EACpB,aAAqB,EACrB,KAAa;QAEb,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,oBAAoB,EAAE;YAC5D,QAAQ,EAAE,SAAS;YACnB,cAAc,EAAE,YAAY;YAC5B,eAAe,EAAE,aAAa;YAC9B,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,CAAC,KAAK,CAAC,0CAA0C,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;CACF","sourcesContent":["import { DBRepository } from \"./repository.js\";\r\n\r\nexport class GenTokensRepository extends DBRepository {\r\n  public async persistGenTokens(\r\n    sessionId: string,\r\n    input_tokens: number,\r\n    output_tokens: number,\r\n    model: string,\r\n  ): Promise<void> {\r\n    const { error } = await this.client.rpc(\"persist_gen_tokens\", {\r\n      p_gen_id: sessionId,\r\n      p_input_tokens: input_tokens,\r\n      p_output_tokens: output_tokens,\r\n      p_model: model,\r\n    });\r\n\r\n    if (error) {\r\n      console.error(`Failed calling persist_gen_tokens RPC: ${error.message}`);\r\n    }\r\n  }\r\n}\r\n"]}

package/dist/tests/toolLoopRunner.tokensPersist.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=toolLoopRunner.tokensPersist.test.d.ts.map

package/dist/tests/toolLoopRunner.tokensPersist.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolLoopRunner.tokensPersist.test.d.ts","sourceRoot":"","sources":["../../src/tests/toolLoopRunner.tokensPersist.test.ts"],"names":[],"mappings":""}

package/dist/tests/toolLoopRunner.tokensPersist.test.js

@@ -0,0 +1,52 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+import { FunctionCallingConfigMode } from "@google/genai";
+import { runToolLoop } from "../ai/toolLoop/toolLoopRunner.js";
+test("tool loop: persists gen tokens from usageMetadata", async () => {
+    const persisted = [];
+    let aiCalls = 0;
+    const res = await runToolLoop({
+        initialContents: [],
+        tools: [],
+        workspaceRoot: "/dummy/path",
+        toolCallingMode: FunctionCallingConfigMode.ANY,
+        maxSteps: 2,
+        keepFullTrace: false,
+        logger: async () => { },
+        aiCall: async () => {
+            aiCalls += 1;
+            if (aiCalls === 1) {
+                return {
+                    functionCalls: [{ name: "get_available_routes", args: {} }],
+                    modelVersion: "gemini-2.5-flash-lite",
+                    usageMetadata: { promptTokenCount: 11, candidatesTokenCount: 22 },
+                };
+            }
+            return {
+                functionCalls: [],
+                text: "ok",
+                modelVersion: "gemini-2.5-flash-lite",
+                usageMetadata: { promptTokenCount: 33, candidatesTokenCount: 44 },
+            };
+        },
+        tokenPersistence: {
+            sessionId: "sess_123",
+            repository: {
+                persistGenTokens: async (sessionId, input, output, model) => {
+                    persisted.push({ sessionId, input, output, model });
+                },
+            },
+            model: 'gemini-2.5-flash-lite'
+        },
+    });
+    assert.equal(res.finalText, "ok");
+    assert.deepEqual(persisted, [
+        {
+            sessionId: "sess_123",
+            input: 44,
+            output: 66,
+            model: "gemini-2.5-flash-lite",
+        },
+    ]);
+});
+//# sourceMappingURL=toolLoopRunner.tokensPersist.test.js.map

package/dist/tests/toolLoopRunner.tokensPersist.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolLoopRunner.tokensPersist.test.js","sourceRoot":"","sources":["../../src/tests/toolLoopRunner.tokensPersist.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,yBAAyB,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AAE/D,IAAI,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;IACnE,MAAM,SAAS,GAKV,EAAE,CAAC;IAER,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,MAAM,GAAG,GAAG,MAAM,WAAW,CAAC;QAC5B,eAAe,EAAE,EAAE;QACnB,KAAK,EAAE,EAAE;QACT,aAAa,EAAE,aAAa;QAC5B,eAAe,EAAE,yBAAyB,CAAC,GAAG;QAC9C,QAAQ,EAAE,CAAC;QACX,aAAa,EAAE,KAAK;QACpB,MAAM,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QACtB,MAAM,EAAE,KAAK,IAAI,EAAE;YACjB,OAAO,IAAI,CAAC,CAAC;YACb,IAAI,OAAO,KAAK,CAAC,EAAE,CAAC;gBAClB,OAAO;oBACL,aAAa,EAAE,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;oBAC3D,YAAY,EAAE,uBAAuB;oBACrC,aAAa,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,oBAAoB,EAAE,EAAE,EAAE;iBAClE,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,aAAa,EAAE,EAAE;gBACjB,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,uBAAuB;gBACrC,aAAa,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,oBAAoB,EAAE,EAAE,EAAE;aAClE,CAAC;QACJ,CAAC;QACD,gBAAgB,EAAE;YAChB,SAAS,EAAE,UAAU;YACrB,UAAU,EAAE;gBACV,gBAAgB,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE;oBAC1D,SAAS,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtD,CAAC;aACF;YACD,KAAK,EAAC,uBAAuB;SAC9B;KACF,CAAC,CAAC;IAEH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAClC,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE;QAC1B;YACE,SAAS,EAAE,UAAU;YACrB,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,EAAE;YACV,KAAK,EAAE,uBAAuB;SAC/B;KACF,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport test from \"node:test\";\nimport { FunctionCallingConfigMode } from \"@google/genai\";\nimport { runToolLoop } from \"../ai/toolLoop/toolLoopRunner.js\";\n\ntest(\"tool loop: persists gen tokens from usageMetadata\", async () => {\n  const persisted: Array<{\n    sessionId: string;\n    input: number;\n    output: number;\n    model: string;\n  }> = [];\n\n  let aiCalls = 0;\n  const res = await runToolLoop({\n    initialContents: [],\n    tools: [],\n    workspaceRoot: \"/dummy/path\",\n    toolCallingMode: FunctionCallingConfigMode.ANY,\n    maxSteps: 2,\n    keepFullTrace: false,\n    logger: async () => {},\n    aiCall: async () => {\n      aiCalls += 1;\n      if (aiCalls === 1) {\n        return {\n          functionCalls: [{ name: \"get_available_routes\", args: {} }],\n          modelVersion: \"gemini-2.5-flash-lite\",\n          usageMetadata: { promptTokenCount: 11, candidatesTokenCount: 22 },\n        };\n      }\n      return {\n        functionCalls: [],\n        text: \"ok\",\n        modelVersion: \"gemini-2.5-flash-lite\",\n        usageMetadata: { promptTokenCount: 33, candidatesTokenCount: 44 },\n      };\n    },\n    tokenPersistence: {\n      sessionId: \"sess_123\",\n      repository: {\n        persistGenTokens: async (sessionId, input, output, model) => {\n          persisted.push({ sessionId, input, output, model });\n        },\n      },\n      model:'gemini-2.5-flash-lite'\n    },\n  });\n\n  assert.equal(res.finalText, \"ok\");\n  assert.deepEqual(persisted, [\n    {\n      sessionId: \"sess_123\",\n      input: 44,\n      output: 66,\n      model: \"gemini-2.5-flash-lite\",\n    },\n  ]);\n});\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vedangiitb/qwintly-core",
-  "version": "1.4.9",
+  "version": "1.4.11",
   "description": "Qwintly Core",
   "homepage": "https://github.com/vedangiitb/qwintly-core#readme",
   "bugs": {