Function createDevToolsMiddleware

src/node/services/devToolsMiddleware.ts:241–727

(
  workspaceId: string,
  service: DevToolsService
)

}

export function createDevToolsMiddleware(
  workspaceId: string,
  service: DevToolsService
): LanguageModelV3Middleware {
  assert(workspaceId.trim().length > 0, "createDevToolsMiddleware requires a workspaceId");
  assert(service, "createDevToolsMiddleware requires a DevToolsService");

  const runId = randomUUID();
  let runCreated = false;
  let runCreationPromise: Promise<void> | null = null;
  let stepCounter = 0;

  async function ensureRun(runMetadataId?: string): Promise<void> {
    if (runCreated) {
      return;
    }

    if (runCreationPromise) {
      await runCreationPromise;
      return;
    }

    runCreationPromise = (async () => {
      try {
        await service.createRun(
          workspaceId,
          {
            id: runId,
            workspaceId,
            startedAt: new Date().toISOString(),
          },
          runMetadataId
        );
        runCreated = true;
      } catch (error) {
        log.warn("DevTools: failed to create run", {
          workspaceId,
          runId,
          error,
        });
      }
    })();

    try {
      await runCreationPromise;
    } finally {
      runCreationPromise = null;
    }
  }

  function extractRunMetadataId(params: LanguageModelV3CallOptions): string | undefined {
    const rawMetadataId = params.headers?.[DEVTOOLS_RUN_METADATA_ID_HEADER];
    if (typeof rawMetadataId !== "string") {
      return undefined;
    }

    const runMetadataId = rawMetadataId.trim();
    return runMetadataId.length > 0 ? runMetadataId : undefined;