@aigne/doc-smith 0.8.12-beta.8 → 0.8.12-beta.9

package/tests/agents/update/check-update-is-single.test.mjs

@@ -1,300 +0,0 @@
-import { afterEach, beforeEach, describe, expect, mock, spyOn, test } from "bun:test";
-import checkUpdateIsSingle from "../../../agents/update/check-update-is-single.mjs";
-
-describe("check-update-is-single", () => {
-  let mockOptions;
-  let consoleSpy;
-
-  beforeEach(() => {
-    mockOptions = {
-      context: {
-        agents: {
-          updateSingleDocument: { mockSingleAgent: true },
-          batchUpdateDocument: { mockBatchAgent: true },
-        },
-        invoke: mock(async () => ({ mockResult: true })),
-      },
-    };
-
-    consoleSpy = spyOn(console, "error").mockImplementation(() => {});
-
-    // Clear context mock call history
-    mockOptions.context.invoke.mockClear();
-  });
-
-  afterEach(() => {
-    consoleSpy?.mockRestore();
-  });
-
-  // INPUT VALIDATION TESTS
-  test("should throw error when selectedDocs is not provided", async () => {
-    await expect(checkUpdateIsSingle({}, mockOptions)).rejects.toThrow(
-      "selectedDocs must be provided as an array",
-    );
-  });
-
-  test("should throw error when selectedDocs is not an array", async () => {
-    await expect(checkUpdateIsSingle({ selectedDocs: "not-array" }, mockOptions)).rejects.toThrow(
-      "selectedDocs must be provided as an array",
-    );
-  });
-
-  test("should throw error when selectedDocs is empty", async () => {
-    await expect(checkUpdateIsSingle({ selectedDocs: [] }, mockOptions)).rejects.toThrow(
-      "selectedDocs cannot be empty",
-    );
-  });
-
-  test("should throw error when selectedDocs is null", async () => {
-    await expect(checkUpdateIsSingle({ selectedDocs: null }, mockOptions)).rejects.toThrow(
-      "selectedDocs must be provided as an array",
-    );
-  });
-
-  test("should throw error when selectedDocs is undefined", async () => {
-    await expect(checkUpdateIsSingle({ selectedDocs: undefined }, mockOptions)).rejects.toThrow(
-      "selectedDocs must be provided as an array",
-    );
-  });
-
-  // AGENT AVAILABILITY TESTS
-  test("should throw error when updateSingleDocument agent is not available", async () => {
-    const optionsWithoutSingleAgent = {
-      context: {
-        agents: {
-          batchUpdateDocument: { mockBatchAgent: true },
-        },
-        invoke: mock(),
-      },
-    };
-
-    await expect(
-      checkUpdateIsSingle({ selectedDocs: ["doc1"] }, optionsWithoutSingleAgent),
-    ).rejects.toThrow('Agent "updateSingleDocument" is not available');
-  });
-
-  test("should throw error when batchUpdateDocument agent is not available", async () => {
-    const optionsWithoutBatchAgent = {
-      context: {
-        agents: {
-          updateSingleDocument: { mockSingleAgent: true },
-        },
-        invoke: mock(),
-      },
-    };
-
-    await expect(
-      checkUpdateIsSingle({ selectedDocs: ["doc1", "doc2"] }, optionsWithoutBatchAgent),
-    ).rejects.toThrow('Agent "batchUpdateDocument" is not available');
-  });
-
-  test("should throw error when no agents are available", async () => {
-    const optionsWithoutAgents = {
-      context: {
-        agents: {},
-        invoke: mock(),
-      },
-    };
-
-    await expect(
-      checkUpdateIsSingle({ selectedDocs: ["doc1"] }, optionsWithoutAgents),
-    ).rejects.toThrow('Agent "updateSingleDocument" is not available');
-  });
-
-  // SINGLE DOCUMENT ROUTING TESTS
-  test("should route to updateSingleDocument when selectedDocs has one item", async () => {
-    const result = await checkUpdateIsSingle(
-      {
-        selectedDocs: ["doc1"],
-        customParam: "test",
-      },
-      mockOptions,
-    );
-
-    expect(mockOptions.context.invoke).toHaveBeenCalledWith(
-      { mockSingleAgent: true },
-      {
-        selectedDocs: ["doc1"],
-        customParam: "test",
-      },
-    );
-    expect(result).toEqual({ mockResult: true });
-  });
-
-  test("should pass through all parameters to single document agent", async () => {
-    await checkUpdateIsSingle(
-      {
-        selectedDocs: [{ path: "/doc1", content: "content" }],
-        docsDir: "./docs",
-        forceRegenerate: true,
-        customData: { key: "value" },
-      },
-      mockOptions,
-    );
-
-    expect(mockOptions.context.invoke).toHaveBeenCalledWith(
-      { mockSingleAgent: true },
-      {
-        selectedDocs: [{ path: "/doc1", content: "content" }],
-        docsDir: "./docs",
-        forceRegenerate: true,
-        customData: { key: "value" },
-      },
-    );
-  });
-
-  // BATCH DOCUMENT ROUTING TESTS
-  test("should route to batchUpdateDocument when selectedDocs has multiple items", async () => {
-    const result = await checkUpdateIsSingle(
-      {
-        selectedDocs: ["doc1", "doc2"],
-        customParam: "test",
-      },
-      mockOptions,
-    );
-
-    expect(mockOptions.context.invoke).toHaveBeenCalledWith(
-      { mockBatchAgent: true },
-      {
-        selectedDocs: ["doc1", "doc2"],
-        customParam: "test",
-      },
-    );
-    expect(result).toEqual({ mockResult: true });
-  });
-
-  test("should route to batchUpdateDocument when selectedDocs has three items", async () => {
-    await checkUpdateIsSingle(
-      {
-        selectedDocs: ["doc1", "doc2", "doc3"],
-      },
-      mockOptions,
-    );
-
-    expect(mockOptions.context.invoke).toHaveBeenCalledWith(
-      { mockBatchAgent: true },
-      {
-        selectedDocs: ["doc1", "doc2", "doc3"],
-      },
-    );
-  });
-
-  test("should pass through all parameters to batch document agent", async () => {
-    await checkUpdateIsSingle(
-      {
-        selectedDocs: [
-          { path: "/doc1", content: "content1" },
-          { path: "/doc2", content: "content2" },
-        ],
-        docsDir: "./docs",
-        forceRegenerate: false,
-        batchSize: 10,
-        customSettings: { parallel: true },
-      },
-      mockOptions,
-    );
-
-    expect(mockOptions.context.invoke).toHaveBeenCalledWith(
-      { mockBatchAgent: true },
-      {
-        selectedDocs: [
-          { path: "/doc1", content: "content1" },
-          { path: "/doc2", content: "content2" },
-        ],
-        docsDir: "./docs",
-        forceRegenerate: false,
-        batchSize: 10,
-        customSettings: { parallel: true },
-      },
-    );
-  });
-
-  // ERROR HANDLING TESTS
-  test("should handle and re-throw agent invocation errors for single document", async () => {
-    const mockError = new Error("Agent execution failed");
-    mockOptions.context.invoke.mockRejectedValue(mockError);
-
-    await expect(checkUpdateIsSingle({ selectedDocs: ["doc1"] }, mockOptions)).rejects.toThrow(
-      "Agent execution failed",
-    );
-
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "Error invoking updateSingleDocument:",
-      "Agent execution failed",
-    );
-  });
-
-  test("should handle and re-throw agent invocation errors for batch documents", async () => {
-    const mockError = new Error("Batch processing failed");
-    mockOptions.context.invoke.mockRejectedValue(mockError);
-
-    await expect(
-      checkUpdateIsSingle({ selectedDocs: ["doc1", "doc2"] }, mockOptions),
-    ).rejects.toThrow("Batch processing failed");
-
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "Error invoking batchUpdateDocument:",
-      "Batch processing failed",
-    );
-  });
-
-  test("should log specific error message for single document failures", async () => {
-    const mockError = new Error("Network timeout");
-    mockOptions.context.invoke.mockRejectedValue(mockError);
-
-    try {
-      await checkUpdateIsSingle({ selectedDocs: ["doc1"] }, mockOptions);
-    } catch {
-      // Expected to throw
-    }
-
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "Error invoking updateSingleDocument:",
-      "Network timeout",
-    );
-  });
-
-  test("should log specific error message for batch document failures", async () => {
-    const mockError = new Error("Memory limit exceeded");
-    mockOptions.context.invoke.mockRejectedValue(mockError);
-
-    try {
-      await checkUpdateIsSingle({ selectedDocs: ["doc1", "doc2", "doc3"] }, mockOptions);
-    } catch {
-      // Expected to throw
-    }
-
-    expect(consoleSpy).toHaveBeenCalledWith(
-      "Error invoking batchUpdateDocument:",
-      "Memory limit exceeded",
-    );
-  });
-
-  // RESULT PASSING TESTS
-  test("should return result from single document agent unchanged", async () => {
-    const mockResult = {
-      success: true,
-      processedDocs: ["doc1"],
-      metadata: { timestamp: "2023-01-01" },
-    };
-    mockOptions.context.invoke.mockResolvedValue(mockResult);
-
-    const result = await checkUpdateIsSingle({ selectedDocs: ["doc1"] }, mockOptions);
-
-    expect(result).toEqual(mockResult);
-  });
-
-  test("should return result from batch document agent unchanged", async () => {
-    const mockResult = {
-      success: true,
-      processedDocs: ["doc1", "doc2"],
-      errors: [],
-      summary: "All documents processed successfully",
-    };
-    mockOptions.context.invoke.mockResolvedValue(mockResult);
-
-    const result = await checkUpdateIsSingle({ selectedDocs: ["doc1", "doc2"] }, mockOptions);
-
-    expect(result).toEqual(mockResult);
-  });
-});

package/tests/agents/update/document-tools/update-document-content.test.mjs

@@ -1,329 +0,0 @@
-import { beforeEach, describe, expect, test } from "bun:test";
-import updateDocumentContent from "../../../../agents/update/document-tools/update-document-content.mjs";
-
-describe("update-document-content", () => {
-  let mockOptions;
-
-  beforeEach(() => {
-    mockOptions = {
-      context: {
-        userContext: {
-          currentContent: "",
-        },
-      },
-    };
-  });
-
-  // Helper function to extract content from page_content wrapper
-  function extractContent(wrappedContent) {
-    const match = wrappedContent.match(/<page_content>\s*([\s\S]*?)\s*<\/page_content>/);
-    return match ? match[1] : wrappedContent;
-  }
-
-  // INPUT VALIDATION TESTS
-  test("should return error when diffPatch is not provided", async () => {
-    mockOptions.context.userContext.currentContent = "original content";
-    const result = await updateDocumentContent({}, mockOptions);
-    expect(result.success).toBe(false);
-    expect(result.error.message).toContain("diffPatch");
-  });
-
-  test("should return error when diffPatch is not a string", async () => {
-    mockOptions.context.userContext.currentContent = "original content";
-    const result = await updateDocumentContent({ diffPatch: null }, mockOptions);
-    expect(result.success).toBe(false);
-    expect(result.error.message).toContain("diffPatch");
-  });
-
-  test("should return error when diffPatch is empty string", async () => {
-    mockOptions.context.userContext.currentContent = "original content";
-    const result = await updateDocumentContent({ diffPatch: "" }, mockOptions);
-    expect(result.success).toBe(false);
-    expect(result.error.message).toContain("Diff patch is required");
-  });
-
-  // SUCCESSFUL PATCH APPLICATION TESTS
-  test("should successfully apply simple diff patch", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3";
-    const diffPatch = "@@ -1,3 +1,3 @@\n line 1\n-line 2\n+updated line 2\n line 3";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe("line 1\nupdated line 2\nline 3");
-    expect(result.message).toContain("Document content updated successfully");
-  });
-
-  test("should handle diff with multiple hunks", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3\nline 4\nline 5";
-    const diffPatch =
-      "@@ -1,2 +1,2 @@\n-line 1\n+updated line 1\n line 2\n@@ -4,2 +4,2 @@\n line 4\n-line 5\n+updated line 5";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe(
-      "updated line 1\nline 2\nline 3\nline 4\nupdated line 5",
-    );
-  });
-
-  test("should handle diff with additions only", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2";
-    const diffPatch = "@@ -2,0 +3,2 @@\n line 2\n+new line 3\n+new line 4";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe("line 1\nline 2\nnew line 3\nnew line 4");
-  });
-
-  test("should handle diff with deletions only", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3\nline 4";
-    const diffPatch = "@@ -2,2 +2,0 @@\n line 1\n-line 2\n-line 3\n line 4";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe("line 1\nline 4");
-  });
-
-  test("should handle diff with context lines", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3\nline 4\nline 5";
-    const diffPatch =
-      "@@ -2,3 +2,3 @@\n line 1\n line 2\n-line 3\n+updated line 3\n line 4\n line 5";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe(
-      "line 1\nline 2\nupdated line 3\nline 4\nline 5",
-    );
-  });
-
-  // DIFF PATCH PARSING TESTS
-  test("should return error for invalid diff format", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2";
-    const diffPatch = "invalid diff format";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result).toEqual({
-      success: false,
-      error: { message: "No valid hunks found in diff" },
-    });
-  });
-
-  test("should return error for empty diff", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2";
-    const diffPatch = "\n\n";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result).toEqual({
-      success: false,
-      error: { message: "No valid hunks found in diff" },
-    });
-  });
-
-  test("should handle malformed hunk headers gracefully", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2";
-    const diffPatch = "@@malformed header@@\n-line 1\n+updated line 1";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(false);
-    expect(result.error.message).toContain("No valid hunks found");
-  });
-
-  // LINE NUMBER FIXING TESTS
-  test("should fix line numbers when patch position is off", async () => {
-    mockOptions.context.userContext.currentContent = "prefix line\nline 1\nline 2\nline 3";
-    // This diff patch expects to find "line 1" at line 1, but it's actually at line 2
-    const diffPatch = "@@ -1,2 +1,2 @@\n-line 1\n+updated line 1\n line 2";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe(
-      "prefix line\nupdated line 1\nline 2\nline 3",
-    );
-  });
-
-  test("should use fuzzy matching when exact match fails", async () => {
-    mockOptions.context.userContext.currentContent =
-      "similar line 1\nsimilar line 2\ndifferent content\nsimilar line 3";
-    // This patch refers to lines that are similar but not exactly matching
-    const diffPatch = "@@ -10,2 +10,2 @@\n-similar line 1\n+updated line 1\n similar line 2";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe(
-      "updated line 1\nsimilar line 2\ndifferent content\nsimilar line 3",
-    );
-  });
-
-  test("should return error when no matching context found", async () => {
-    mockOptions.context.userContext.currentContent =
-      "completely different content\nnothing matches";
-    const diffPatch =
-      "@@ -1,2 +1,2 @@\n-line that doesn't exist\n+updated line\n another nonexistent line";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(false);
-    expect(result.error.message).toContain("Cannot find matching context");
-  });
-
-  // COMPLEX DIFF SCENARIOS
-  test("should handle diff with no context lines", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3";
-    const diffPatch = "@@ -2,1 +2,1 @@\n-line 2\n+updated line 2";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe("line 1\nupdated line 2\nline 3");
-  });
-
-  test("should handle diff with single line count (implicit 1)", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3";
-    const diffPatch = "@@ -2 +2 @@\n-line 2\n+updated line 2";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe("line 1\nupdated line 2\nline 3");
-  });
-
-  test("should handle multiple changes in single hunk", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3\nline 4\nline 5";
-    const diffPatch =
-      "@@ -2,3 +2,4 @@\n line 1\n-line 2\n-line 3\n+updated line 2\n+new line\n+updated line 3\n line 4";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe(
-      "line 1\nupdated line 2\nnew line\nupdated line 3\nline 4\nline 5",
-    );
-  });
-
-  // EDGE CASES
-  test("should handle empty lines in diff", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\n\nline 3";
-    const diffPatch = "@@ -1,3 +1,3 @@\n line 1\n \n-line 3\n+updated line 3";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe("line 1\n\nupdated line 3");
-  });
-
-  test("should handle diff with trailing newlines", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\n";
-    const diffPatch = "@@ -1,2 +1,2 @@\n-line 1\n+updated line 1\n line 2";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    // Note: applyPatch may normalize trailing newlines
-    const content = extractContent(result.updatedContent);
-    expect(content).toContain("updated line 1");
-    expect(content).toContain("line 2");
-  });
-
-  // REALISTIC DOCUMENTATION UPDATE SCENARIOS
-  test("should handle updating code blocks", async () => {
-    mockOptions.context.userContext.currentContent =
-      "# API Reference\n\n```javascript\nfunction oldFunction() {\n  return 'old';\n}\n```\n\nDescription here.";
-    const diffPatch =
-      "@@ -3,3 +3,3 @@\n # API Reference\n \n ```javascript\n-function oldFunction() {\n-  return 'old';\n+function newFunction() {\n+  return 'new';\n }\n ```";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    const content = extractContent(result.updatedContent);
-    expect(content).toContain("function newFunction()");
-    expect(content).toContain("return 'new';");
-  });
-
-  test("should handle adding new sections", async () => {
-    mockOptions.context.userContext.currentContent =
-      "# Documentation\n\n## Section 1\n\nContent 1\n\n## Section 3\n\nContent 3";
-    const diffPatch =
-      "@@ -4,0 +5,3 @@\n Content 1\n \n+## Section 2\n+\n+Content 2\n+\n ## Section 3";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    const content = extractContent(result.updatedContent);
-    expect(content).toContain("## Section 2");
-    expect(content).toContain("Content 2");
-  });
-
-  test("should handle removing outdated content", async () => {
-    mockOptions.context.userContext.currentContent =
-      "# Guide\n\n## Current Feature\n\nThis is current.\n\n## Deprecated Feature\n\nThis is deprecated.\n\n## Another Feature\n\nThis is also current.";
-    const diffPatch =
-      "@@ -5,4 +5,0 @@\n This is current.\n \n-## Deprecated Feature\n-\n-This is deprecated.\n-\n ## Another Feature";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    const content = extractContent(result.updatedContent);
-    expect(content).not.toContain("Deprecated Feature");
-    expect(content).toContain("Current Feature");
-    expect(content).toContain("Another Feature");
-  });
-
-  // ERROR SCENARIOS WITH REALISTIC CONTENT
-  test("should handle conflicting patches gracefully", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3";
-    // This patch tries to modify content that doesn't match exactly
-    const diffPatch =
-      "@@ -1,3 +1,3 @@\n-different line 1\n-different line 2\n+updated line 1\n+updated line 2\n line 3";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(false);
-    expect(result.error.message).toContain("Cannot find matching context");
-  });
-
-  test("should handle patches with incorrect line counts", async () => {
-    mockOptions.context.userContext.currentContent = "line 1\nline 2\nline 3\nline 4";
-    // Incorrect line count in hunk header (says 5 lines but content has 4)
-    const diffPatch = "@@ -1,5 +1,3 @@\n-line 1\n-line 2\n line 3\n line 4";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toBe("line 3\nline 4");
-  });
-
-  // FUZZY MATCHING EDGE CASES
-  test("should succeed fuzzy matching with high similarity", async () => {
-    mockOptions.context.userContext.currentContent =
-      "function calculateTotal(items) {\n  let total = 0;\n  for (item of items) {\n    total += item.price;\n  }\n  return total;\n}";
-    // Patch has slight differences but should match with fuzzy matching
-    const diffPatch =
-      "@@ -2,2 +2,2 @@\n function calculateTotal(items) {\n-  let total = 0;\n+  let sum = 0;\n   for (item of items) {";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(true);
-    expect(extractContent(result.updatedContent)).toContain("let sum = 0;");
-  });
-
-  test("should fail fuzzy matching with low similarity", async () => {
-    mockOptions.context.userContext.currentContent =
-      "completely different content\nwith nothing similar\nat all";
-    const diffPatch =
-      "@@ -1,2 +1,2 @@\n-some totally different text\n+updated text\n that has no relation";
-
-    const result = await updateDocumentContent({ diffPatch }, mockOptions);
-
-    expect(result.success).toBe(false);
-    expect(result.error.message).toContain("Cannot find matching context");
-  });
-});

package/tests/agents/update/generate-document.test.mjs

@@ -1,51 +0,0 @@
-import { afterAll, beforeAll, describe, expect, test } from "bun:test";
-import { join } from "node:path";
-import { AIAgent } from "@aigne/core";
-import { loadAgent } from "@aigne/core/loader/index.js";
-import { loadModel } from "../../utils/mock-chat-model.mjs";
-
-describe("generateDocument Agent", () => {
-  beforeAll(() => {
-    process.env.AIGNE_OBSERVABILITY_DISABLED = "true";
-  });
-
-  afterAll(() => {
-    delete process.env.AIGNE_OBSERVABILITY_DISABLED;
-  });
-  test("should load agent correctly with proper configuration", async () => {
-    const agent = await loadAgent(
-      join(import.meta.dirname, "../../../agents/update/generate-document.yaml"),
-      {
-        model: loadModel,
-      },
-    );
-
-    expect(agent).toBeDefined();
-
-    // Verify agent exists and is correct type
-    expect(agent).toBeDefined();
-    expect(agent).toBeInstanceOf(AIAgent);
-    expect(agent.name).toBe("generateDocument");
-  });
-
-  test("should have instructions loaded from file", async () => {
-    const agent = await loadAgent(
-      join(import.meta.dirname, "../../../agents/update/generate-document.yaml"),
-      {
-        model: loadModel,
-      },
-    );
-
-    expect(agent).toBeDefined();
-
-    // Verify instructions are loaded
-    expect(agent.instructions).toBeDefined();
-    const instructions = await agent.instructions.build({});
-    expect(instructions.messages).toBeDefined();
-    expect(instructions.messages.length).toBeGreaterThan(0);
-
-    // The instructions should contain content from the prompt file
-    const systemMessage = instructions.messages.find((m) => m.role === "system");
-    expect(systemMessage).toBeDefined();
-  });
-});