windmill/ai_evals/modes/global.ts

import { readFile, stat } from "node:fs/promises";
import { basename } from "node:path";
import { loadAppFixtureForEval } from "../adapters/frontend/core/app/appFixtureLoader";
import {
  runGlobalEval,
  type GlobalLiveEditorDraftFixture,
  type GlobalUserFixture,
} from "../adapters/frontend/core/global/globalEvalRunner";
import type { SeededArtifact } from "../adapters/frontend/core/global/evalArtifactStore";
import type { EvalPreviewTabFixture } from "../adapters/frontend/core/global/evalPreviewTabs";
import type { BenchmarkWorkspaceRunnables } from "../adapters/frontend/mockBackend";
import type { FrontendEvalModelConfig } from "../core/models";
import type {
  BenchmarkArtifactFile,
  GlobalValidationSpec,
  ModeRunner,
} from "../core/types";
import { validateGlobalState, type GlobalDraftState } from "../core/validators";
import type { WindmillBackendSettings } from "../core/windmillBackendSettings";
import { getFrontendApiKey } from "./frontendCommon";

export interface GlobalInitialFixture {
  workspace?: BenchmarkWorkspaceRunnables;
  liveEditorDrafts?: GlobalLiveEditorDraftFixture[];
  user?: GlobalUserFixture;
  artifacts?: SeededArtifact[];
  previewTabs?: EvalPreviewTabFixture[];
}

export function createGlobalModeRunner(
  modelConfig: FrontendEvalModelConfig,
  backendSettings: WindmillBackendSettings,
): ModeRunner<GlobalInitialFixture, GlobalDraftState, GlobalDraftState> {
  return {
    mode: "global",
    concurrency: 3,
    judgeThreshold: 80,
    async loadInitial(path) {
      return path ? await loadGlobalInitialFixture(path) : undefined;
    },
    async loadExpected(path) {
      return path ? await loadGlobalExpectedFixture(path) : undefined;
    },
    async run(prompt, initial, context) {
      const result = await runGlobalEval(
        prompt,
        getFrontendApiKey(modelConfig.provider),
        {
          workspaceFixtures: initial?.workspace,
          liveEditorDrafts: initial?.liveEditorDrafts,
          user: initial?.user,
          artifacts: initial?.artifacts,
          previewTabs: initial?.previewTabs,
          sessionChat: context.evalCase?.runtime?.sessionChat,
          planMode: context.evalCase?.runtime?.planMode,
          maxIterations: context.evalCase?.runtime?.maxTurns,
          userAnswers: context.evalCase?.runtime?.userAnswers,
          provider: modelConfig.provider,
          model: modelConfig.model,
          backend: backendSettings,
          runContext: context,
        },
      );

      return {
        success: result.success,
        actual: result.state,
        error: result.error,
        assistantMessageCount: result.assistantMessageCount,
        toolCallCount: result.toolCallCount,
        toolsUsed: result.toolsUsed,
        toolCallDetails: result.toolCallDetails,
        skillsInvoked: [],
        assistantText: result.assistantText,
        tokenUsage: result.tokenUsage,
        finalContextTokens: result.finalContextTokens,
      };
    },
    validate({ evalCase, actual, expected }) {
      return validateGlobalState({
        actual,
        expected,
        validate: evalCase.validate as GlobalValidationSpec | undefined,
      });
    },
    buildArtifacts(actual): BenchmarkArtifactFile[] {
      return [
        {
          path: "global-drafts.json",
          content: JSON.stringify(actual, null, 2) + "\n",
        },
      ];
    },
  };
}

async function loadGlobalInitialFixture(
  path: string,
): Promise<GlobalInitialFixture> {
  if ((await stat(path)).isDirectory()) {
    const { initialFrontend, initialBackend, initialDatatables } =
      await loadAppFixtureForEval(path);
    const name = basename(path);
    return {
      workspace: {
        apps: [
          {
            path: `f/evals/global/${name}`,
            summary: name,
            value: {
              files: initialFrontend,
              runnables: initialBackend,
              data: initialDatatables,
            },
          },
        ],
      },
      liveEditorDrafts: [],
    };
  }

  const parsed = JSON.parse(
    await readFile(path, "utf8"),
  ) as GlobalInitialFixture;
  return {
    workspace: parsed.workspace ?? {},
    liveEditorDrafts: parsed.liveEditorDrafts ?? [],
    user: parsed.user,
    artifacts: parsed.artifacts,
    previewTabs: parsed.previewTabs ?? [],
  };
}

async function loadGlobalExpectedFixture(
  path: string,
): Promise<GlobalDraftState> {
  return JSON.parse(await readFile(path, "utf8")) as GlobalDraftState;
}