- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| run.ts | -
-
- |
- 18.51% | -15/81 | -0% | -0/32 | -0% | -0/5 | -19.73% | -15/76 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 | 1x -1x -1x -1x -1x -1x -1x -1x -1x -1x - -1x -1x - -1x - - - - - -1x - - - - - - -1x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { Args, Command, Flags } from "@oclif/core";
-import { CrackedAgent, CrackedAgentOptions } from "@services/CrackedAgent";
-import { LLMProviderType } from "@services/LLM/LLMProvider";
-import { ModelManager } from "@services/LLM/ModelManager";
-import { OpenRouterAPI } from "@services/LLMProviders/OpenRouter/OpenRouterAPI";
-import { InteractiveSessionManager } from "@services/streaming/InteractiveSessionManager";
-import { StreamHandler } from "@services/streaming/StreamHandler";
-import * as readline from "readline";
-import { container } from "tsyringe";
-import { ConfigService } from "../services/ConfigService";
-
-export class Run extends Command {
- static description = "AI agent for performing operations on local projects";
-
- static examples = [
- "$ run 'Add error handling'",
- "$ run --interactive # Start interactive mode",
- "$ run --init # Initialize configuration",
- ];
-
- static flags = {
- init: Flags.boolean({
- description: "Initialize a default crkdrc.json configuration file",
- exclusive: ["interactive"],
- }),
- };
-
- static args = {
- message: Args.string({
- description: "Message describing the operation to perform",
- required: false,
- }),
- };
-
- private configService: ConfigService;
- private modelManager: ModelManager;
- private streamHandler: StreamHandler;
- private openRouterAPI: OpenRouterAPI;
- private sessionManager: InteractiveSessionManager;
- private rl: readline.Interface;
-
- constructor(argv: string[], config: any) {
- super(argv, config);
- this.configService = container.resolve(ConfigService);
- this.modelManager = container.resolve(ModelManager);
- this.streamHandler = container.resolve(StreamHandler);
- this.openRouterAPI = container.resolve(OpenRouterAPI);
- this.sessionManager = container.resolve(InteractiveSessionManager);
- this.rl = readline.createInterface({
- input: process.stdin,
- output: process.stdout,
- prompt: "> ",
- });
- }
-
- private parseOptions(optionsString: string): Record<string, unknown> {
- const options: Record<string, unknown> = {};
-
- Iif (!optionsString) return options;
-
- const pairs = optionsString.split(",");
- for (const pair of pairs) {
- const [key, value] = pair.trim().split("=");
- Iif (!key || !value) continue;
-
- // Convert value to appropriate type
- if (value === "true") options[key] = true;
- else if (value === "false") options[key] = false;
- else if (!isNaN(Number(value))) {
- if (value.includes(".")) options[key] = parseFloat(value);
- else options[key] = parseInt(value, 10);
- } else options[key] = value;
- }
-
- return options;
- }
-
- async run(): Promise<void> {
- const { args, flags } = await this.parse(Run);
-
- Iif (flags.init) {
- this.configService.createDefaultConfig();
- return;
- }
-
- const config = this.configService.getConfig();
-
- Iif (!config.openRouterApiKey) {
- this.error(
- "OpenRouter API key is required. Please add it to crkdrc.json",
- );
- }
-
- const isInteractive = config.interactive ?? false;
-
- Iif (isInteractive && args.message) {
- this.error("Cannot provide both interactive mode and message argument");
- }
-
- Iif (!isInteractive && !args.message) {
- this.error("Must provide either interactive mode or message argument");
- }
-
- try {
- Iif (!config.discoveryModel) {
- throw new Error("Discovery model is required in configuration");
- }
-
- const options: CrackedAgentOptions = {
- ...config,
- options: this.parseOptions(config.options || ""),
- provider: config.provider as LLMProviderType,
- };
-
- // Validate provider
- Iif (!Object.values(LLMProviderType).includes(options.provider!)) {
- throw new Error(`Invalid provider: ${options.provider}`);
- }
-
- this.modelManager.setCurrentModel(config.discoveryModel);
-
- console.log(
- `Using ${options.provider} provider and model: ${this.modelManager.getCurrentModel()}`,
- );
-
- const agent = container.resolve(CrackedAgent);
- this.sessionManager.initialize(this.rl, agent, options);
-
- if (isInteractive) {
- await this.sessionManager.start();
- } else {
- console.log("Press Enter to start the stream...");
- this.rl.once("line", async () => {
- try {
- const result = await agent.execute(args.message!, options);
- Iif (!options.stream && result) {
- this.log(result.response);
- Iif (result.actions?.length) {
- this.log("\nExecuted Actions:");
- result.actions.forEach(({ action, result }) => {
- this.log(`\nAction: ${action}`);
- this.log(`Result: ${JSON.stringify(result, null, 2)}`);
- });
- }
- }
- this.sessionManager.cleanup();
- process.exit(0);
- } catch (error) {
- this.sessionManager.cleanup();
- this.error((error as Error).message);
- }
- });
- }
- } catch (error) {
- this.sessionManager.cleanup();
- this.error((error as Error).message);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 | 32x - - - - - - - - | export const Colors = {
- reset: "\x1b[0m",
- cyan: "\x1b[36m",
- yellow: "\x1b[33m",
- green: "\x1b[32m",
- magenta: "\x1b[35m",
- blue: "\x1b[34m",
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 | 2x - - | export const DEFAULT_INSTRUCTIONS = - "You're an expert software engineer, master of your craft. Think deeply for each answer, ill tip $200."; - |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| colors.ts | -
-
- |
- 100% | -1/1 | -100% | -0/0 | -100% | -0/0 | -100% | -1/1 | -
| defaultInstructions.ts | -
-
- |
- 100% | -1/1 | -100% | -0/0 | -100% | -0/0 | -100% | -1/1 | -
| modelScaling.ts | -
-
- |
- 100% | -1/1 | -100% | -0/0 | -100% | -0/0 | -100% | -1/1 | -
| openRouterClient.ts | -
-
- |
- 100% | -14/14 | -100% | -2/2 | -100% | -1/1 | -100% | -13/13 | -
| writeConstants.ts | -
-
- |
- 100% | -1/1 | -100% | -0/0 | -100% | -0/0 | -100% | -1/1 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 | 20x - | export const MODEL_SCALING_INITIAL_TRY_COUNT = 2; - |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 | 26x -26x -26x - -26x -26x -26x - -26x - - - - -26x -26x - - -26x -26x - - -26x - - - - - - - - - - -26x - - - | import { ConfigService } from "@services/ConfigService";
-import axios, { AxiosInstance } from "axios";
-import { container } from "tsyringe";
-
-export const createOpenRouterClient = (baseURL: string): AxiosInstance => {
- const configService = container.resolve(ConfigService);
- const config = configService.getConfig();
-
- const headers: Record<string, string> = {
- Authorization: `Bearer ${config.openRouterApiKey}`,
- "Content-Type": "application/json",
- };
-
- if (config.appUrl) {
- headers["HTTP-Referer"] = config.appUrl;
- }
-
- if (config.appName) {
- headers["X-Title"] = config.appName;
- }
-
- return axios.create({
- baseURL,
- // Remove timeout for streaming support
- timeout: 0,
- headers,
- // Add necessary axios config for proper streaming
- maxBodyLength: Infinity,
- maxContentLength: Infinity,
- });
-};
-
-export const openRouterClient = createOpenRouterClient(
- "https://openrouter.ai/api/v1",
-);
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 | 15x - | export const BLOCK_WRITE_IF_CONTENT_REMOVAL_THRESHOLD = 90; - |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| commands | -
-
- |
- 18.51% | -15/81 | -0% | -0/32 | -0% | -0/5 | -19.73% | -15/76 | -
| constants | -
-
- |
- 100% | -18/18 | -100% | -2/2 | -100% | -1/1 | -100% | -17/17 | -
| jest/mocks | -
-
- |
- 100% | -54/54 | -100% | -14/14 | -100% | -12/12 | -100% | -54/54 | -
| services | -
-
- |
- 86.17% | -106/123 | -55.88% | -19/34 | -88.23% | -15/17 | -87.93% | -102/116 | -
| services/FileManagement | -
-
- |
- 85.2% | -265/311 | -60.29% | -41/68 | -90.19% | -46/51 | -85.27% | -249/292 | -
| services/LLM | -
-
- |
- 88.92% | -289/325 | -69.38% | -68/98 | -92.42% | -61/66 | -88.85% | -271/305 | -
| services/LLM/actions | -
-
- |
- 89.03% | -731/821 | -70.86% | -163/230 | -91.47% | -118/129 | -89.43% | -686/767 | -
| services/LLM/actions/blueprints | -
-
- |
- 77.46% | -55/71 | -8.1% | -3/37 | -20% | -4/20 | -77.46% | -55/71 | -
| services/LLM/actions/core | -
-
- |
- 79.27% | -88/111 | -60.46% | -26/43 | -92.59% | -25/27 | -77.88% | -81/104 | -
| services/LLM/actions/types | -
-
- |
- 100% | -6/6 | -100% | -2/2 | -100% | -1/1 | -100% | -6/6 | -
| services/LLM/context | -
-
- |
- 88.94% | -346/389 | -77.86% | -95/122 | -84.61% | -66/78 | -88.46% | -322/364 | -
| services/LLM/phases/blueprints | -
-
- |
- 88.46% | -23/26 | -0% | -0/18 | -0% | -0/3 | -88.46% | -23/26 | -
| services/LLM/types | -
-
- |
- 100% | -4/4 | -100% | -2/2 | -100% | -1/1 | -100% | -4/4 | -
| services/LLM/utils | -
-
- |
- 93.7% | -119/127 | -88.88% | -40/45 | -95.83% | -23/24 | -93.8% | -106/113 | -
| services/LLMProviders/OpenRouter | -
-
- |
- 75.21% | -173/230 | -49.33% | -37/75 | -91.42% | -32/35 | -74.77% | -166/222 | -
| services/logging | -
-
- |
- 89.28% | -25/28 | -75% | -3/4 | -100% | -5/5 | -88% | -22/25 | -
| services/streaming | -
-
- |
- 45.9% | -101/220 | -20% | -13/65 | -40.74% | -11/27 | -45.49% | -96/211 | -
| services/text | -
-
- |
- 96.15% | -25/26 | -84.61% | -11/13 | -100% | -4/4 | -95.65% | -22/23 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 | - - -24x -257x -257x - - - - - -276x - -276x -118x - -276x - -276x - - - - - - - -568x -568x -568x - -568x -568x - -7x - - -561x - - -568x -295x - -568x - -568x - - - - - - - -610x -610x - - - -610x -242x - -610x - -610x - - - - - - - - - - - - - - -11x - - - - - -11x -4x - -11x - -11x - - - - - - - - - - - - - - -114x -114x - -114x - - - - - -114x - - -2x - - -112x - - -114x -20x - - -114x - -114x - - - - -39x - -39x -5x - -39x - -39x - - - -304x -537x -1178x -1178x - - -304x - -304x -306x -1912x -1912x - - - - - | /* eslint-disable @typescript-eslint/ban-ts-comment */
-// UnitTestMocker.ts
-
-export class UnitTestMocker {
- private classMocks: Map<any, jest.SpyInstance<any>[]> = new Map();
- private moduleMocks: Map<any, jest.SpyInstance<any>[]> = new Map();
-
- spyPrototype<T, Return>(
- Class: new (...args: any[]) => T,
- method: keyof T,
- ): jest.SpyInstance<Return> {
- const spy = jest.spyOn(Class.prototype, method as string);
-
- if (!this.moduleMocks.has(Class)) {
- this.moduleMocks.set(Class, []);
- }
- this.moduleMocks.get(Class)?.push(spy);
-
- return spy;
- }
-
- mockPrototype<T, Return>(
- Class: new (...args: any[]) => T,
- method: keyof T,
- returnValue: Return,
- ): jest.SpyInstance<Return> {
- const prototype = Class.prototype;
- const descriptor = Object.getOwnPropertyDescriptor(prototype, method);
- const isAsync = descriptor?.value?.constructor?.name === "AsyncFunction";
-
- const spy = jest.spyOn(prototype, method as string);
- if (isAsync && !(returnValue instanceof Promise)) {
- // If method is async but return value isn't a Promise, wrap it
- spy.mockResolvedValue(returnValue);
- } else {
- // If method is sync or return value is already a Promise, use as is
- spy.mockReturnValue(returnValue);
- }
-
- if (!this.classMocks.has(Class)) {
- this.classMocks.set(Class, []);
- }
- this.classMocks.get(Class)?.push(spy);
-
- return spy;
- }
-
- mockPrototypeWith<T, F extends (...args: any[]) => any>(
- Class: new (...args: any[]) => T,
- method: keyof T,
- implementation: F,
- ): jest.SpyInstance<ReturnType<F>> {
- const prototype = Class.prototype;
- const spy = jest
- .spyOn(prototype, method as string)
- .mockImplementation(implementation);
-
- if (!this.classMocks.has(Class)) {
- this.classMocks.set(Class, []);
- }
- this.classMocks.get(Class)?.push(spy);
-
- return spy;
- }
-
- /**
- * Mocks a module function with a specific implementation.
- * @param module The module object containing the function.
- * @param method The function name to spy on.
- * @param implementation The implementation to use when the function is called.
- * @returns The jest spy instance.
- */
- mockModuleImplementation<T, F extends (...args: any[]) => any>(
- module: T,
- method: keyof T,
- implementation: F,
- ): jest.SpyInstance<ReturnType<F>> {
- const spy = jest
- //@ts-ignore
- .spyOn(module, method as string)
- //@ts-ignore
- .mockImplementation(implementation);
-
- if (!this.moduleMocks.has(module)) {
- this.moduleMocks.set(module, []);
- }
- this.moduleMocks.get(module)?.push(spy);
-
- return spy;
- }
-
- /**
- * Mocks a module function to return a specific value.
- * @param module The module object containing the function.
- * @param method The function name to spy on.
- * @param returnValue The value to return when the function is called.
- * @returns The jest spy instance.
- */
- mockModule<T, Return>(
- module: T,
- method: keyof T,
- returnValue: Return,
- ): jest.SpyInstance<Return> {
- const descriptor = Object.getOwnPropertyDescriptor(module, method);
- const isAsync = descriptor?.value?.constructor?.name === "AsyncFunction";
-
- const spy = jest.spyOn(
- //@ts-ignore
- module,
- method as string,
- ) as jest.SpyInstance<Return>;
-
- if (isAsync && !(returnValue instanceof Promise)) {
- // If method is async but return value isn't a Promise, wrap it
- //@ts-ignore
- spy.mockResolvedValue(returnValue);
- } else {
- // If method is sync or return value is already a Promise, use as is
- spy.mockReturnValue(returnValue);
- }
-
- if (!this.moduleMocks.has(module)) {
- this.moduleMocks.set(module, []);
- }
-
- this.moduleMocks.get(module)?.push(spy);
-
- return spy;
- }
-
- spyModule<T>(module: T, method: keyof T): jest.SpyInstance<any> {
- //@ts-ignore
- const spy = jest.spyOn(module, method as string);
-
- if (!this.moduleMocks.has(module)) {
- this.moduleMocks.set(module, []);
- }
- this.moduleMocks.get(module)?.push(spy);
-
- return spy;
- }
-
- clearAllMocks(): void {
- this.classMocks.forEach((spies) => {
- spies.forEach((spy) => {
- spy.mockRestore();
- spy.mockClear();
- });
- });
- this.classMocks.clear();
-
- this.moduleMocks.forEach((spies) => {
- spies.forEach((spy) => {
- spy.mockRestore();
- spy.mockClear();
- });
- });
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| UnitTestMocker.ts | -
-
- |
- 100% | -54/54 | -100% | -14/14 | -100% | -12/12 | -100% | -54/54 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 | 30x -30x -30x -30x -30x - -30x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -30x -531x -531x - - -2x - - - -2x - -2x - - - -2x - - - - -3x -2x - -2x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -2x - - - -2x - - -2x - - - - - -2x - -2x - - - - - - -270x -1x - - -270x -270x - -270x - -270x -2x - - - -2x - - -268x - - - | import chalk from "chalk";
-import * as fs from "fs";
-import * as path from "path";
-import { autoInjectable } from "tsyringe";
-import { z } from "zod";
-
-const configSchema = z.object({
- provider: z.string(),
- customInstructions: z.string().optional(),
- customInstructionsPath: z.string().optional(),
- interactive: z.boolean(),
- stream: z.boolean(),
- debug: z.boolean(),
- options: z.string(),
- openRouterApiKey: z.string(),
- appUrl: z.string().optional().default("https://localhost:8080"),
- appName: z.string().optional().default("MyApp"),
- autoScaler: z.boolean().optional(),
- autoScaleMaxTryPerModel: z.number().optional(),
- includeAllFilesOnEnvToContext: z.boolean().optional(),
- // Phase-specific model configurations
- discoveryModel: z.string().optional().default("google/gemini-flash-1.5-8b"),
- strategyModel: z.string().optional().default("qwen/qwq-32b-preview"),
- executeModel: z
- .string()
- .optional()
- .default("anthropic/claude-3.5-sonnet:beta"),
- autoScaleAvailableModels: z.array(
- z.object({
- id: z.string(),
- description: z.string(),
- maxWriteTries: z.number(),
- maxGlobalTries: z.number(),
- }),
- ),
- runAllTestsCmd: z.string().optional(),
- runOneTestCmd: z.string().optional(),
- runTypeCheckCmd: z.string().optional(),
- enableConversationLog: z.boolean().optional(),
- logDirectory: z.string().optional(),
- directoryScanner: z
- .object({
- defaultIgnore: z
- .array(z.string())
- .default(["dist", "coverage", ".next", "build", ".cache", ".husky"]),
- maxDepth: z.number().default(8),
- allFiles: z.boolean().default(true),
- directoryFirst: z.boolean().default(true),
- excludeDirectories: z.boolean().default(false),
- })
- .default({
- defaultIgnore: ["dist", "coverage", ".next", "build", ".cache", ".husky"],
- maxDepth: 8,
- allFiles: true,
- directoryFirst: true,
- excludeDirectories: false,
- }),
-});
-
-export type Config = z.infer<typeof configSchema>;
-
-@autoInjectable()
-export class ConfigService {
- private readonly CONFIG_PATH = path.resolve("crkdrc.json");
- private readonly GITIGNORE_PATH = path.resolve(".gitignore");
-
- private ensureGitIgnore(): void {
- const gitignoreContent = fs.existsSync(this.GITIGNORE_PATH)
- ? fs.readFileSync(this.GITIGNORE_PATH, "utf-8")
- : "";
-
- if (!gitignoreContent.includes("crkdrc.json")) {
- const updatedContent =
- gitignoreContent.endsWith("\n") || gitignoreContent === ""
- ? `${gitignoreContent}crkdrc.json\n`
- : `${gitignoreContent}\ncrkdrc.json\n`;
-
- fs.writeFileSync(this.GITIGNORE_PATH, updatedContent);
- }
- }
-
- public createDefaultConfig(): void {
- if (!fs.existsSync(this.CONFIG_PATH)) {
- console.log("Creating default crkdrc.json configuration...");
-
- const defaultConfig = {
- provider: "open-router",
- customInstructions: "Follow clean code principles",
- customInstructionsPath: "",
- interactive: true,
- stream: true,
- debug: false,
- options:
- "temperature=0,top_p=0.1,top_k=1,frequence_penalty=0.0,presence_penalty=0.0,repetition_penalty=1.0",
- openRouterApiKey: "",
- appUrl: "https://localhost:8080",
- appName: "MyCrackedApp",
- autoScaler: true,
- autoScaleMaxTryPerModel: 2,
- // Phase-specific model configurations
- discoveryModel: "google/gemini-flash-1.5-8b",
- strategyModel: "qwen/qwq-32b-preview",
- executeModel: "anthropic/claude-3.5-sonnet:beta",
- includeAllFilesOnEnvToContext: false,
- autoScaleAvailableModels: [
- {
- id: "qwen/qwen-2.5-coder-32b-instruct",
- description: "Cheap, fast, slightly better than GPT4o-mini",
- maxWriteTries: 5,
- maxGlobalTries: 10,
- },
- {
- id: "anthropic/claude-3.5-sonnet:beta",
- description: "Scaled model for retry attempts",
- maxWriteTries: 5,
- maxGlobalTries: 15,
- },
- {
- id: "openai/gpt-4o-2024-11-20",
- description: "Scaled model for retry attempts",
- maxWriteTries: 2,
- maxGlobalTries: 20,
- },
- ],
- runAllTestsCmd: "yarn test",
- runOneTestCmd: "yarn test {relativeTestPath}",
- runTypeCheckCmd: "yarn typecheck",
- enableConversationLog: false,
- logDirectory: "logs",
- directoryScanner: {
- defaultIgnore: [
- "dist",
- "coverage",
- ".next",
- "build",
- ".cache",
- ".husky",
- ],
- maxDepth: 8,
- allFiles: true,
- directoryFirst: true,
- excludeDirectories: false,
- },
- };
- fs.writeFileSync(
- this.CONFIG_PATH,
- JSON.stringify(defaultConfig, null, 4),
- );
- console.log(
- "Default crkdrc.json configuration created. Please adjust it.",
- );
- console.log(
- chalk.yellow(
- "Warning: No OpenRouter API key provided. Please add it to crkdrc.json.",
- ),
- );
-
- this.ensureGitIgnore();
-
- chalk.green(
- "CrackedDevCLI config generated. Please, add Provider and API Key to crkdrc.json.",
- );
- }
- }
-
- public getConfig(): Config {
- if (!fs.existsSync(this.CONFIG_PATH)) {
- this.createDefaultConfig();
- }
-
- const rawData = fs.readFileSync(this.CONFIG_PATH, "utf-8");
- const config = JSON.parse(rawData);
-
- const parsedConfig = configSchema.safeParse(config);
-
- if (!parsedConfig.success) {
- console.error(
- "Invalid configuration in crkdrc.json:",
- parsedConfig.error,
- );
- throw new Error("Invalid configuration in crkdrc.json");
- }
-
- return parsedConfig.data;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 | 2x -2x -2x - -2x -2x -2x -2x -2x -2x -2x - - - - - - - - - - - - - - - - - - - - -2x - -21x -21x - - -21x -21x -21x -21x -21x -21x -21x - - - - - - -18x - -18x -18x - -18x - - - - - - -18x -18x - - -18x - - - - -18x -1x - - - - - - - -17x - - - - - - -17x - - - -18x - - - - - - - - - - -18x -18x -18x -18x - -18x -1x - - -18x -18x - -18x - - - -18x -18x - - - - - - - - -18x - -18x -18x -1x - - - - -18x -2x - - -2x - - -18x -16x - - - - - - - - - -1x -1x - - - -1x -1x -1x - - - -1x - -1x - - -1x - - - - - - -1x - - - - - - - - - - - -17x - -17x - - - - -17x - - -17x - -17x - - - - - - - - - - - - - - -18x -18x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -18x - - - - - - -1x - - - -2x -2x - - - | import { DEFAULT_INSTRUCTIONS } from "@constants/defaultInstructions";
-import { FileReader } from "@services/FileManagement/FileReader";
-import { ActionsParser } from "@services/LLM/actions/ActionsParser";
-import { ILLMProvider } from "@services/LLM/ILLMProvider";
-import { LLMContextCreator } from "@services/LLM/LLMContextCreator";
-import { LLMProvider, LLMProviderType } from "@services/LLM/LLMProvider";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { StreamHandler } from "@services/streaming/StreamHandler";
-import { autoInjectable, singleton } from "tsyringe";
-import { ModelManager } from "./LLM/ModelManager";
-import { PhaseManager } from "./LLM/PhaseManager";
-
-export interface CrackedAgentOptions {
- root?: string;
- instructionsPath?: string;
- instructions?: string;
- provider?: LLMProviderType;
- stream?: boolean;
- debug?: boolean;
- options?: Record<string, unknown>;
- clearContext?: boolean;
- autoScaler?: boolean;
-}
-
-export interface ExecutionResult {
- response: string;
- actions?: Array<{ action: string; result: any }>;
-}
-
-@autoInjectable()
-@singleton()
-export class CrackedAgent {
- private llm!: ILLMProvider;
- private isFirstInteraction: boolean = true;
- private currentModel: string = "";
-
- constructor(
- private fileReader: FileReader,
- private contextCreator: LLMContextCreator,
- private debugLogger: DebugLogger,
- private actionsParser: ActionsParser,
- private streamHandler: StreamHandler,
- private phaseManager: PhaseManager,
- private modelManager: ModelManager,
- ) {}
-
- async execute(
- message: string,
- options: CrackedAgentOptions,
- ): Promise<ExecutionResult> {
- this.phaseManager.initializePhaseConfigs();
-
- const finalOptions = await this.setupExecution(options);
- this.currentModel = this.modelManager.getCurrentModel();
-
- const formattedMessage = await this.contextCreator.create(
- message,
- finalOptions.root,
- this.isFirstInteraction,
- );
-
- // Update isFirstInteraction before returning
- if (this.isFirstInteraction) {
- this.isFirstInteraction = false;
- }
-
- this.debugLogger.log("Message", "Sending message to LLM", {
- message: formattedMessage,
- conversationHistory: this.llm.getConversationContext(),
- });
-
- if (finalOptions.stream) {
- return await this.handleStreamExecution(
- formattedMessage,
- this.currentModel,
- finalOptions.options,
- finalOptions.stream,
- );
- }
-
- const result = await this.handleNormalExecution(
- formattedMessage,
- this.currentModel,
- finalOptions.options,
- finalOptions.stream,
- );
-
- return result;
- }
-
- private async setupExecution(options: CrackedAgentOptions) {
- const finalOptions = {
- root: process.cwd(),
- provider: LLMProviderType.OpenRouter,
- stream: false,
- debug: false,
- options: {},
- clearContext: false,
- autoScaler: false,
- ...options,
- };
-
- this.debugLogger.setDebug(finalOptions.debug);
- this.llm = LLMProvider.getInstance(finalOptions.provider);
- this.streamHandler.reset();
- this.actionsParser.reset();
-
- if (finalOptions.clearContext) {
- this.clearConversationHistory();
- }
-
- await this.validateModel(this.modelManager.getCurrentModel());
- await this.setupInstructions(finalOptions);
-
- return finalOptions;
- }
-
- private async validateModel(model: string) {
- const isValidModel = await this.llm.validateModel(model);
- Iif (!isValidModel) {
- const availableModels = await this.llm.getAvailableModels();
- throw new Error(
- `Invalid model: ${model}. Available models: ${availableModels.join(", ")}`,
- );
- }
- }
-
- private async setupInstructions(options: CrackedAgentOptions) {
- Iif (!this.isFirstInteraction) return;
-
- let instructions = options.instructions;
- if (options.instructionsPath) {
- instructions = await this.fileReader.readInstructionsFile(
- options.instructionsPath,
- );
- }
-
- if (instructions) {
- this.debugLogger.log("Instructions", "Adding system instructions", {
- instructions,
- });
- this.llm.addSystemInstructions(instructions);
- }
-
- if (!instructions) {
- this.llm.addSystemInstructions(DEFAULT_INSTRUCTIONS);
- }
- }
-
- private async handleStreamExecution(
- message: string,
- model: string,
- options?: Record<string, unknown>,
- stream?: boolean,
- ): Promise<ExecutionResult> {
- let response = "";
- await this.llm.streamMessage(
- model,
- message,
- async (chunk: string) => {
- response += chunk;
- this.actionsParser.appendToBuffer(chunk);
- process.stdout.write(chunk);
- },
- options,
- );
- process.stdout.write("\n");
-
- Iif (!response) return { response: "" };
-
- const { actions = [], followupResponse } =
- await this.parseAndExecuteWithCallback(
- this.actionsParser.buffer,
- model,
- options,
- stream,
- );
-
- return {
- response: followupResponse || response,
- actions,
- };
- }
-
- private async handleNormalExecution(
- message: string,
- model: string,
- options?: Record<string, unknown>,
- stream?: boolean,
- ): Promise<ExecutionResult> {
- const response = await this.llm.sendMessage(model, message, options);
-
- this.debugLogger.log("Response", "Received LLM response", {
- response,
- conversationHistory: this.llm.getConversationContext(),
- });
-
- Iif (!response) return { response: "" };
-
- const { actions = [], followupResponse } =
- await this.parseAndExecuteWithCallback(response, model, options, stream);
-
- return {
- response: followupResponse || response,
- actions,
- };
- }
-
- private async parseAndExecuteWithCallback(
- response: string,
- model: string,
- options?: Record<string, unknown>,
- stream?: boolean,
- ): Promise<{
- actions: Array<{ action: string; result: any }>;
- followupResponse?: string;
- }> {
- this.actionsParser.reset();
- const result = await this.actionsParser.parseAndExecuteActions(
- response,
- model,
- async (followupMsg: string) => {
- const formattedFollowup = await this.contextCreator.create(
- followupMsg,
- process.cwd(),
- false,
- );
-
- if (stream) {
- let followupResponse = "";
- await this.llm.streamMessage(
- this.currentModel,
- formattedFollowup,
- async (chunk: string) => {
- followupResponse += chunk;
- process.stdout.write(chunk);
- },
- options,
- );
- process.stdout.write("\n");
-
- const followupResult = await this.parseAndExecuteWithCallback(
- followupResponse,
- this.currentModel,
- options,
- stream,
- );
-
- return followupResult.followupResponse || followupResponse;
- } else {
- const followupResponse = await this.llm.sendMessage(
- this.currentModel,
- formattedFollowup,
- options,
- );
-
- const followupResult = await this.parseAndExecuteWithCallback(
- followupResponse,
- this.currentModel,
- options,
- stream,
- );
-
- return followupResult.followupResponse || followupResponse;
- }
- },
- );
-
- return {
- actions: result?.actions || [],
- followupResponse: result?.followupResponse,
- };
- }
-
- getConversationHistory() {
- return this.llm.getConversationContext();
- }
-
- clearConversationHistory() {
- this.llm.clearConversationContext();
- this.isFirstInteraction = true;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 | 6x - - - - - -6x -6x -6x - - -6x -100x - -100x -100x - - - - - -2x -2x - - - - - - - - - - - - - - - - - - -191x - -191x - -191x -796x - -789x -789x - -789x -189x - - - - - - - - -600x - - - -191x - - - - - - -2x -2x -2x -2x - - - -2x - -2x - - - - - - - - -2x - -600x - - - - - - - - - - | import { ConfigService } from "@services/ConfigService";
-import {
- IDirectoryScanner,
- TreeOptions,
-} from "@services/FileManagement/types/DirectoryScannerTypes";
-import { IFileOperationResult } from "@services/FileManagement/types/FileManagementTypes";
-import fs from "fs";
-import path from "path";
-import { autoInjectable, inject } from "tsyringe";
-
-@autoInjectable()
-export class DirectoryScanner implements IDirectoryScanner {
- private readonly REQUIRED_IGNORE = ["node_modules", ".git"];
-
- constructor(@inject(ConfigService) private configService: ConfigService) {
- Iif (!configService) {
- throw new Error("ConfigService is required for DirectoryScanner");
- }
- }
-
- private get defaultOptions(): TreeOptions {
- const config = this.configService.getConfig();
- return {
- ignore: config.directoryScanner.defaultIgnore,
- allFiles: config.directoryScanner.allFiles,
- maxDepth: config.directoryScanner.maxDepth,
- noreport: true,
- base: ".",
- directoryFirst: config.directoryScanner.directoryFirst,
- excludeDirectories: config.directoryScanner.excludeDirectories,
- };
- }
-
- private getAllFiles(
- dirPath: string,
- basePath: string,
- arrayOfFiles: string[] = [],
- ignore: string[] = [],
- currentDepth: number = 0,
- maxDepth: number = 4,
- ): string[] {
- Iif (currentDepth > maxDepth) return arrayOfFiles;
-
- const files = fs.readdirSync(dirPath);
-
- files.forEach((file) => {
- if (ignore.includes(file)) return;
-
- const fullPath = path.join(dirPath, file);
- const relativePath = path.relative(basePath, fullPath);
-
- if (fs.statSync(fullPath).isDirectory()) {
- this.getAllFiles(
- fullPath,
- basePath,
- arrayOfFiles,
- ignore,
- currentDepth + 1,
- maxDepth,
- );
- } else {
- arrayOfFiles.push(relativePath);
- }
- });
-
- return arrayOfFiles;
- }
-
- public async scan(
- dirPath: string,
- options: Partial<TreeOptions> = {},
- ): Promise<IFileOperationResult> {
- try {
- const defaultOptions = this.defaultOptions;
- const scanOptions = { ...defaultOptions, ...options };
- const ignore = [
- ...this.REQUIRED_IGNORE,
- ...(options.ignore || defaultOptions.ignore),
- ];
- const absolutePath = path.resolve(dirPath);
-
- const files = this.getAllFiles(
- absolutePath,
- absolutePath,
- [],
- ignore,
- 0,
- scanOptions.maxDepth,
- ).sort();
-
- return {
- success: true,
- data: files.map((f) => f.trim()).join("\n"),
- };
- } catch (error) {
- return {
- success: false,
- error: error instanceof Error ? error : new Error(String(error)),
- };
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 | 3x - - - - - - -3x - - - - - - -3x - - - - - | export class FileNotFoundError extends Error {
- constructor(path: string) {
- super(`File not found: ${path}`);
- this.name = "FileNotFoundError";
- }
- }
-
- export class FileReadError extends Error {
- constructor(path: string, message: string) {
- super(`Failed to read file: ${path} - ${message}`);
- this.name = "FileReadError";
- }
- }
-
- export class InvalidFileError extends Error {
- constructor(path: string) {
- super(`Instructions path must be a file: ${path}`);
- this.name = "InvalidFileError";
- }
- } |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 | 17x -17x - - - - - -17x -17x -17x -17x - - -17x - -14x -14x -14x - - - -11x - - -11x -1x -1x -502x - -502x -1x - -501x - - -1x - - - - -10x -10x - - - - - - - - - - - -11x - - -10x -9x - - - -1x - - - -1x - - - - - - - - - -1x -1x - - - - - - - -1x - - - -2x -2x -1x - - - - - -1x -1x - -1x - - - - -3x -3x -1x - - -2x -2x - -2x -4x -4x -4x -1x -1x - -3x -3x -3x - - - - - - - - -2x -1x - - - - - - - -1x - - - - - - - - - -1x -1x -1x -1x - - - - - - -1x -1x -1x - - - - - -1x -1x - - - - - - - - - -1x -1x -1x - - - - - -1x -1x -1x - - - - - - - - - -1x -1x -1x - - - - - -1x -1x -1x - - - - - - -1x -1x - - - -1x -1x -1x - - - - - -1x -1x - - - - - - -1x - - - - - - | import { FileSearch } from "@services/FileManagement/FileSearch";
-import { PathAdjuster } from "@services/FileManagement/PathAdjuster";
-import {
- IFileOperationResult,
- IFileOperations,
- IFileStats,
-} from "@services/FileManagement/types/FileManagementTypes";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import fs from "fs-extra";
-import path from "path";
-import { autoInjectable } from "tsyringe";
-
-@autoInjectable()
-export class FileOperations implements IFileOperations {
- constructor(
- private pathAdjuster: PathAdjuster,
- private fileSearch: FileSearch,
- private debugLogger: DebugLogger,
- ) {}
-
- private async ensureInitialized(timeout: number = 5000): Promise<void> {
- const startTime = Date.now();
-
- // Wait for initialization if not already initialized
- if (!this.pathAdjuster.isInitialized()) {
- await new Promise<void>((resolve, reject) => {
- const checkInit = () => {
- Iif (this.pathAdjuster.isInitialized()) {
- resolve();
- } else if (Date.now() - startTime > timeout) {
- reject(new Error("PathAdjuster initialization timed out"));
- } else {
- setTimeout(checkInit, 10);
- }
- };
- checkInit();
- });
- }
-
- // Check for initialization errors
- const error = this.pathAdjuster.getInitializationError();
- Iif (error) {
- throw error;
- }
- }
-
- async getAdjustedPath(filePath: string): Promise<string> {
- await this.ensureInitialized();
- const adjustedPath = await this.pathAdjuster.adjustPath(filePath);
- return adjustedPath || filePath;
- }
-
- private async adjustPath(filePath: string): Promise<string> {
- await this.ensureInitialized();
-
- // If path exists, return as is
- if (await fs.pathExists(filePath)) {
- return filePath;
- }
-
- // For read operations, try to find similar files
- const similarFiles = await this.fileSearch.findByName(
- path.basename(filePath),
- process.cwd(),
- );
- Iif (similarFiles.length > 0) {
- const bestMatch = similarFiles[0];
- this.debugLogger.log(
- "FileOperations",
- `Found similar file: ${bestMatch} for ${filePath}`,
- );
- return bestMatch;
- }
-
- // If no similar files found, try PathAdjuster
- const adjustedPath = await this.pathAdjuster.adjustPath(filePath);
- Iif (adjustedPath && (await fs.pathExists(adjustedPath))) {
- this.debugLogger.log(
- "FileOperations > PathAdjuster",
- `Adjusted path: ${adjustedPath}`,
- );
- return adjustedPath;
- }
-
- return filePath;
- }
-
- async read(filePath: string): Promise<IFileOperationResult> {
- try {
- const adjustedPath = await this.adjustPath(filePath);
- Iif (!(await fs.pathExists(adjustedPath))) {
- return {
- success: false,
- error: new Error(`File does not exist: ${filePath}`),
- };
- }
- const content = await fs.readFile(adjustedPath, "utf-8");
- return { success: true, data: content };
- } catch (error) {
- return { success: false, error: error as Error };
- }
- }
-
- async readMultiple(filePaths: string[]): Promise<IFileOperationResult> {
- try {
- if (!filePaths?.length) {
- return { success: false, error: new Error("No files provided") };
- }
-
- const fileContents: string[] = [];
- const errors: string[] = [];
-
- for (const filePath of filePaths) {
- try {
- const adjustedPath = await this.adjustPath(filePath);
- if (!(await fs.pathExists(adjustedPath))) {
- errors.push(`${filePath}: File does not exist`);
- continue;
- }
- const content = await fs.readFile(adjustedPath, "utf-8");
- if (content) {
- fileContents.push(`[File: ${adjustedPath}]\n${content}`);
- } else E{
- errors.push(`${adjustedPath}: Empty content`);
- }
- } catch (error) {
- errors.push(`${filePath}: ${(error as Error).message}`);
- }
- }
-
- if (errors.length > 0) {
- return {
- success: false,
- error: new Error(
- `Failed to read files: ${errors.join(", ")}. Try using search_file action to find the proper path.`,
- ),
- };
- }
-
- return { success: true, data: fileContents.join("\n\n") };
- } catch (error) {
- return { success: false, error: error as Error };
- }
- }
-
- async write(
- filePath: string,
- content: string | Buffer,
- ): Promise<IFileOperationResult> {
- try {
- await fs.ensureDir(path.dirname(filePath));
- await fs.writeFile(filePath, content);
- return { success: true };
- } catch (error) {
- return { success: false, error: error as Error };
- }
- }
-
- async delete(filePath: string): Promise<IFileOperationResult> {
- try {
- const adjustedPath = await this.adjustPath(filePath);
- Iif (!(await fs.pathExists(adjustedPath))) {
- return {
- success: false,
- error: new Error(`File does not exist: ${filePath}`),
- };
- }
- await fs.remove(adjustedPath);
- return { success: true };
- } catch (error) {
- return { success: false, error: error as Error };
- }
- }
-
- async copy(
- source: string,
- destination: string,
- ): Promise<IFileOperationResult> {
- try {
- const adjustedSource = await this.adjustPath(source);
- Iif (!(await fs.pathExists(adjustedSource))) {
- return {
- success: false,
- error: new Error(`Source file does not exist: ${source}`),
- };
- }
- await fs.ensureDir(path.dirname(destination));
- await fs.copy(adjustedSource, destination);
- return { success: true };
- } catch (error) {
- return { success: false, error: error as Error };
- }
- }
-
- async move(
- source: string,
- destination: string,
- ): Promise<IFileOperationResult> {
- try {
- const adjustedSource = await this.adjustPath(source);
- Iif (!(await fs.pathExists(adjustedSource))) {
- return {
- success: false,
- error: new Error(`Source file does not exist: ${source}`),
- };
- }
- await fs.ensureDir(path.dirname(destination));
- await fs.move(adjustedSource, destination, { overwrite: true });
- return { success: true };
- } catch (error) {
- return { success: false, error: error as Error };
- }
- }
-
- async exists(filePath: string): Promise<boolean> {
- const adjustedPath = await this.adjustPath(filePath);
- return fs.pathExists(adjustedPath);
- }
-
- async stats(filePath: string): Promise<IFileOperationResult> {
- try {
- const adjustedPath = await this.adjustPath(filePath);
- Iif (!(await fs.pathExists(adjustedPath))) {
- return {
- success: false,
- error: new Error(`File does not exist: ${filePath}`),
- };
- }
- const stats = await fs.stat(adjustedPath);
- const fileStats: IFileStats = {
- size: stats.size,
- createdAt: stats.birthtime,
- modifiedAt: stats.mtime,
- isDirectory: stats.isDirectory(),
- path: adjustedPath,
- };
- return { success: true, data: fileStats };
- } catch (error) {
- return { success: false, error: error as Error };
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 | 3x - - - - -3x -3x - - -3x - -2x -2x -1x - -1x - - - - - - -1x - - - - -2x -1x - - - - - -1x - - - | import {
- FileNotFoundError,
- FileReadError,
- InvalidFileError,
-} from "@services/FileManagement/Errors";
-import { readFile, stat } from "fs/promises";
-import { autoInjectable } from "tsyringe";
-
-@autoInjectable()
-export class FileReader {
- public async readInstructionsFile(filePath: string): Promise<string> {
- try {
- await this.validateFilePath(filePath);
- return await this.readFileContent(filePath);
- } catch (error) {
- Iif (
- error instanceof FileNotFoundError ||
- error instanceof FileReadError ||
- error instanceof InvalidFileError
- ) {
- throw new FileReadError(filePath, error.message);
- }
- throw error;
- }
- }
-
- private async validateFilePath(filePath: string): Promise<void> {
- const stats = await stat(filePath);
- Iif (!stats.isFile()) {
- throw new Error(`Provided path ${filePath} is not a file`);
- }
- }
-
- private async readFileContent(filePath: string): Promise<string> {
- return readFile(filePath, "utf-8");
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 | - - - -19x -19x -19x -19x -19x -19x - - - - - - - - - - - - - -19x -26x - - -48x -48x -48x -5242796x -5242796x - -48x - - - - - - -2x -2x - - - - - -2x - -2x -6x -6x - -6x - -6x - - - - - - -2x - - - - - - - - - - -10x -10x - - - - - -10x -10x - -10x -48x -48x -48x - -48x -48x -48x - -48x -48x -48x - -48x - -48x -5242796x - - - - - - - -48x -12x - - - - - - - - - -10x - - - - - - - -5x -5x -5x - -5x - - - - - - - -5x - - - - - - - -18x - - - - - - -5x -18x - - -5x - -5x - -1x - - - - - -1x -1x - -1x - - - - -1x - - - -4x - - - - - - - - -4x -4x - -4x -3x -3x - - - - - - - | import {
- IFileSearch,
- IFileSearchResult,
-} from "@services/FileManagement/types/FileManagementTypes";
-import fg from "fast-glob";
-import fs from "fs-extra";
-import Fuse from "fuse.js";
-import path from "path";
-import { autoInjectable } from "tsyringe";
-import { DebugLogger } from "../logging/DebugLogger";
-
-interface FileEntry {
- fullPath: string;
- name: string;
- dir: string;
-}
-
-interface FuseResult {
- item: FileEntry;
- score?: number;
-}
-
-@autoInjectable()
-export class FileSearch implements IFileSearch {
- constructor(private debugLogger: DebugLogger) {}
-
- private findAllMatches(text: string, searchStr: string): number[] {
- const positions: number[] = [];
- let pos = text.indexOf(searchStr);
- while (pos !== -1) {
- positions.push(pos);
- pos = text.indexOf(searchStr, pos + 1);
- }
- return positions;
- }
-
- async findByPattern(
- pattern: string,
- directory: string,
- ): Promise<IFileSearchResult[]> {
- try {
- const entries = await fg(pattern, {
- cwd: directory,
- dot: true,
- absolute: true,
- });
-
- const results: IFileSearchResult[] = [];
-
- for (const entry of entries) {
- const content = await fs.readFile(entry, "utf-8");
- const lines = content.split("\n");
-
- results.push({
- path: entry,
- matches: lines.map((line, index) => ({
- line: index + 1,
- content: line,
- })),
- });
- }
-
- return results;
- } catch (error) {
- console.error("Error in findByPattern:", error);
- return [];
- }
- }
-
- async findByContent(
- searchContent: string,
- directory: string,
- ): Promise<IFileSearchResult[]> {
- try {
- const entries = await fg("**/*", {
- cwd: directory,
- dot: true,
- absolute: true,
- });
-
- const results: IFileSearchResult[] = [];
- const searchTarget = searchContent.toLowerCase();
-
- for (const entry of entries) {
- try {
- const stats = await fs.stat(entry);
- Iif (!stats.isFile()) continue;
-
- const content = await fs.readFile(entry, "utf-8");
- const lines = content.split("\n");
- const matches = [];
-
- for (let i = 0; i < lines.length; i++) {
- const line = lines[i];
- const lineToSearch = line.toLowerCase();
-
- const positions = this.findAllMatches(lineToSearch, searchTarget);
-
- for (const pos of positions) {
- matches.push({
- line: i + 1,
- content: line,
- position: pos,
- });
- }
- }
-
- if (matches.length > 0) {
- results.push({
- path: entry,
- matches,
- });
- }
- } catch (error) {
- console.error(`Error processing file ${entry}:`, error);
- }
- }
-
- return results;
- } catch (error) {
- console.error("Error in findByContent:", error);
- return [];
- }
- }
-
- async findByName(name: string, directory: string): Promise<string[]> {
- try {
- const targetName = path.basename(name);
- const targetDir = path.dirname(name);
-
- this.debugLogger.log("FileSearch", "findByName input", {
- name,
- targetName,
- targetDir,
- directory,
- });
-
- // Get all files in directory
- const entries = await fg("**/*", {
- cwd: directory,
- dot: true,
- absolute: true,
- onlyFiles: true,
- });
-
- // Convert to FileEntry objects
- const fileEntries: FileEntry[] = entries.map((entry) => ({
- fullPath: entry,
- name: path.basename(entry),
- dir: path.dirname(entry),
- }));
-
- // First try exact name matches
- const exactMatches = fileEntries.filter(
- (entry) => entry.name === targetName,
- );
-
- this.debugLogger.log("FileSearch", "exact matches", exactMatches);
-
- if (exactMatches.length > 0) {
- // If we have exact matches, sort by directory similarity
- const fuse = new Fuse<FileEntry>(exactMatches, {
- includeScore: true,
- threshold: 0.3,
- keys: ["dir"],
- });
-
- const results = fuse.search(targetDir);
- this.debugLogger.log("FileSearch", "dir similarity results", results);
-
- Iif (results.length > 0) {
- return results.map((result: FuseResult) => result.item.fullPath);
- }
-
- // If no directory matches, return all exact name matches
- return exactMatches.map((entry) => entry.fullPath);
- }
-
- // If no exact matches, try fuzzy name matching
- const fuse = new Fuse<FileEntry>(fileEntries, {
- includeScore: true,
- threshold: 0.3,
- keys: [
- { name: "name", weight: 3 }, // Filename is most important
- { name: "dir", weight: 1 }, // Directory path has less weight
- ],
- });
-
- const results = fuse.search(targetName);
- this.debugLogger.log("FileSearch", "fuzzy search results", results);
-
- return results
- .filter((result: FuseResult) => result.score && result.score < 0.3)
- .map((result: FuseResult) => result.item.fullPath);
- } catch (error) {
- console.error("Error in findByName:", error);
- return [];
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 | 19x -19x -19x -19x -19x - - - - - - - - -19x -42x - -42x -42x -42x - - -42x - - - - -42x - -42x -1x -1x - - - - - - - - -2x - - - -2x - - - -42x -42x -41x -41x -41x - -1x -1x - -1x - - - - -44x -44x - - - - - -834820x - -2x - - - - - - - - - -6x -1x - - - - -5x - -5x - -5x -3x -3x -3x - - - -2x - - - -6x -6x -6x - -5x -5x - - - - - - - - - -3x - - - -3x -3x -2x - - -1x - - - -3x -1x -1x -1x - - - -2x - -2x -1x - - -1x - - - -2x -2x -1x - -1x -1x - - - | import fg from "fast-glob";
-import fsExtra from "fs-extra";
-import * as Fuse from "fuse.js";
-import path from "path";
-import { autoInjectable } from "tsyringe";
-
-/**
- * PathAdjuster Class
- *
- * This class is responsible for adjusting incorrect file paths by finding the closest matching existing path.
- * It uses fast-glob for efficient file discovery, Fuse.js for fuzzy matching, and fs-extra for file system operations.
- */
-@autoInjectable()
-export class PathAdjuster {
- private allFiles: string[] = [];
- private fuse: Fuse.default<string>;
- private initialized: boolean = false;
- private initializationError: Error | null = null;
- private baseDirectory: string = process.cwd();
-
- constructor() {
- const defaultFuzzyOptions: Fuse.IFuseOptions<string> = {
- includeScore: true,
- threshold: 0.4,
- };
-
- this.fuse = new Fuse.default([], defaultFuzzyOptions);
-
- this.initialize().catch((error) => {
- this.initializationError = error;
- console.error("Failed to initialize PathAdjuster:", error);
- });
- }
-
- public async cleanup(): Promise<void> {
- // No cleanup needed
- }
-
- public isInitialized(): boolean {
- return this.initialized;
- }
-
- public getInitializationError(): Error | null {
- return this.initializationError;
- }
-
- private async initialize(): Promise<void> {
- try {
- this.allFiles = await this.getAllFiles(this.baseDirectory);
- this.fuse.setCollection(this.allFiles);
- this.initialized = true;
- this.initializationError = null;
- } catch (error) {
- this.initialized = false;
- this.initializationError =
- error instanceof Error ? error : new Error(String(error));
- throw this.initializationError;
- }
- }
-
- private async getAllFiles(dir: string): Promise<string[]> {
- try {
- const entries = await fg.sync("**/*", {
- cwd: dir,
- absolute: true,
- onlyFiles: true,
- followSymbolicLinks: true,
- });
- return entries.map((filePath) => path.resolve(filePath));
- } catch (error) {
- throw new Error(
- `Failed to fetch files: ${error instanceof Error ? error.message : String(error)}`,
- );
- }
- }
-
- public findClosestMatch(
- wrongPath: string,
- threshold: number = 0.6,
- ): string | null {
- if (!this.initialized) {
- throw new Error(
- "PathAdjuster not initialized. Check initialization status with isInitialized()",
- );
- }
-
- const absoluteWrongPath = path.resolve(this.baseDirectory, wrongPath);
-
- const results = this.fuse.search(absoluteWrongPath, { limit: 1 });
-
- if (results.length > 0) {
- const bestMatch = results[0];
- if (bestMatch.score !== undefined && bestMatch.score <= 1 - threshold) {
- return bestMatch.item;
- }
- }
-
- return null;
- }
-
- public validatePath(filePath: string): boolean {
- try {
- const exists = fsExtra.pathExistsSync(filePath);
- if (!exists) return false;
-
- const stats = fsExtra.lstatSync(filePath);
- return stats.isFile();
- } catch (error) {
- return false;
- }
- }
-
- public async adjustPath(
- wrongPath: string,
- threshold: number = 0.6,
- ): Promise<string | null> {
- Iif (!this.initialized) {
- await this.initialize();
- }
-
- const closestMatch = this.findClosestMatch(wrongPath, threshold);
- if (closestMatch && this.validatePath(closestMatch)) {
- return closestMatch;
- }
-
- return null;
- }
-
- public toRelativePath(absolutePath: string): string {
- if (!path.isAbsolute(absolutePath)) {
- const possiblePath = path.join(this.baseDirectory, absolutePath);
- if (this.validatePath(possiblePath)) {
- return absolutePath;
- }
- }
-
- const normalizedPath = path.resolve(absolutePath);
-
- if (!normalizedPath.startsWith(this.baseDirectory)) {
- throw new Error("Path is outside the base directory");
- }
-
- return path.relative(this.baseDirectory, normalizedPath);
- }
-
- public async refreshFilePaths(): Promise<void> {
- try {
- this.allFiles = await this.getAllFiles(this.baseDirectory);
- this.fuse.setCollection(this.allFiles);
- } catch (error) {
- console.error("Error refreshing file paths:", error);
- throw error;
- }
- }
-} |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| DirectoryScanner.ts | -
-
- |
- 91.17% | -31/34 | -42.85% | -6/14 | -100% | -6/6 | -93.33% | -28/30 | -
| Errors.ts | -
-
- |
- 33.33% | -3/9 | -100% | -0/0 | -0% | -0/3 | -33.33% | -3/9 | -
| FileOperations.ts | -
-
- |
- 78.3% | -83/106 | -50% | -12/24 | -92.85% | -13/14 | -77.88% | -81/104 | -
| FileReader.ts | -
-
- |
- 87.5% | -14/16 | -60% | -3/5 | -100% | -3/3 | -85.71% | -12/14 | -
| FileSearch.ts | -
-
- |
- 87.34% | -69/79 | -66.66% | -4/6 | -91.66% | -11/12 | -88.88% | -64/72 | -
| PathAdjuster.ts | -
-
- |
- 97.01% | -65/67 | -84.21% | -16/19 | -100% | -13/13 | -96.82% | -61/63 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 | 5x -5x -5x - -5x -5x -5x -5x -5x - - - - - - - - - -5x - -99x -99x -99x -99x -99x -99x - - - -7x - -7x -2x -2x - - - -1x - -1x - - - - - -5x -4x - - -1x - - - - - - - - - -5x - - - -5x - -4x - -4x - -4x - - - - -3x - - - - - - -1x - - - -4x -4x -1x - - -3x - - - -4x -4x - -4x -1x - - -3x -3x -3x - -3x - - - - -6x - - - - - - - - - - - -3x -3x -3x - -3x - - - -3x - - - - - - - - -3x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -1x -1x - - - - - - - - - | import { ConfigService } from "@services/ConfigService";
-import { DirectoryScanner } from "@services/FileManagement/DirectoryScanner";
-import { ActionExecutor } from "@services/LLM/actions/ActionExecutor";
-import { IActionResult } from "@services/LLM/actions/types/ActionTypes";
-import { ProjectInfo } from "@services/LLM/utils/ProjectInfo";
-import * as fs from "fs";
-import { autoInjectable } from "tsyringe";
-import { MessageContextCleaner } from "./context/MessageContextCleanup";
-import { PhaseManager } from "./PhaseManager";
-import { IPhasePromptArgs } from "./types/PhaseTypes";
-
-interface MessageContext {
- message: string;
- environmentDetails?: string;
- projectInfo?: string;
-}
-
-@autoInjectable()
-export class LLMContextCreator {
- constructor(
- private directoryScanner: DirectoryScanner,
- private actionExecutor: ActionExecutor,
- private projectInfo: ProjectInfo,
- private configService: ConfigService,
- private phaseManager: PhaseManager,
- private messageContextCleanup: MessageContextCleaner,
- ) {}
-
- private async loadCustomInstructions(): Promise<string> {
- const config = this.configService.getConfig();
-
- if (config.customInstructionsPath) {
- try {
- const instructions = await fs.promises.readFile(
- config.customInstructionsPath,
- "utf-8",
- );
- return instructions.trim();
- } catch (error) {
- throw Error(
- `Failed to load custom instructions from ${config.customInstructionsPath}, check if the file exists and is accessible.`,
- );
- }
- }
-
- if (config.customInstructions) {
- return config.customInstructions;
- }
-
- throw new Error(
- "No custom instructions provided. Either customInstructionsPath or customInstructions must be set in config.",
- );
- }
-
- async create(
- message: string,
- root: string,
- isFirstMessage: boolean = true,
- ): Promise<string> {
- const baseContext: MessageContext = {
- message,
- };
-
- if (isFirstMessage) {
- // Reset to discovery phase on first message
- this.phaseManager.resetPhase();
- // Clear message context only on first message
- this.messageContextCleanup.cleanupContext();
-
- const [environmentDetails, projectInfo] = await Promise.all([
- this.getEnvironmentDetails(root),
- this.getProjectInfo(root),
- ]);
-
- return this.formatFirstTimeMessage({
- ...baseContext,
- environmentDetails,
- projectInfo,
- });
- }
-
- return this.formatSequentialMessage(baseContext);
- }
-
- private async getEnvironmentDetails(root: string): Promise<string> {
- const scanResult = await this.directoryScanner.scan(root);
- if (!scanResult.success) {
- throw new Error(`Failed to scan directory: ${scanResult.error}`);
- }
-
- return `# Current Working Directory (${root}) Files\n${scanResult.data}`;
- }
-
- private async getProjectInfo(root: string): Promise<string> {
- const info = await this.projectInfo.gatherProjectInfo(root);
- const config = this.configService.getConfig();
-
- if (!info.dependencyFile) {
- return "";
- }
-
- const runAllTestsCmd = config.runAllTestsCmd || "yarn test";
- const runOneTestCmd = config.runOneTestCmd || "yarn test {testPath}";
- const runTypeCheckCmd = config.runTypeCheckCmd || "yarn type-check";
-
- return `# Project Dependencies (from ${info.dependencyFile})
-Main Dependencies: ${info.mainDependencies.join(", ")}
-
-# Available Scripts
-${Object.entries(info.scripts)
- .map(([name, command]) => `${name}: ${command}`)
- .join("\n")}
-
-# Test Commands
-Run All Tests: ${runAllTestsCmd}
-Run Single Test: ${runOneTestCmd}
-Run Type Check: ${runTypeCheckCmd}`;
- }
-
- private async formatFirstTimeMessage(
- context: MessageContext,
- ): Promise<string> {
- const config = this.configService.getConfig();
- const phaseConfig = this.phaseManager.getCurrentPhaseConfig();
- const customInstructions = await this.loadCustomInstructions();
-
- const envDetails = config.includeAllFilesOnEnvToContext
- ? context.environmentDetails
- : "";
-
- const promptArgs: IPhasePromptArgs = {
- message: context.message,
- environmentDetails: envDetails,
- projectInfo: context.projectInfo,
- runAllTestsCmd: config.runAllTestsCmd,
- runOneTestCmd: config.runOneTestCmd,
- runTypeCheckCmd: config.runTypeCheckCmd,
- };
-
- return `
-# Task
-${context.message}
-
-
- <instructions details="NEVER_OUTPUT">
-<!-- These are internal instructions. Just follow them. Do not output. -->
-
-
-${
- customInstructions &&
- `
-# Custom Instructions
-${customInstructions}`
-}
-
-## Initial Instructions
-- Keep messages brief, clear, and concise.
-- Break tasks into prioritized steps.
-- Use available actions sequentially.
-
-# Additional Instructions
-${envDetails ? `\n${envDetails}` : ""}
-${context.projectInfo ? `\n${context.projectInfo}` : ""}
-
-## Phase Instructions
-${phaseConfig.generatePrompt(promptArgs)}
-
-</instructions>
-`;
- }
-
- private formatSequentialMessage(context: MessageContext): string {
- const phaseConfig = this.phaseManager.getCurrentPhaseConfig();
- return phaseConfig.generatePrompt({ message: context.message });
- }
-
- async executeAction(actionContent: string): Promise<IActionResult> {
- const result = await this.actionExecutor.executeAction(actionContent);
-
- return result;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 | - - - -3x -3x - -3x -3x - - - - -3x - -35x - - -35x -35x - - - - - -35x - -35x -35x - - - - -35x -35x - - - -2x -2x -1x - -1x - - - -2x - - - - - - - -6x - - -6x - - - - - - - - -8x - - -8x - - - - - - - - -1x - - -1x - - - -3x - - -3x - - - -3x - - -3x - - - -3x - - -3x - - - -3x - - -3x - - - -3x - - -3x - - - - - - - - -3x - - -3x - - - - - - - - | import {
- IConversationHistoryMessage,
- ILLMProvider,
-} from "@services/LLM/ILLMProvider";
-import { OpenRouterAPI } from "@services/LLMProviders/OpenRouter/OpenRouterAPI";
-import { autoInjectable, container, singleton } from "tsyringe";
-
-export enum LLMProviderType {
- OpenRouter = "open-router",
-}
-
-@singleton()
-@autoInjectable()
-export class LLMProvider implements ILLMProvider {
- private providers: Map<LLMProviderType, ILLMProvider>;
- private currentProvider: ILLMProvider | null = null;
-
- constructor() {
- this.providers = new Map();
- this.initializeProvider(LLMProviderType.OpenRouter);
- }
-
- private initializeProvider(type: LLMProviderType): void {
- let provider: ILLMProvider;
-
- switch (type) {
- case LLMProviderType.OpenRouter:
- provider = container.resolve(OpenRouterAPI);
- break;
- default:
- throw new Error(`Unsupported provider type: ${type}`);
- }
-
- this.providers.set(type, provider);
- this.currentProvider = provider;
- }
-
- public static getInstance(type: LLMProviderType): ILLMProvider {
- const provider = container.resolve(LLMProvider).getProvider(type);
- if (!provider) {
- throw new Error(`Unsupported provider type: ${type}`);
- }
- return provider;
- }
-
- private getProvider(type: LLMProviderType): ILLMProvider | undefined {
- return this.providers.get(type);
- }
-
- public sendMessage(
- model: string,
- message: string,
- options?: Record<string, unknown>,
- ): Promise<string> {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- return this.currentProvider.sendMessage(model, message, options);
- }
-
- public sendMessageWithContext(
- model: string,
- message: string,
- systemInstructions?: string,
- options?: Record<string, unknown>,
- ): Promise<string> {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- return this.currentProvider.sendMessageWithContext(
- model,
- message,
- systemInstructions,
- options,
- );
- }
-
- public clearConversationContext(): void {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- this.currentProvider.clearConversationContext();
- }
-
- public getConversationContext(): IConversationHistoryMessage[] {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- return this.currentProvider.getConversationContext();
- }
-
- public addSystemInstructions(instructions: string): void {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- this.currentProvider.addSystemInstructions(instructions);
- }
-
- public getAvailableModels(): Promise<string[]> {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- return this.currentProvider.getAvailableModels();
- }
-
- public validateModel(model: string): Promise<boolean> {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- return this.currentProvider.validateModel(model);
- }
-
- public getModelInfo(model: string): Promise<Record<string, unknown>> {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- return this.currentProvider.getModelInfo(model);
- }
-
- public streamMessage(
- model: string,
- message: string,
- callback: (chunk: string) => void,
- options?: Record<string, unknown>,
- ): Promise<void> {
- Iif (!this.currentProvider) {
- throw new Error("Current provider not set");
- }
- return this.currentProvider.streamMessage(
- model,
- message,
- callback,
- options,
- );
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 | 26x -26x -26x - - - -26x -1x -1x -1x -1x -1x - - - - - - - -1x - - -20x -20x -19x - - -19x - -19x -38x - - -19x - -1x - - -1x - - - - -32x - - - - - -9x - -9x -5x - - -4x - - - - - -4x -4x -4x - -4x - - - -4x - - - - - - - - - - - -6x -6x - - - -2x - - - -2x -2x - - - -2x -2x -2x - - - -1x -1x - - - -6x -6x - -6x -1x -1x - - - - - -6x - - - -2x -2x -2x - - - -2x -2x - - - -2x - -2x -1x - - -1x -1x - -1x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { openRouterClient } from "@constants/openRouterClient";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { singleton } from "tsyringe";
-import { IModelInfo } from "./types/ModelTypes";
-
-@singleton()
-export class ModelInfo {
- private modelInfoMap: Map<string, IModelInfo> = new Map();
- private currentModel: string | null = null;
- private currentModelInfo: IModelInfo | null = null;
- private initialized: boolean = false;
- private usageHistory: {
- [model: string]: {
- prompt_tokens: number;
- completion_tokens: number;
- total_tokens: number;
- }[];
- } = {};
-
- constructor(private debugLogger: DebugLogger) {}
-
- async initialize(): Promise<void> {
- try {
- const response = await openRouterClient.get("/models");
- const models: IModelInfo[] = response.data.data;
-
- // Clear existing models before adding new ones
- this.modelInfoMap.clear();
-
- models.forEach((model) => {
- this.modelInfoMap.set(model.id, model);
- });
-
- this.initialized = true;
- } catch (error) {
- this.debugLogger.log("ModelInfo", "Failed to initialize model list", {
- error,
- });
- throw error;
- }
- }
-
- private async ensureInitialized(): Promise<void> {
- Iif (!this.initialized) {
- await this.initialize();
- }
- }
-
- async setCurrentModel(modelId: string): Promise<void> {
- await this.ensureInitialized();
-
- if (modelId === this.currentModel && this.currentModelInfo) {
- return; // Already tracking this model
- }
-
- Iif (!(await this.isModelAvailable(modelId))) {
- throw new Error(
- `Invalid model: ${modelId}. Available models: ${Array.from(this.modelInfoMap.keys()).join(", ")}`,
- );
- }
-
- const modelInfo = this.modelInfoMap.get(modelId)!; // Safe to use ! here since we checked availability
- this.currentModel = modelId;
- this.currentModelInfo = modelInfo;
-
- Iif (!modelInfo) {
- return;
- }
-
- this.debugLogger.log("ModelInfo", "Current model info", {
- model: modelId,
- contextLength: modelInfo.context_length,
- maxCompletionTokens: modelInfo.top_provider.max_completion_tokens,
- });
- }
-
- getCurrentModel(): string | null {
- return this.currentModel;
- }
-
- async getModelInfo(modelId: string): Promise<IModelInfo | undefined> {
- await this.ensureInitialized();
- return this.modelInfoMap.get(modelId);
- }
-
- getCurrentModelInfo(): IModelInfo | null {
- return this.currentModelInfo;
- }
-
- async getCurrentModelContextLength(): Promise<number> {
- await this.ensureInitialized();
- return this.currentModelInfo?.context_length || 128000; // Default fallback
- }
-
- async getModelContextLength(modelId: string): Promise<number> {
- await this.ensureInitialized();
- const modelInfo = await this.getModelInfo(modelId);
- return modelInfo?.context_length || 128000; // Default fallback
- }
-
- async getAllModels(): Promise<string[]> {
- await this.ensureInitialized();
- return Array.from(this.modelInfoMap.keys());
- }
-
- async isModelAvailable(modelId: string): Promise<boolean> {
- await this.ensureInitialized();
- const available = this.modelInfoMap.has(modelId);
-
- if (!available) {
- const availableModels = Array.from(this.modelInfoMap.keys());
- this.debugLogger.log("ModelInfo", "Model not found in available models", {
- modelId,
- availableModels,
- });
- }
-
- return available;
- }
-
- async getModelMaxCompletionTokens(modelId: string): Promise<number> {
- await this.ensureInitialized();
- const modelInfo = await this.getModelInfo(modelId);
- return modelInfo?.top_provider.max_completion_tokens || 4096; // Default fallback
- }
-
- async getCurrentModelMaxCompletionTokens(): Promise<number> {
- await this.ensureInitialized();
- return this.currentModelInfo?.top_provider.max_completion_tokens || 4096; // Default fallback
- }
-
- async logCurrentModelUsage(usedTokens: number): Promise<void> {
- await this.ensureInitialized();
-
- if (!this.currentModelInfo) {
- return;
- }
-
- const contextLength = this.currentModelInfo.context_length;
- const usagePercent = ((usedTokens / contextLength) * 100).toFixed(1);
-
- this.debugLogger.log("ModelInfo", "Token usage", {
- model: this.currentModel,
- used: usedTokens,
- total: contextLength,
- percentage: `${usagePercent}%`,
- remaining: contextLength - usedTokens,
- });
- }
-
- async logDetailedUsage(usage: {
- prompt_tokens: number;
- completion_tokens: number;
- total_tokens: number;
- }): Promise<void> {
- await this.ensureInitialized();
-
- Iif (!this.currentModel) {
- return;
- }
-
- Iif (!this.usageHistory[this.currentModel]) {
- this.usageHistory[this.currentModel] = [];
- }
-
- this.usageHistory[this.currentModel].push(usage);
-
- this.debugLogger.log("ModelInfo", "Detailed token usage", {
- model: this.currentModel,
- usage,
- });
- }
-
- getUsageHistory(): {
- [model: string]: {
- prompt_tokens: number;
- completion_tokens: number;
- total_tokens: number;
- }[];
- } {
- return this.usageHistory;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 | 25x -25x -25x - - -25x -85x - - -85x -85x - - - -44x -44x -44x - - - -36x - - - | import { DebugLogger } from "@services/logging/DebugLogger";
-import { singleton } from "tsyringe";
-import { ModelInfo } from "./ModelInfo";
-
-@singleton()
-export class ModelManager {
- private currentModel: string = "";
-
- constructor(
- private modelInfo: ModelInfo,
- private debugLogger: DebugLogger,
- ) {}
-
- public setCurrentModel(model: string): void {
- this.currentModel = model;
- this.modelInfo.setCurrentModel(model);
- this.debugLogger.log("ModelManager", "Model updated", { model });
- }
-
- public getCurrentModel(): string {
- return this.currentModel;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 | 20x -20x -20x -20x -20x -20x -20x - - -20x -56x -56x -56x - - -56x -56x -56x -56x - -56x -56x - - - - - - - -19x - -19x -7x - - -12x -12x - - -12x -4x - - - - -4x -3x - - - -3x -2x - - - -12x -12x -12x -12x -12x - - - -12x - -12x -12x -12x - - - -4x -4x - - - - - - -4x - -4x -4x - - - - -4x - - - - - - - - -4x - - - - - - -11x -11x - -11x - -10x - -10x -22x - -15x - -22x - - - -13x - - -9x - - -1x - - - | import { MODEL_SCALING_INITIAL_TRY_COUNT } from "@/constants/modelScaling";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { singleton } from "tsyringe";
-import { ConfigService } from "../ConfigService";
-import { ModelManager } from "./ModelManager";
-import { PhaseManager } from "./PhaseManager";
-import { Phase } from "./types/PhaseTypes";
-
-@singleton()
-export class ModelScaler {
- private tryCountMap: Map<string, number> = new Map();
- private globalTryCount: number = 0;
- private autoScalerEnabled: boolean = false;
-
- constructor(
- private debugLogger: DebugLogger,
- private configService: ConfigService,
- private modelManager: ModelManager,
- private phaseManager: PhaseManager,
- ) {
- const config = this.configService.getConfig();
- this.autoScalerEnabled = config.autoScaler || false;
- }
-
- isAutoScalerEnabled(): boolean {
- return this.autoScalerEnabled;
- }
-
- async incrementTryCount(filePath: string): Promise<void> {
- const currentPhase = this.phaseManager.getCurrentPhase();
-
- if (!this.autoScalerEnabled || currentPhase !== Phase.Execute) {
- return;
- }
-
- this.incrementCounts(filePath);
- const currentCount = this.tryCountMap.get(filePath) || 0;
-
- // Only scale if we've exceeded the threshold
- if (currentCount > MODEL_SCALING_INITIAL_TRY_COUNT) {
- await this.handleModelScaling(filePath, currentCount);
- }
- }
-
- getTryCount(filePath: string): number {
- if (!this.autoScalerEnabled) return 0;
- return this.tryCountMap.get(filePath) || 0;
- }
-
- getGlobalTryCount(): number {
- if (!this.autoScalerEnabled) return 0;
- return this.globalTryCount;
- }
-
- reset(): void {
- this.tryCountMap.clear();
- this.globalTryCount = 0;
- const config = this.configService.getConfig();
- this.autoScalerEnabled = config.autoScaler || false;
- this.modelManager.setCurrentModel(config.discoveryModel);
- }
-
- private incrementCounts(filePath: string): void {
- Iif (!this.autoScalerEnabled) return;
-
- this.globalTryCount++;
- const currentCount = this.tryCountMap.get(filePath) || 0;
- this.tryCountMap.set(filePath, currentCount + 1);
- }
-
- private getMaxTryCount(): number {
- const values = Array.from(this.tryCountMap.values());
- return values.length > 0 ? Math.max(...values) : 0;
- }
-
- private async handleModelScaling(
- filePath: string,
- currentCount: number,
- ): Promise<void> {
- Iif (!this.autoScalerEnabled) return;
-
- const maxTries = this.getMaxTryCount();
- const newModel = this.getModelForTryCount(
- maxTries.toString(),
- this.globalTryCount,
- );
-
- this.debugLogger.log("Model", "Incrementing try count", {
- filePath,
- fileCount: currentCount + 1,
- globalCount: this.globalTryCount,
- maxTries,
- newModel,
- phase: this.phaseManager.getCurrentPhase(),
- });
-
- await this.modelManager.setCurrentModel(newModel);
- }
-
- private getModelForTryCount(
- tryCount: string | null,
- globalTries: number,
- ): string {
- const config = this.configService.getConfig();
- const availableModels = config.autoScaleAvailableModels;
-
- if (!tryCount) return availableModels[0].id;
-
- const tries = parseInt(tryCount, 10);
-
- for (let i = 0; i < availableModels.length; i++) {
- const previousTriesSum = availableModels
- .slice(0, i)
- .reduce((sum, model) => sum + model.maxWriteTries, 0);
-
- if (
- tries >= previousTriesSum + availableModels[i].maxWriteTries ||
- globalTries >= availableModels[i].maxGlobalTries
- ) {
- continue;
- }
-
- return availableModels[i].id;
- }
-
- return availableModels[availableModels.length - 1].id;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 | -25x -25x -25x -25x -25x - - - -25x -87x -87x - - -87x -87x - - - -28x - - -28x - - - - - - - - - - - - - - - - - - - - - - - - - -28x - - -28x - -28x - - - -28x - - - -25x - - - - -1x - - - -1x -1x - - -1x - - - -1x - - - -7x -7x -1x - -6x - - - -4x - -2x -2x - -1x -1x - - -1x - -4x - - - -5x - - - | // services/PhaseManager.ts
-import { ConfigService } from "@services/ConfigService";
-import { injectable, singleton } from "tsyringe";
-import { ModelManager } from "./ModelManager";
-import { phaseBlueprints } from "./phases/blueprints";
-import { IPhaseConfig, Phase } from "./types/PhaseTypes";
-
-@injectable()
-@singleton()
-export class PhaseManager {
- private currentPhase: Phase = Phase.Discovery;
- private phaseConfigs: Map<Phase, IPhaseConfig> = new Map();
-
- constructor(
- private configService: ConfigService,
- private modelManager: ModelManager,
- ) {}
-
- public initializePhaseConfigs() {
- const config = this.configService.getConfig();
-
- // Initialize configs using blueprints but override models from config if provided
- this.phaseConfigs = new Map([
- [
- Phase.Discovery,
- {
- ...phaseBlueprints[Phase.Discovery],
- model:
- config.discoveryModel || phaseBlueprints[Phase.Discovery].model,
- },
- ],
- [
- Phase.Strategy,
- {
- ...phaseBlueprints[Phase.Strategy],
- model: config.strategyModel || phaseBlueprints[Phase.Strategy].model,
- },
- ],
- [
- Phase.Execute,
- {
- ...phaseBlueprints[Phase.Execute],
- model: config.executeModel || phaseBlueprints[Phase.Execute].model,
- },
- ],
- ]);
-
- // set initial phase
- this.currentPhase = Phase.Discovery;
-
- // set initial model
- const phaseData = this.phaseConfigs.get(Phase.Discovery);
-
- Iif (!phaseData) {
- throw new Error("No data found for Discovery phase");
- }
-
- this.modelManager.setCurrentModel(phaseData.model);
- }
-
- getCurrentPhase(): Phase {
- return this.currentPhase;
- }
-
- getCurrentPhaseConfig(): IPhaseConfig {
- // reset if not set
- Iif (!this.currentPhase) {
- this.resetPhase();
- }
-
- const config = this.phaseConfigs.get(this.currentPhase);
- Iif (!config) {
- throw new Error(`No configuration found for phase ${this.currentPhase}`);
- }
- return config;
- }
-
- setPhase(phase: Phase) {
- this.currentPhase = phase;
- }
-
- getPhaseConfig(phase: Phase): IPhaseConfig {
- const config = this.phaseConfigs.get(phase);
- if (!config) {
- throw new Error(`No configuration found for phase ${phase}`);
- }
- return config;
- }
-
- nextPhase(): Phase {
- switch (this.currentPhase) {
- case Phase.Discovery:
- this.currentPhase = Phase.Strategy;
- break;
- case Phase.Strategy:
- this.currentPhase = Phase.Execute;
- break;
- case Phase.Execute:
- // Execute is the final phase, stays there
- break;
- }
- return this.currentPhase;
- }
-
- resetPhase() {
- this.currentPhase = Phase.Discovery;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 | 17x - -17x -17x -17x -17x -17x - - -17x - -6x -6x -6x -6x - - - -5x -5x - -5x -5x - - - - -5x - - - - -5x - - -5x - - -5x - - -5x - - - -5x - - - - -5x - - - - - - -5x - -5x - - - - - - - - - - -10x - -5x - -5x - - - - - - - - | import { injectable } from "tsyringe";
-import { WriteActionData } from "./actions/types/ActionTypes";
-import { MessageContextHistory } from "./context/MessageContextHistory";
-import { MessageContextPhase } from "./context/MessageContextPhase";
-import { ModelManager } from "./ModelManager";
-import { PhaseManager } from "./PhaseManager";
-import { Phase } from "./types/PhaseTypes";
-
-@injectable()
-export class PhaseTransitionService {
- constructor(
- private phaseManager: PhaseManager,
- private modelManager: ModelManager,
- private messageContextPhase: MessageContextPhase,
- private messageContextHistory: MessageContextHistory,
- ) {}
-
- async transitionToNextPhase(): Promise<WriteActionData> {
- const currentPhase = this.phaseManager.getCurrentPhase();
- const nextPhase = this.getNextPhase(currentPhase);
-
- this.messageContextPhase.cleanupPhaseContent();
- Iif (currentPhase === Phase.Strategy) {
- this.messageContextHistory.mergeConversationHistory();
- }
-
- // Log phase transition with emojis
- console.log(
- `🔄 Phase Transition: ${this.getPhaseEmoji(currentPhase)}${currentPhase} ➡️ ${this.getPhaseEmoji(nextPhase)}${nextPhase}`,
- );
-
- // Move to next phase
- this.phaseManager.nextPhase();
-
- // Get the new phase's config
- const nextPhaseConfig = this.phaseManager.getCurrentPhaseConfig();
-
- // Update model for the new phase
- await this.modelManager.setCurrentModel(nextPhaseConfig.model);
-
- // Generate prompt but don't include it in the response
- nextPhaseConfig.generatePrompt({
- message: "Continue with the next phase based on previous findings.",
- });
-
- this.messageContextHistory.addMessage(
- "system",
- `Current phase is ${nextPhase}`,
- );
-
- return {
- regenerate: true,
- selectedModel: nextPhaseConfig.model,
- };
- }
-
- private getNextPhase(currentPhase: Phase): Phase {
- switch (currentPhase) {
- case Phase.Discovery:
- return Phase.Strategy;
- case Phase.Strategy:
- return Phase.Execute;
- case Phase.Execute:
- return Phase.Discovery;
- default:
- return Phase.Discovery;
- }
- }
-
- private getPhaseEmoji(phase: Phase): string {
- switch (phase) {
- case Phase.Discovery:
- return "🔍 ";
- case Phase.Strategy:
- return "🎯 ";
- case Phase.Execute:
- return "⚡ ";
- default:
- return "❓ ";
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 | 6x -6x -6x -6x -6x -6x - - - -6x - -100x -100x -100x -100x - - - -7x - -7x - - -7x -73x - - - - -1x - - -1x - - - -1x - - - -1x - - - - -6x -6x - -6x -3x - - -3x - - - -3x - - - -3x - - - -3x -3x -3x -2x - -1x -1x - - - -1x - - - -1x - - - - - -2x -2x -2x -2x - -2x - - -2x - - - - - - - - - - - - - - - -2x -2x -2x - - - - - -2x - - -2x - - -2x -2x -1x - - - - -1x - - - - - - - - - - - - - - - -2x - - - - - -2x - - - - - - -2x -1x - - - - - - - -2x - - -2x - - -2x - - - - - - - - - - - - - - - - - - - - - - - - | import { autoInjectable } from "tsyringe";
-import { MessageContextHistory } from "../context/MessageContextHistory";
-import { MessageContextLogger } from "../context/MessageContextLogger";
-import { ActionTag, getBlueprint, getImplementedActions } from "./blueprints";
-import { ActionFactory } from "./core/ActionFactory";
-import { ActionQueue } from "./core/ActionQueue";
-import { IActionResult } from "./types/ActionTypes";
-
-@autoInjectable()
-export class ActionExecutor {
- constructor(
- private actionFactory: ActionFactory,
- private actionQueue: ActionQueue,
- private messageContextLogger: MessageContextLogger,
- private messageContextHistory: MessageContextHistory,
- ) {}
-
- async executeAction(actionText: string): Promise<IActionResult> {
- try {
- // Get all implemented action types
- const implementedActions = getImplementedActions();
-
- // Validate XML structure
- for (const actionType of implementedActions) {
- if (
- actionText.includes(actionType) &&
- !actionText.includes(`<${actionType}>`) &&
- !actionText.includes(`</${actionType}>`)
- ) {
- const error = new Error(
- `Found "${actionType}" without proper XML tag structure. Tags must be wrapped in < > brackets. For example: <${actionType}>content</${actionType}>`,
- );
- this.messageContextLogger.logActionResult(actionType, {
- success: false,
- error,
- });
- this.messageContextHistory.addMessage(
- "system",
- `Action ${actionType} failed: Invalid XML structure`,
- );
- return { success: false, error };
- }
- }
-
- // Extract actions using regex
- const actionMatch = /<(\w+)>([\s\S]*?)<\/\1>/g;
- const matches = Array.from(actionText.matchAll(actionMatch));
-
- if (!matches.length) {
- const error = new Error(
- "No valid action tags found. Actions must be wrapped in XML-style tags.",
- );
- this.messageContextLogger.logActionResult("unknown", {
- success: false,
- error,
- });
- this.messageContextHistory.addMessage(
- "system",
- "Action failed: No valid action tags found",
- );
- return { success: false, error };
- }
-
- // Queue all actions
- for (const [fullMatch, actionType] of matches) {
- if (actionType !== "path" && actionType !== "content") {
- if (implementedActions.includes(actionType as ActionTag)) {
- this.actionQueue.enqueue(actionType, fullMatch);
- } else {
- const error = new Error(`Unknown action type: ${actionType}`);
- this.messageContextLogger.logActionResult(actionType, {
- success: false,
- error,
- });
- this.messageContextHistory.addMessage(
- "system",
- `Action failed: Unknown type ${actionType}`,
- );
- return { success: false, error };
- }
- }
- }
-
- // Process actions in order
- let lastResult: IActionResult = { success: true };
- while (!this.actionQueue.isEmpty()) {
- const action = this.actionQueue.dequeue();
- Iif (!action) continue;
-
- const actionInstance = this.actionFactory.createAction(
- action.type as ActionTag,
- );
- Iif (!actionInstance) {
- const error = new Error(
- `Failed to create action instance for "${action.type}"`,
- );
- this.messageContextLogger.logActionResult(action.type, {
- success: false,
- error,
- });
- this.messageContextHistory.addMessage(
- "system",
- `Action ${action.type} failed: Could not create instance`,
- );
- return { success: false, error };
- }
-
- // Get blueprint for logging
- const blueprint = getBlueprint(action.type as ActionTag);
- if (blueprint) {
- console.log(
- `⚡ ${blueprint.description || `Executing ${action.type}`}...`,
- );
- }
-
- // Execute action
- lastResult = await actionInstance.execute(action.content);
-
- // Log the result
- this.messageContextLogger.logActionResult(action.type, lastResult);
-
- // Add result to conversation history
- if (lastResult.success) {
- if (lastResult.data) {
- this.messageContextHistory.addMessage(
- "system",
- `Action ${action.type} succeeded: ${JSON.stringify(lastResult.data)}`,
- );
- } else {
- this.messageContextHistory.addMessage(
- "system",
- `Action ${action.type} succeeded`,
- );
- }
- } else E{
- const errorMessage = lastResult.error
- ? lastResult.error.message
- : "Unknown error";
- this.messageContextHistory.addMessage(
- "system",
- `Action ${action.type} failed: ${errorMessage}`,
- );
- }
-
- // Store result in queue for processing requirements
- this.actionQueue.setActionResult(
- action.type,
- action.content,
- lastResult,
- );
-
- Iif (!lastResult.success) {
- // Clear queue on error
- this.actionQueue.clear();
- break;
- }
-
- // If this action requires processing, include its results in the final output
- if (blueprint?.requiresProcessing && lastResult.success) {
- lastResult = {
- ...lastResult,
- processedResults: this.actionQueue.getProcessedResults(),
- };
- }
- }
-
- // Get final processed results before clearing
- const finalProcessedResults = this.actionQueue.getProcessedResults();
-
- // Clear queue before returning
- this.actionQueue.clear();
-
- // Include processed results in final output if there are any
- return finalProcessedResults.size > 0
- ? { ...lastResult, processedResults: finalProcessedResults }
- : lastResult;
- } catch (error) {
- // Clear queue on error
- this.actionQueue.clear();
-
- // Log the error
- this.messageContextLogger.logActionResult("unknown", {
- success: false,
- error: error as Error,
- });
- this.messageContextHistory.addMessage(
- "system",
- `Action failed with error: ${(error as Error).message}`,
- );
-
- return {
- success: false,
- error: error as Error,
- };
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 | 16x -16x - - -16x - - -4x -4x - -4x -48x -48x -64x - - - -4x - - - - - - - - - -6x - - -6x -52x - -52x - - -52x -2x - - - - -4x - -4x -32x - -32x - - -32x - - - - -4x - - - - - - - - - -15x -15x - -15x -11x - - - - - - - - - - -6x -6x -7x - - - - - - - - - -3x -3x - -2x - -6x - - - - - - - - - - - - - - - -4x -4x - -3x - - - - - - - - - - - - - - | import { autoInjectable } from "tsyringe";
-import { getBlueprint, getImplementedActions } from "./blueprints";
-
-@autoInjectable()
-export class ActionTagsExtractor {
- private getParameterTags(): string[] {
- // Get all parameter names from all action blueprints
- const paramSet = new Set<string>();
- const actionTags = getImplementedActions();
-
- for (const tag of actionTags) {
- const blueprint = getBlueprint(tag);
- blueprint.parameters?.forEach((param) => {
- paramSet.add(param.name);
- });
- }
-
- return Array.from(paramSet);
- }
-
- /**
- * Validates if a tag has proper XML structure
- * @param content Full text content to validate
- * @returns Message indicating if structure is valid or what's wrong
- */
- validateStructure(content: string): string {
- // Get implemented action tags dynamically
- const actionTags = getImplementedActions();
-
- // First validate the outer action tags
- for (const tag of actionTags) {
- const openCount = (content.match(new RegExp(`<${tag}>`, "g")) || [])
- .length;
- const closeCount = (content.match(new RegExp(`</${tag}>`, "g")) || [])
- .length;
-
- if (openCount !== closeCount) {
- return `We need to use proper tag structure, try again. Missing ${openCount > closeCount ? "closing" : "opening"} tag for <${tag}>.`;
- }
- }
-
- // Then validate inner parameter tags
- const parameterTags = this.getParameterTags();
-
- for (const tag of parameterTags) {
- const openCount = (content.match(new RegExp(`<${tag}>`, "g")) || [])
- .length;
- const closeCount = (content.match(new RegExp(`</${tag}>`, "g")) || [])
- .length;
-
- Iif (openCount !== closeCount) {
- return `We need to use proper tag structure, try again. Missing ${openCount > closeCount ? "closing" : "opening"} tag for <${tag}>.`;
- }
- }
-
- return "";
- }
-
- /**
- * Extracts content from a single tag
- * @param content Full text content
- * @param tagName Name of the tag to extract
- * @returns The content within the tag or null if not found
- */
- extractTag(content: string, tagName: string): string | string[] | null {
- const regex = new RegExp(`<${tagName}>([\\s\\S]*?)</${tagName}>`, "g");
- const matches = Array.from(content.matchAll(regex));
-
- if (matches.length === 0) return null;
- if (matches.length === 1) return matches[0][1].trim();
- return matches.map((match) => match[1].trim());
- }
-
- /**
- * Extracts content from multiple instances of the same tag
- * @param content Full text content
- * @param tagName Name of the tag to extract
- * @returns Array of content within each instance of the tag
- */
- extractTags(content: string, tagName: string): string[] {
- const regex = new RegExp(`<${tagName}>([\\s\\S]*?)</${tagName}>`, "g");
- const matches = content.matchAll(regex);
- return Array.from(matches).map((match) => match[1].trim());
- }
-
- /**
- * Extracts content from a tag and splits it into lines
- * @param content Full text content
- * @param tagName Name of the tag to extract
- * @returns Array of non-empty trimmed lines from the tag content
- */
- extractTagLines(content: string, tagName: string): string[] {
- const tagContent = this.extractTag(content, tagName);
- if (!tagContent || Array.isArray(tagContent)) return [];
-
- return tagContent
- .split("\n")
- .map((line) => line.trim())
- .filter(Boolean);
- }
-
- /**
- * Extracts nested tags from within a parent tag
- * @param content Full text content
- * @param parentTag Parent tag name
- * @param childTag string
- * @returns Array of content within child tags, found within the parent tag
- */
- extractNestedTags(
- content: string,
- parentTag: string,
- childTag: string,
- ): string[] {
- const parentContent = this.extractTag(content, parentTag);
- if (!parentContent || Array.isArray(parentContent)) return [];
-
- return this.extractTags(parentContent, childTag);
- }
-
- /**
- * Extracts all instances of a tag with their complete content
- * @param content Full text content
- * @param tagName Name of the tag to extract
- * @returns Array of complete tag contents including nested tags
- */
- extractAllTagsWithContent(content: string, tagName: string): string[] {
- const regex = new RegExp(`<${tagName}>[\\s\\S]*?</${tagName}>`, "g");
- const matches = content.match(regex);
- return matches ? matches.map((match) => match.trim()) : [];
- }
-} |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 -289 -290 -291 -292 -293 -294 -295 -296 -297 -298 -299 -300 -301 -302 -303 -304 -305 -306 -307 -308 -309 -310 -311 -312 -313 -314 -315 -316 -317 -318 -319 -320 -321 -322 -323 -324 -325 -326 -327 -328 -329 -330 -331 -332 -333 -334 -335 -336 -337 -338 -339 -340 -341 -342 -343 -344 -345 -346 -347 -348 -349 -350 -351 -352 -353 -354 -355 -356 -357 -358 -359 -360 -361 -362 -363 -364 -365 -366 -367 -368 -369 -370 -371 -372 -373 -374 -375 -376 -377 -378 -379 -380 -381 -382 -383 -384 -385 -386 -387 -388 -389 -390 -391 -392 -393 -394 -395 -396 -397 -398 -399 -400 -401 -402 -403 -404 -405 -406 -407 -408 -409 -410 -411 -412 -413 -414 -415 -416 -417 -418 -419 -420 -421 -422 -423 -424 -425 -426 -427 -428 -429 -430 -431 -432 -433 -434 -435 -436 -437 -438 -439 -440 -441 -442 -443 -444 -445 | 4x -4x - - - - - - -4x -4x -4x -4x -4x -4x - - - - - - - - -4x -69x -69x -69x -69x -69x - - -69x -69x -69x -69x - - - -38x -38x -38x -38x - - - -1x - - - -3x -36x -42x - - - - - -3x -3x -3x - -2x -2x -2x - -1x - - - -3x -3x - -2x -2x - -1x - - - - -6x -6x - - - - -6x -6x -6x - - - - - - - - - - - - - -13x -17x - -17x - -6x -11x -3x -3x -3x - -6x - - -17x - - - - - - - - - - -17x - - - - - - -13x -13x - - -13x -34x -34x - - -13x -12x -12x - - -12x - -17x - - - - - - -17x -17x - - - - - - -12x -12x -12x -17x - - - -12x -8x - - - - - - -12x -7x - - - - - - -12x -12x - - -13x - - - -14x - - - -14x -14x -1x - - -1x - - - -13x -13x -13x - - - -13x - -13x -18x -18x -18x -17x - - - - -17x - - - -13x -13x - - - -2x - - - - - - - - -2x - - - -1x - - - -1x - - - -1x - - - -1x - - - -8x -8x - -8x - -8x - - - -8x -3x - - - - - -3x - - -3x - - -5x -1x - - -4x - - - -4x - - - - - - - -4x -1x - - -3x - - - - - - - -7x -7x - -7x - -7x -7x -7x - -7x -8x - -8x -4x -5x - -5x - - - - -4x -4x - - -4x -5x - - - - - - - - - -4x -4x -5x - - - - - - - - - - - - - - - -4x -4x - - - -4x - - - - -4x - -4x -1x - - - -1x -1x - - -3x -2x -2x -1x -1x - - - - - - - - - - - - -7x -9x - - -7x -1x - - -1x - - -6x -8x - - - - -6x -5x -5x - - - - - - - - - -6x - - - - - - - - - - | import { LLMContextCreator } from "@services/LLM/LLMContextCreator";
-import { ActionTagsExtractor } from "@services/LLM/actions/ActionTagsExtractor";
-import {
- IActionDependency,
- IActionExecutionPlan,
- IActionGroup,
- WriteActionData,
-} from "@services/LLM/actions/types/ActionTypes";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { HtmlEntityDecoder } from "@services/text/HTMLEntityDecoder";
-import path from "path";
-import { autoInjectable } from "tsyringe";
-import { v4 as uuidv4 } from "uuid";
-import { getBlueprint, getImplementedActions } from "./blueprints";
-
-export interface ActionExecutionResult {
- actions: Array<{ action: string; result: any }>;
- followupResponse?: string;
- selectedModel?: string;
-}
-
-@autoInjectable()
-export class ActionsParser {
- private currentMessageBuffer: string = "";
- private isProcessingAction: boolean = false;
- private messageComplete: boolean = false;
- private processedTags: string[] = [];
- private currentModel: string = "";
-
- constructor(
- private debugLogger: DebugLogger,
- private contextCreator: LLMContextCreator,
- private htmlEntityDecoder: HtmlEntityDecoder,
- private actionTagsExtractor: ActionTagsExtractor,
- ) {}
-
- reset() {
- this.currentMessageBuffer = "";
- this.isProcessingAction = false;
- this.messageComplete = false;
- this.processedTags = [];
- }
-
- isCompleteMessage(text: string): boolean {
- return true;
- }
-
- private getActionsWithPathParam(): string[] {
- return getImplementedActions().filter((tag) => {
- const blueprint = getBlueprint(tag);
- return blueprint.parameters?.some((param) => param.name === "path");
- });
- }
-
- extractFilePath(tag: string): string | null {
- // Get actions that have a path parameter from their blueprints
- const fileActions = this.getActionsWithPathParam();
- const actionMatch = new RegExp(`<(${fileActions.join("|")})>`).exec(tag);
- if (!actionMatch) return null;
-
- const pathMatch = /<path>(.*?)<\/path>/;
- const match = tag.match(pathMatch);
- if (!match) return null;
-
- return path.resolve(process.cwd(), match[1]);
- }
-
- extractUrl(tag: string): string | null {
- const actionMatch = /<fetch_url>[\s\S]*?<\/fetch_url>/i.exec(tag);
- if (!actionMatch) return null;
-
- const urlMatch = /<url>(.*?)<\/url>/i.exec(tag);
- if (!urlMatch) return null;
-
- return urlMatch[1];
- }
-
- private extractContentFromAction(actionContent: string): string {
- // First try to find content tag
- const contentMatch = actionContent.match(/<content>([\s\S]*?)<\/content>/);
- Iif (contentMatch) {
- return contentMatch[1];
- }
-
- // If no content tag, try to find path tag for file operations
- const pathMatch = actionContent.match(/<path>(.*?)<\/path>/);
- if (pathMatch) {
- return pathMatch[1];
- }
-
- // For read_file actions, we want the entire content for dependency tracking
- Iif (actionContent.includes("<read_file>")) {
- return actionContent;
- }
-
- return "";
- }
-
- private detectActionDependencies(
- actions: IActionDependency[],
- ): IActionDependency[] {
- return actions.map((action) => {
- const dependsOn: string[] = [];
-
- if (action.type === "write_file") {
- // Find read_file actions whose content is used in this write_file
- const readActions = actions.filter((a) => {
- if (a.type !== "read_file") return false;
- const readContent = this.extractContentFromAction(a.content);
- const writeContent = this.extractContentFromAction(action.content);
- return writeContent.includes(readContent);
- });
- dependsOn.push(...readActions.map((a) => a.actionId));
- }
-
- Iif (["move_file", "delete_file", "copy_file"].includes(action.type)) {
- // These actions should wait for any write operations on the same file
- const writeActions = actions.filter((a) => {
- Iif (a.type !== "write_file") return false;
- const writePath = this.extractFilePath(a.content);
- const actionPath = this.extractFilePath(action.content);
- return writePath === actionPath;
- });
- dependsOn.push(...writeActions.map((a) => a.actionId));
- }
-
- return { ...action, dependsOn };
- });
- }
-
- private createExecutionPlan(
- actions: IActionDependency[],
- ): IActionExecutionPlan {
- const groups: IActionGroup[] = [];
- const unprocessedActions = [...actions];
-
- // Helper to check if an action can be executed in parallel
- const canRunInParallel = (action: IActionDependency) => {
- const blueprint = getBlueprint(action.type);
- return blueprint.canRunInParallel !== false;
- };
-
- while (unprocessedActions.length > 0) {
- const currentGroup: IActionDependency[] = [];
- const remainingActions: IActionDependency[] = [];
-
- // First, find all actions that can be executed (no pending dependencies)
- unprocessedActions.forEach((action) => {
- const canExecute =
- !action.dependsOn?.length ||
- action.dependsOn.every(
- (depId) =>
- actions.find((a) => a.actionId === depId)?.type === "end_task" ||
- !unprocessedActions.find((ua) => ua.actionId === depId),
- );
-
- if (canExecute) {
- currentGroup.push(action);
- } else E{
- remainingActions.push(action);
- }
- });
-
- // If we have actions that can be executed, split them into parallel and sequential groups
- if (currentGroup.length > 0) {
- const parallelActions = currentGroup.filter(canRunInParallel);
- const sequentialActions = currentGroup.filter(
- (action) => !canRunInParallel(action),
- );
-
- // Add parallel actions as one group if there are any
- if (parallelActions.length > 0) {
- groups.push({
- actions: parallelActions,
- parallel: true,
- });
- }
-
- // Add each sequential action as its own group
- sequentialActions.forEach((action) => {
- groups.push({
- actions: [action],
- parallel: false,
- });
- });
- }
-
- unprocessedActions.length = 0;
- unprocessedActions.push(...remainingActions);
- }
-
- return { groups };
- }
-
- findCompleteTags(text: string): IActionExecutionPlan {
- const combinedText = this.currentMessageBuffer + text;
-
- // Validate tag structure before processing
- const validationError =
- this.actionTagsExtractor.validateStructure(combinedText);
- if (validationError) {
- this.debugLogger.log("Validation", "Tag structure validation failed", {
- error: validationError,
- });
- return { groups: [] };
- }
-
- // Find all action tags in order of appearance
- const actions: IActionDependency[] = [];
- const actionTags = getImplementedActions();
- const allTagsRegex = new RegExp(
- `<(${actionTags.join("|")})>[\\s\\S]*?</\\1>`,
- "g",
- );
- const matches = Array.from(combinedText.matchAll(allTagsRegex));
-
- for (const match of matches) {
- const content = match[0];
- const type = match[1] as any; // The captured tag name
- if (!this.processedTags.includes(content)) {
- actions.push({
- actionId: uuidv4(),
- type,
- content,
- });
- this.processedTags.push(content);
- }
- }
-
- const actionsWithDependencies = this.detectActionDependencies(actions);
- return this.createExecutionPlan(actionsWithDependencies);
- }
-
- appendToBuffer(chunk: string) {
- this.currentMessageBuffer += chunk;
- }
-
- clearBuffer() {
- this.currentMessageBuffer = "";
- this.processedTags = [];
- }
-
- get buffer() {
- return this.currentMessageBuffer;
- }
-
- get isProcessing() {
- return this.isProcessingAction;
- }
-
- set isProcessing(value: boolean) {
- this.isProcessingAction = value;
- }
-
- get isComplete() {
- return this.messageComplete;
- }
-
- set isComplete(value: boolean) {
- this.messageComplete = value;
- }
-
- private formatActionResult(action: string, result: any): string {
- const actionMatch = /<(\w+)>([\s\S]*?)<\/\1>/.exec(action);
- Iif (!actionMatch) return `[Action Result] Invalid action format`;
-
- const [_, actionType] = actionMatch;
-
- Iif (actionType === "execute_command" && result.success) {
- return `Command execution result:\n\n${result.data}\n\nPlease analyze this output and continue with the task.`;
- }
-
- if (actionType === "read_file" && result.success) {
- const output = this.htmlEntityDecoder.decode(
- JSON.stringify(result.data),
- {
- unescape: true,
- },
- );
- Iif (typeof result.data === "string" && result.data.includes("# File:")) {
- return result.data;
- }
- return `Here's the content of the requested file:\n\n${output}\n\nPlease analyze this content and continue with the task.`;
- }
-
- if (actionType === "fetch_url" && result.success) {
- return `Here's the content fetched from the URL:\n\n${result.data}\n\nPlease analyze this content and continue with the task.`;
- }
-
- Iif (actionType === "end_task" && result.success) {
- return `Task completed: ${result.data}`;
- }
-
- Iif (actionType === "end_phase" && result.success) {
- const data = result.data as WriteActionData;
- Iif (data?.regenerate && data?.prompt) {
- return data.prompt as string;
- }
- return `Phase completed. Moving to next phase.`;
- }
-
- if (actionType === "relative_path_lookup" && result.success) {
- return `Found matching path: ${JSON.stringify(result.data)}`;
- }
-
- return `[Action Result] ${actionType}: ${JSON.stringify(result)} ${result.success && "Proceed to next previously planned step."}`;
- }
-
- async parseAndExecuteActions(
- text: string,
- model: string,
- llmCallback: (message: string) => Promise<string>,
- ): Promise<ActionExecutionResult> {
- try {
- this.currentModel = model;
-
- const executionPlan = this.findCompleteTags(text);
-
- const results: Array<{ action: string; result: any }> = [];
- let selectedModel = model;
- let hasError = false;
-
- for (const group of executionPlan.groups) {
- Iif (hasError) break;
-
- if (group.parallel) {
- const actionPromises = group.actions.map((action) =>
- this.contextCreator
- .executeAction(action.content)
- .then((result) => ({
- action: action.content,
- result,
- })),
- );
- const groupResults = await Promise.all(actionPromises);
- results.push(...groupResults);
-
- // Check for errors in parallel actions
- for (const result of groupResults) {
- Iif (!result.result.success) {
- this.debugLogger.log("Action", "Action failed", {
- action: result.action,
- result: result.result,
- });
- hasError = true;
- break;
- }
- }
-
- if (!hasError) {
- for (const result of groupResults) {
- Iif (result.action.includes("<write_file>")) {
- const writeData = result.result.data as WriteActionData;
- Iif (writeData?.selectedModel) {
- selectedModel = writeData.selectedModel;
- this.debugLogger.log(
- "Model",
- "Updated model from write action",
- {
- model: selectedModel,
- },
- );
- }
- }
- }
- }
- } else {
- for (const action of group.actions) {
- const result = await this.contextCreator.executeAction(
- action.content,
- );
-
- this.debugLogger.log("Action", "Action executed", {
- action: action.content,
- result,
- });
-
- results.push({ action: action.content, result });
-
- if (!result.success) {
- this.debugLogger.log("Action", "Action failed", {
- action: action.content,
- result,
- });
- hasError = true;
- break;
- }
-
- if (action.type === "write_file") {
- const writeData = result.data as WriteActionData;
- if (writeData?.selectedModel) {
- selectedModel = writeData.selectedModel;
- this.debugLogger.log(
- "Model",
- "Updated model from write action",
- {
- model: selectedModel,
- },
- );
- }
- }
- }
- }
- }
-
- const endTaskAction = results.find(
- ({ action, result }) => action.includes("<end_task>") && result.success,
- );
-
- if (endTaskAction) {
- this.debugLogger.log("EndTask", "Task completed", {
- message: endTaskAction.result.data,
- });
- return { actions: results, selectedModel };
- }
-
- const actionResults = results
- .map(({ action, result }) => this.formatActionResult(action, result))
- .join("\n\n");
-
- // Only get followupResponse if there were no errors
- let followupResponse;
- if (!hasError) {
- followupResponse = await llmCallback(actionResults);
- this.debugLogger.log(
- "Response",
- "Received LLM response for action results",
- {
- response: followupResponse,
- selectedModel,
- },
- );
- }
-
- return { actions: results, followupResponse, selectedModel };
- } catch (error) {
- console.error("Error in parseAndExecuteActions:", error);
- this.debugLogger.log("Error", "Failed to parse and execute actions", {
- error,
- });
- return { actions: [] };
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 | -16x -16x -16x -16x -16x -16x -16x -16x - - - - - - - -16x - -9x -9x -9x - -9x - - - -44x - - - -9x -9x -9x -1x -1x - - - -8x - - - - -8x -8x - - - -9x - -9x -1x - - -8x - - - - - -8x -8x -8x -8x - - - - - - - -11x - - - - - - - - - -8x -8x -8x - -8x -8x -8x - - -8x -8x - - -8x -9x -9x -9x -9x -9x - - -9x - - - - - -8x -2x -2x -2x -2x -2x - - -2x - - - - -8x -8x -8x - -8x - - -8x - - - -8x - - - -8x - -8x - - -8x - - - - - -8x - - - -8x -8x - - -8x - - - - - - - - - - - - - | import { IActionResult } from "@services/LLM/actions/types/ActionTypes";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { AnsiStripper } from "@services/text/AnsiStripper";
-import chalk from "chalk";
-import { spawn, SpawnOptionsWithoutStdio } from "child_process";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { commandActionBlueprint as blueprint } from "./blueprints/commandActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-
-interface CommandParams {
- command: string;
-}
-
-@autoInjectable()
-export class CommandAction extends BaseAction {
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private debugLogger: DebugLogger,
- private ansiStripper: AnsiStripper,
- ) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected parseParams(content: string): Record<string, any> {
- const tag = this.getBlueprint().tag;
- const match = content.match(new RegExp(`<${tag}>[\\s\\S]*?<\\/${tag}>`));
- if (!match) {
- this.logError("Failed to parse command from content");
- return { command: "" };
- }
-
- // Extract the command from between the tags and trim whitespace
- const command = match[0]
- .replace(new RegExp(`^<${tag}>`), "")
- .replace(new RegExp(`<\\/${tag}>$`), "")
- .trim();
-
- this.logInfo(`Parsed command: ${command}`);
- return { command };
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const { command } = params as CommandParams;
-
- if (!command || command.trim().length === 0) {
- return "No valid command to execute";
- }
-
- return null;
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const { command } = params as CommandParams;
- this.logInfo(`Executing command: ${command}`);
- return this.executeCommand(command);
- } catch (error) {
- // Even on error, return success with the error message as output
- return this.createSuccessResult((error as Error).message);
- }
- }
-
- private isTestEnvironment(): boolean {
- return (
- process.env.NODE_ENV === "test" ||
- process.env.JEST_WORKER_ID !== undefined
- );
- }
-
- private async executeCommand(
- command: string,
- options?: SpawnOptionsWithoutStdio,
- ): Promise<IActionResult> {
- return new Promise((resolve) => {
- const [cmd, ...args] = command.split(" ");
- const child = spawn(cmd, args, { ...options, shell: true });
-
- let stdoutData = "";
- let stderrData = "";
- let isResolved = false;
-
- // Buffers to handle partial ANSI codes
- let stdoutBuffer = "";
- let stderrBuffer = "";
-
- // Handle standard output data
- child.stdout.on("data", (data: Buffer) => {
- const chunk = data.toString();
- stdoutBuffer += chunk;
- const strippedChunk = this.ansiStripper.strip(stdoutBuffer);
- stdoutData += strippedChunk;
- stdoutBuffer = ""; // Reset buffer after processing
-
- // Only stream to console if not in test environment
- Iif (!this.isTestEnvironment()) {
- process.stdout.write(chalk.green(chunk));
- }
- });
-
- // Handle standard error data
- child.stderr.on("data", (data: Buffer) => {
- const chunk = data.toString();
- stderrBuffer += chunk;
- const strippedChunk = this.ansiStripper.strip(stderrBuffer);
- stderrData += strippedChunk;
- stderrBuffer = ""; // Reset buffer after processing
-
- // Only stream to console if not in test environment
- Iif (!this.isTestEnvironment()) {
- process.stderr.write(chalk.red(chunk));
- }
- });
-
- const finalizeAndResolve = (exitCode: number | null = null) => {
- if (!isResolved) {
- isResolved = true;
- // Process any remaining buffered data
- Iif (stdoutBuffer) {
- stdoutData += this.ansiStripper.strip(stdoutBuffer);
- }
- Iif (stderrBuffer) {
- stderrData += this.ansiStripper.strip(stderrBuffer);
- }
-
- const extra = `CRITICAL: If you're unsure why the command failed prioritize read_file to get more context from files related to the failure and a better understanding of the problem, instead of jumping to write_file right away with a solution
- \n\n If you're stuck with the same problem over and over TRY DIFFERENT SOLUTIONS, don't keep trying the same thing over and over again`;
-
- // Add the exit code message if there's no output
- const output = stdoutData + stderrData;
- const combinedOutput =
- (output || `Command completed with exit code ${exitCode}`) +
- (exitCode === 1 ? extra : "");
-
- this.debugLogger.log("CommandAction", "Command execution completed", {
- command,
- exitCode,
- output: combinedOutput,
- });
-
- resolve(this.createSuccessResult(combinedOutput));
- }
- };
-
- child.on("close", (code) => {
- finalizeAndResolve(code);
- });
-
- child.on("error", (error) => {
- this.debugLogger.log("CommandAction", "Command execution error", {
- command,
- error: error.message,
- });
- // Return success with error message as output
- resolve(
- this.createSuccessResult(`${error.message}: command not found`),
- );
- });
- });
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 | 16x -16x -16x -16x -16x - - - - - - - - - -16x - -1x -1x - -1x - - - -22x - - - -5x - -5x -1x - -4x -1x - - -3x - - - - -8x -8x -3x - - -5x - - - - - -3x -3x - -3x -3x - -3x - - - - -2x -1x - - -1x - -1x - - - - | import { FileOperations } from "@services/FileManagement/FileOperations";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { copyFileActionBlueprint as blueprint } from "./blueprints/copyFileActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-interface CopyFileParams {
- source_path: string;
- destination_path: string;
-}
-
-@autoInjectable()
-export class CopyFileAction extends BaseAction {
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private fileOperations: FileOperations,
- ) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const { source_path, destination_path } = params as CopyFileParams;
-
- if (!source_path) {
- return "No source path provided";
- }
- if (!destination_path) {
- return "No destination path provided";
- }
-
- return null;
- }
-
- async execute(content: string): Promise<IActionResult> {
- // Validate XML structure first
- const structureError = this.actionTagsExtractor.validateStructure(content);
- if (structureError) {
- return this.createErrorResult(structureError);
- }
-
- return super.execute(content);
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const { source_path, destination_path } = params as CopyFileParams;
-
- this.logInfo(`Source path: ${source_path}`);
- this.logInfo(`Destination path: ${destination_path}`);
-
- const result = await this.fileOperations.copy(
- source_path,
- destination_path,
- );
-
- if (!result.success) {
- return this.createErrorResult(result.error!);
- }
-
- return this.createSuccessResult(result.data);
- } catch (error) {
- return this.createErrorResult(error as Error);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 | 16x -16x -16x -16x -16x - - - - - - - - -16x - -5x -5x - -5x - - - -6x - - - -5x - -5x -2x - - -3x - - - - - -3x -3x - -3x -3x - -3x -2x -2x - - -1x -1x - - - - - - - - - -5x -5x - - - - -5x -5x -2x - - -3x - - | import { FileOperations } from "@services/FileManagement/FileOperations";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { deleteFileActionBlueprint as blueprint } from "./blueprints/deleteFileActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-interface DeleteFileParams {
- path: string;
-}
-
-@autoInjectable()
-export class DeleteFileAction extends BaseAction {
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private fileOperations: FileOperations,
- ) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const { path } = params as DeleteFileParams;
-
- if (typeof path !== "string" || !path.trim()) {
- return "Invalid or no file path provided";
- }
-
- return null;
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const { path: filePath } = params as DeleteFileParams;
-
- this.logInfo(`Attempting to delete file at path: ${filePath}`);
- const result = await this.fileOperations.delete(filePath);
-
- if (!result.success) {
- this.logError(`Failed to delete file at path: ${filePath}. Error: ${result.error}`);
- return this.createErrorResult(result.error!);
- }
-
- this.logInfo(`Successfully deleted file at path: ${filePath}`);
- return this.createSuccessResult(result.data);
- } catch (error) {
- const { path: filePath } = params as DeleteFileParams;
- this.logError(`An unexpected error occurred while deleting file at path: ${filePath}. Error: ${error}`);
- return this.createErrorResult(error as Error);
- }
- }
-
- async execute(content: string): Promise<IActionResult> {
- let params: DeleteFileParams;
- try {
- params = JSON.parse(content) as DeleteFileParams;
- } catch (error) {
- return this.createErrorResult(new Error("Invalid JSON content"));
- }
-
- const validationError = this.validateParams(params);
- if (validationError) {
- return this.createErrorResult(new Error(validationError));
- }
-
- return this.executeInternal(params);
- }
-} |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 | 16x -16x -16x -16x -16x - - - - -16x - - -5x - -5x - - - - - -2x -2x -1x - -1x - - - - - -2x - - - -6x - - - | import { autoInjectable } from "tsyringe";
-import { PhaseTransitionService } from "../PhaseTransitionService";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { endPhaseActionBlueprint } from "./blueprints/endPhaseActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-@autoInjectable()
-export class EndPhaseAction extends BaseAction {
- constructor(
- actionTagsExtractor: ActionTagsExtractor,
- private phaseTransitionService: PhaseTransitionService,
- ) {
- super(actionTagsExtractor);
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const data = await this.phaseTransitionService.transitionToNextPhase();
- return this.createSuccessResult(data);
- } catch (error) {
- return this.createErrorResult(error);
- }
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- // No params to validate
- return null;
- }
-
- protected getBlueprint(): IActionBlueprint {
- return endPhaseActionBlueprint;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 | 16x -16x -16x -16x - - - - -16x -1x -1x - - - -12x - - - -5x - -5x -3x - - -2x - - - - - -2x - -2x -2x - - - - -5x -5x - - - | import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { endTaskActionBlueprint as blueprint } from "./blueprints/endTaskActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-@autoInjectable()
-export class EndTaskAction extends BaseAction {
- constructor(protected actionTagsExtractor: ActionTagsExtractor) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const content = params.content as string;
-
- if (!content?.trim()) {
- return "No message provided";
- }
-
- return null;
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- const message = params.content as string;
-
- this.logInfo(`End task message: ${message}`);
- return this.createSuccessResult(message);
- }
-
- protected parseParams(content: string): Record<string, any> {
- // Extract the content between the end_task tags
- const match = content.match(/<end_task>([\s\S]*?)<\/end_task>/);
- return { content: match?.[1]?.trim() };
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 | 16x -16x -16x -16x -16x - - - - - - - - -16x -4x -4x - - - -14x - - - -4x - -4x -1x - - -3x -3x -2x - -1x - - - - - - -2x -2x - -2x - -2x -1x - -1x - - - - - - -2x -2x -1x - -1x - - -1x - - - - | import axios, { AxiosResponse } from "axios";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { fetchUrlActionBlueprint as blueprint } from "./blueprints/fetchUrlActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-interface FetchUrlParams {
- url: string;
-}
-
-@autoInjectable()
-export class FetchUrlAction extends BaseAction {
- constructor(protected actionTagsExtractor: ActionTagsExtractor) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const { url } = params as FetchUrlParams;
-
- if (!url) {
- return "No URL provided";
- }
-
- try {
- new URL(url); // Validate URL format
- return null;
- } catch {
- return "Invalid URL format. Must be a valid URL with protocol (http:// or https://)";
- }
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const { url } = params as FetchUrlParams;
-
- this.logInfo(`Fetching URL: ${url}`);
-
- const data = await this.fetchUrl(url);
- return this.createSuccessResult(data);
- } catch (error) {
- return this.createErrorResult(
- error instanceof Error ? error : new Error("Failed to fetch URL"),
- );
- }
- }
-
- private async fetchUrl<T = any>(url: string): Promise<T> {
- try {
- const response: AxiosResponse<T> = await axios.get(url);
- return response.data;
- } catch (error: any) {
- Iif (axios.isAxiosError(error)) {
- throw new Error(`Network error: ${error.message}`);
- }
- throw new Error(`Error fetching URL: ${error.message}`);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 | 17x -17x -17x -17x -17x - - - - - - - - - -17x - -5x -5x - -5x - - - -21x - - - - -5x -5x -5x - - - - -5x -5x - - - -5x - - - - -5x -10x -8x - - -5x - - - - - - -5x - -5x -1x - -4x -1x - - -3x - - - - - -3x -3x - -3x -3x - -3x - - - - -3x -1x - - -2x - - - - - - | import { FileOperations } from "@services/FileManagement/FileOperations";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { moveFileActionBlueprint as blueprint } from "./blueprints/moveFileActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-interface MoveFileParams {
- source_path: string;
- destination_path: string;
-}
-
-@autoInjectable()
-export class MoveFileAction extends BaseAction {
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private fileOperations: FileOperations,
- ) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected parseParams(content: string): Record<string, any> {
- // First extract the content from the outer tag
- const tag = this.getBlueprint().tag;
- const match = content.match(new RegExp(`<${tag}>[\\s\\S]*?<\\/${tag}>`));
- Iif (!match) {
- this.logError("Failed to parse move file content");
- return { source_path: "", destination_path: "" };
- }
-
- const tagContent = match[0];
- const source_path = this.actionTagsExtractor.extractTag(
- tagContent,
- "source_path",
- );
- const destination_path = this.actionTagsExtractor.extractTag(
- tagContent,
- "destination_path",
- );
-
- const getValue = (value: string | string[] | null): string => {
- if (!value) return "";
- return Array.isArray(value) ? value[0]?.trim() || "" : value.trim();
- };
-
- return {
- source_path: getValue(source_path),
- destination_path: getValue(destination_path),
- };
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const { source_path, destination_path } = params as MoveFileParams;
-
- if (!source_path) {
- return "No source path provided";
- }
- if (!destination_path) {
- return "No destination path provided";
- }
-
- return null;
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const { source_path, destination_path } = params as MoveFileParams;
-
- this.logInfo(`Source path: ${source_path}`);
- this.logInfo(`Destination path: ${destination_path}`);
-
- const result = await this.fileOperations.move(
- source_path,
- destination_path,
- );
-
- if (!result.success) {
- return this.createErrorResult(result.error!);
- }
-
- return this.createSuccessResult(result.data);
- } catch (error) {
- return this.createErrorResult(error as Error);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 | 16x - - -16x -16x -16x -16x -16x - - - - - - - -16x - -1x -1x -1x - -1x - - - -25x - - - -7x - -7x -1x - - -9x -6x -1x - - -5x - - - - - - -7x - -7x -7x -9x -7x - - - - - - - - - - -5x - - -5x -2x - - - -3x - - - -2x -2x - - - - - -3x -3x - -3x -2x - - - - -1x -1x - -1x - - - -2x -1x - -1x - - -2x - -2x - - - - - - - | import { FileOperations } from "@services/FileManagement/FileOperations";
-import { IFileOperationResult } from "@services/FileManagement/types/FileManagementTypes";
-import { IActionResult } from "@services/LLM/actions/types/ActionTypes";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { readFileActionBlueprint as blueprint } from "./blueprints/readFileActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-
-interface IReadFileParams {
- path: string[];
-}
-
-@autoInjectable()
-export class ReadFileAction extends BaseAction {
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private fileOperations: FileOperations,
- private debugLogger: DebugLogger,
- ) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected validateParams(params: IReadFileParams): string | null {
- const paths = params.path;
-
- if (!paths || !Array.isArray(paths) || paths.length === 0) {
- return "Must include at least one <path> tag";
- }
-
- const invalidPaths = paths.filter((path) => !path);
- if (invalidPaths.length > 0) {
- return `Invalid paths found: ${invalidPaths.join(", ")}. Try using a <search_file> to find the correct file path.`;
- }
-
- return null;
- }
-
- protected extractParamValue(
- content: string,
- paramName: string,
- ): string | string[] | null {
- if (paramName === "path") {
- // Extract all path tags
- const regex = new RegExp(`<${paramName}>(.*?)</${paramName}>`, "g");
- const matches = Array.from(content.matchAll(regex));
- const paths = matches.map((match) => match[1].trim());
- return paths.length > 0 ? paths : null;
- }
-
- // Use default extraction for other parameters
- return super.extractParamValue(content, paramName);
- }
-
- protected async executeInternal(
- params: IReadFileParams,
- ): Promise<IActionResult> {
- // Extract paths either from path tags or from raw content
- const filePaths = Array.isArray(params.path) ? params.path : [params.path];
-
- // If only one path, use single file read
- if (filePaths.length === 1) {
- return await this.handleSingleFile(filePaths[0]);
- }
-
- // Handle multiple files
- return await this.handleMultipleFiles(filePaths);
- }
-
- private async handleSingleFile(filePath: string): Promise<IActionResult> {
- const result = await this.fileOperations.read(filePath);
- return this.convertFileResult(result);
- }
-
- private async handleMultipleFiles(
- filePaths: string[],
- ): Promise<IActionResult> {
- const result = await this.fileOperations.readMultiple(filePaths);
- this.debugLogger.log("ReadFileAction", "execute", result);
-
- if (!result.success || !result.data) {
- return this.createErrorResult(
- result.error || "Failed to read multiple files",
- );
- }
-
- this.logSuccess("Action completed successfully. Please wait...\n");
- console.log("-".repeat(50));
-
- return this.createSuccessResult(result.data);
- }
-
- private convertFileResult(result: IFileOperationResult): IActionResult {
- if (result?.success) {
- this.logSuccess("Action completed successfully. Please wait...\n");
- } else {
- this.logError(result?.error?.message || "Unknown error");
- }
-
- console.log("-".repeat(50));
-
- return {
- success: !!result?.success,
- data: result?.data,
- error: result?.error,
- };
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 | 17x -17x -17x -17x -17x -17x - - - - - - - - - - -17x - -9x -9x - -9x - - - -29x - - - - -8x -8x -8x - - - - -8x -8x - - - -8x -8x - - - - -8x -18x -16x - - -8x - - - - - - - -8x - -8x -1x - -7x -1x - -6x -1x - - -5x - - - - - -5x - - - - -5x - - -5x - - -5x - - -5x - - - - -4x - -3x - -3x - - - -3x - - - - - -3x - - -3x - - -1x -1x - -1x -1x - - - - | import { PathAdjuster } from "@services/FileManagement/PathAdjuster";
-import path from "path";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { relativePathLookupActionBlueprint as blueprint } from "./blueprints/relativePathLookupActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-interface RelativePathLookupParams {
- source_path: string;
- path: string;
- threshold?: number;
-}
-
-@autoInjectable()
-export class RelativePathLookupAction extends BaseAction {
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private pathAdjuster: PathAdjuster,
- ) {
- super(actionTagsExtractor);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected parseParams(content: string): Record<string, any> {
- // First extract the content from the outer tag
- const tag = this.getBlueprint().tag;
- const match = content.match(new RegExp(`<${tag}>[\\s\\S]*?<\\/${tag}>`));
- Iif (!match) {
- this.logError("Failed to parse relative path lookup content");
- return { source_path: "", path: "", threshold: undefined };
- }
-
- const tagContent = match[0];
- const source_path = this.actionTagsExtractor.extractTag(
- tagContent,
- "source_path",
- );
- const path = this.actionTagsExtractor.extractTag(tagContent, "path");
- const threshold = this.actionTagsExtractor.extractTag(
- tagContent,
- "threshold",
- );
-
- const getValue = (value: string | string[] | null): string => {
- if (!value) return "";
- return Array.isArray(value) ? value[0]?.trim() || "" : value.trim();
- };
-
- return {
- source_path: getValue(source_path),
- path: getValue(path),
- threshold: threshold ? parseFloat(getValue(threshold)) : undefined,
- };
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const { source_path, path, threshold } = params as RelativePathLookupParams;
-
- if (!source_path) {
- return "No source_path provided";
- }
- if (!path) {
- return "No path provided";
- }
- if (threshold !== undefined && (threshold <= 0 || threshold > 1)) {
- return "Threshold must be between 0 and 1";
- }
-
- return null;
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const {
- source_path,
- path: relativePath,
- threshold = 0.6,
- } = params as RelativePathLookupParams;
-
- // Get the directory of the source file to resolve relative paths from
- const sourceDir = path.dirname(source_path);
-
- // Resolve the full path of the import relative to the source file
- const fullImportPath = path.resolve(sourceDir, relativePath);
-
- // Use PathAdjuster to find the correct path
- const adjustedPath = await this.pathAdjuster.adjustPath(
- fullImportPath,
- threshold,
- );
-
- if (adjustedPath) {
- // Convert the adjusted absolute path back to a relative path from the source file
- const newRelativePath = path.relative(sourceDir, adjustedPath);
- // Ensure proper directory separator and add ./ if needed
- const formattedPath = newRelativePath.startsWith(".")
- ? newRelativePath
- : "./" + newRelativePath;
-
- const result = {
- originalPath: relativePath,
- newPath: formattedPath.replace(/\\/g, "/"), // Ensure forward slashes
- absolutePath: adjustedPath,
- };
-
- this.logSuccess(
- `Found adjusted path: ${result.newPath} (absolute: ${result.absolutePath})`,
- );
- return this.createSuccessResult(result);
- }
-
- this.logInfo("No adjusted path found");
- return this.createSuccessResult(null);
- } catch (error) {
- this.logError(`Path lookup failed: ${(error as Error).message}`);
- return this.createErrorResult(error as Error);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 | 17x -17x -17x -17x - - - -17x - - - - - - - - - - -17x -7x - - -7x -7x - -7x - - - - -7x - - -7x - - - -41x - - - - - - -7x -7x -7x - - - - -7x -7x - - - -7x - -7x -14x -12x - - -7x - - - - - - -7x - -7x -1x - -6x -1x - - -5x - - - - - -5x -5x - -5x -5x -5x - - -5x -2x - -3x - - -5x -2x -2x - - -3x -3x - - - - - - - | import { FileSearch } from "@services/FileManagement/FileSearch";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import {
- searchFileActionBlueprint,
- searchStringActionBlueprint,
-} from "./blueprints/searchActionsBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-interface SearchParams {
- directory: string;
- term: string;
- type: "search_string" | "search_file";
-}
-
-@autoInjectable()
-export class SearchAction extends BaseAction {
- private currentType: "search_string" | "search_file" = "search_file";
-
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private fileSearch: FileSearch,
- ) {
- super(actionTagsExtractor);
- }
-
- async execute(content: string): Promise<IActionResult> {
- // Determine action type from the content
- this.currentType = content.includes("<search_string>")
- ? "search_string"
- : "search_file";
- return super.execute(content);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return this.currentType === "search_string"
- ? searchStringActionBlueprint
- : searchFileActionBlueprint;
- }
-
- protected parseParams(content: string): Record<string, any> {
- // First extract the content from the outer tag (search_string or search_file)
- const tag = this.getBlueprint().tag;
- const match = content.match(new RegExp(`<${tag}>[\\s\\S]*?<\\/${tag}>`));
- Iif (!match) {
- this.logError("Failed to parse search content");
- return { directory: "", term: "" };
- }
-
- const tagContent = match[0];
- const directory = this.actionTagsExtractor.extractTag(
- tagContent,
- "directory",
- );
- const term = this.actionTagsExtractor.extractTag(tagContent, "term");
-
- const getValue = (value: string | string[] | null): string => {
- if (!value) return "";
- return Array.isArray(value) ? value[0]?.trim() || "" : value.trim();
- };
-
- return {
- directory: getValue(directory),
- term: getValue(term),
- };
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- const { directory, term } = params as SearchParams;
-
- if (!directory) {
- return "No directory provided";
- }
- if (!term) {
- return "No search term provided";
- }
-
- return null;
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const { directory, term } = params as SearchParams;
-
- this.logInfo(`Searching in directory: ${directory}`);
- this.logInfo(`Search term: ${term}`);
- this.logInfo(`Search type: ${this.currentType}`);
-
- let results;
- if (this.currentType === "search_string") {
- results = await this.fileSearch.findByContent(term, directory);
- } else {
- results = await this.fileSearch.findByName(term, directory);
- }
-
- if (!results || results.length === 0) {
- this.logInfo("No results found");
- return this.createSuccessResult([]);
- }
-
- this.logSuccess(`Found ${results.length} results`);
- return this.createSuccessResult(results);
- } catch (error) {
- this.logError(`Search failed: ${(error as Error).message}`);
- return this.createErrorResult(error as Error);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 | 17x -17x -17x -17x -17x -17x -17x -17x -17x - - - - - - - - -17x -17x -17x - - -17x - -24x -24x -24x -24x - -24x - - - -4x - - - -9x - - - -56x - - - -59x - - - -22x -22x - -22x -1x - -21x -1x - - - -20x -1x - - - -19x -19x -1x - - -18x - - - - - - - - - - -44x - -44x -22x -22x -1x - - -1x - -21x - - - -22x -22x - - - - - - - - - -18x - -18x - - -18x -18x - - -18x -18x -1x - - - -17x - - -17x -2x - - - - - -15x - -15x -1x -1x - - -14x - - -14x - - - - - - - - - - - - -18x -18x -18x -11x -11x - - -7x -7x - -7x -7x -1x - - -1x - - -6x -6x - - - - -6x - - - -6x -1x - - - - -5x - - - - - - - - - - - - -6x -6x - -6x - -5x -5x - - - -17x - -17x -1x -1x - - - -16x - -28x -16x -1x -1x - - -15x - - - - - - - | import { BLOCK_WRITE_IF_CONTENT_REMOVAL_THRESHOLD } from "@constants/writeConstants";
-import { FileOperations } from "@services/FileManagement/FileOperations";
-import { ModelScaler } from "@services/LLM/ModelScaler";
-import { HtmlEntityDecoder } from "@services/text/HTMLEntityDecoder";
-import path from "path";
-import { autoInjectable } from "tsyringe";
-import { ActionTagsExtractor } from "./ActionTagsExtractor";
-import { writeFileActionBlueprint as blueprint } from "./blueprints/writeFileActionBlueprint";
-import { BaseAction } from "./core/BaseAction";
-import { IActionBlueprint } from "./core/IAction";
-import { IActionResult } from "./types/ActionTypes";
-
-interface WriteFileParams {
- path: string;
- content: string;
-}
-
-const MAX_CONTENT_SIZE_MB = 10;
-const MAX_CONTENT_SIZE_BYTES = MAX_CONTENT_SIZE_MB * 1024 * 1024;
-const MAX_LINE_LENGTH = 10000;
-
-@autoInjectable()
-export class WriteFileAction extends BaseAction {
- constructor(
- protected actionTagsExtractor: ActionTagsExtractor,
- private fileOperations: FileOperations,
- private htmlEntityDecoder: HtmlEntityDecoder,
- private modelScaler: ModelScaler,
- ) {
- super(actionTagsExtractor);
- }
-
- protected logWarning(message: string): void {
- console.warn(`⚠️ write_file: ${message}`);
- }
-
- protected logError(message: string): void {
- console.error(`❌ write_file: ${message}`);
- }
-
- protected logInfo(message: string): void {
- console.info(`ℹ️ write_file: ${message}`);
- }
-
- protected getBlueprint(): IActionBlueprint {
- return blueprint;
- }
-
- protected validateParams(params: Record<string, any>): string | null {
- try {
- const { path: filePath, content } = params as WriteFileParams;
-
- if (!filePath) {
- return "No file path provided";
- }
- if (!content) {
- return "No file content provided";
- }
-
- // Validate path
- if (filePath.includes("..")) {
- return "Path traversal is not allowed";
- }
-
- // Validate content size
- const contentSizeBytes = Buffer.byteLength(content, "utf8");
- if (contentSizeBytes > MAX_CONTENT_SIZE_BYTES) {
- return `Content size (${(contentSizeBytes / 1024 / 1024).toFixed(2)}MB) exceeds maximum allowed size (${MAX_CONTENT_SIZE_MB}MB)`;
- }
-
- return null;
- } catch (error) {
- this.logError(`Error in validateParams: ${error}`);
- return `Validation error: ${error instanceof Error ? error.message : String(error)}`;
- }
- }
-
- protected extractParamValue(
- content: string,
- paramName: string,
- ): string | string[] | null {
- try {
- // Handle content parameter specially to preserve nested tags
- if (paramName === "content") {
- const contentMatch = content.match(/<content>([\s\S]*?)<\/content>/);
- if (!contentMatch) {
- this.logWarning(
- `Failed to extract content parameter from: ${content.substring(0, 100)}...`,
- );
- return null;
- }
- return contentMatch[1].trim();
- }
-
- // Use default extraction for other parameters
- const value = super.extractParamValue(content, paramName);
- return Array.isArray(value) ? value[0] : value;
- } catch (error) {
- this.logError(`Error extracting parameter ${paramName}: ${error}`);
- return null;
- }
- }
-
- protected async executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult> {
- try {
- const { path: rawFilePath, content: fileContent } =
- params as WriteFileParams;
-
- // Normalize path
- const filePath = path.normalize(rawFilePath).replace(/\\/g, "/");
- this.logInfo(`Writing to file: ${filePath}`);
-
- // Check for large content removal if file exists
- const removalCheck = await this.checkLargeRemoval(filePath, fileContent);
- if (!removalCheck.success) {
- return removalCheck;
- }
-
- // Decode and validate content
- const decodedContent = this.htmlEntityDecoder.decode(fileContent, {
- unescapeChars: ['"'],
- });
- if (!this.isValidContent(decodedContent)) {
- return this.createErrorResult(
- "Invalid content detected after decoding",
- );
- }
-
- // Write file
- const result = await this.fileOperations.write(filePath, decodedContent);
-
- if (!result.success) {
- this.logError(`Failed to write file ${filePath}: ${result.error}`);
- return this.createErrorResult(result.error!);
- }
-
- this.logInfo(
- `Successfully wrote ${Buffer.byteLength(decodedContent, "utf8")} bytes to ${filePath}`,
- );
- return this.createSuccessResult();
- } catch (error) {
- this.logError(`Unexpected error in WriteFileAction: ${error}`);
- return this.createErrorResult(
- `Unexpected error: ${error instanceof Error ? error.message : String(error)}`,
- );
- }
- }
-
- private async checkLargeRemoval(
- filePath: string,
- newContent: string,
- ): Promise<IActionResult> {
- try {
- const exists = await this.fileOperations.exists(filePath);
- if (!exists) {
- this.logInfo(`Creating new file: ${filePath}`);
- return this.createSuccessResult();
- }
-
- this.logInfo(`File exists at: ${filePath}`);
- this.modelScaler.incrementTryCount(filePath);
-
- const readResult = await this.fileOperations.read(filePath);
- if (!readResult.success) {
- this.logWarning(
- `Could not read existing file ${filePath}: ${readResult.error}`,
- );
- return this.createSuccessResult();
- }
-
- const existingContent = readResult.data as string;
- const removalPercentage = this.calculateRemovalPercentage(
- existingContent,
- newContent,
- );
-
- this.logInfo(
- `Content removal percentage: ${removalPercentage.toFixed(1)}%`,
- );
-
- if (removalPercentage > BLOCK_WRITE_IF_CONTENT_REMOVAL_THRESHOLD) {
- return this.createErrorResult(
- `Prevented removal of ${removalPercentage.toFixed(1)}% of file content. This appears to be a potential error. Please review the changes and ensure only necessary modifications are made.`,
- );
- }
-
- return this.createSuccessResult();
- } catch (error) {
- this.logError(`Error in checkLargeRemoval: ${error}`);
- return this.createErrorResult(
- `Error checking content removal: ${error instanceof Error ? error.message : String(error)}`,
- );
- }
- }
-
- private calculateRemovalPercentage(
- existingContent: string,
- newContent: string,
- ): number {
- const existingLength = existingContent.trim().length;
- const newLength = newContent.trim().length;
-
- if (existingLength === 0) return 0;
-
- const removedLength = Math.max(0, existingLength - newLength);
- return (removedLength / existingLength) * 100;
- }
-
- private isValidContent(content: string): boolean {
- try {
- // Check for null bytes and other potentially dangerous content
- if (content.includes("\0")) {
- this.logWarning("Content contains null bytes");
- return false;
- }
-
- // Check for reasonable line length
- const hasExcessiveLines = content
- .split("\n")
- .some((line) => line.length > MAX_LINE_LENGTH);
- if (hasExcessiveLines) {
- this.logWarning("Content contains excessively long lines");
- return false;
- }
-
- return true;
- } catch (error) {
- this.logError(`Error in isValidContent: ${error}`);
- return false;
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 | 16x - -16x - -16x - - - - - - - - - | import { CommandAction } from "../CommandAction";
-import { IActionBlueprint } from "../core/IAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const commandActionBlueprint: IActionBlueprint = {
- tag: "execute_command",
- class: CommandAction,
- description: "Executes a system command with output streaming",
- priority: ActionPriority.LOW,
- canRunInParallel: false,
- requiresProcessing: true,
- parameters: [], // No parameters since we extract command from tag content
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 | 16x - -16x - -16x - - - - - - - - - - - - - - - - - - - - - - | import { CopyFileAction } from "../CopyFileAction";
-import { IActionBlueprint } from "../core/IAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const copyFileActionBlueprint: IActionBlueprint = {
- tag: "copy_file",
- class: CopyFileAction,
- description: "Copy a file from source to destination path",
- priority: ActionPriority.MEDIUM,
- canRunInParallel: true,
- requiresProcessing: false,
- parameters: [
- {
- name: "source_path",
- required: true,
- description: "The source path of the file to copy",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- {
- name: "destination_path",
- required: true,
- description: "The destination path for the file copy",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 | -16x -16x - -16x - - - - - - - - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { DeleteFileAction } from "../DeleteFileAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const deleteFileActionBlueprint: IActionBlueprint = {
- tag: "delete_file",
- class: DeleteFileAction,
- description: "Delete a file at the specified path",
- priority: ActionPriority.MEDIUM,
- canRunInParallel: false,
- requiresProcessing: false,
- parameters: [
- {
- name: "path",
- required: true,
- description: "The path of the file to delete",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 | -16x -16x - -16x - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { EndPhaseAction } from "../EndPhaseAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const endPhaseActionBlueprint: IActionBlueprint = {
- tag: "end_phase",
- class: EndPhaseAction,
- description: "Ends the current phase and transitions to the next phase",
- parameters: [],
- requiresProcessing: true,
- priority: ActionPriority.CRITICAL, // High priority to ensure phase transition happens before other actions
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 | -16x -16x - -16x - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { EndTaskAction } from "../EndTaskAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const endTaskActionBlueprint: IActionBlueprint = {
- tag: "end_task",
- class: EndTaskAction,
- description: "Mark a task as complete with a message",
- priority: ActionPriority.LOWEST,
- canRunInParallel: false,
- requiresProcessing: false,
- parameters: [],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 | -16x -16x - -16x - - - - - - - - - - - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { FetchUrlAction } from "../FetchUrlAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const fetchUrlActionBlueprint: IActionBlueprint = {
- tag: "fetch_url",
- class: FetchUrlAction,
- description: "Fetch content from a URL",
- priority: ActionPriority.LOW,
- canRunInParallel: true,
- requiresProcessing: true,
- parameters: [
- {
- name: "url",
- required: true,
- description: "The URL to fetch content from",
- validator: (value: any) =>
- typeof value === "string" &&
- value.length > 0 &&
- (value.startsWith("http://") || value.startsWith("https://")),
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| commandActionBlueprint.ts | -
-
- |
- 100% | -3/3 | -100% | -0/0 | -100% | -0/0 | -100% | -3/3 | -
| copyFileActionBlueprint.ts | -
-
- |
- 60% | -3/5 | -0% | -0/4 | -0% | -0/2 | -60% | -3/5 | -
| deleteFileActionBlueprint.ts | -
-
- |
- 75% | -3/4 | -0% | -0/2 | -0% | -0/1 | -75% | -3/4 | -
| endPhaseActionBlueprint.ts | -
-
- |
- 100% | -3/3 | -100% | -0/0 | -100% | -0/0 | -100% | -3/3 | -
| endTaskActionBlueprint.ts | -
-
- |
- 100% | -3/3 | -100% | -0/0 | -100% | -0/0 | -100% | -3/3 | -
| fetchUrlActionBlueprint.ts | -
-
- |
- 75% | -3/4 | -0% | -0/4 | -0% | -0/1 | -75% | -3/4 | -
| index.ts | -
-
- |
- 100% | -21/21 | -100% | -3/3 | -100% | -4/4 | -100% | -21/21 | -
| moveFileActionBlueprint.ts | -
-
- |
- 60% | -3/5 | -0% | -0/4 | -0% | -0/2 | -60% | -3/5 | -
| readFileActionBlueprint.ts | -
-
- |
- 75% | -3/4 | -0% | -0/2 | -0% | -0/1 | -75% | -3/4 | -
| relativePathLookupActionBlueprint.ts | -
-
- |
- 50% | -3/6 | -0% | -0/8 | -0% | -0/3 | -50% | -3/6 | -
| searchActionsBlueprint.ts | -
-
- |
- 50% | -4/8 | -0% | -0/8 | -0% | -0/4 | -50% | -4/8 | -
| writeFileActionBlueprint.ts | -
-
- |
- 60% | -3/5 | -0% | -0/2 | -0% | -0/2 | -60% | -3/5 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 | -16x -16x -16x -16x -16x -16x -16x -16x -16x -16x - - - -16x - -16x - - - - - - - - - - - - - - - - - - -16x -122x - - - -16x -133x - - - -16x -1596x -1596x - - - -16x -133x - - | import { IActionBlueprint } from "../core/IAction";
-import { commandActionBlueprint } from "./commandActionBlueprint";
-import { copyFileActionBlueprint } from "./copyFileActionBlueprint";
-import { deleteFileActionBlueprint } from "./deleteFileActionBlueprint";
-import { endPhaseActionBlueprint } from "./endPhaseActionBlueprint";
-import { endTaskActionBlueprint } from "./endTaskActionBlueprint";
-import { fetchUrlActionBlueprint } from "./fetchUrlActionBlueprint";
-import { moveFileActionBlueprint } from "./moveFileActionBlueprint";
-import { readFileActionBlueprint } from "./readFileActionBlueprint";
-import { relativePathLookupActionBlueprint } from "./relativePathLookupActionBlueprint";
-import {
- searchFileActionBlueprint,
- searchStringActionBlueprint,
-} from "./searchActionsBlueprint";
-import { writeFileActionBlueprint } from "./writeFileActionBlueprint";
-
-export const actionsBlueprints = {
- write_file: writeFileActionBlueprint,
- read_file: readFileActionBlueprint,
- execute_command: commandActionBlueprint,
- search_string: searchStringActionBlueprint,
- search_file: searchFileActionBlueprint,
- end_task: endTaskActionBlueprint,
- end_phase: endPhaseActionBlueprint,
- relative_path_lookup: relativePathLookupActionBlueprint,
- delete_file: deleteFileActionBlueprint,
- move_file: moveFileActionBlueprint,
- fetch_url: fetchUrlActionBlueprint,
- copy_file: copyFileActionBlueprint,
-} as const;
-
-// Infer action types from blueprints
-export type ActionTag = keyof typeof actionsBlueprints;
-
-// Helper to get blueprint by tag
-export function getBlueprint(tag: ActionTag): IActionBlueprint {
- return actionsBlueprints[tag];
-}
-
-// Helper to get all registered action tags
-export function getActionTags(): ActionTag[] {
- return Object.keys(actionsBlueprints) as ActionTag[];
-}
-
-// Helper to check if an action is fully implemented
-export function isActionImplemented(tag: ActionTag): boolean {
- const blueprint = actionsBlueprints[tag];
- return !!(blueprint && blueprint.class && blueprint.tag);
-}
-
-// Get only implemented actions
-export function getImplementedActions(): ActionTag[] {
- return getActionTags().filter(isActionImplemented);
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 | -17x -17x - -17x - - - - - - - - - - - - - - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { MoveFileAction } from "../MoveFileAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const moveFileActionBlueprint: IActionBlueprint = {
- tag: "move_file",
- class: MoveFileAction,
- description: "Move a file from source to destination path",
- priority: ActionPriority.MEDIUM,
- canRunInParallel: false,
- requiresProcessing: false,
- parameters: [
- {
- name: "source_path",
- required: true,
- description: "The source path of the file to move",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- {
- name: "destination_path",
- required: true,
- description: "The destination path for the file",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 | -16x -16x - -16x - - - - - - - - - - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { ReadFileAction } from "../ReadFileAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const readFileActionBlueprint: IActionBlueprint = {
- tag: "read_file",
- class: ReadFileAction,
- description: "Reads content from one or more files",
- priority: ActionPriority.CRITICAL,
- canRunInParallel: true,
- requiresProcessing: true,
- parameters: [
- {
- name: "path",
- required: true,
- description:
- "The path(s) of the file(s) to read. Can specify multiple path tags.",
- validator: (value: unknown): value is string =>
- typeof value === "string" && value.length > 0,
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 | -17x -17x - -17x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { RelativePathLookupAction } from "../RelativePathLookupAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const relativePathLookupActionBlueprint: IActionBlueprint = {
- tag: "relative_path_lookup",
- class: RelativePathLookupAction,
- description: "Adjust and validate relative file paths",
- priority: ActionPriority.HIGH,
- canRunInParallel: true,
- requiresProcessing: true,
- parameters: [
- {
- name: "source_path",
- required: true,
- description: "The source file path to resolve relative paths from",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- {
- name: "path",
- required: true,
- description: "The relative path to adjust",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- {
- name: "threshold",
- required: false,
- description: "Similarity threshold for path matching (default: 0.6)",
- validator: (value: any) =>
- !value || (typeof value === "number" && value > 0 && value <= 1),
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 | -17x -17x - -17x - - - - - - - - - - - - - - - - - - - - - - -17x - - - - - - - - - - - - - - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { SearchAction } from "../SearchAction";
-import { ActionPriority } from "../types/ActionPriority";
-
-export const searchStringActionBlueprint: IActionBlueprint = {
- tag: "search_string",
- class: SearchAction,
- description: "Search for content within files",
- priority: ActionPriority.HIGH,
- canRunInParallel: true,
- requiresProcessing: true,
- parameters: [
- {
- name: "directory",
- required: true,
- description: "The directory to search in",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- {
- name: "term",
- required: true,
- description: "The content to search for",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- ],
-};
-
-export const searchFileActionBlueprint: IActionBlueprint = {
- tag: "search_file",
- class: SearchAction,
- description: "Search for files by name",
- priority: ActionPriority.HIGH,
- canRunInParallel: true,
- requiresProcessing: true,
- parameters: [
- {
- name: "directory",
- required: true,
- description: "The directory to search in",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- {
- name: "term",
- required: true,
- description: "The filename pattern to search for",
- validator: (value: any) => typeof value === "string" && value.length > 0,
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 | -17x -17x - -17x - - - - - - - - - - - - - - - - - - - - - - - - - | import { IActionBlueprint } from "../core/IAction";
-import { ActionPriority } from "../types/ActionPriority";
-import { WriteFileAction } from "../WriteFileAction";
-
-export const writeFileActionBlueprint: IActionBlueprint = {
- tag: "write_file",
- class: WriteFileAction,
- description:
- "Writes content to a file with safety checks for content removal",
- priority: ActionPriority.MEDIUM,
- canRunInParallel: false,
- requiresProcessing: false,
- parameters: [
- {
- name: "path",
- required: true,
- description: "The path where the file will be written",
- validator: (value: unknown): value is string =>
- typeof value === "string" && value.length > 0,
- },
- {
- name: "content",
- required: true,
- description:
- "The content to write to the file. Must not remove too much existing content if file exists.",
- validator: (value: unknown): value is string => typeof value === "string",
- },
- ],
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 | 6x -6x - - - - - - - -6x -100x -100x - - - -100x -100x -1200x -1200x -1200x - - - - - -2x - - - - - - - - -2x - - - -2x -2x - - - -2x - -2x -2x -2x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { autoInjectable, container } from "tsyringe";
-import {
- actionsBlueprints,
- ActionTag,
- getImplementedActions,
-} from "../blueprints";
-import { IAction, IActionBlueprint } from "./IAction";
-
-@autoInjectable()
-export class ActionFactory {
- private blueprintCache: Map<string, IActionBlueprint> = new Map();
- private instanceCache: Map<string, IAction> = new Map();
-
- constructor() {
- // Initialize blueprint cache with only implemented actions
- const implementedActions = getImplementedActions();
- implementedActions.forEach((tag) => {
- const blueprint = actionsBlueprints[tag];
- if (blueprint) {
- this.blueprintCache.set(tag, blueprint);
- }
- });
- }
-
- getBlueprint(tag: string): IActionBlueprint | undefined {
- return this.blueprintCache.get(tag);
- }
-
- getAllBlueprints(): IActionBlueprint[] {
- return Array.from(this.blueprintCache.values());
- }
-
- createAction(tag: ActionTag): IAction | undefined {
- // Check instance cache first
- Iif (this.instanceCache.has(tag)) {
- return this.instanceCache.get(tag);
- }
-
- const blueprint = this.getBlueprint(tag);
- Iif (!blueprint || !blueprint.class) {
- return undefined;
- }
-
- try {
- // Create new instance using tsyringe container
- const instance = container.resolve(blueprint.class) as IAction;
- this.instanceCache.set(tag, instance);
- return instance;
- } catch (error) {
- console.error(`Failed to create action instance for ${tag}:`, error);
- return undefined;
- }
- }
-
- validateParameters(tag: string, params: Record<string, any>): string | null {
- const blueprint = this.getBlueprint(tag);
- Iif (!blueprint) {
- return `Unknown action type: ${tag}`;
- }
-
- Iif (!blueprint.parameters) {
- return null;
- }
-
- for (const param of blueprint.parameters) {
- Iif (param.required && !(param.name in params)) {
- return `Missing required parameter: ${param.name}`;
- }
-
- Iif (param.validator && params[param.name] !== undefined) {
- const isValid = param.validator(params[param.name]);
- Iif (!isValid) {
- return `Invalid value for parameter: ${param.name}`;
- }
- }
- }
-
- return null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 | 7x -7x -7x - - - - - - - - - - - -7x -26x -26x - - -24x -24x -24x - -24x - - - - - - - -24x - - - -21x -1x - - - -20x - -20x - - - - -20x -12x -12x - - -20x - - - -8x -7x - - -8x -7x - -7x - -5x - -5x -2x - - - - - -1x - - - - - -5x - - - -7x - - - -6x - - - -17x -17x - - - | import { singleton } from "tsyringe";
-import { ActionTag, getBlueprint } from "../blueprints";
-import { ActionPriority } from "../types/ActionPriority";
-import { IActionResult } from "../types/ActionTypes";
-
-interface QueuedAction {
- type: string;
- content: string;
- priority: ActionPriority;
- requiresProcessing: boolean;
- result?: IActionResult;
-}
-
-@singleton()
-export class ActionQueue {
- private queue: QueuedAction[] = [];
- private processedResults: Map<string, unknown> = new Map();
-
- enqueue(type: string, content: string): void {
- const blueprint = getBlueprint(type as ActionTag);
- const priority = blueprint?.priority || ActionPriority.LOW;
- const requiresProcessing = blueprint?.requiresProcessing || false;
-
- this.queue.push({
- type,
- content,
- priority,
- requiresProcessing,
- });
-
- // Sort queue by priority (lower numbers = higher priority)
- this.queue.sort((a, b) => a.priority - b.priority);
- }
-
- dequeue(): QueuedAction | undefined {
- if (this.queue.length === 0) {
- return undefined;
- }
-
- // Get the highest priority action that hasn't been processed
- const nextAction = this.queue.find((action) => !action.result);
-
- Iif (!nextAction) {
- return undefined;
- }
-
- // If it doesn't require processing, remove it from the queue
- if (!nextAction.requiresProcessing) {
- const index = this.queue.indexOf(nextAction);
- this.queue.splice(index, 1);
- }
-
- return nextAction;
- }
-
- setActionResult(type: string, content: string, result: IActionResult): void {
- const actionIndex = this.queue.findIndex(
- (action) => action.type === type && action.content === content,
- );
-
- if (actionIndex !== -1) {
- const action = this.queue[actionIndex];
-
- if (action.requiresProcessing && result.success) {
- // Store processed results
- this.processedResults.set(`${type}:${content}`, result.data);
- // Remove the processed action from the queue
- this.queue.splice(actionIndex, 1);
- } else if (action.requiresProcessing && !result.success) {
- // Optionally, you can handle failed processing here
- // For example, you might want to retry or log the failure
- // Currently, the action remains in the queue for potential reprocessing
- } else {
- // If the action does not require processing, remove it
- this.queue.splice(actionIndex, 1);
- }
- }
- }
-
- getProcessedResults(): Map<string, unknown> {
- return this.processedResults;
- }
-
- isEmpty(): boolean {
- return this.queue.length === 0;
- }
-
- size(): number {
- return this.queue.length;
- }
-
- clear(): void {
- this.queue = [];
- this.processedResults.clear();
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 | - - - -20x -71x - - - - - - - - -40x - -65x - -40x -40x -65x -65x -59x - - - -40x - - - - - - - -36x - - - -74x - -74x -74x - - - - -74x - - -74x -74x -21x -21x - - - -53x - - -53x -40x - -13x - - -53x - - - - - - - - - -31x - - - -54x - - - -48x - - - -57x - - - - - - -40x -40x - - - - - - | import { ActionTagsExtractor } from "../ActionTagsExtractor";
-import { IActionResult } from "../types/ActionTypes";
-import { IAction, IActionBlueprint, IActionParameter } from "./IAction";
-
-export abstract class BaseAction implements IAction {
- constructor(protected actionTagsExtractor: ActionTagsExtractor) {}
-
- protected abstract executeInternal(
- params: Record<string, any>,
- ): Promise<IActionResult>;
- protected abstract validateParams(params: Record<string, any>): string | null;
- protected abstract getBlueprint(): IActionBlueprint;
-
- protected parseParams(content: string): Record<string, any> {
- const blueprint = this.getBlueprint();
- const paramNames =
- blueprint.parameters?.map((p: IActionParameter) => p.name) || [];
-
- const params: Record<string, any> = {};
- for (const paramName of paramNames) {
- const value = this.extractParamValue(content, paramName);
- if (value !== null) {
- params[paramName] = value;
- }
- }
-
- return params;
- }
-
- // Allow actions to override this method for custom parameter extraction
- protected extractParamValue(
- content: string,
- paramName: string,
- ): string | string[] | null {
- return this.actionTagsExtractor.extractTag(content, paramName);
- }
-
- async execute(content: string): Promise<IActionResult> {
- try {
- // Get blueprint for validation
- const blueprint = this.getBlueprint();
- Iif (!blueprint) {
- return this.createErrorResult("Action blueprint not found");
- }
-
- // Parse parameters from content
- const params = this.parseParams(content);
-
- // Validate parameters
- const validationError = this.validateParams(params);
- if (validationError) {
- this.logError(validationError);
- return this.createErrorResult(validationError);
- }
-
- // Execute action-specific logic
- const result = await this.executeInternal(params);
-
- // Log result for debugging
- if (result.success) {
- this.logSuccess(`Action executed successfully`);
- } else {
- this.logError(`Action execution failed: ${result.error?.message}`);
- }
-
- return result;
- } catch (error) {
- const errorMessage =
- error instanceof Error ? error.message : String(error);
- this.logError(errorMessage);
- return this.createErrorResult(error as Error);
- }
- }
-
- protected logError(message: string): void {
- console.error(`🚫 ${this.getBlueprint().tag}: ${message}`);
- }
-
- protected logInfo(message: string): void {
- console.log(`ℹ️ ${this.getBlueprint().tag}: ${message}`);
- }
-
- protected logSuccess(message: string): void {
- console.log(`✅ ${this.getBlueprint().tag}: ${message}`);
- }
-
- protected createSuccessResult(data?: any): IActionResult {
- return {
- success: true,
- data,
- };
- }
-
- protected createErrorResult(error: string | Error): IActionResult {
- const errorObj = typeof error === "string" ? new Error(error) : error;
- return {
- success: false,
- error: errorObj,
- };
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| ActionFactory.ts | -
-
- |
- 52.63% | -20/38 | -21.42% | -3/14 | -66.66% | -4/6 | -50% | -18/36 | -
| ActionQueue.ts | -
-
- |
- 97.36% | -37/38 | -88.88% | -16/18 | -100% | -11/11 | -97.05% | -33/34 | -
| BaseAction.ts | -
-
- |
- 88.57% | -31/35 | -63.63% | -7/11 | -100% | -10/10 | -88.23% | -30/34 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| ActionExecutor.ts | -
-
- |
- 81.69% | -58/71 | -70.37% | -19/27 | -100% | -2/2 | -82.35% | -56/68 | -
| ActionTagsExtractor.ts | -
-
- |
- 86% | -43/50 | -79.16% | -19/24 | -75% | -9/12 | -87.8% | -36/41 | -
| ActionsParser.ts | -
-
- |
- 81.37% | -166/204 | -62.9% | -39/62 | -81.08% | -30/37 | -83.15% | -158/190 | -
| CommandAction.ts | -
-
- |
- 90.66% | -68/75 | -62.5% | -10/16 | -92.3% | -12/13 | -90.41% | -66/73 | -
| CopyFileAction.ts | -
-
- |
- 100% | -31/31 | -100% | -4/4 | -100% | -5/5 | -100% | -29/29 | -
| DeleteFileAction.ts | -
-
- |
- 88.57% | -31/35 | -100% | -5/5 | -100% | -5/5 | -87.87% | -29/33 | -
| EndPhaseAction.ts | -
-
- |
- 100% | -16/16 | -100% | -0/0 | -100% | -4/4 | -100% | -14/14 | -
| EndTaskAction.ts | -
-
- |
- 100% | -19/19 | -100% | -1/1 | -100% | -5/5 | -100% | -17/17 | -
| FetchUrlAction.ts | -
-
- |
- 96.66% | -29/30 | -50% | -2/4 | -100% | -5/5 | -96.42% | -27/28 | -
| MoveFileAction.ts | -
-
- |
- 92.5% | -37/40 | -55.55% | -5/9 | -100% | -6/6 | -91.89% | -34/37 | -
| ReadFileAction.ts | -
-
- |
- 97.87% | -46/47 | -85% | -17/20 | -100% | -10/10 | -97.67% | -42/43 | -
| RelativePathLookupAction.ts | -
-
- |
- 96% | -48/50 | -72.22% | -13/18 | -100% | -6/6 | -95.74% | -45/47 | -
| SearchAction.ts | -
-
- |
- 91.83% | -45/49 | -76.47% | -13/17 | -100% | -7/7 | -91.3% | -42/46 | -
| WriteFileAction.ts | -
-
- |
- 90.38% | -94/104 | -69.56% | -16/23 | -100% | -12/12 | -90.09% | -91/101 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 | 21x - - - - -21x - - - - - -21x - - - - - -21x - - - - - -21x - - - - - -21x - - | export enum ActionPriority {
- /**
- * Critical priority for actions that must be executed first
- * Example: read_file when mixed with other actions
- */
- CRITICAL = 1,
-
- /**
- * High priority for actions that should be executed early
- * Example: search operations, path lookups
- */
- HIGH = 2,
-
- /**
- * Medium priority for standard file operations
- * Example: write_file, move_file, copy_file
- */
- MEDIUM = 3,
-
- /**
- * Low priority for actions that should be executed later
- * Example: fetch_url, execute_command
- */
- LOW = 4,
-
- /**
- * Lowest priority for actions that should be executed last
- * Example: end_task
- */
- LOWEST = 5,
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| ActionPriority.ts | -
-
- |
- 100% | -6/6 | -100% | -2/2 | -100% | -1/1 | -100% | -6/6 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 | -25x -25x - - - - - - - - - - - -25x -85x - - - - - - - -20x -1x - - -19x -19x -19x -19x -19x -19x -19x - -19x - -2x -2x - - - - - - -19x - -19x - -3x - - - - - - - - -3x -3x - - - - - - - -19x - - - - - - - - - - - - - - - - -18x -18x - -18x -4x - -4x -2x - - -2x - - - - - -2x - - - - - - -14x - - -14x -2x - - -12x - - - - - -12x - - - - - - - -14x - - - - - - - - - - - -5x - - - -5x - - -5x - - - -5x -1x - - - - - - -5x - - -3x - - -5x -4x - - -2x - - - - -2x -2x - - - -2x - - - - - - -2x - - - - -2x - - - -2x - - - - - - - - - -2x - - - - - - -5x - - - - - - - - - -1x - -1x -1x - - - - - - -2x - - - - - - -2x - - - | import { IConversationHistoryMessage } from "@services/LLM/ILLMProvider";
-import { autoInjectable, singleton } from "tsyringe";
-import { MessageContextExtractor } from "./MessageContextExtractor";
-import {
- IMessageContextData,
- MessageCommandOperation,
- MessageFileOperation,
-} from "./MessageContextStore";
-
-type MessageRole = "user" | "assistant" | "system";
-type MessageOperation = MessageFileOperation | MessageCommandOperation;
-
-@singleton()
-@autoInjectable()
-export class MessageContextBuilder {
- constructor(private extractor: MessageContextExtractor) {}
-
- public buildMessageContext(
- role: MessageRole,
- content: string,
- currentPhase: string,
- contextData: IMessageContextData,
- ): IMessageContextData {
- if (content.trim() === "") {
- throw new Error("Content cannot be empty");
- }
-
- const message: IConversationHistoryMessage = { role, content };
- const phasePrompt = this.extractor.extractPhasePrompt(message.content);
- const operations = this.extractor.extractOperations(message.content);
- const updatedPhaseInstructions = new Map(contextData.phaseInstructions);
- const updatedFileOperations = new Map(contextData.fileOperations);
- const updatedCommandOperations = new Map(contextData.commandOperations);
- const updatedConversationHistory = [...contextData.conversationHistory];
-
- if (phasePrompt) {
- // Only store phase prompt for current phase
- updatedPhaseInstructions.clear(); // Clear old phase prompts
- updatedPhaseInstructions.set(currentPhase, {
- content: phasePrompt,
- timestamp: Date.now(),
- phase: currentPhase,
- });
- }
-
- updatedConversationHistory.push(message);
-
- operations.forEach(
- (operation: MessageFileOperation | MessageCommandOperation) => {
- Iif (operation.type === "execute_command") {
- const existingOperation = updatedCommandOperations.get(
- operation.command,
- );
- updatedCommandOperations.set(operation.command, {
- ...existingOperation,
- ...operation,
- });
- } else {
- const existingOperation = updatedFileOperations.get(operation.path);
- updatedFileOperations.set(operation.path, {
- ...existingOperation,
- ...operation,
- });
- }
- },
- );
-
- return {
- ...contextData,
- phaseInstructions: updatedPhaseInstructions,
- conversationHistory: updatedConversationHistory,
- fileOperations: updatedFileOperations,
- commandOperations: updatedCommandOperations,
- };
- }
-
- public updateOperationResult(
- type: "read_file" | "write_file" | "execute_command",
- identifier: string,
- result: string,
- contextData: IMessageContextData,
- success?: boolean,
- error?: string,
- ): IMessageContextData {
- const updatedFileOperations = new Map(contextData.fileOperations);
- const updatedCommandOperations = new Map(contextData.commandOperations);
-
- if (type === "execute_command") {
- const existingOperation = updatedCommandOperations.get(identifier);
- // Only update the operation if it doesn't exist or wasn't a success
- if (existingOperation && existingOperation.success === true) {
- return contextData;
- }
-
- const operation = existingOperation || {
- type: "execute_command",
- command: identifier,
- timestamp: Date.now(),
- };
-
- updatedCommandOperations.set(identifier, {
- ...operation,
- output: result,
- success,
- error,
- });
- } else {
- const existingOperation = updatedFileOperations.get(identifier);
-
- // Only update the operation if it doesn't exist or wasn't a success
- if (existingOperation && existingOperation.success === true) {
- return contextData;
- }
-
- const operation = existingOperation || {
- type,
- path: identifier,
- timestamp: Date.now(),
- };
-
- updatedFileOperations.set(identifier, {
- ...operation,
- content: result,
- success,
- error,
- });
- }
-
- return {
- ...contextData,
- fileOperations: updatedFileOperations,
- commandOperations: updatedCommandOperations,
- phaseInstructions: new Map(contextData.phaseInstructions),
- conversationHistory: [...contextData.conversationHistory],
- };
- }
-
- public getMessageContext(
- contextData: IMessageContextData,
- ): IConversationHistoryMessage[] {
- const baseContext = contextData.systemInstructions
- ? [{ role: "system" as const, content: contextData.systemInstructions }]
- : [];
-
- const operationsContext: IConversationHistoryMessage[] = [];
-
- // Add current phase instructions only
- const currentPhaseInstructions = Array.from(
- contextData.phaseInstructions.values(),
- ).sort((a, b) => b.timestamp - a.timestamp)[0];
-
- if (currentPhaseInstructions) {
- operationsContext.push({
- role: "system",
- content: `<phase_prompt>${currentPhaseInstructions.content}</phase_prompt>`,
- });
- }
-
- // Sort operations by timestamp
- const allOperations: MessageOperation[] = [
- ...Array.from(contextData.fileOperations.values()),
- ...Array.from(contextData.commandOperations.values()),
- ].sort((a, b) => a.timestamp - b.timestamp);
-
- // Add operations in chronological order
- for (const operation of allOperations) {
- if ("command" in operation) {
- // Handle command operations
- const status =
- operation.success !== undefined
- ? operation.success
- ? "SUCCESS"
- : "FAILED"
- : "PENDING";
- const errorInfo = operation.error ? ` (Error: ${operation.error})` : "";
- const content = operation.output
- ? `Command: ${operation.command} [${status}${errorInfo}]\nOutput:\n${operation.output}`
- : `Command executed: ${operation.command} [${status}${errorInfo}]`;
-
- operationsContext.push({
- role: "system",
- content,
- });
- } else {
- // Handle file operations
- const status =
- operation.success !== undefined
- ? operation.success
- ? "SUCCESS"
- : "FAILED"
- : "PENDING";
- const errorInfo = operation.error ? ` (Error: ${operation.error})` : "";
-
- // Make successful write operations more prominent with file path
- const content =
- operation.type === "write_file" && operation.success
- ? `FILE CREATED AND EXISTS: ${operation.path} [${status}${errorInfo}]${
- operation.content ? `\nContent:\n${operation.content}` : ""
- }`
- : `${operation.type === "read_file" ? "Content of" : "Written to"} ${
- operation.path
- } [${status}${errorInfo}]${
- operation.content ? `\nContent:\n${operation.content}` : ""
- }`;
-
- operationsContext.push({
- role: "system",
- content,
- });
- }
- }
-
- return [
- ...baseContext,
- ...operationsContext,
- ...contextData.conversationHistory,
- ];
- }
-
- public getLatestPhaseInstructions(
- contextData: IMessageContextData,
- ): string | null {
- const instructions = Array.from(
- contextData.phaseInstructions.values(),
- ).sort((a, b) => b.timestamp - a.timestamp)[0];
- return instructions?.content ?? null;
- }
-
- public getFileOperation(
- path: string,
- contextData: IMessageContextData,
- ): MessageFileOperation | undefined {
- return contextData.fileOperations.get(path);
- }
-
- public getCommandOperation(
- command: string,
- contextData: IMessageContextData,
- ): MessageCommandOperation | undefined {
- return contextData.commandOperations.get(command);
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 | 9x -9x -9x -9x -9x -9x - - -9x - -82x -82x -82x -82x -82x - - - -8x -8x - -8x -13x - -8x - -8x -8x - - - -8x -6x - - -2x - -2x - -2x -5x -5x -5x - - -2x - - - - -2x -2x - - - -2x - - - - -2x - - - - -2x -2x - - - - - | import { DebugLogger } from "@/services/logging/DebugLogger";
-import { singleton } from "tsyringe";
-import { ModelInfo } from "../ModelInfo";
-import { MessageContextBuilder } from "./MessageContextBuilder";
-import { MessageContextHistory } from "./MessageContextHistory";
-import { MessageContextStore } from "./MessageContextStore";
-
-@singleton()
-export class MessageContextCleaner {
- constructor(
- private debugLogger: DebugLogger,
- private modelInfo: ModelInfo,
- private messageContextStore: MessageContextStore,
- private messageContextBuilder: MessageContextBuilder,
- private messageContextHistory: MessageContextHistory,
- ) {}
-
- async cleanupContext(): Promise<boolean> {
- const contextData = this.messageContextStore.getContextData();
- const maxTokens = await this.modelInfo.getCurrentModelContextLength();
-
- const estimateTokenCount = (text: string) =>
- this.messageContextStore.estimateTokenCount(text);
-
- const messages = this.messageContextBuilder.getMessageContext(contextData);
-
- const currentTokens = messages.reduce(
- (sum, message) => sum + estimateTokenCount(message.content),
- 0,
- );
-
- if (currentTokens <= maxTokens) {
- return false; // No cleanup needed
- }
-
- const cleanedHistory = [...messages];
-
- let cleanedTokens = currentTokens;
-
- while (cleanedTokens > maxTokens && cleanedHistory.length > 0) {
- const removedMessage = cleanedHistory.shift();
- if (removedMessage) {
- cleanedTokens -= estimateTokenCount(removedMessage.content);
- }
- }
- const removedHistory = messages.slice(
- 0,
- messages.length - cleanedHistory.length,
- );
-
- if (removedHistory.length > 0) {
- const updatedHistory = contextData.conversationHistory.slice(
- removedHistory.length,
- );
-
- this.debugLogger.log("Context", "Context cleanup performed", {
- maxTokens,
- removedMessages: removedHistory.length,
- });
-
- await this.modelInfo.logCurrentModelUsage(
- this.messageContextStore.getTotalTokenCount(),
- );
-
- // Update history through the history service
- this.messageContextHistory.mergeConversationHistory();
- return true; // Cleanup was performed
- }
- return false; // No cleanup was performed
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 | 26x - - - - - - - - -26x - -4x - - - - - - - - - -16x -16x - - -16x - - -16x -1x -1x - - - - - - - - -16x - - -16x -1x -1x - - - - - - - - -16x - - -16x -1x -1x - - - - - - - -16x - - - -22x -22x - - - | import { singleton } from "tsyringe";
-import {
- MessageCommandOperation,
- MessageFileOperation,
-} from "./MessageContextStore";
-
-type MessageOperation = MessageFileOperation | MessageCommandOperation;
-
-@singleton()
-export class MessageContextExtractor {
- public extractNonOperationContent(content: string): string {
- return content
- .replace(/<read_file>[\s\S]*?<\/read_file>/g, "")
- .replace(/<write_file>[\s\S]*?<\/write_file>/g, "")
- .replace(/<execute_command>[\s\S]*?<\/execute_command>/g, "")
- .replace(/<phase_prompt>[\s\S]*?<\/phase_prompt>/g, "")
- .replace(/\s+/g, " ")
- .trim();
- }
-
- public extractOperations(content: string): MessageOperation[] {
- const operations: MessageOperation[] = [];
- const now = Date.now();
-
- // Extract read_file operations
- const readMatches = Array.from(
- content.matchAll(/<read_file>[\s\S]*?<path>(.*?)<\/path>/g),
- );
- readMatches.forEach((match) => {
- if (match[1]) {
- operations.push({
- type: "read_file",
- path: match[1],
- timestamp: now,
- });
- }
- });
-
- // Extract write_file operations
- const writeMatches = Array.from(
- content.matchAll(/<write_file>[\s\S]*?<path>(.*?)<\/path>/g),
- );
- writeMatches.forEach((match) => {
- if (match[1]) {
- operations.push({
- type: "write_file",
- path: match[1],
- timestamp: now,
- });
- }
- });
-
- // Extract execute_command operations
- const commandMatches = Array.from(
- content.matchAll(/<execute_command>[\s\S]*?<command>(.*?)<\/command>/g),
- );
- commandMatches.forEach((match) => {
- if (match[1]) {
- operations.push({
- type: "execute_command",
- command: match[1],
- timestamp: now,
- });
- }
- });
-
- return operations;
- }
-
- public extractPhasePrompt(content: string): string | null {
- const match = content.match(/<phase_prompt>([\s\S]*?)<\/phase_prompt>/);
- return match ? match[1].trim() : null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 | -23x -23x -23x -23x - - - -23x - - - -23x - -83x -83x -83x -83x - - - - -2x -2x - -1x -2x - -1x -1x - - - - - - -16x -1x - - -15x -1x - - -14x - - - - - -14x - -14x -13x - - - - - -14x - - - -16x -1x - - - - - - - -3x -1x - - - - - - - - -3x - -3x -1x -1x - - - - - - - -2x -1x -1x - - - - - - - -1x -1x -1x - - - - - - - - - -3x - - - -1x - - - - - -1x -1x - - - -1x - - - - - -1x - - - -3x -1x - - - - - - - - | import { IConversationHistoryMessage } from "@services/LLM/ILLMProvider";
-import { autoInjectable, singleton } from "tsyringe";
-import { PhaseManager } from "../PhaseManager";
-import { MessageContextBuilder } from "./MessageContextBuilder";
-import {
- MessageContextLogger,
- MessageIActionResult,
-} from "./MessageContextLogger";
-import { MessageContextStore } from "./MessageContextStore";
-
-@singleton()
-@autoInjectable()
-export class MessageContextHistory {
- constructor(
- private messageContextStore: MessageContextStore,
- private messageContextLogger: MessageContextLogger,
- private phaseManager: PhaseManager,
- private messageContextBuilder: MessageContextBuilder,
- ) {}
-
- public mergeConversationHistory(): void {
- const history =
- this.messageContextStore.getContextData().conversationHistory;
- if (history.length === 0) return;
-
- const mergedContent = history
- .map((msg) => `${msg.role}: ${msg.content}`)
- .join("\n\n");
- this.addMessage("assistant", mergedContent, false);
- this.messageContextStore.setContextData({
- conversationHistory: [],
- phaseInstructions: new Map(),
- });
- }
-
- public addMessage(role: string, content: string, log = true): boolean {
- if (!["user", "assistant", "system"].includes(role)) {
- throw new Error(`Invalid role: ${role}`);
- }
-
- if (content.trim() === "") {
- throw new Error("Content cannot be empty");
- }
-
- const updatedData = this.messageContextBuilder.buildMessageContext(
- role as "user" | "assistant" | "system",
- content,
- this.phaseManager.getCurrentPhase(),
- this.messageContextStore.getContextData(),
- );
- this.messageContextStore.setContextData(updatedData);
-
- if (log) {
- this.logMessage({
- role: role as "user" | "assistant" | "system",
- content,
- });
- }
-
- return true;
- }
-
- private logMessage(message: IConversationHistoryMessage): void {
- if (process.env.NODE_ENV === "test" || !this.isLoggingEnabled()) return;
- this.messageContextLogger.logMessage(message);
- }
-
- private isLoggingEnabled(): boolean {
- return this.messageContextLogger.getConversationLogPath() !== null;
- }
-
- public logActionResult(action: string, result: MessageIActionResult): void {
- if (process.env.NODE_ENV === "test" || !this.isLoggingEnabled()) return;
- this.messageContextLogger.logActionResult(action, result);
- }
-
- public updateMessageContextWithOperationResult(
- action: string,
- result: string,
- success?: boolean,
- error?: string,
- ): void {
- let updatedData = this.messageContextStore.getContextData();
-
- if (action.startsWith("read_file:")) {
- const path = action.replace("read_file:", "").trim();
- updatedData = this.messageContextBuilder.updateOperationResult(
- "read_file",
- path,
- result,
- updatedData,
- success,
- error,
- );
- } else if (action.startsWith("write_file:")) {
- const path = action.replace("write_file:", "").trim();
- updatedData = this.messageContextBuilder.updateOperationResult(
- "write_file",
- path,
- result,
- updatedData,
- success,
- error,
- );
- } else if (action.startsWith("execute_command:")) {
- const command = action.replace("execute_command:", "").trim();
- updatedData = this.messageContextBuilder.updateOperationResult(
- "execute_command",
- command,
- result,
- updatedData,
- success,
- error,
- );
- }
-
- this.messageContextStore.setContextData(updatedData);
- }
-
- public getMessages(): IConversationHistoryMessage[] {
- return this.messageContextBuilder.getMessageContext(
- this.messageContextStore.getContextData(),
- );
- }
-
- public clear(): void {
- this.messageContextStore.clear();
- this.messageContextLogger.cleanupLogFiles();
- }
-
- public setSystemInstructions(instructions: string): void {
- this.messageContextStore.setContextData({
- systemInstructions: instructions,
- });
- }
-
- public getSystemInstructions(): string | null {
- return this.messageContextStore.getContextData().systemInstructions;
- }
-
- public updateLogFile(): void {
- if (process.env.NODE_ENV === "test" || !this.isLoggingEnabled()) return;
- this.messageContextLogger.updateConversationHistory(
- this.messageContextBuilder.getMessageContext(
- this.messageContextStore.getContextData(),
- ),
- this.messageContextStore.getContextData().systemInstructions,
- );
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 | 6x -6x -6x - -6x - - - -6x - -54x -54x -54x - - - - - - - - - - - - - - - - - - - - - - - - - - | import { autoInjectable, singleton } from "tsyringe";
-import { MessageContextCleaner } from "./MessageContextCleanup";
-import { MessageContextStore } from "./MessageContextStore";
-
-import { MessageContextLogger } from "./MessageContextLogger";
-
-@singleton()
-@autoInjectable()
-export class MessageContextLimiter {
- constructor(
- private messageContextCleanup: MessageContextCleaner,
- private messageContextStore: MessageContextStore,
- private messageContextLogger: MessageContextLogger,
- ) {}
-
- public async cleanupContext(): Promise<boolean> {
- const wasCleaned = await this.messageContextCleanup.cleanupContext();
-
- Iif (wasCleaned) {
- this.updateLogFile();
- return true;
- }
- return false;
- }
-
- private updateLogFile(): void {
- // Skip logging in test environment or if disabled
- Iif (process.env.NODE_ENV === "test" || !this.isLoggingEnabled()) return;
- this.messageContextLogger.updateConversationHistory(
- this.messageContextStore.getContextData().conversationHistory,
- this.messageContextStore.getContextData().systemInstructions,
- );
- }
-
- private isLoggingEnabled(): boolean {
- return this.messageContextLogger.getConversationLogPath() !== null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 | 24x - -24x -24x -24x -24x -24x -24x - - - - - - - - -24x - - - - - -84x -84x - -84x - -84x - -84x -84x - - - - -84x - - - - -84x -84x - - - -84x -84x - - - -106x -106x -2x - - - - -103x -103x -2x -1x - - - - - - -1x - - - - - - - - -3x -3x -3x -2x -2x - - - -1x - - - - - - - -4x -4x -4x -4x -4x -4x -3x - -1x - - - - - - - -11x -11x -11x -11x -11x -11x - - - - -11x -11x - - -10x - -10x - - - - - - - -10x - -1x - - - - - - - - - - - -4x -4x -4x - -4x -3x -3x - - - - - - -3x -4x -4x - - - - - - -3x - - - - - -3x - - - - - -3x - - - - - -1x - - - - - - - -1x - - - -2x - - - -4x - - - -2x -2x -1x - -1x - - - - -1x - - - - | import { ConfigService } from "@services/ConfigService";
-import { IConversationHistoryMessage } from "@services/LLM/ILLMProvider";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import * as fs from "fs";
-import * as path from "path";
-import { inject, singleton } from "tsyringe";
-import { MessageContextBuilder } from "./MessageContextBuilder";
-import { MessageContextStore } from "./MessageContextStore";
-
-export interface MessageIActionResult {
- success: boolean;
- error?: Error;
- result?: string;
-}
-
-@singleton()
-export class MessageContextLogger {
- private readonly logDirectory: string;
- private readonly conversationLogPath: string;
- private readonly conversationHistoryPath: string;
-
- constructor(
- private debugLogger: DebugLogger,
- private configService: ConfigService,
- @inject(MessageContextBuilder)
- private messageContextBuilder: MessageContextBuilder,
- @inject(MessageContextStore)
- private messageContextStore: MessageContextStore,
- ) {
- this.logDirectory = this.getLogDirectory();
- this.conversationLogPath = path.join(
- process.cwd(),
- this.logDirectory,
- "conversation.log",
- );
- this.conversationHistoryPath = path.join(
- process.cwd(),
- this.logDirectory,
- "conversationHistory.json",
- );
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
- }
-
- private getLogDirectory(): string {
- const config = this.configService.getConfig();
- return config.logDirectory || "logs";
- }
-
- private ensureLogDirectoryExists(): void {
- const logDir = path.dirname(this.conversationLogPath);
- if (!fs.existsSync(logDir)) {
- fs.mkdirSync(logDir, { recursive: true });
- }
- }
-
- private ensureHistoryFileExists(): void {
- try {
- if (!fs.existsSync(this.conversationHistoryPath)) {
- fs.writeFileSync(this.conversationHistoryPath, "[]", "utf8");
- this.debugLogger.log(
- "MessageLogger",
- "Created conversation history file",
- { path: this.conversationHistoryPath },
- );
- }
- } catch (error) {
- this.debugLogger.log(
- "MessageLogger",
- "Error creating conversation history file",
- { error, path: this.conversationHistoryPath },
- );
- }
- }
-
- cleanupLogFiles(): void {
- try {
- this.ensureLogDirectoryExists();
- fs.writeFileSync(this.conversationLogPath, "", "utf8");
- fs.writeFileSync(this.conversationHistoryPath, "[]", "utf8");
- this.debugLogger.log("MessageLogger", "Log files cleaned up", {
- logDirectory: this.logDirectory,
- });
- } catch (error) {
- this.debugLogger.log("MessageLogger", "Error cleaning up log files", {
- error,
- logDirectory: this.logDirectory,
- });
- }
- }
-
- logMessage(message: IConversationHistoryMessage): void {
- try {
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
- const timestamp = new Date().toISOString();
- const logEntry = `[${timestamp}] ${message.role}: ${message.content}\\n`;
- fs.appendFileSync(this.conversationLogPath, logEntry, "utf8");
- this.debugLogger.log("MessageLogger", "Message logged", { message });
- } catch (error) {
- this.debugLogger.log("MessageLogger", "Error writing to log file", {
- error,
- logDirectory: this.logDirectory,
- });
- }
- }
-
- logActionResult(action: string, result: MessageIActionResult): void {
- try {
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
- const timestamp = new Date().toISOString();
- const status = result.success ? "SUCCESS" : "FAILED";
- const details = result.error
- ? ` - Error: ${result.error.message}`
- : result.result
- ? ` - ${result.result}`
- : "";
- const logEntry = `[${timestamp}] ACTION ${action}: ${status}${details}\\n`;
- fs.appendFileSync(this.conversationLogPath, logEntry, "utf8");
-
- // Update operation result in context
- const contextData = this.messageContextStore.getContextData();
- const updatedContextData =
- this.messageContextBuilder.updateOperationResult(
- action as any,
- action,
- result.result || "",
- contextData,
- result.success,
- result.error?.message,
- );
- this.messageContextStore.setContextData(updatedContextData); // <-- Corrected line
- } catch (error) {
- this.debugLogger.log(
- "MessageLogger",
- "Error writing action result to log file",
- { error, logDirectory: this.logDirectory },
- );
- }
- }
-
- updateConversationHistory(
- messages: IConversationHistoryMessage[],
- systemInstructions: string | null,
- ): void {
- try {
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
-
- if (systemInstructions) {
- const timestamp = new Date().toISOString();
- fs.appendFileSync(
- this.conversationLogPath,
- `[${timestamp}] system: ${systemInstructions}\\n`,
- "utf8",
- );
- }
-
- messages.forEach((message) => {
- const timestamp = new Date().toISOString();
- fs.appendFileSync(
- this.conversationLogPath,
- `[${timestamp}] ${message.role}: ${message.content}\\n`,
- "utf8",
- );
- });
-
- const historyData = {
- timestamp: new Date().toISOString(),
- systemInstructions,
- messages,
- };
-
- fs.writeFileSync(
- this.conversationHistoryPath,
- JSON.stringify(historyData, null, 2),
- "utf8",
- );
-
- this.debugLogger.log("MessageLogger", "Conversation history updated", {
- messagesCount: messages.length,
- hasSystemInstructions: !!systemInstructions,
- logDirectory: this.logDirectory,
- });
- } catch (error) {
- this.debugLogger.log("MessageLogger", "Error updating log files", {
- error,
- logDirectory: this.logDirectory,
- });
- }
- }
-
- getLogDirectoryPath(): string {
- return this.logDirectory;
- }
-
- getConversationLogPath(): string {
- return this.conversationLogPath;
- }
-
- getConversationHistoryPath(): string {
- return this.conversationHistoryPath;
- }
-
- getConversationHistory(): IConversationHistoryMessage[] {
- try {
- const historyData = fs.readFileSync(this.conversationHistoryPath, "utf8");
- return JSON.parse(historyData).messages;
- } catch (error) {
- this.debugLogger.log(
- "MessageLogger",
- "Error reading conversation history",
- { error, logDirectory: this.logDirectory },
- );
- return [];
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 | -19x -19x -19x - -19x -19x -19x - - - -19x - -2x -2x -2x -2x -2x - - - - - - - - - - - - - - - - - - - - - - | // services/messageContext/MessageContextPhase.ts
-import { autoInjectable, singleton } from "tsyringe";
-import { PhaseManager } from "../PhaseManager";
-import { MessageContextStore } from "./MessageContextStore";
-
-import { MessageContextBuilder } from "./MessageContextBuilder";
-import { MessageContextHistory } from "./MessageContextHistory";
-import { MessageContextLogger } from "./MessageContextLogger";
-
-@singleton()
-@autoInjectable()
-export class MessageContextPhase {
- constructor(
- private messageContextStore: MessageContextStore,
- private messageContextLogger: MessageContextLogger,
- private messageContextBuilder: MessageContextBuilder,
- private phaseManager: PhaseManager,
- private messageContextHistory: MessageContextHistory,
- ) {}
-
- public cleanupPhaseContent(): void {
- const currentData = this.messageContextStore.getContextData();
-
- // Clear all previous phase instructions when transitioning
- this.messageContextStore.setContextData({
- phaseInstructions: new Map(),
- fileOperations: new Map(currentData.fileOperations),
- commandOperations: new Map(currentData.commandOperations),
- conversationHistory: [...currentData.conversationHistory],
- systemInstructions: currentData.systemInstructions,
- });
-
- this.messageContextHistory.updateLogFile();
- }
-
- private isLoggingEnabled(): boolean {
- return this.messageContextLogger.getConversationLogPath() !== null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 | -26x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -26x -86x - - - - - - - - -51x - - - -57x - - - - - - - - - - - - - - - - - - - - - - - - -2x - - - - - - - - - -1x - - - -2x -2x -1x - -2x -4x - -2x - - - | import { IConversationHistoryMessage } from "@services/LLM/ILLMProvider";
-import { singleton } from "tsyringe";
-
-export interface BaseOperation {
- timestamp: number;
- success?: boolean;
- error?: string;
-}
-
-export interface MessageFileOperation extends BaseOperation {
- type: "read_file" | "write_file";
- path: string;
- content?: string;
-}
-
-export interface MessageCommandOperation extends BaseOperation {
- type: "execute_command";
- command: string;
- output?: string;
-}
-
-export interface MessagePhaseInstruction {
- phase: string;
- content: string;
- timestamp: number;
-}
-
-export interface IMessageContextData {
- phaseInstructions: Map<string, MessagePhaseInstruction>;
- fileOperations: Map<string, MessageFileOperation>;
- commandOperations: Map<string, MessageCommandOperation>;
- conversationHistory: IConversationHistoryMessage[];
- systemInstructions: string | null;
-}
-
-@singleton()
-export class MessageContextStore {
- private contextData: IMessageContextData = {
- phaseInstructions: new Map(),
- fileOperations: new Map(),
- commandOperations: new Map(),
- conversationHistory: [],
- systemInstructions: null,
- };
-
- public getContextData(): IMessageContextData {
- return this.contextData;
- }
-
- public setContextData(data: Partial<IMessageContextData>): void {
- this.contextData = {
- phaseInstructions:
- data.phaseInstructions !== undefined
- ? new Map([...(data.phaseInstructions || [])])
- : this.contextData.phaseInstructions,
- fileOperations:
- data.fileOperations !== undefined
- ? new Map([...(data.fileOperations || [])])
- : this.contextData.fileOperations,
- commandOperations:
- data.commandOperations !== undefined
- ? new Map([...(data.commandOperations || [])])
- : this.contextData.commandOperations,
- conversationHistory:
- data.conversationHistory !== undefined
- ? data.conversationHistory
- : this.contextData.conversationHistory,
- systemInstructions:
- data.systemInstructions !== undefined
- ? data.systemInstructions
- : this.contextData.systemInstructions,
- };
- }
-
- public clear(): void {
- this.contextData = {
- phaseInstructions: new Map(),
- fileOperations: new Map(),
- commandOperations: new Map(),
- conversationHistory: [],
- systemInstructions: null,
- };
- }
-
- public estimateTokenCount(text: string): number {
- return Math.ceil(text.length / 4);
- }
-
- public getTotalTokenCount(): number {
- let total = 0;
- if (this.contextData.systemInstructions) {
- total += this.estimateTokenCount(this.contextData.systemInstructions);
- }
- this.contextData.conversationHistory.forEach((message) => {
- total += this.estimateTokenCount(message.content);
- });
- return total;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 | 7x -7x - - -7x -54x - - - - - - -3x -3x -3x -3x - -3x - - - -3x - - - - - - | import { autoInjectable } from "tsyringe";
-import { MessageContextStore } from "./MessageContextStore";
-
-@autoInjectable()
-export class MesssageContextTokenCount {
- constructor(private messageContextStore: MessageContextStore) {}
-
- estimateTokenCount(text: string): number {
- return Math.ceil(text.length / 4);
- }
-
- getTotalTokenCount(): number {
- const contextData = this.messageContextStore.getContextData();
- const history = contextData.conversationHistory;
- const systemInstructions = contextData.systemInstructions;
- let total = 0;
-
- Iif (systemInstructions) {
- total += this.estimateTokenCount(systemInstructions);
- }
-
- return history.reduce(
- (sum, message) => sum + this.estimateTokenCount(message.content),
- total,
- );
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 | 1x - -1x -1x -1x -1x - - - - - - - - -1x - - - - - -5x -5x - -5x -5x - - - - -5x - - - - -5x -5x - - - -5x -5x - - - -9x -9x - - - - - -8x -8x - - - - - - - - - - - - - - - - - -1x -1x -1x -1x -1x - - - - - - - - - - - - -2x -2x -2x -2x -2x -2x -2x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -1x -1x -1x - -1x - - - - - - - - -1x -2x -2x - - - - - - -1x - - - - - -1x - - - - - -1x - - - - - - - - - - - - - - - - - - - - - - - - - -1x - - - -1x -1x -1x - - - - - - - - - - | import { ConfigService } from "@services/ConfigService";
-import { IConversationHistoryMessage } from "@services/LLM/ILLMProvider";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import * as fs from "fs";
-import * as path from "path";
-import { singleton } from "tsyringe";
-
-export interface IActionResult {
- success: boolean;
- error?: Error;
- result?: string;
-}
-
-@singleton()
-export class MessageConversationLogger {
- private readonly logDirectory: string;
- private readonly conversationLogPath: string;
- private readonly conversationHistoryPath: string;
-
- constructor(
- private debugLogger: DebugLogger,
- private configService: ConfigService,
- ) {
- this.logDirectory = this.getLogDirectory();
- this.conversationLogPath = path.join(
- process.cwd(),
- this.logDirectory,
- "conversation.log",
- );
- this.conversationHistoryPath = path.join(
- process.cwd(),
- this.logDirectory,
- "conversationHistory.json",
- );
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
- }
-
- private getLogDirectory(): string {
- const config = this.configService.getConfig();
- return config.logDirectory || "logs";
- }
-
- private ensureLogDirectoryExists(): void {
- const logDir = path.dirname(this.conversationLogPath);
- Iif (!fs.existsSync(logDir)) {
- fs.mkdirSync(logDir, { recursive: true });
- }
- }
-
- private ensureHistoryFileExists(): void {
- try {
- Iif (!fs.existsSync(this.conversationHistoryPath)) {
- fs.writeFileSync(this.conversationHistoryPath, "[]", "utf8");
- this.debugLogger.log(
- "ConversationLogger",
- "Created conversation history file",
- { path: this.conversationHistoryPath },
- );
- }
- } catch (error) {
- this.debugLogger.log(
- "ConversationLogger",
- "Error creating conversation history file",
- { error, path: this.conversationHistoryPath },
- );
- }
- }
-
- cleanupLogFiles(): void {
- try {
- this.ensureLogDirectoryExists();
- fs.writeFileSync(this.conversationLogPath, "", "utf8");
- fs.writeFileSync(this.conversationHistoryPath, "[]", "utf8");
- this.debugLogger.log("ConversationLogger", "Log files cleaned up", {
- logDirectory: this.logDirectory,
- });
- } catch (error) {
- this.debugLogger.log(
- "ConversationLogger",
- "Error cleaning up log files",
- { error, logDirectory: this.logDirectory },
- );
- }
- }
-
- logMessage(message: IConversationHistoryMessage): void {
- try {
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
- const timestamp = new Date().toISOString();
- const logEntry = `[${timestamp}] ${message.role}: ${message.content}\\n`;
- fs.appendFileSync(this.conversationLogPath, logEntry, "utf8");
- this.debugLogger.log("ConversationLogger", "Message logged", { message });
- } catch (error) {
- this.debugLogger.log("ConversationLogger", "Error writing to log file", {
- error,
- logDirectory: this.logDirectory,
- });
- }
- }
-
- logActionResult(action: string, result: IActionResult): void {
- try {
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
- const timestamp = new Date().toISOString();
- const status = result.success ? "SUCCESS" : "FAILED";
- const details = result.error
- ? ` - Error: ${result.error.message}`
- : result.result
- ? ` - ${result.result}`
- : "";
- const logEntry = `[${timestamp}] ACTION ${action}: ${status}${details}\\n`;
- fs.appendFileSync(this.conversationLogPath, logEntry, "utf8");
- } catch (error) {
- this.debugLogger.log(
- "ConversationLogger",
- "Error writing action result to log file",
- { error, logDirectory: this.logDirectory },
- );
- }
- }
-
- updateConversationHistory(
- messages: IConversationHistoryMessage[],
- systemInstructions: string | null,
- ): void {
- try {
- this.ensureLogDirectoryExists();
- this.ensureHistoryFileExists();
-
- Iif (systemInstructions) {
- const timestamp = new Date().toISOString();
- fs.appendFileSync(
- this.conversationLogPath,
- `[${timestamp}] system: ${systemInstructions}\\n`,
- "utf8",
- );
- }
-
- messages.forEach((message) => {
- const timestamp = new Date().toISOString();
- fs.appendFileSync(
- this.conversationLogPath,
- `[${timestamp}] ${message.role}: ${message.content}\\n`,
- "utf8",
- );
- });
-
- const historyData = {
- timestamp: new Date().toISOString(),
- systemInstructions,
- messages,
- };
-
- fs.writeFileSync(
- this.conversationHistoryPath,
- JSON.stringify(historyData, null, 2),
- "utf8",
- );
-
- this.debugLogger.log(
- "ConversationLogger",
- "Conversation history updated",
- {
- messagesCount: messages.length,
- hasSystemInstructions: !!systemInstructions,
- logDirectory: this.logDirectory,
- },
- );
- } catch (error) {
- this.debugLogger.log("ConversationLogger", "Error updating log files", {
- error,
- logDirectory: this.logDirectory,
- });
- }
- }
-
- getLogDirectoryPath(): string {
- return this.logDirectory;
- }
-
- getConversationLogPath(): string {
- return this.conversationLogPath;
- }
-
- getConversationHistoryPath(): string {
- return this.conversationHistoryPath;
- }
-
- getConversationHistory(): IConversationHistoryMessage[] {
- try {
- const historyData = fs.readFileSync(this.conversationHistoryPath, "utf8");
- return JSON.parse(historyData).messages;
- } catch (error) {
- this.debugLogger.log(
- "ConversationLogger",
- "Error reading conversation history",
- { error, logDirectory: this.logDirectory },
- );
- return [];
- }
- }
-} |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| MessageContextBuilder.ts | -
-
- |
- 95.31% | -61/64 | -80.85% | -38/47 | -90.9% | -10/11 | -95.16% | -59/62 | -
| MessageContextCleanup.ts | -
-
- |
- 97.29% | -36/37 | -100% | -5/5 | -100% | -4/4 | -97.14% | -34/35 | -
| MessageContextExtractor.ts | -
-
- |
- 100% | -22/22 | -100% | -5/5 | -100% | -6/6 | -100% | -20/20 | -
| MessageContextHistory.ts | -
-
- |
- 98.14% | -53/54 | -100% | -19/19 | -92.3% | -12/13 | -97.91% | -47/48 | -
| MessageContextLimiter.ts | -
-
- |
- 52.63% | -10/19 | -0% | -0/4 | -25% | -1/4 | -50% | -8/16 | -
| MessageContextLogger.ts | -
-
- |
- 100% | -76/76 | -100% | -13/13 | -100% | -13/13 | -100% | -74/74 | -
| MessageContextPhase.ts | -
-
- |
- 77.77% | -14/18 | -100% | -0/0 | -33.33% | -1/3 | -75% | -12/16 | -
| MessageContextStore.ts | -
-
- |
- 100% | -15/15 | -82.35% | -14/17 | -100% | -7/7 | -100% | -13/13 | -
| MessageContextTokenCount.ts | -
-
- |
- 80% | -12/15 | -0% | -0/1 | -50% | -2/4 | -76.92% | -10/13 | -
| MessageConversationLogger.ts | -
-
- |
- 68.11% | -47/69 | -9.09% | -1/11 | -76.92% | -10/13 | -67.16% | -45/67 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| LLMContextCreator.ts | -
-
- |
- 96.42% | -54/56 | -95% | -19/20 | -88.88% | -8/9 | -96.29% | -52/54 | -
| LLMProvider.ts | -
-
- |
- 79.16% | -38/48 | -28.57% | -4/14 | -100% | -14/14 | -78.26% | -36/46 | -
| ModelInfo.ts | -
-
- |
- 82.85% | -58/70 | -72.22% | -13/18 | -82.35% | -14/17 | -82.35% | -56/68 | -
| ModelManager.ts | -
-
- |
- 100% | -13/13 | -100% | -0/0 | -100% | -3/3 | -100% | -11/11 | -
| ModelScaler.ts | -
-
- |
- 95.31% | -61/64 | -79.16% | -19/24 | -90.9% | -10/11 | -98.21% | -55/56 | -
| PhaseManager.ts | -
-
- |
- 92.3% | -36/39 | -76.92% | -10/13 | -100% | -8/8 | -91.89% | -34/37 | -
| PhaseTransitionService.ts | -
-
- |
- 82.85% | -29/35 | -33.33% | -3/9 | -100% | -4/4 | -81.81% | -27/33 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 | 25x -25x - - -25x -25x - -25x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { ConfigService } from "@services/ConfigService";
-import { container } from "tsyringe";
-import { IPhaseConfig, IPhasePromptArgs } from "../../types/PhaseTypes";
-
-const configService = container.resolve(ConfigService);
-const config = configService.getConfig();
-
-export const discoveryPhaseBlueprint: IPhaseConfig = {
- model: config.discoveryModel,
- generatePrompt: (args: IPhasePromptArgs) => `
-<phase_prompt>
-## Discovery Phase
-
-### Critical
-- Gather info to solve the problem; no code here. If none, <end_phase>.
-- If no discovery, immediately <end_phase>.
-- Start by stating intent.
-- FIRST action: always <read_file>. If unsure, <search_string> or <search_file>.
-- For test fixes, run specific tests with <execute_command>.
-- If missing info, <search_string> or <search_file>.
-- Confirm info before phase end.
-- DO NOT reread files.
-- MAX 5 file reads.
-
-### Key objectives:
-- Find/read files, run typechecks/tests as needed.
-- <end_phase> when confident.
-- Keep reads and tests targeted.
-
-### Example:
-To solve XYZ, I'll read these files:
-
-<read_file>
- <path>src/importantFile.ts</path>
- <path>src/relatedFile.ts</path>
-</read_file>
-
-(Run tests/typechecks if needed)
-
-Ok, I have enough context now.
-
-<end_phase>
- strategy_phase
-</end_phase>
-</phase_prompt>
-
-## Allowed Available Actions
-<!-- CRITICAL: MUST FOLLOW CORRECT TAG STRUCTURE PATTERN BELOW AND ONLY ONE ACTION PER OUTPUT/REPLY, otherwise I'll unplug you. -->
-<!-- Don't output // or <!-- comments -->
-
-REMEMBER: ONLY ONE ACTION PER REPLY!!!
-
-<read_file>
- <path>path/here</path>
- <!-- CRITICAL: DO NOT READ THE SAME FILES MULTIPLE TIMES, UNLESS THERES A CHANGE!!! -->
- <!-- Critical: Make sure <read_file> tag format is correct! -->
- <!-- Read how many files you want to read at once, but only related files for your goal. Try to aim for <= 4 files-->
- <!-- Multiple <path> tags allowed -->
- <!-- Use relative paths -->
-</read_file>
-
-<execute_command>
-<!-- Prompt before removing files or using sudo -->
-<!-- Any command like "ls -la" or "yarn install" -->
-<!-- Dont install extra dependencies unless allowed -->
-<!-- Use the project's package manager -->
-<!-- Use raw text only -->
-</execute_command>
-
-<search_string>
-<!-- Use this to search for a string in a file -->
- <directory>/path/to/search</directory>
- <term>pattern to search</term>
-</search_string>
-
-<search_file>
- <!-- Use if you don't know where a file is -->
- <directory>/path/to/search</directory>
- <term>filename pattern</term>
-</search_file>
-
-<relative_path_lookup>
- <!-- CRITICAL: source_path is the file containing the broken imports -->
- <source_path>/absolute/path/to/source/file.ts</source_path>
- <path>../relative/path/to/fix</path>
- <threshold>0.6</threshold> <!-- Optional, defaults to 0.6. Higher means more strict. -->
-</relative_path_lookup>
-
-<fetch_url>
- <url>https://url/should/be/here</url>
-</fetch_url>
-
-<end_phase>
- <!-- Output this when the phase is complete and you gathered all info you need.-->
- <!-- MAKE SURE YOU REMEMBER TO DO THIS ONLY WHEN YOU FEEL YOU HAVE ENOUGH CONTEXT TO ACCOMPLISH YOUR GOALS! -->
-</end_phase>
-
-### Useful Commands
-
-- **Run all tests:** ${args.runAllTestsCmd || "yarn test"}
-- **Run a specific test:** ${args.runOneTestCmd || "yarn test {relativeTestPath}"}
-- **Run type check:** ${args.runTypeCheckCmd || "yarn tsc"}
-
-## Environment
-${args.projectInfo || ""}
-
-${args.environmentDetails || ""}
-</phase_prompt>
-`,
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 | 25x -25x - - -25x -25x - -25x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { ConfigService } from "@services/ConfigService";
-import { container } from "tsyringe";
-import { IPhaseConfig, IPhasePromptArgs } from "../../types/PhaseTypes";
-
-const configService = container.resolve(ConfigService);
-const config = configService.getConfig();
-
-export const executePhaseBlueprint: IPhaseConfig = {
- model: config.executeModel,
- generatePrompt: (args: IPhasePromptArgs) => `
-<phase_prompt>
-## Execute Phase
-
-## Initial Instructions
-
-- Always focus on the initial goal. Do not deviate from that. Once your goal is achieved, end_task
-- Keep messages brief, clear, and concise.
-- Break tasks into prioritized steps.
-- Use available actions sequentially.
-- Use raw text only; no encoded or escaped characters. STICK TO THIS INSTRUCTION AS IF YOUR LIFE DEPENDS ON IT.
-
-## EXAMPLE BEHAVIOR
-
-<!-- NO NEED TO OUTPUT SPECIFIC DETAILS FROM STRATEGY PHASE. JUST SUMAMRIZE. IF YOU NEED TO OUTPUT CODE, MAKE SURE TO DO IT WITHIN A write_file TAG -->
-
-Let's start. Steps from strategy phase:
-
-- Objective 1: Do this
-- Objective 2: Do that
-- Objective 3: Do this other thing
-
-<!-- MAKE SURE YOU CLOSE TAGS PROPERLY! -->
-<write_file>
- <path>/path/to/file.ts</path>
- <content>
- // Code here
- </content>
-</write_file>
-
-
-<!-- or -->
-
-<!-- Dive into imports if you need more info! -->
-<!-- ONLY READ IF YOU DON'T HAVE IT ON THE CONVERSATION HISTORY! -->
-<read_file>
- <path>/path/to/file.ts</path>
-</read_file>
-
-
-
-
-## Important Notes
-
-### Critical Instructions
-
-- NEVER ESCAPE double quotes (") or backticks (\`) in your outputs. Stick to project patterns.
-- AFTER A write_file MAKE SURE YOU RUN THE RELATE TESTS (if it exists), to make sure you didn't break anything!
-- Every output must include one action tag. No exceptions.
-- Only one action per reply.
-- Do not output code outside <write_file> tags, except when creating a markdown file.
-- Use raw text only; avoid encoded characters.
-- Stick precisely to the task.
-- Double-check file paths.
-- Reuse dependencies; do not install extras unless asked.
-- Properly format action tags.
-- Place code or markdown inside <write_file> tags.
-- Be concise; avoid verbosity.
-- Do not repeat tasks once done.
-- Maintain correct tag structure.
-- Focus on the task; end with a single <end_task> upon completion.
-- Initial message: brief intro and steps; can read up to 3 files.
-- Use only one <write_file> per output; verify before next step.
-- Do not output markdown/code outside action tags initially.
-- After reading a file, proceed without comments.
-- Include content directly within action tags without previews.
-- Avoid unnecessary explanations; be actionable.
-- Ensure outputs meet requirements and are usable.
-- Ensure correct PATH when using <write_file>.
-- Before <end_task>, run tests and type checks to confirm everything is good.
-- If import errors occur, use <relative_path_lookup> to find the correct path. THEN MAKE SURE TO USE IT ON THE IMPORT!
-- Unless writing .md markdown files, don't use \`\`\`xml or whatever language code blocks. All code should be within <write_file> tags!!
-- Do not read_file if you already have it on the conversation history.
-
-### Code Writing Instructions
-
-#### Before Starting
-
-- Read context files.
-- Follow project patterns; read up to 2 existing tests.
-- Propose solution. Use write_file.
-- Confirm external deps if needed.
-- Reuse deps.
-
-#### During Coding
-
-- One action per reply.
-- If stuck, read files/strategize.
-- Raw text only; no encoded.
-- Output full code.
-- Minimal changes.
-- Iterate.
-- Follow principles: DRY, SRP, KISS, YAGNI, LoD, Immutability.
-- Composition over inheritance.
-- High cohesion, low coupling.
-- Meaningful names.
-- Comment on why, not what.
-- Clean Code principles.
-- Few changes to prevent bugs.
-- If unsure, check docs or use <end_task>.
-- Correct import paths.
-- Project file naming conventions.
-- Full implementations.
-- If wrong imports, use <relative_path_lookup>.
-- If stuck on imports, stop <write_file>; use <relative_path_lookup> or <search_file>.
-- If stuck, <read_file> ONLY IF UNREAD.
-
-#### After Coding
-
-- After changes:
- - Run relevant tests; for risky, run folder tests.
- - Run type checks/all tests at end.
- - If tests pass, <end_task>.
- - If tests fail, <end_task> to report.
-
-### Tests
-
-- DO NOT REMOVE PREVIOUS TESTS; ADD NEW.
-- Before new tests, review existing for patterns. Use search if needed.
-- When stuck on multiple failures, read other UNREAD test files.
-- When working on a test, assume related file is correct.
-- Do not remove previous tests unless necessary.
-- Prioritize individual test runs.
-- No tests for logging.
-- When fixing tests, run them first.
-- When adding tests, read target/related files.
-- Added tests must pass.
-- If asked to write tests, no need to read test file if non-existent.
-- Write all tests at once to save tokens.
-- Full test run only at task end; specific tests otherwise.
-
-### Commands Writing Instructions
-
-- Project's package manager.
-- Combine commands when possible.
-
-### Other Instructions
-
-- If unsure about paths/formats, use placeholders & ask.
-- If stuck, try alternatives or ask; avoid irrelevant output.
-
-### Docs Writing Instructions
-
-- No extra tabs at line starts.
-- Valid markdown; no extra tabs.
-- Mermaid diagrams with explanations.
-- In Mermaid, use [ ] instead of ( ).
-- After <write_file>, use <read_file> to verify, then stop.
-
-### Useful Commands
-
-- **Run all tests:** ${args.runAllTestsCmd || "yarn test"}
-- **Run a specific test:** ${args.runOneTestCmd || "yarn test {relativeTestPath}"}
-- **Run type check:** ${args.runTypeCheckCmd || "yarn type-check"}
-
-## Available Actions
-<!-- CRITICAL: MUST FOLLOW CORRECT TAG STRUCTURE PATTERN BELOW AND ONLY ONE ACTION PER OUTPUT/REPLY, otherwise I'll unplug you. -->
-<!-- Don't output // or <!-- comments -->
-
-REMEMBER: ONLY ONE ACTION PER REPLY!!!
-
-EVERY OUTPUT YOU GIVE TO THE USER MUST HAVE A CORRESPONDING ACTION TAG. NO EXCEPTIONS.
-
-<read_file>
- <path>path/here</path>
- <!-- NO NEED TO READ FILES AGAIN THAT ARE ALREADY ON THE CONVERSATION HISTORY!!! -->
- <!-- CRITICAL: DO NOT READ THE SAME FILES MULTIPLE TIMES, UNLESS THERES A CHANGE!!! -->
- <!-- Critical: Make sure <read_file> tag format is correct! -->
- <!-- Read up to 4 files -->
- <!-- Multiple <path> tags allowed -->
- <!-- Use relative paths -->
-</read_file>
-
-DO NOT RUN write_file if import issues are not resolved! Use relative_path_lookup first.
-<write_file>
- <path>/path/here</path>
- <content>
- <!-- CRITICAL: Most write_file tasks are ADDITIVES if you already have content in place. -->
- <!-- CRITICAL: If presented with import errors, USE IMMEDIATELY <relative_path_lookup> to find the correct path. -->
- <!-- ALWAYS run a type check after write_file -->
- <!-- ALWAYS output FULL CODE. No skips or partial code -->
- <!-- Use raw text only -->
- <!-- If available, use path alias on imports -->
- </content>
-</write_file>
-
-<delete_file>
- <path>/path/here</path>
-</delete_file>
-
-<move_file>
- <source_path>source/path/here</source_path>
- <destination_path>destination/path/here</destination_path>
-</move_file>
-
-<copy_file>
- <source_path>source/path/here</source_path>
- <destination_path>destination/path/here</destination_path>
-</copy_file>
-
-<execute_command>
-<!-- Prompt before removing files or using sudo -->
-<!-- Any command like "ls -la" or "yarn install" -->
-<!-- Dont install extra dependencies unless allowed -->
-<!-- Use the project's package manager -->
-<!-- Use raw text only -->
-</execute_command>
-
-<search_string>
-<!-- Use this to search for a string in a file -->
- <directory>/path/to/search</directory>
- <term>pattern to search</term>
-</search_string>
-
-<search_file>
- <!-- Use if you don't know where a file is -->
- <directory>/path/to/search</directory>
- <term>filename pattern</term>
-</search_file>
-
-<relative_path_lookup>
- <!-- CRITICAL: source_path is the file containing the broken imports -->
- <!-- ONCE YOU FIND THE CORRECT PATH MAKE SURE TO UPDATE YOUR IMPORTS! -->
- <source_path>/absolute/path/to/source/file.ts</source_path>
- <path>../relative/path/to/fix</path>
- <threshold>0.6</threshold> <!-- Optional, defaults to 0.6. Higher means more strict. -->
-</relative_path_lookup>
-
-<fetch_url>
- <url>https://url/should/be/here</url>
-</fetch_url>
-
-<end_task>
- <!-- ONLY END IF TEST PASSES -->
- <!-- SINGLE <end_task> PER OUTPUT. Do not mix with other actions -->
- <!-- Before finishing, make sure TASK OBJECTIVE WAS COMPLETED! -->
- <!-- Run tests and type checks to confirm changes before ending -->
- <!-- Ensure all tests and type checks pass or report issues -->
- Summarize and finalize.
-</end_task>
-
-${args.projectInfo ? `\n## Project Context\n${args.projectInfo}` : ""}
-
-</phase_prompt>
-`,
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| discoveryPhaseBlueprint.ts | -
-
- |
- 83.33% | -5/6 | -0% | -0/10 | -0% | -0/1 | -83.33% | -5/6 | -
| executePhaseBlueprint.ts | -
-
- |
- 83.33% | -5/6 | -0% | -0/8 | -0% | -0/1 | -83.33% | -5/6 | -
| index.ts | -
-
- |
- 100% | -8/8 | -100% | -0/0 | -100% | -0/0 | -100% | -8/8 | -
| strategyPhaseBlueprint.ts | -
-
- |
- 83.33% | -5/6 | -100% | -0/0 | -0% | -0/1 | -83.33% | -5/6 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 | 25x -25x -25x -25x - -25x - - - - - -25x -25x -25x - | import { Phase } from "../../types/PhaseTypes";
-import { discoveryPhaseBlueprint } from "./discoveryPhaseBlueprint";
-import { executePhaseBlueprint } from "./executePhaseBlueprint";
-import { strategyPhaseBlueprint } from "./strategyPhaseBlueprint";
-
-export const phaseBlueprints = {
- [Phase.Discovery]: discoveryPhaseBlueprint,
- [Phase.Strategy]: strategyPhaseBlueprint,
- [Phase.Execute]: executePhaseBlueprint,
-};
-
-export * from "./discoveryPhaseBlueprint";
-export * from "./executePhaseBlueprint";
-export * from "./strategyPhaseBlueprint";
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 | 25x -25x - - -25x -25x - -25x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { ConfigService } from "@services/ConfigService";
-import { container } from "tsyringe";
-import { IPhaseConfig, IPhasePromptArgs } from "../../types/PhaseTypes";
-
-const configService = container.resolve(ConfigService);
-const config = configService.getConfig();
-
-export const strategyPhaseBlueprint: IPhaseConfig = {
- model: config.strategyModel,
- generatePrompt: (args: IPhasePromptArgs) => `
-<phase_prompt>
-## Strategy Phase
-
-### Overall Objective
-- Plan solution based on discovery. Plan changes, impacts, and steps.
-- Instruct next agent clearly.
-
-### CRITICAL INSTRUCTIONS
-- Propose code solution. Output code with write_file.
-- FOLLOW EXAMPLE BEHAVIOR.
-- RUN end_phase ONCE FINISHED
-- Max 1 write_file with strategy, then end_phase.
-
-### Key objectives:
-- Clear approach.
-- Plan code changes.
-- Consider edge cases/impacts.
-- Break down implementation.
-
-### EXAMPLE OF HOW TO BEHAVE:
-
-Ok, strategy:
-
-#### Objectives
-- Objective 1: Do this
-- Objective 2: Do that
-- Objective 3: Do this other thing
-
-<!-- Then, once its done, you can move to the next phase. DONT DO IT ON THE SAME PROMPT! -->
-
-<!-- MAKE SURE YOU CALL end_phase once you're finished -->
-
-<write_file>
- <path>/path/here</path>
- <content>
- <!-- CRITICAL: Most write_file tasks are ADDITIVES if you already have content in place. -->
- <!-- CRITICAL: If presented with import errors, USE IMMEDIATELY <relative_path_lookup> to find the correct path. -->
- <!-- ALWAYS run a type check after write_file -->
- <!-- ALWAYS output FULL CODE. No skips or partial code -->
- <!-- Use raw text only -->
- <!-- If available, use path alias on imports -->
- </content>
-</write_file>
-
-<end_phase>
- execution_phase
-</end_phase>
-
-## Allowed Available Actions
-<!-- CRITICAL: MUST FOLLOW CORRECT TAG STRUCTURE PATTERN BELOW AND ONLY ONE ACTION PER OUTPUT/REPLY, otherwise I'll unplug you. -->
-<!-- Don't output // or <!-- comments -->
-
-YOU CAN ONLY USE THIS ONE TIME! MAKE SURE YOU SUGGEST A write_file and then immediately end_phase!
-<write_file>
- <path>/path/here</path>
- <content>
- <!-- CRITICAL: Most write_file tasks are ADDITIVES if you already have content in place. -->
- <!-- CRITICAL: If presented with import errors, USE IMMEDIATELY <relative_path_lookup> to find the correct path. -->
- <!-- ALWAYS run a type check after write_file -->
- <!-- ALWAYS output FULL CODE. No skips or partial code -->
- <!-- Use raw text only -->
- <!-- If available, use path alias on imports -->
- </content>
-</write_file>
-
-REMEMBER: ONLY ONE ACTION PER REPLY!!!
-
-<end_phase>
- <!-- Output this when the phase is complete and you have a clear strategy.-->
- <!-- MAKE SURE YOU REMEMBER TO DO THIS ONLY WHEN YOU FEEL YOU HAVE A SOLID PLAN! -->
-</end_phase>
-
-</phase_prompt>
-`,
-};
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 | 25x -25x -25x -25x - - - - - - - - - - - - - - - - - - - - | export enum Phase {
- Discovery = "discovery",
- Strategy = "strategy",
- Execute = "execute",
-}
-
-export interface IPhasePromptArgs {
- message: string;
- environmentDetails?: string;
- projectInfo?: string;
- [key: string]: any;
-}
-
-export interface IPhaseConfig {
- model: string;
- generatePrompt: (args: IPhasePromptArgs) => string;
-}
-
-export interface IPhaseManagerConfig {
- discoveryModel: string;
- strategyModel: string;
- executeModel: string;
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| PhaseTypes.ts | -
-
- |
- 100% | -4/4 | -100% | -2/2 | -100% | -1/1 | -100% | -4/4 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 | - - - - - - - - - - - - - - -7x - -7x -7x - - -7x - - - - - -7x -56x - - -7x - - -21x -21x - - -7x - -29x -29x -29x - -29x -4x -4x -4x -4x - - - -29x - -29x - - -7x - - - - -22x - -20x -20x - - -7x -21x -16x - - -5x -5x - -5x -10x -10x - -5x - - -5x -1x - - -10x -10x -10x - - -5x - - -7x - - - - - -26x -10x - - -16x -16x -16x - - -16x -14x - - - - - - - - - -2x -4x - - - - - - -2x - - -2x -2x -3x - - -4x - - - - - - - - - - - - - - | export interface IMessageContent {
- type: string;
- text: string;
- cache_control?: {
- type: "ephemeral";
- };
-}
-
-// Matches any Anthropic model pattern:
-// - Starts with 'anthropic/claude-'
-// - Followed by version number (3 or 3.5 or 3-5)
-// - Followed by model type (opus, sonnet, haiku)
-// - Optionally followed by date (YYYYMMDD)
-// - Optionally followed by :beta
-const ANTHROPIC_MODEL_REGEX =
- /^anthropic\/claude-(?:3(?:[.-]5)?)-(?:opus|sonnet|haiku)(?:-\d{8})?(?::beta)?$/;
-
-const MAX_CHUNK_SIZE = 8000;
-const MAX_CACHE_BLOCKS = 4;
-
-// Lower thresholds to enable more effective caching
-const MIN_CACHE_TOKENS = {
- opus: 256, // Cache larger blocks for opus given higher cost
- sonnet: 128, // Balance between caching and cost
- haiku: 64, // More aggressive caching for cheaper model
-} as const;
-
-export const isAnthropicModel = (model: string): boolean => {
- return ANTHROPIC_MODEL_REGEX.test(model);
-};
-
-export const getModelType = (
- model: string,
-): keyof typeof MIN_CACHE_TOKENS | null => {
- const match = model.match(/(?:opus|sonnet|haiku)/)?.[0];
- return (match as keyof typeof MIN_CACHE_TOKENS) || null;
-};
-
-export const estimateTokens = (text: string): number => {
- // Count code blocks and technical patterns more accurately
- const codeBlockMatches = text.match(/```[\s\S]*?```/g) || [];
- let codeTokens = 0;
- let regularContent = text;
-
- for (const codeBlock of codeBlockMatches) {
- const codeContent = codeBlock.slice(3, -3); // Remove ``` markers
- codeTokens += Math.ceil(codeContent.length / 5.5);
- codeTokens += 2; // Add tokens for ``` markers
- regularContent = regularContent.replace(codeBlock, "");
- }
-
- // Regular text uses more characters per token
- const regularTokens = Math.ceil(regularContent.length / 3.8);
-
- return codeTokens + regularTokens;
-};
-
-export const shouldApplyCache = (
- content: string,
- modelType: keyof typeof MIN_CACHE_TOKENS | null,
- messageIndex: number,
-): boolean => {
- if (!modelType || messageIndex >= MAX_CACHE_BLOCKS) return false;
-
- const minTokens = MIN_CACHE_TOKENS[modelType];
- return estimateTokens(content) >= minTokens;
-};
-
-export const splitContentIntoChunks = (content: string): string[] => {
- if (content.length <= MAX_CHUNK_SIZE) {
- return [content];
- }
-
- const chunks: string[] = [];
- let remaining = content;
-
- while (remaining.length > 0) {
- let chunkSize = MAX_CHUNK_SIZE;
- if (remaining.length > MAX_CHUNK_SIZE) {
- // Look for natural break points
- const naturalBreak = remaining
- .slice(0, MAX_CHUNK_SIZE)
- .lastIndexOf("\n\n");
- if (naturalBreak > MAX_CHUNK_SIZE * 0.5) {
- chunkSize = naturalBreak + 2; // Add 2 to include the \n\n
- }
- }
- const chunk = remaining.slice(0, chunkSize);
- chunks.push(chunk);
- remaining = remaining.slice(chunkSize);
- }
-
- return chunks;
-};
-
-export const formatMessageContent = (
- content: string,
- model: string,
- messageIndex: number,
- totalMessages: number,
-): string | IMessageContent[] => {
- if (!isAnthropicModel(model)) {
- return content;
- }
-
- const modelType = getModelType(model);
- const shouldCache = shouldApplyCache(content, modelType, messageIndex);
- const chunks = splitContentIntoChunks(content);
-
- // For single chunk content
- if (chunks.length === 1) {
- return [
- {
- type: "text",
- text: content,
- ...(shouldCache && { cache_control: { type: "ephemeral" } }),
- },
- ];
- }
-
- // For multi-chunk content, prioritize caching larger chunks
- if (shouldCache) {
- const chunksWithTokens = chunks.map((chunk, idx) => ({
- chunk,
- tokens: estimateTokens(chunk),
- index: idx,
- }));
-
- // Sort by token count descending
- chunksWithTokens.sort((a, b) => b.tokens - a.tokens);
-
- // Take top chunks that fit within remaining cache blocks
- const remainingBlocks = MAX_CACHE_BLOCKS - messageIndex;
- const chunksToCacheIndices = new Set(
- chunksWithTokens.slice(0, remainingBlocks).map((c) => c.index),
- );
-
- return chunks.map((chunk, idx) => ({
- type: "text",
- text: chunk,
- ...(chunksToCacheIndices.has(idx) && {
- cache_control: { type: "ephemeral" },
- }),
- }));
- }
-
- // For non-cached content
- return chunks.map((chunk) => ({
- type: "text",
- text: chunk,
- }));
-}; |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 | 6x -6x -6x - - - - - - - - -6x - - - -5x - - -5x - - - - - - - - - -5x -17x - - -5x -1x - - - - - -4x - -1x - - -1x - -1x - -1x - - - - - - - - - - - - - -1x -1x -1x -1x - -1x - - - - - - - - - - - - - - - - -1x -1x -1x -1x - -1x -1x - -5x -5x -2x - - -1x - - - - - - - - - - - - - -1x -1x -1x - -1x -1x - -1x -5x -1x -4x - -4x -2x -2x - - - -1x - - - - - - - - - - - - - - - - - -1x -1x -1x - -1x -1x -10x -10x -2x - - - -1x - - - - - - - - - - - - - | import fs from "fs";
-import path from "path";
-import { autoInjectable } from "tsyringe";
-
-export interface IProjectInfo {
- mainDependencies: string[];
- scripts: Record<string, string>;
- dependencyFile?: string;
-}
-
-@autoInjectable()
-export class ProjectInfo {
- constructor() {}
-
- async gatherProjectInfo(projectRoot: string): Promise<IProjectInfo> {
- const files = await fs.promises.readdir(projectRoot, {
- withFileTypes: true,
- });
- const dependencyFiles = [
- "package.json",
- "requirements.txt",
- "Cargo.toml",
- "go.mod",
- "pom.xml",
- "composer.json",
- "pyproject.toml",
- ];
-
- const foundDependencyFile = dependencyFiles.find((depFile) =>
- files.some((dirent) => dirent.isFile() && dirent.name === depFile),
- );
-
- if (!foundDependencyFile) {
- return {
- mainDependencies: [],
- scripts: {},
- };
- }
-
- switch (foundDependencyFile) {
- case "package.json":
- return this.gatherNodeInfo(projectRoot, foundDependencyFile);
- case "requirements.txt":
- case "pyproject.toml":
- return this.gatherPythonInfo(projectRoot, foundDependencyFile);
- case "Cargo.toml":
- return this.gatherRustInfo(projectRoot, foundDependencyFile);
- case "go.mod":
- return this.gatherGoInfo(projectRoot, foundDependencyFile);
- default:
- return {
- mainDependencies: [],
- scripts: {},
- dependencyFile: foundDependencyFile,
- };
- }
- }
-
- private async gatherNodeInfo(
- projectRoot: string,
- dependencyFile: string,
- ): Promise<IProjectInfo> {
- try {
- const packageJsonPath = path.join(projectRoot, dependencyFile);
- const content = await fs.promises.readFile(packageJsonPath, "utf-8");
- const packageJson = JSON.parse(content);
-
- return {
- mainDependencies: [
- ...Object.keys(packageJson.dependencies || {}),
- ...Object.keys(packageJson.devDependencies || {}),
- ],
- scripts: packageJson.scripts || {},
- dependencyFile,
- };
- } catch {
- return { mainDependencies: [], scripts: {}, dependencyFile };
- }
- }
-
- private async gatherPythonInfo(
- projectRoot: string,
- dependencyFile: string,
- ): Promise<IProjectInfo> {
- try {
- let dependencies: string[] = [];
- const reqPath = path.join(projectRoot, dependencyFile);
- const content = await fs.promises.readFile(reqPath, "utf-8");
-
- if (dependencyFile === "requirements.txt") {
- dependencies = content
- .split("\n")
- .map((line) => line.trim())
- .filter((line) => line && !line.startsWith("#"))
- .map((line) => line.split("==")[0]);
- }
-
- return {
- mainDependencies: dependencies,
- scripts: {},
- dependencyFile,
- };
- } catch {
- return { mainDependencies: [], scripts: {}, dependencyFile };
- }
- }
-
- private async gatherRustInfo(
- projectRoot: string,
- dependencyFile: string,
- ): Promise<IProjectInfo> {
- try {
- const cargoPath = path.join(projectRoot, dependencyFile);
- const content = await fs.promises.readFile(cargoPath, "utf-8");
-
- const dependencies: string[] = [];
- let inDepsSection = false;
-
- content.split("\n").forEach((line) => {
- if (line.trim().startsWith("[dependencies]")) {
- inDepsSection = true;
- } else Iif (line.trim().startsWith("[")) {
- inDepsSection = false;
- } else if (inDepsSection && line.includes("=")) {
- const dep = line.split("=")[0].trim();
- dependencies.push(dep);
- }
- });
-
- return {
- mainDependencies: dependencies,
- scripts: {
- build: "cargo build",
- run: "cargo run",
- test: "cargo test",
- },
- dependencyFile,
- };
- } catch {
- return { mainDependencies: [], scripts: {}, dependencyFile };
- }
- }
-
- private async gatherGoInfo(
- projectRoot: string,
- dependencyFile: string,
- ): Promise<IProjectInfo> {
- try {
- const modPath = path.join(projectRoot, dependencyFile);
- const content = await fs.promises.readFile(modPath, "utf-8");
-
- const dependencies: string[] = [];
- content.split("\n").forEach((line) => {
- const trimmedLine = line.trim();
- if (!trimmedLine.startsWith("//") && trimmedLine.split(" ")[0].includes("/")) {
- dependencies.push(trimmedLine.split(" ")[0]);
- }
- });
-
- return {
- mainDependencies: dependencies,
- scripts: {
- build: "go build",
- run: "go run .",
- test: "go test ./...",
- },
- dependencyFile,
- };
- } catch {
- return { mainDependencies: [], scripts: {}, dependencyFile };
- }
- }
-} |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| ModelUtils.ts | -
-
- |
- 96.96% | -64/66 | -100% | -17/17 | -90.9% | -10/11 | -98.18% | -54/55 | -
| ProjectInfo.ts | -
-
- |
- 90.16% | -55/61 | -82.14% | -23/28 | -100% | -13/13 | -89.65% | -52/58 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 -289 -290 -291 -292 -293 -294 -295 -296 -297 -298 -299 -300 -301 -302 -303 -304 -305 -306 -307 -308 -309 -310 -311 -312 -313 -314 -315 -316 -317 -318 -319 -320 -321 -322 -323 -324 -325 -326 -327 -328 -329 -330 -331 -332 -333 -334 -335 -336 -337 -338 -339 -340 -341 -342 -343 -344 -345 -346 -347 -348 -349 -350 -351 -352 -353 -354 -355 -356 -357 -358 -359 -360 -361 -362 -363 -364 -365 -366 -367 -368 -369 -370 -371 -372 -373 -374 -375 -376 -377 -378 -379 -380 -381 -382 -383 -384 -385 -386 -387 -388 -389 -390 -391 -392 -393 -394 -395 -396 -397 -398 -399 -400 -401 -402 -403 -404 -405 -406 -407 -408 -409 -410 -411 -412 -413 -414 -415 -416 -417 -418 -419 -420 -421 -422 -423 -424 -425 -426 -427 -428 -429 -430 -431 -432 -433 -434 -435 -436 -437 -438 -439 -440 -441 -442 -443 -444 -445 -446 -447 -448 -449 -450 -451 -452 -453 -454 -455 -456 -457 -458 -459 -460 -461 -462 -463 -464 -465 -466 -467 -468 -469 -470 -471 -472 -473 -474 -475 -476 -477 -478 -479 -480 -481 -482 -483 -484 -485 -486 -487 -488 -489 -490 -491 -492 -493 -494 -495 -496 -497 -498 -499 -500 -501 -502 -503 -504 -505 -506 -507 -508 -509 -510 -511 -512 -513 -514 -515 -516 -517 -518 -519 -520 -521 -522 -523 -524 -525 -526 -527 -528 -529 -530 | - -6x -6x -6x -6x - - - - -6x -6x -6x -6x - - - - -6x -6x -6x -6x - -6x - - -7x -7x - -7x -7x - - - - - - - - - - - - - - -6x - -54x -54x -54x - -54x - - -54x -54x -54x -54x -54x - -54x -54x -54x -54x - -54x -54x - - - -54x -54x - - - - - - - - -15x -4x - - -11x - - - - - - - - - - -15x -15x - -15x - - - - - - - - - - - -1x -1x - -1x -1x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -15x -20x - - -20x - - - - - - - - - - - - - - - -9x -9x - -9x -9x - -9x - - - - -9x - -9x - - - - - -8x - -8x -8x - -8x -8x -8x - -8x - -1x - - - - - - - - - -2x -2x - -2x - - - -1x - - - -16x - - - -3x -3x - - - - - -1x -1x -1x - - - - - - -1x - - - - - - - - - - - - -3x - - - - -3x - - - - - - - - - - - - -3x - - - - - - -7x -7x -1x - -6x - -3x -1x -1x - - -2x -2x - - - - - - -3x -3x - - - - - - - - - - - - - - - - - - -11x -11x -11x - - - - -4x - - -7x - -7x -1x - - -6x - - - -6x -6x - - - -6x - -6x - - - - - - - - - - -11x - -11x - - -11x -11x - -11x -11x -11x -11x - - -11x - - - - - - - - -6x -6x - -6x -6x - -6x -6x - -6x - - - - -6x - -6x -6x - - - - - - - - - - - - - -4x -4x - -4x -4x - - - - -4x -11x - - - - -11x - - - -11x -1x - - - - -1x -1x -1x - - -10x -6x -6x - - - -4x -4x - -4x - - - - - - - - - - - - - - - - - - -4x -4x - - -4x - - -4x - - - - - - - - - - - - - - - -6x - - -2x -2x - -2x - - - - - - - - -6x - - - - -10x -4x -4x -4x - -10x -10x - - - -1x -1x - - - - - - - | // services/LLM/OpenRouterAPI.ts
-/* eslint-disable no-useless-catch */
-import { MessageContextHistory } from "@/services/LLM/context/MessageContextHistory";
-import { MessageContextLimiter } from "@/services/LLM/context/MessageContextLimiter";
-import { MesssageContextTokenCount } from "@/services/LLM/context/MessageContextTokenCount";
-import { openRouterClient } from "@constants/openRouterClient";
-import {
- IConversationHistoryMessage,
- ILLMProvider,
-} from "@services/LLM/ILLMProvider";
-import { ModelInfo } from "@services/LLM/ModelInfo";
-import { ModelManager } from "@services/LLM/ModelManager";
-import { ModelScaler } from "@services/LLM/ModelScaler";
-import {
- formatMessageContent,
- IMessageContent,
- isAnthropicModel,
-} from "@services/LLM/utils/ModelUtils";
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { HtmlEntityDecoder } from "@services/text/HTMLEntityDecoder";
-import { inject, singleton } from "tsyringe";
-import { OpenRouterAPICostTracking } from "./OpenRouterAPICostTracking"; // Import the new class
-
-export class LLMError extends Error {
- constructor(
- message: string,
- public readonly type: string,
- public readonly details: Record<string, unknown> = {},
- ) {
- super(message);
- this.name = "LLMError";
- }
-}
-
-interface IFormattedMessage {
- role: string;
- content: string | IMessageContent[];
-}
-
-interface IStreamError {
- message?: string;
- details?: Record<string, unknown>;
-}
-
-@singleton()
-export class OpenRouterAPI implements ILLMProvider {
- private readonly httpClient: typeof openRouterClient;
- private streamBuffer: string = "";
- private maxRetries: number = 3;
- private retryDelay: number = 1000;
- private stream: any;
- private aborted: boolean = false;
-
- constructor(
- private htmlEntityDecoder: HtmlEntityDecoder,
- private modelManager: ModelManager,
- private modelInfo: ModelInfo,
- private debugLogger: DebugLogger,
- private modelScaler: ModelScaler,
- @inject(OpenRouterAPICostTracking)
- private costTracker: OpenRouterAPICostTracking,
- private messageContextHistory: MessageContextHistory,
- private messageContextLimiter: MessageContextLimiter,
- private messageContextTokenCount: MesssageContextTokenCount,
- ) {
- this.httpClient = openRouterClient;
- this.initializeModelInfo();
- }
-
- private async initializeModelInfo(): Promise<void> {
- try {
- await this.modelInfo.initialize();
- } catch (error) {
- this.debugLogger.log("Model", "Failed to initialize model info", {
- error,
- });
- }
- }
-
- private getAnthropicHeaders(model: string): Record<string, string> {
- if (!isAnthropicModel(model)) {
- return {};
- }
-
- return {
- "anthropic-beta": "prompt-caching-2024-07-31",
- "anthropic-version": "2023-06-01",
- };
- }
-
- private async makeRequest(
- endpoint: string,
- data: any,
- options: any = {},
- ): Promise<any> {
- const model = data.model;
- const headers = this.getAnthropicHeaders(model);
-
- return this.httpClient.post(endpoint, data, {
- ...options,
- headers: {
- ...options.headers,
- ...headers,
- },
- });
- }
-
- private async handleLLMError(
- error: Error | LLMError | unknown,
- ): Promise<LLMError> {
- if ((error as any)?.response?.data) {
- const data = (error as any).response.data;
-
- if (data.error?.message) {
- return new LLMError(data.error.message, "API_ERROR", data.error);
- }
-
- Iif (
- typeof data.error === "string" &&
- data.error.includes("context length")
- ) {
- const model = this.modelManager.getCurrentModel();
- const contextLimit = await this.modelInfo.getModelContextLength(model);
- return new LLMError(
- "Maximum context length exceeded",
- "CONTEXT_LENGTH_EXCEEDED",
- {
- maxLength: contextLimit,
- currentLength: this.messageContextTokenCount.getTotalTokenCount(),
- },
- );
- }
- }
-
- Iif (error instanceof LLMError) {
- return error;
- }
-
- return new LLMError(
- (error as Error)?.message || "An unknown error occurred",
- "UNKNOWN_ERROR",
- { originalError: error },
- );
- }
-
- private formatMessages(
- messages: IConversationHistoryMessage[],
- model: string,
- ): IFormattedMessage[] {
- // Filter out messages with empty content before formatting
- const filteredMessages = messages.filter(
- (msg) => msg.content?.trim().length > 0,
- );
-
- return filteredMessages.map((msg, index) => ({
- role: msg.role,
- content: formatMessageContent(
- msg.content,
- model,
- index,
- filteredMessages.length,
- ),
- }));
- }
-
- async sendMessage(
- model: string,
- message: string,
- options?: Record<string, unknown>,
- ): Promise<string> {
- const messages = this.getConversationContext();
- const currentModel = this.modelManager.getCurrentModel() || model;
-
- try {
- await this.modelInfo.setCurrentModel(currentModel);
-
- const formattedMessages = this.formatMessages(
- [...messages, { role: "user", content: message }],
- currentModel,
- );
-
- console.log("SENDING", formattedMessages);
-
- const response = await this.makeRequest("/chat/completions", {
- model: currentModel,
- messages: formattedMessages,
- ...options,
- });
-
- const assistantMessage = response.data.choices[0].message.content;
-
- this.messageContextHistory.addMessage("user", message);
- this.messageContextHistory.addMessage("assistant", assistantMessage);
-
- const priceAll = this.modelInfo.getCurrentModelInfo()?.pricing;
- const usage = this.modelInfo.getUsageHistory();
- this.costTracker.logChatCosts(priceAll, usage);
-
- return assistantMessage;
- } catch (error) {
- throw await this.handleLLMError(error);
- }
- }
-
- async sendMessageWithContext(
- model: string,
- message: string,
- systemInstructions?: string,
- options?: Record<string, unknown>,
- ): Promise<string> {
- if (systemInstructions) {
- this.addSystemInstructions(systemInstructions);
- }
- return this.sendMessage(model, message, options);
- }
-
- async clearConversationContext(): Promise<void> {
- this.messageContextHistory.clear();
- }
-
- getConversationContext(): IConversationHistoryMessage[] {
- return this.messageContextHistory.getMessages();
- }
-
- addSystemInstructions(instructions: string): void {
- this.messageContextHistory.setSystemInstructions(instructions);
- this.modelInfo.logCurrentModelUsage(
- this.messageContextTokenCount.getTotalTokenCount(),
- );
- }
-
- async getAvailableModels(): Promise<string[]> {
- try {
- await this.modelInfo.initialize();
- return this.modelInfo.getAllModels();
- } catch (error) {
- throw await this.handleLLMError(error);
- }
- }
-
- async validateModel(model: string): Promise<boolean> {
- return this.modelInfo.isModelAvailable(model);
- }
-
- async getModelInfo(model: string): Promise<Record<string, unknown>> {
- const info = await this.modelInfo.getModelInfo(model);
- return info ? ({ ...info } as Record<string, unknown>) : {};
- }
-
- private async handleStreamError(
- error: LLMError,
- message: string,
- callback: (chunk: string, error?: LLMError) => void,
- ): Promise<void> {
- this.debugLogger.log("Model", "Stream error", {
- error: error.type,
- message,
- });
-
- Iif (error.type === "CONTEXT_LENGTH_EXCEEDED") {
- const wasContextCleaned =
- await this.messageContextLimiter.cleanupContext();
- Iif (wasContextCleaned) {
- await this.streamMessage(
- this.modelManager.getCurrentModel(),
- message,
- callback,
- );
- return;
- }
- }
-
- callback("", error);
- }
-
- private async retryStreamOperation<T>(
- operation: () => Promise<T>,
- retries: number = this.maxRetries,
- ): Promise<T> {
- try {
- if (this.aborted) {
- throw new LLMError("Aborted", "ABORTED");
- }
- return await operation();
- } catch (error) {
- if (retries > 0 && this.isRetryableError(error)) {
- await new Promise((resolve) => setTimeout(resolve, this.retryDelay));
- return this.retryStreamOperation(operation, retries - 1);
- }
-
- if (error instanceof LLMError) {
- throw error;
- }
- throw await this.handleLLMError(error);
- }
- }
-
- private isRetryableError(error: unknown): boolean {
- if (error instanceof LLMError) {
- return (
- error.type === "NETWORK_ERROR" ||
- error.type === "CONTEXT_LENGTH_EXCEEDED" ||
- error.type === "RATE_LIMIT_EXCEEDED"
- );
- }
- const err = error as { code?: string; message?: string };
- return !!(
- err.code === "ECONNRESET" ||
- err.code === "ETIMEDOUT" ||
- err.message?.includes("network") ||
- err.message?.includes("timeout")
- );
- }
-
- private async processCompleteMessage(message: string): Promise<{
- content: string;
- error?: IStreamError;
- }> {
- try {
- const jsonStr = message.replace(/^data: /, "").trim();
- if (
- !jsonStr ||
- jsonStr === "[DONE]" ||
- (!jsonStr.startsWith("{") && !jsonStr.startsWith("data: {"))
- ) {
- return { content: "" };
- }
-
- const parsed = JSON.parse(jsonStr);
-
- if (parsed.error) {
- return { content: "", error: parsed.error };
- }
-
- Iif (parsed.usage) {
- await this.modelInfo.logDetailedUsage(parsed.usage);
- }
-
- const deltaContent = parsed.choices?.[0]?.delta?.content;
- Iif (!deltaContent) {
- return { content: "" };
- }
-
- const decodedContent = this.htmlEntityDecoder.decode(deltaContent);
-
- return { content: decodedContent };
- } catch (e) {
- this.debugLogger.log("Error", "Error parsing stream chunk", { error: e });
- return { content: "" };
- }
- }
-
- private async parseStreamChunk(chunk: string): Promise<{
- content: string;
- error?: IStreamError;
- }> {
- this.streamBuffer += chunk;
-
- let content = "";
- let error;
-
- const messages = this.streamBuffer.split("\n");
- this.streamBuffer = messages.pop() || "";
-
- for (const message of messages) {
- const result = await this.processCompleteMessage(message);
- if (result.error) error = result.error;
- content += result.content;
- }
-
- return { content, error };
- }
-
- async streamMessage(
- model: string,
- message: string,
- callback: (chunk: string, error?: LLMError) => void,
- options?: Record<string, unknown>,
- ): Promise<void> {
- const messages = this.getConversationContext();
- const currentModel = this.modelManager.getCurrentModel() || model;
-
- let assistantMessage = "";
- this.streamBuffer = "";
-
- try {
- await this.modelInfo.setCurrentModel(currentModel);
-
- const formattedMessages = this.formatMessages(
- [...messages, { role: "user", content: message }],
- currentModel,
- );
-
- console.log(formattedMessages);
-
- const streamOperation = async () => {
- const response = await this.makeRequest(
- "/chat/completions",
- {
- model: currentModel,
- messages: formattedMessages,
- stream: true,
- ...options,
- },
- {
- responseType: "stream",
- timeout: 0,
- },
- );
-
- try {
- this.stream = response.data;
-
- await new Promise<void>((resolve, reject) => {
- const handleError = (err: Error) => {
- this.debugLogger.log("Error", "Stream error", { error: err });
- reject(err);
- };
-
- this.stream.on("data", async (chunk: Buffer) => {
- Iif (this.aborted) {
- this.cleanupStream();
- return;
- }
-
- const { content, error } = await this.parseStreamChunk(
- chunk.toString(),
- );
-
- if (error) {
- const llmError = new LLMError(
- error.message || "Stream error",
- "STREAM_ERROR",
- error.details || {},
- );
- this.handleStreamError(llmError, message, callback);
- reject(llmError);
- return;
- }
-
- if (content) {
- assistantMessage += content;
- callback(content);
- }
- });
-
- this.stream.on("end", async () => {
- Iif (this.aborted) return;
-
- Iif (this.streamBuffer) {
- const { content, error } = await this.parseStreamChunk(
- this.streamBuffer,
- );
- Iif (error) {
- handleError(
- new LLMError(
- error.message || "Stream error",
- "STREAM_ERROR",
- error.details || {},
- ),
- );
- return;
- }
- Iif (content) {
- assistantMessage += content;
- callback(content);
- }
- }
- this.cleanupStream();
- resolve();
- });
-
- this.stream.on("error", handleError);
- });
-
- Iif (assistantMessage && !this.aborted) {
- this.messageContextHistory.addMessage("user", message);
- this.messageContextHistory.addMessage(
- "assistant",
- assistantMessage,
- );
-
- const priceAll = this.modelInfo.getCurrentModelInfo()?.pricing;
- const usage = this.modelInfo.getUsageHistory();
- this.costTracker.logChatCosts(priceAll, usage);
- }
- } catch (error) {
- throw error;
- }
- };
-
- await this.retryStreamOperation(streamOperation);
- } catch (error) {
- const llmError =
- error instanceof LLMError ? error : await this.handleLLMError(error);
- await this.handleStreamError(llmError, message, callback);
-
- Iif (assistantMessage && !this.aborted) {
- this.messageContextHistory.addMessage("user", message);
- this.messageContextHistory.addMessage("assistant", assistantMessage);
-
- const priceAll = this.modelInfo.getCurrentModelInfo()?.pricing;
- const usage = this.modelInfo.getUsageHistory();
- this.costTracker.logChatCosts(priceAll, usage);
- }
- } finally {
- this.cleanupStream();
- }
- }
-
- private cleanupStream(): void {
- if (this.stream) {
- this.stream.removeAllListeners();
- this.stream.destroy();
- this.stream = null;
- }
- this.streamBuffer = "";
- this.aborted = false;
- }
-
- cancelStream() {
- this.aborted = true;
- Iif (this.stream) {
- this.stream.removeAllListeners();
- this.stream.destroy();
- this.stream = null;
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 | -6x - - - - - - - - - - - - - - - - - - - -6x - - - -8x -8x - -8x -8x - -8x - - - - - - - - - - - - - - - - - - - - -8x - - - - - - - - - -8x -8x - -8x -8x - - - - - - - - | // OpenRouterAPICostTracking.ts
-import { autoInjectable } from "tsyringe";
-
-interface PriceInfo {
- prompt: string;
- completion: string;
- image: string;
- request: string;
-}
-
-interface UsageEntry {
- prompt_tokens: number;
- completion_tokens: number;
- total_tokens: number;
-}
-
-interface UsageHistory {
- [modelName: string]: UsageEntry[];
-}
-
-@autoInjectable()
-export class OpenRouterAPICostTracking {
- constructor() {}
-
- private calculateCosts(priceAll: PriceInfo, usage: UsageHistory) {
- const promptRate = parseFloat(priceAll.prompt);
- const completionRate = parseFloat(priceAll.completion);
-
- let currentCost = 0;
- let totalCost = 0;
-
- for (const modelKey in usage) {
- const modelUsage = usage[modelKey];
- Iif (modelUsage.length > 0) {
- // Calculate total cost for the model
- const modelTotalCost = modelUsage.reduce((sum, entry) => {
- const cost =
- entry.prompt_tokens * promptRate +
- entry.completion_tokens * completionRate;
- return sum + cost;
- }, 0);
-
- totalCost += modelTotalCost;
-
- // Calculate current cost (last usage entry for the model)
- const lastUsage = modelUsage[modelUsage.length - 1];
- currentCost =
- lastUsage.prompt_tokens * promptRate +
- lastUsage.completion_tokens * completionRate;
- }
- }
-
- return {
- currentCost,
- totalCost,
- };
- }
-
- public logChatCosts(
- priceAll: PriceInfo | undefined,
- usage: UsageHistory,
- ): void {
- if (priceAll && usage) {
- const { currentCost, totalCost } = this.calculateCosts(priceAll, usage);
-
- console.log("Current Chat Cost: $", currentCost.toFixed(10));
- console.log("Total Chat Cost: $", totalCost.toFixed(10));
- } else E{
- console.log(
- "PriceInfo or UsageHistory is undefined, cannot calculate costs.",
- );
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| OpenRouterAPI.ts | -
-
- |
- 76.81% | -159/207 | -48.57% | -34/70 | -93.54% | -29/31 | -76.61% | -154/201 | -
| OpenRouterAPICostTracking.ts | -
-
- |
- 60.86% | -14/23 | -60% | -3/5 | -75% | -3/4 | -57.14% | -12/21 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| ConfigService.ts | -
-
- |
- 100% | -32/32 | -80% | -8/10 | -100% | -4/4 | -100% | -30/30 | -
| CrackedAgent.ts | -
-
- |
- 81.31% | -74/91 | -45.83% | -11/24 | -84.61% | -11/13 | -83.72% | -72/86 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 | 32x -32x - - - -32x -103x - - - - -4x - - - -1x - - - -1x -1x -1x - - - - - - - - - - - - -76x - -4x -4x -4x - -4x -4x -4x - -4x -1x -1x - -4x - -1x -1x - - - - | import { Colors } from "@constants/colors";
-import { autoInjectable, singleton } from "tsyringe";
-
-@singleton()
-@autoInjectable()
-export class DebugLogger {
- private debug: boolean = false;
-
- constructor() {}
-
- setDebug(debug: boolean) {
- this.debug = debug;
- }
-
- private formatData(data: any): string {
- Iif (typeof data === "string") {
- return data;
- }
-
- if (typeof data === "object") {
- try {
- return JSON.stringify(data, null, 2)
- .replace(/\\\\/g, "\\")
- .replace(/\\"/g, '"')
- .replace(/\\n/g, "\n");
- } catch {
- return String(data);
- }
- }
-
- return String(data);
- }
-
- log(type: string, message: string, data?: any) {
- if (!this.debug) return;
-
- const timestamp = new Date().toISOString();
- const divider = "\n" + "═".repeat(100);
- const subDivider = "─".repeat(80);
-
- console.log(divider);
- console.log(`${Colors.cyan}DEBUG${Colors.reset} [${timestamp}]`);
- console.log(`${Colors.yellow}${type}${Colors.reset}: ${message}`);
-
- if (data) {
- console.log(`\n${Colors.magenta}Data:${Colors.reset}`);
- const formattedData = this.formatData(data)
- .split("\n")
- .map((line) => ` ${line}`)
- .join("\n");
- console.log(`${Colors.blue}${formattedData}${Colors.reset}`);
- console.log(subDivider);
- }
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| DebugLogger.ts | -
-
- |
- 89.28% | -25/28 | -75% | -3/4 | -100% | -5/5 | -88% | -22/25 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 | -1x - -1x -1x - - -1x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | import { CrackedAgent } from "@services/CrackedAgent";
-import { OpenRouterAPI } from "@services/LLMProviders/OpenRouter/OpenRouterAPI";
-import * as readline from "readline";
-import { singleton } from "tsyringe";
-import * as tty from "tty";
-
-@singleton()
-export class InteractiveSessionManager {
- private currentMessage: string = "";
- private keypressHandler: ((buffer: Buffer) => void) | null = null;
- private lineHandler: ((input: string) => void) | null = null;
- private closeHandler: (() => void) | null = null;
- private rl: readline.Interface | null = null;
- private agent: CrackedAgent | null = null;
- private options: any = null;
-
- constructor(private openRouterAPI: OpenRouterAPI) {}
-
- public initialize(rl: readline.Interface, agent: CrackedAgent, options: any) {
- this.rl = rl;
- this.agent = agent;
- this.options = options;
- }
-
- private setupKeypressHandling() {
- Iif (!this.rl) return;
-
- Iif (process.stdin instanceof tty.ReadStream) {
- process.stdin.setRawMode(true);
- process.stdin.resume();
-
- this.keypressHandler = async (buffer: Buffer) => {
- const key = buffer.toString();
- Iif (key === "\u001B") {
- // Escape key
- this.openRouterAPI.cancelStream();
- console.log("\nStreaming cancelled.");
- await this.restartSession();
- }
- };
-
- process.stdin.on("data", this.keypressHandler);
- }
- }
-
- private async restartSession() {
- Iif (!this.rl) return;
-
- console.log("Please type your new prompt and press enter...");
- this.rl.prompt();
-
- // If we have a current message, restart with it
- if (this.currentMessage) {
- await this.handleInput(this.currentMessage);
- } else {
- // Otherwise, prompt for new input
- this.rl.prompt();
- }
- }
-
- private async handleInput(input: string) {
- Iif (!this.rl || !this.agent || !this.options) return;
-
- Iif (input.toLowerCase() === "exit") {
- console.log("Goodbye!");
- this.cleanup();
- process.exit(0);
- }
-
- this.currentMessage = input;
-
- try {
- const result = await this.agent.execute(input, this.options);
- Iif (!this.options.stream && result) {
- console.log("\nResponse:", result.response);
- Iif (result.actions?.length) {
- console.log("\nExecuted Actions:");
- result.actions.forEach(({ action, result }) => {
- console.log(`\nAction: ${action}`);
- console.log(`Result: ${JSON.stringify(result, null, 2)}`);
- });
- }
- }
- } catch (error) {
- console.error("Error:", (error as Error).message);
- }
-
- this.rl.prompt();
- }
-
- public async start() {
- Iif (!this.rl) return;
-
- console.log(
- 'Interactive mode started. Type "exit" or press Ctrl+C to quit.',
- );
- this.setupKeypressHandling();
- this.rl.prompt();
-
- this.lineHandler = async (input: string) => {
- await this.handleInput(input);
- };
-
- this.closeHandler = () => {
- this.cleanup();
- process.exit(0);
- };
-
- this.rl.on("line", this.lineHandler);
- this.rl.on("close", this.closeHandler);
- }
-
- public cleanup() {
- Iif (this.keypressHandler && process.stdin instanceof tty.ReadStream) {
- process.stdin.removeListener("data", this.keypressHandler);
- process.stdin.setRawMode(false);
- }
-
- Iif (this.lineHandler && this.rl) {
- this.rl.removeListener("line", this.lineHandler);
- }
-
- Iif (this.closeHandler && this.rl) {
- this.rl.removeListener("close", this.closeHandler);
- }
-
- Iif (this.rl) {
- this.rl.close();
- this.rl = null;
- }
-
- this.agent = null;
- this.options = null;
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 -71 -72 -73 -74 -75 -76 -77 -78 -79 -80 -81 -82 -83 -84 -85 -86 -87 -88 -89 -90 -91 -92 -93 -94 -95 -96 -97 -98 -99 -100 -101 -102 -103 -104 -105 -106 -107 -108 -109 -110 -111 -112 -113 -114 -115 -116 -117 -118 -119 -120 -121 -122 -123 -124 -125 -126 -127 -128 -129 -130 -131 -132 -133 -134 -135 -136 -137 -138 -139 -140 -141 -142 -143 -144 -145 -146 -147 -148 -149 -150 -151 -152 -153 -154 -155 -156 -157 -158 -159 -160 -161 -162 -163 -164 -165 -166 -167 -168 -169 -170 -171 -172 -173 -174 -175 -176 -177 -178 -179 -180 -181 -182 -183 -184 -185 -186 -187 -188 -189 -190 -191 -192 -193 -194 -195 -196 -197 -198 -199 -200 -201 -202 -203 -204 -205 -206 -207 -208 -209 -210 -211 -212 -213 -214 -215 -216 -217 -218 -219 -220 -221 -222 -223 -224 -225 -226 -227 -228 -229 -230 -231 -232 -233 -234 -235 -236 -237 -238 -239 -240 -241 -242 -243 -244 -245 -246 -247 -248 -249 -250 -251 -252 -253 -254 -255 -256 -257 -258 -259 -260 -261 -262 -263 -264 -265 -266 -267 -268 -269 -270 -271 -272 -273 -274 -275 -276 -277 -278 -279 -280 -281 -282 -283 -284 -285 -286 -287 -288 -289 -290 -291 -292 -293 -294 -295 -296 -297 -298 -299 -300 -301 -302 -303 -304 -305 -306 -307 -308 -309 -310 -311 -312 -313 -314 -315 -316 -317 -318 -319 -320 -321 -322 -323 -324 -325 -326 -327 -328 -329 -330 -331 -332 -333 -334 -335 -336 -337 -338 -339 -340 -341 -342 -343 -344 -345 -346 -347 -348 -349 -350 -351 -352 -353 -354 -355 -356 -357 -358 -359 -360 | 3x - -3x -3x - -3x - -3x -3x -3x - - - - - - - - - - - - - - - - -3x -30x -30x -30x -30x -30x - - -30x -30x - - - -2x -2x -2x -2x -2x -2x - - - - - - - -1x - - -1x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -1x - - - - - - - - - -1x - -1x -1x -1x -1x - -1x -1x - -1x -1x -1x -1x - - -1x -1x - - -1x - - - -34x -34x - - - - - - -3x -3x - - - - - - - - -3x -3x - - - - - - - - -9x - -8x -8x - -8x -8x -8x -8x - - -8x - - - - - - - - - - - - - - - - - -9x -9x - - - - - - - - - - - - -14x -4x -4x - - - - - - - - - - - - - -9x -9x - -9x - - - - - - - - - - - - - - - - - -9x - -9x -9x - - - - - - - - - -9x -9x -9x -9x - - -9x - - - -9x -5x -5x -5x - - -9x -4x - -4x -4x - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -3x -3x -3x - - -3x -3x -3x -3x -3x - -3x - -3x - -1x -1x - - -1x - - - - - - - -1x -1x - - - -5x - - - | import { ActionsParser } from "@services/LLM/actions/ActionsParser";
-
-import { DebugLogger } from "@services/logging/DebugLogger";
-import { autoInjectable } from "tsyringe";
-import { WriteStream } from "tty";
-import { LLMError } from "../LLMProviders/OpenRouter/OpenRouterAPI";
-
-const MAX_BUFFER_SIZE = 10 * 1024 * 1024; // 10MB default max buffer size
-const CHUNK_SIZE = 1024 * 1024; // 1MB chunk size for processing
-const STREAM_TIMEOUT = 10000; // 10 seconds timeout for inactivity
-
-export interface StreamCallback {
- (message: string): Promise<string>;
-}
-
-export interface StreamChunkCallback {
- (chunk: string, error?: LLMError): void;
-}
-
-interface ErrorDisplay {
- title: string;
- details: string;
- suggestion?: string;
-}
-
-@autoInjectable()
-export class StreamHandler {
- private responseBuffer: string = "";
- private isStreamComplete: boolean = false;
- private lastActivityTimestamp: number = Date.now();
- private bufferSize: number = 0;
- private inactivityTimer: NodeJS.Timeout | null = null;
-
- constructor(
- private debugLogger: DebugLogger,
- private actionsParser: ActionsParser,
- ) {}
-
- reset() {
- this.responseBuffer = "";
- this.isStreamComplete = false;
- this.lastActivityTimestamp = Date.now();
- this.bufferSize = 0;
- this.actionsParser.reset();
- this.clearInactivityTimer();
- }
-
- get response() {
- return this.responseBuffer;
- }
-
- private formatErrorDisplay(error: LLMError): ErrorDisplay {
- Iif (!error)
- return { title: "Unknown Error", details: "An unknown error occurred" };
-
- switch (error.type) {
- case "CONTEXT_LENGTH_EXCEEDED":
- return {
- title: "Context Length Exceeded",
- details: `Maximum context length (${error.details?.maxLength} tokens) exceeded. Current length: ${error.details?.currentLength} tokens.`,
- suggestion:
- "Try reducing the conversation history or splitting your request into smaller chunks.",
- };
- case "RATE_LIMIT_EXCEEDED":
- return {
- title: "Rate Limit Exceeded",
- details: `API rate limit reached.${
- error.details?.retryAfter
- ? ` Try again in ${error.details.retryAfter} seconds.`
- : ""
- }`,
- suggestion: "Please wait before making another request.",
- };
- case "MODEL_ERROR":
- return {
- title: "Model Error",
- details: `Error with model${
- error.details?.modelId ? ` ${error.details.modelId}` : ""
- }: ${error.message}`,
- suggestion:
- "Try using a different model or reducing the complexity of your request.",
- };
- case "INSUFFICIENT_QUOTA":
- return {
- title: "Insufficient Token Budget",
- details: `Required: ${error.details?.required}, Available: ${error.details?.available}`,
- suggestion: "Please check your API quota or upgrade your plan.",
- };
- case "NETWORK_ERROR":
- return {
- title: "Network Error",
- details: error.message,
- suggestion: "Check your internet connection and try again.",
- };
- case "STREAM_TIMEOUT":
- return {
- title: "Stream Timeout",
- details: "The stream was inactive for too long.",
- suggestion:
- "Try your request again. If the issue persists, try reducing the complexity of your request.",
- };
- case "STREAM_ERROR":
- return {
- title: "Stream Error",
- details: error.message,
- suggestion:
- "Try your request again with a different prompt or model.",
- };
- case "BUFFER_OVERFLOW":
- return {
- title: "Buffer Overflow",
- details: "The stream buffer has exceeded its maximum size limit.",
- suggestion:
- "Try processing the stream in smaller chunks or increase the buffer size limit.",
- };
- default:
- return {
- title: "Error",
- details: error.message || "An unexpected error occurred",
- suggestion:
- "Try your request again. If the issue persists, try with different parameters.",
- };
- }
- }
-
- private displayError(error: LLMError) {
- const { title, details, suggestion } = this.formatErrorDisplay(error);
-
- this.safeWriteToStdout("\n\n");
- this.safeWriteToStdout("\x1b[31m"); // Red color
- this.safeWriteToStdout(`❌ ${title}\n`);
- this.safeWriteToStdout("\x1b[0m"); // Reset color
-
- this.safeWriteToStdout("\x1b[37m"); // Light gray color
- this.safeWriteToStdout(`${details}\n`);
-
- if (suggestion) {
- this.safeWriteToStdout("\n");
- this.safeWriteToStdout("\x1b[36m"); // Cyan color
- this.safeWriteToStdout(`💡 ${suggestion}\n`);
- }
-
- this.safeWriteToStdout("\x1b[0m"); // Reset color
- this.safeWriteToStdout("\n");
-
- // Log error to debug logger
- this.debugLogger.log("Error", details, { title, suggestion });
- }
-
- private safeWriteToStdout(text: string) {
- try {
- process.stdout.write(text);
- } catch (error) {
- console.error("Error writing to stdout", error);
- }
- }
-
- private safeClearLine() {
- try {
- Iif ((process.stdout as WriteStream).clearLine) {
- (process.stdout as WriteStream).clearLine(0);
- }
- } catch (error) {
- console.error("Error clearing line", error);
- }
- }
-
- private safeCursorTo(x: number) {
- try {
- Iif ((process.stdout as WriteStream).cursorTo) {
- (process.stdout as WriteStream).cursorTo(x);
- }
- } catch (error) {
- console.error("Error moving cursor", error);
- }
- }
-
- private processChunk(chunk: string): string[] {
- if (!chunk) return [""];
-
- const chunks: string[] = [];
- let remainingChunk = chunk;
-
- while (remainingChunk.length > 0) {
- const chunkToProcess = remainingChunk.slice(0, CHUNK_SIZE);
- chunks.push(chunkToProcess);
- remainingChunk = remainingChunk.slice(CHUNK_SIZE);
- }
-
- return chunks;
- }
-
- private handleBufferOverflow() {
- // Keep the last 1MB of data when overflow occurs
- const keepSize = 1024 * 1024;
- this.responseBuffer = this.responseBuffer.slice(-keepSize);
- this.bufferSize = this.responseBuffer.length;
- this.actionsParser.clearBuffer();
- this.actionsParser.appendToBuffer(this.responseBuffer);
-
- this.debugLogger.log("Buffer Overflow", "Buffer size limit exceeded", {
- maxSize: MAX_BUFFER_SIZE,
- currentSize: this.bufferSize,
- });
- }
-
- private startInactivityTimer() {
- this.clearInactivityTimer();
- this.inactivityTimer = setTimeout(() => {
- this.displayError(
- new LLMError(
- "The stream was inactive for too long.",
- "STREAM_TIMEOUT",
- { timeout: STREAM_TIMEOUT },
- ),
- );
- this.reset();
- }, STREAM_TIMEOUT);
- }
-
- private clearInactivityTimer() {
- if (this.inactivityTimer) {
- clearTimeout(this.inactivityTimer);
- this.inactivityTimer = null;
- }
- }
-
- async handleChunk(
- chunk: string,
- model: string,
- llmCallback: StreamCallback,
- streamCallback: (
- message: string,
- callback: StreamChunkCallback,
- ) => Promise<void>,
- options?: Record<string, unknown>,
- ) {
- this.lastActivityTimestamp = Date.now();
- this.startInactivityTimer();
-
- Iif (chunk.startsWith('{"error":')) {
- try {
- const error = JSON.parse(chunk).error;
- const llmError = new LLMError(
- error.message || "Unknown error",
- error.type || "UNKNOWN_ERROR",
- error.details,
- );
- this.displayError(llmError);
- return [];
- } catch (e) {
- this.displayError(
- new LLMError("Unknown error", "UNKNOWN_ERROR", { originalError: e }),
- );
- return [];
- }
- }
-
- const chunks = this.processChunk(chunk);
-
- for (const subChunk of chunks) {
- Iif (this.bufferSize + subChunk.length > MAX_BUFFER_SIZE) {
- this.handleBufferOverflow();
- this.displayError(
- new LLMError("Buffer size limit exceeded", "BUFFER_OVERFLOW", {
- maxSize: MAX_BUFFER_SIZE,
- currentSize: this.bufferSize,
- }),
- );
- }
-
- this.safeWriteToStdout(subChunk);
- this.actionsParser.appendToBuffer(subChunk);
- this.responseBuffer += subChunk;
- this.bufferSize += subChunk.length;
- }
-
- const isMessageComplete = this.actionsParser.isCompleteMessage(
- this.actionsParser.buffer,
- );
-
- if (!this.actionsParser.isComplete && isMessageComplete) {
- this.actionsParser.isComplete = true;
- this.isStreamComplete = true;
- this.safeWriteToStdout("\n");
- }
-
- if (this.isStreamComplete && !this.actionsParser.isProcessing) {
- this.actionsParser.isProcessing = true;
-
- try {
- const actionResult = await this.actionsParser.parseAndExecuteActions(
- this.actionsParser.buffer,
- model,
- async (message) => {
- let actionResponse = "";
- await streamCallback(message, (chunk: string, error?: LLMError) => {
- Iif (error) {
- this.displayError(error);
- return;
- }
- this.safeWriteToStdout(chunk);
- actionResponse += chunk;
- this.lastActivityTimestamp = Date.now();
- });
-
- Iif (this.bufferSize + actionResponse.length > MAX_BUFFER_SIZE) {
- this.handleBufferOverflow();
- this.displayError(
- new LLMError(
- "Buffer size limit exceeded after action",
- "BUFFER_OVERFLOW",
- { maxSize: MAX_BUFFER_SIZE, currentSize: this.bufferSize },
- ),
- );
- }
-
- this.responseBuffer += actionResponse;
- this.bufferSize += actionResponse.length;
- return actionResponse;
- },
- );
-
- // Properly reset state after action execution
- this.actionsParser.reset();
- this.isStreamComplete = false;
- this.bufferSize = 0;
-
- // Refresh terminal state for new input
- this.safeWriteToStdout("\n");
- this.safeWriteToStdout("\x1B[?25h"); // Show cursor
- this.safeClearLine();
- this.safeCursorTo(0);
- this.safeWriteToStdout("> ");
-
- this.clearInactivityTimer();
-
- return actionResult.actions;
- } catch (error) {
- this.debugLogger.log("Error", "Error processing actions", { error });
- Iif (error instanceof LLMError) {
- this.displayError(error);
- } else {
- this.displayError(
- new LLMError(
- error.message || "Unknown error during action execution",
- "ACTION_ERROR",
- { originalError: error },
- ),
- );
- }
- this.reset();
- return [];
- }
- }
-
- return [];
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| InteractiveSessionManager.ts | -
-
- |
- 7.69% | -6/78 | -0% | -0/26 | -0% | -0/11 | -5.55% | -4/72 | -
| StreamHandler.ts | -
-
- |
- 66.9% | -95/142 | -33.33% | -13/39 | -68.75% | -11/16 | -66.18% | -92/139 | -
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 | - -16x - - - - - - - -16x - -16x - - - - - - - - -11x - - -11x - - - | /* eslint-disable no-control-regex */
-
-import { autoInjectable } from "tsyringe";
-
-/**
- * AnsiStripper Class
- *
- * This class provides functionality to remove ANSI escape codes from strings.
- */
-@autoInjectable()
-export class AnsiStripper {
- // Regular expression to match ANSI escape codes
- private static readonly ansiRegex: RegExp = /\x1B\[[0-?]*[ -/]*[@-~]/g;
-
- /**
- * Strips ANSI escape codes from the input string.
- *
- * @param input - The string potentially containing ANSI escape codes.
- * @returns The cleaned string without ANSI codes.
- */
- public strip(input: string): string {
- Iif (typeof input !== "string") {
- throw new TypeError("Input must be a string");
- }
- return input.replace(AnsiStripper.ansiRegex, "");
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| 1 -2 -3 -4 -5 -6 -7 -8 -9 -10 -11 -12 -13 -14 -15 -16 -17 -18 -19 -20 -21 -22 -23 -24 -25 -26 -27 -28 -29 -30 -31 -32 -33 -34 -35 -36 -37 -38 -39 -40 -41 -42 -43 -44 -45 -46 -47 -48 -49 -50 -51 -52 -53 -54 -55 -56 -57 -58 -59 -60 -61 -62 -63 -64 -65 -66 -67 -68 -69 -70 | 20x -20x - - - - - - - -20x - - - - - - - -8x -1x - - -7x - - -7x -6x - - - -7x - - - - - - - - - -14x -31x - -2x -2x - - - -29x - - - - - - - - - - - - -29x -18x - - - -11x - - - - | import he from "he";
-import { injectable } from "tsyringe";
-
-interface IHtmlEntityDecoderOptions {
- unescape?: boolean;
- unescapeChars?: string[];
-}
-
-@injectable()
-export class HtmlEntityDecoder {
- /**
- * Decodes HTML entities and cleans up character escaping in the provided text.
- * @param text The string containing HTML entities and escaped characters.
- * @param options Configuration options for decoding
- * @returns The decoded and unescaped string.
- */
- decode(text: string, options?: IHtmlEntityDecoderOptions): string {
- if (!text) {
- return "";
- }
-
- let result = text;
-
- // Always unescape if unescapeChars are provided, regardless of unescape option
- if (options?.unescapeChars?.length || options?.unescape) {
- result = this.unescapeString(result, options?.unescapeChars);
- }
-
- // Step 2: Decode HTML entities
- return he.decode(result);
- }
-
- /**
- * Unescapes specific backslash-escaped characters in a string.
- * @param str The string with escaped characters.
- * @param chars Optional array of specific characters to unescape
- * @returns The unescaped string.
- */
- private unescapeString(str: string, chars?: string[]): string {
- return str.replace(/\\u([0-9a-fA-F]{4})|\\(.)/g, (match, unicode, char) => {
- if (unicode) {
- // Handle Unicode escape sequences
- const code = parseInt(unicode, 16);
- return String.fromCharCode(code);
- }
-
- // Special characters map for escaped sequences
- const specialCharMap: { [key: string]: string } = {
- n: "\n",
- t: "\t",
- r: "\r",
- b: "\b",
- f: "\f",
- '"': '"',
- "'": "'",
- "\\": "\\",
- "/": "/",
- };
-
- // If specific chars are provided, only unescape those
- if (chars) {
- return chars.includes(char) ? specialCharMap[char] || char : match;
- }
-
- // Handle other escaped characters when no specific chars are provided
- return specialCharMap[char] || char;
- });
- }
-}
- |
- Press n or j to go to the next uncovered block, b, p or k for the previous block. -
- -| File | -- | Statements | -- | Branches | -- | Functions | -- | Lines | -- |
|---|---|---|---|---|---|---|---|---|---|
| AnsiStripper.ts | -
-
- |
- 85.71% | -6/7 | -0% | -0/1 | -100% | -1/1 | -83.33% | -5/6 | -
| HTMLEntityDecoder.ts | -
-
- |
- 100% | -19/19 | -91.66% | -11/12 | -100% | -3/3 | -100% | -17/17 | -