@autobe/agent/src/orchestrate/test/transformTestCorrectHistories.ts

AI backend server code generator

import { IAgenticaHistoryJson } from "@agentica/core";
import { IAutoBeTypeScriptCompileResult } from "@autobe/interface";
import { v4 } from "uuid";

import { AutoBeSystemPromptConstant } from "../../constants/AutoBeSystemPromptConstant";
import { IAutoBeTestFunction } from "./structures/IAutoBeTestFunction";
import { transformTestWriteHistories } from "./transformTestWriteHistories";

export const transformTestCorrectHistories = (
  func: IAutoBeTestFunction,
  failure: IAutoBeTypeScriptCompileResult.IFailure,
): Array<
  IAgenticaHistoryJson.IAssistantMessage | IAgenticaHistoryJson.ISystemMessage
> => [
  ...transformTestWriteHistories(func.scenario, func.artifacts),
  {
    id: v4(),
    created_at: new Date().toISOString(),
    type: "assistantMessage",
    text: [
      "## Generated TypeScript Code",
      "```typescript",
      func.script,
      "```",
      "",
      "## Compile Errors",
      "Fix the compilation error in the provided code.",
      "",
      "```json",
      JSON.stringify(failure.diagnostics),
      "```",
    ].join("\n"),
  },
  {
    id: v4(),
    created_at: new Date().toISOString(),
    type: "systemMessage",
    text: AutoBeSystemPromptConstant.TEST_CORRECT.replace(
      "{{API_DTO_SCHEMAS}}",
      transformTestWriteHistories.structures(func.artifacts),
    ).replace(
      "{{API_SDK_FUNCTIONS}}",
      transformTestWriteHistories.functional(func.artifacts),
    ),
  },
];