Add referenceExample

Author: mmssajith

src/__tests__/ConfigService.test.ts

@@ -107,6 +107,7 @@ describe("ConfigService", () => {
           directoryFirst: true,
           excludeDirectories: false,
         },
+        referenceExamples: {}, // Added referenceExamples
       };
 
       (fs.existsSync as jest.Mock).mockReturnValue(true);
@@ -154,4 +155,4 @@ describe("ConfigService", () => {
       expect(fs.existsSync).toHaveBeenCalledWith(mockConfigPath);
     });
   });
-});
+});

src/services/ConfigService.ts

@@ -53,6 +53,7 @@ const configSchema = z.object({
       directoryFirst: true,
       excludeDirectories: false,
     }),
+  referenceExamples: z.record(z.string(), z.string()).optional().default({}),
 });
 
 export type Config = z.infer<typeof configSchema>;
@@ -137,6 +138,12 @@ export class ConfigService {
           directoryFirst: true,
           excludeDirectories: false,
         },
+        referenceExamples: {
+          example1: "path/to/example1/file.ts",
+          example2: "path/to/example2/file.ts",
+          myService: "src/services/MyService.ts",
+          anotherKey: "path/to/some/other/example.ts",
+        },
       };
       fs.writeFileSync(
         this.CONFIG_PATH,

src/services/LLM/LLMContextCreator.ts

@@ -77,6 +77,13 @@ export class LLMContextCreator {
     const runOneTestCmd = config.runOneTestCmd || "yarn test {testPath}";
     const runTypeCheckCmd = config.runTypeCheckCmd || "yarn type-check";
 
+    // Add flexible reference examples section
+    const referenceExamplesSection = Object.entries(
+      config.referenceExamples || {},
+    )
+      .map(([key, path]) => `${key}: ${path}`)
+      .join("\n");
+
     return `# Project Dependencies (from ${info.dependencyFile})
 Main Dependencies: ${info.mainDependencies.join(", ")}
 
@@ -88,7 +95,10 @@ ${Object.entries(info.scripts)
 # Test Commands
 Run All Tests: ${runAllTestsCmd}
 Run Single Test: ${runOneTestCmd}
-Run Type Check: ${runTypeCheckCmd}`;
+Run Type Check: ${runTypeCheckCmd}
+
+# Reference Examples
+${referenceExamplesSection}`;
   }
 
   private formatFirstTimeMessage(context: MessageContext): string {

src/services/LLM/__tests__/LLMContextCreator.test.ts

@@ -134,6 +134,49 @@ describe("LLMContextCreator", () => {
       expect(result).toContain(message);
       expect(result).not.toContain("file1\nfile2");
     });
+
+    it("should load config with flexible referenceExamples", async () => {
+      const mockConfig = {
+        includeAllFilesOnEnvToContext: true,
+        referenceExamples: {
+          serviceA: "src/services/ServiceA.ts",
+          utilityB: "src/utils/UtilityB.ts",
+          someOtherKey: "path/to/another/example.ts",
+        },
+      };
+
+      mocker.spyOnPrototypeAndReturn(ConfigService, "getConfig", mockConfig);
+
+      const result = await contextCreator.create(
+        "test message",
+        "/test/root",
+        true,
+      );
+
+      expect(result).toContain("Reference Examples");
+      expect(result).toContain("serviceA: src/services/ServiceA.ts");
+      expect(result).toContain("utilityB: src/utils/UtilityB.ts");
+      expect(result).toContain("someOtherKey: path/to/another/example.ts");
+    });
+
+    it("should use default empty object for referenceExamples when not provided", async () => {
+      const mockConfig = {
+        includeAllFilesOnEnvToContext: true,
+        referenceExamples: {},
+      };
+
+      mocker.spyOnPrototypeAndReturn(ConfigService, "getConfig", mockConfig);
+
+      const result = await contextCreator.create(
+        "test message",
+        "/test/root",
+        true,
+      );
+
+      expect(result).toContain("Reference Examples");
+      expect(result).not.toContain("serviceA:");
+      expect(result).not.toContain("utilityB:");
+    });
   });
 
   afterEach(() => {

src/services/LLM/__tests__/PhaseManager.test.ts

@@ -46,6 +46,7 @@ describe("PhaseManager", () => {
         maxGlobalTries: 20,
       },
     ],
+    referenceExamples: {}, // Added referenceExamples
   };
 
   beforeAll(() => {