added changeset and updated test cases

Author: Prashant-7718

.changeset/smart-nails-mix.md

@@ -0,0 +1,5 @@
+---
+'@mermaidchart/sdk': patch
+---
+
+Add a regenerate method to the SDK that complements our Mermaid Diagram Sync GitHub App functionality. This method should accept the existing Mermaid diagram content along with source file changes, and regenerate the diagram based on those updates while preserving the overall diagram structure and intent.

packages/sdk/src/index.e2e.test.ts

@@ -276,6 +276,8 @@ describe('regenerateDiagram', () => {
     ];
 
     try {
+      const creditsBefore = await client.getAICredits();
+
       const result = await client.regenerateDiagram({
         code,
         sourceFiles,
@@ -285,6 +287,21 @@ describe('regenerateDiagram', () => {
       expect(result).toHaveProperty('result');
       expect(result).toHaveProperty('code');
       expect(['ok', 'fail']).toContain(result.result);
+
+      // When the AI successfully regenerates the diagram, creditUsage should be present
+      // and credits should have been deducted
+      if (result.result === 'ok' && result.solved === true) {
+        expect(result.creditUsage).toMatchObject({
+          creditsToDeduct: expect.any(Number),
+          baseCost: expect.any(Number),
+          reason: expect.any(String),
+        });
+
+        const creditsAfter = await client.getAICredits();
+        expect(creditsAfter.aiCredits.remaining).toBe(
+          creditsBefore.aiCredits.remaining - result.creditUsage!.creditsToDeduct,
+        );
+      }
     } catch (error) {
       if (error instanceof AICreditsLimitExceededError) {
         return; // Credits exceeded is acceptable for E2E test

packages/sdk/src/index.test.ts

@@ -221,74 +221,4 @@ describe('MermaidChart', () => {
     });
   });
 
-  describe('#regenerateDiagram', () => {
-    beforeEach(async () => {
-      await client.setAccessToken('test-access-token');
-    });
-
-    it('should POST to the regenerate endpoint with the request body and return response.data', async () => {
-      const jsonResponse = {
-        result: 'ok' as const,
-        code: '```mermaid\nflowchart TD\n  A --> B --> C\n```',
-        solved: true,
-      };
-
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      const postSpy = vi.spyOn((client as any).axios, 'post').mockResolvedValue({
-        data: jsonResponse,
-      });
-
-      const requestBody = {
-        code: 'flowchart TD\n  A --> B',
-        sourceFiles: ['const x = 1;', 'function foo() {}'],
-      };
-
-      const result = await client.regenerateDiagram(requestBody);
-
-      expect(postSpy).toHaveBeenCalledWith(URLS.rest.openai.regenerate, requestBody);
-      expect(result).toEqual(jsonResponse);
-    });
-
-    it('should include creditUsage in response when present', async () => {
-      const jsonResponse = {
-        result: 'ok' as const,
-        code: '```mermaid\nflowchart TD\n  A --> B\n```',
-        solved: true,
-        creditUsage: {
-          creditsToDeduct: 1,
-          baseCost: 1,
-          reason: 'regeneration',
-        },
-      };
-
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      vi.spyOn((client as any).axios, 'post').mockResolvedValue({
-        data: jsonResponse,
-      });
-
-      const result = await client.regenerateDiagram({
-        code: 'flowchart TD\n  A --> B',
-        sourceFiles: ['const x = 1;'],
-      });
-
-      expect(result.creditUsage).toEqual(jsonResponse.creditUsage);
-    });
-
-    it('should throw AICreditsLimitExceededError on 402', async () => {
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      vi.spyOn((client as any).axios, 'post').mockRejectedValue({
-        response: {
-          status: 402,
-          data: 'AI credits limit exceeded',
-        },
-      });
-
-      await expect(
-        client.regenerateDiagram({
-          code: 'flowchart TD\n  A --> B',
-          sourceFiles: [],
-        }),
-      ).rejects.toThrow(AICreditsLimitExceededError);
-    });
-  });
 });