@superblocksteam/vite-plugin-file-sync 1.99.99-next.2 → 2.0.0-SNAPSHOT.2143e2e95d0107ab

package/dist/ai-service/template-renderer.d.ts

@@ -0,0 +1,45 @@
+import { templates } from "@superblocksteam/ai-service-templates/templates";
+import { type NpmRegistryClient, type NpmRegistryFetchResult } from "./app-interface/npm-registry.js";
+type TemplateName = keyof typeof templates;
+export declare class TemplateRenderer {
+    workDir: string;
+    nodeModuleDirs: Record<TemplateName, Promise<string | null>>;
+    private readonly detectionCwd;
+    private readonly npmRegistryClient;
+    constructor({ prefetchNodeModules, workDirSeed, detectionCwd, npmRegistryClient, }: {
+        prefetchNodeModules?: boolean;
+        workDirSeed?: string;
+        /**
+         * cwd passed to package-manager-detector for layer-2 lockfile/packageManager
+         * walks. Should be the workspace root (e.g. AiService.appRootDirPath) so
+         * detection finds the surrounding workspace's pnpm-workspace.yaml /
+         * pnpm-lock.yaml. Defaults to `process.cwd()`, which is fragile under
+         * monorepo sub-invocations and CI matrix shards (a stray ancestor
+         * lockfile can pick the wrong agent). See PR #19282 (F3).
+         */
+        detectionCwd?: string;
+        /**
+         * Optional npm registry client. The AiService always constructs one in
+         * production (SaaS, local dev, ephemeral pods); the LD flag inside the
+         * client decides whether to short-circuit before a server call.
+         * `undefined` here is the test-only opt-out — the install runs against
+         * npm's default resolution.
+         */
+        npmRegistryClient?: NpmRegistryClient;
+    });
+    getInstallationTasks(): Promise<void>;
+    renderTemplate(templateName: TemplateName): Promise<string>;
+    private templateModulesDir;
+}
+export declare function installTemplateModules(dir: string, options?: {
+    detectionCwd?: string;
+    /**
+     * Result of the immediately-preceding `maybeWriteNpmrcForDir` for the
+     * template dir. Used solely to attach `registry_host` to the
+     * `superblocks.npm.install.*` metric emit (APPS-4189 / P6.2). Passed
+     * through unchanged — no side-effects on the install itself.
+     */
+    registryFetchResult?: NpmRegistryFetchResult;
+}): Promise<string>;
+export {};
+//# sourceMappingURL=template-renderer.d.ts.map

package/dist/ai-service/template-renderer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"template-renderer.d.ts","sourceRoot":"","sources":["../../src/ai-service/template-renderer.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,SAAS,EAAE,MAAM,iDAAiD,CAAC;AAK5E,OAAO,EAEL,KAAK,iBAAiB,EACtB,KAAK,sBAAsB,EAC5B,MAAM,iCAAiC,CAAC;AAGzC,KAAK,YAAY,GAAG,MAAM,OAAO,SAAS,CAAC;AAa3C,qBAAa,gBAAgB;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,MAAM,CAAC,YAAY,EAAE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAqB;IAClD,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAgC;gBAEtD,EACV,mBAA2B,EAC3B,WAAW,EACX,YAAY,EACZ,iBAAiB,GAClB,EAAE;QACD,mBAAmB,CAAC,EAAE,OAAO,CAAC;QAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB;;;;;;;WAOG;QACH,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB;;;;;;WAMG;QACH,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;KACvC;IA8BK,oBAAoB;IAKpB,cAAc,CAAC,YAAY,EAAE,YAAY;IAkC/C,OAAO,CAAC,kBAAkB;CA6D3B;AAwDD,wBAAsB,sBAAsB,CAC1C,GAAG,EAAE,MAAM,EACX,OAAO,CAAC,EAAE;IACR,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;;;;OAKG;IACH,mBAAmB,CAAC,EAAE,sBAAsB,CAAC;CAC9C,GACA,OAAO,CAAC,MAAM,CAAC,CAuBjB"}

package/dist/ai-service/template-renderer.js

@@ -0,0 +1,316 @@
+var __runInitializers = (this && this.__runInitializers) || function (thisArg, initializers, value) {
+    var useValue = arguments.length > 2;
+    for (var i = 0; i < initializers.length; i++) {
+        value = useValue ? initializers[i].call(thisArg, value) : initializers[i].call(thisArg);
+    }
+    return useValue ? value : void 0;
+};
+var __esDecorate = (this && this.__esDecorate) || function (ctor, descriptorIn, decorators, contextIn, initializers, extraInitializers) {
+    function accept(f) { if (f !== void 0 && typeof f !== "function") throw new TypeError("Function expected"); return f; }
+    var kind = contextIn.kind, key = kind === "getter" ? "get" : kind === "setter" ? "set" : "value";
+    var target = !descriptorIn && ctor ? contextIn["static"] ? ctor : ctor.prototype : null;
+    var descriptor = descriptorIn || (target ? Object.getOwnPropertyDescriptor(target, contextIn.name) : {});
+    var _, done = false;
+    for (var i = decorators.length - 1; i >= 0; i--) {
+        var context = {};
+        for (var p in contextIn) context[p] = p === "access" ? {} : contextIn[p];
+        for (var p in contextIn.access) context.access[p] = contextIn.access[p];
+        context.addInitializer = function (f) { if (done) throw new TypeError("Cannot add initializers after decoration has completed"); extraInitializers.push(accept(f || null)); };
+        var result = (0, decorators[i])(kind === "accessor" ? { get: descriptor.get, set: descriptor.set } : descriptor[key], context);
+        if (kind === "accessor") {
+            if (result === void 0) continue;
+            if (result === null || typeof result !== "object") throw new TypeError("Object expected");
+            if (_ = accept(result.get)) descriptor.get = _;
+            if (_ = accept(result.set)) descriptor.set = _;
+            if (_ = accept(result.init)) initializers.unshift(_);
+        }
+        else if (_ = accept(result)) {
+            if (kind === "field") initializers.unshift(_);
+            else descriptor[key] = _;
+        }
+    }
+    if (target) Object.defineProperty(target, contextIn.name, descriptor);
+    done = true;
+};
+import { spawn } from "child_process";
+import { randomUUID } from "crypto";
+import * as fs from "node:fs/promises";
+import * as os from "node:os";
+import * as path from "node:path";
+import { resolveCommand } from "package-manager-detector";
+import { detect, getUserAgent } from "package-manager-detector/detect";
+import { reifyTemplate, } from "@superblocksteam/ai-service-templates/reify-template";
+import { templates } from "@superblocksteam/ai-service-templates/templates";
+import { recordNpmInstall } from "@superblocksteam/telemetry";
+import { generatePredictableId } from "../ids.js";
+import { getErrorMeta, getLogger } from "../util/logger.js";
+import { maybeWriteNpmrcForDir, } from "./app-interface/npm-registry.js";
+import llmobs from "./llmobs/index.js";
+const SHARED_DIR = "__shared__";
+const TEMPLATE_INSTALL_FILES = [
+    "package.json",
+    "pnpm-workspace.yaml",
+    "pnpm-lock.yaml",
+    "package-lock.json",
+    "yarn.lock",
+    "bun.lock",
+    "bun.lockb",
+];
+let TemplateRenderer = (() => {
+    let _instanceExtraInitializers = [];
+    let _renderTemplate_decorators;
+    return class TemplateRenderer {
+        static {
+            const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(null) : void 0;
+            _renderTemplate_decorators = [llmobs.decorate({ kind: "task", name: "TemplateRenderer.renderTemplate" })];
+            __esDecorate(this, null, _renderTemplate_decorators, { kind: "method", name: "renderTemplate", static: false, private: false, access: { has: obj => "renderTemplate" in obj, get: obj => obj.renderTemplate }, metadata: _metadata }, null, _instanceExtraInitializers);
+            if (_metadata) Object.defineProperty(this, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
+        }
+        workDir = __runInitializers(this, _instanceExtraInitializers);
+        nodeModuleDirs;
+        detectionCwd;
+        npmRegistryClient;
+        constructor({ prefetchNodeModules = false, workDirSeed, detectionCwd, npmRegistryClient, }) {
+            // Use a stable work dir derived from a caller-provided seed (typically app root).
+            // This keeps template cache location deterministic for a given app context.
+            const stableSeed = workDirSeed ?? process.cwd();
+            const stableId = generatePredictableId(stableSeed);
+            this.workDir = path.join(os.tmpdir(), `superblocks-ai-service-${stableId}`);
+            this.detectionCwd = detectionCwd;
+            this.npmRegistryClient = npmRegistryClient;
+            this.nodeModuleDirs = {};
+            if (prefetchNodeModules) {
+                for (const [templateName, template] of Object.entries(templates)) {
+                    getLogger().debug("[ai-service] Pre-fetching node_modules for", templateName);
+                    // prefetch starts here, we don't wait for it to finish
+                    this.templateModulesDir(templateName, template).catch((error) => {
+                        getLogger().error(`Error pre-fetching node_modules for ${templateName}`, getErrorMeta(error));
+                    });
+                }
+            }
+        }
+        async getInstallationTasks() {
+            await Promise.all(Object.values(this.nodeModuleDirs));
+        }
+        async renderTemplate(templateName) {
+            if (!(templateName in templates)) {
+                throw new Error(`Template ${templateName} not found`);
+            }
+            const template = templates[templateName];
+            const templateNodeModules = await this.templateModulesDir(templateName, template);
+            const templateDir = path.join(this.workDir, templateName, `gen-${randomUUID()}`);
+            getLogger().debug(`[ai-service] Rendering template ${templateName} to ${templateDir}`);
+            reifyTemplate(templateDir, templates[templateName]);
+            if (templateNodeModules) {
+                // copy shared node_modules into the rendered template
+                await fs.cp(templateNodeModules, path.join(templateDir, "node_modules"), {
+                    recursive: true,
+                });
+            }
+            return templateDir;
+        }
+        // resolve shared node_modules for a template
+        templateModulesDir(templateName, template) {
+            const cached = this.nodeModuleDirs[templateName];
+            if (cached) {
+                return cached;
+            }
+            if (!("package.json" in template)) {
+                // no node_modules for this template
+                const nullPromise = Promise.resolve(null);
+                this.nodeModuleDirs[templateName] = nullPromise;
+                return nullPromise;
+            }
+            // Hardcoding `npm install` here breaks any environment where pnpm is the
+            // canonical package manager — npm rejects pnpm-only protocols (workspace:,
+            // catalog:) with EUNSUPPORTEDPROTOCOL even when they originate from
+            // ambient pnpm config rather than the template's own package.json. Detect
+            // the surrounding package manager and route through it instead, matching
+            // the rest of the CLI's install paths (see automatic-upgrades.ts,
+            // cli-replacement/dev.mts:installPackages).
+            //
+            // Populate the cache slot synchronously with the IIFE promise before any
+            // `await` fires (PR #19282 W1 / cr1-cache-race). The previous version
+            // awaited fs.mkdir/writeFile before assigning to `nodeModuleDirs`, which
+            // let two concurrent callers for the same template both pass the cache
+            // check and both spawn installs — under pnpm, duplicate installs in the
+            // same directory risk ENOTEMPTY / half-written symlink trees.
+            //
+            // A rejected install promise stays cached: this is intentional
+            // fail-fast behavior. Recovery is owned by the surrounding AiService
+            // lifecycle, which constructs a fresh TemplateRenderer on restart. We
+            // deliberately do NOT clear the slot on failure here — re-running the
+            // same broken `pnpm install --ignore-scripts` against the same dir
+            // immediately would just hide the underlying problem (registry down,
+            // disk full, pm not on PATH) behind retry noise.
+            const dir = path.join(this.workDir, templateName, SHARED_DIR);
+            const installDfd = (async () => {
+                await fs.mkdir(dir, { recursive: true });
+                await writeTemplateInstallFiles(dir, template);
+                // Write the configured private-registry .npmrc into the shared template
+                // dir before running the install. No-op when the LD flag is off inside
+                // the client (which is the default for SaaS / local dev today) or
+                // when the org has no registry rows. The template dir has no lockfile
+                // yet on first install, so `prepareForPrivateRegistry`'s strip step
+                // would also be a no-op — we use the .npmrc-only helper here to make
+                // that intent explicit.
+                const registryFetchResult = await maybeWriteNpmrcForDir(dir, this.npmRegistryClient);
+                return installTemplateModules(dir, {
+                    detectionCwd: this.detectionCwd,
+                    registryFetchResult,
+                });
+            })();
+            this.nodeModuleDirs[templateName] = installDfd;
+            return installDfd;
+        }
+    };
+})();
+export { TemplateRenderer };
+async function writeTemplateInstallFiles(dir, template) {
+    await Promise.all(TEMPLATE_INSTALL_FILES.flatMap((fileName) => {
+        const content = template[fileName];
+        return content === undefined
+            ? []
+            : [fs.writeFile(path.join(dir, fileName), content, "utf-8")];
+    }));
+}
+// Env-var escape hatch for environments where automatic detection picks the
+// wrong agent (e.g. CI matrix shards with stray ancestor lockfiles, or users
+// who explicitly want to force `npm`). Highest precedence — overrides both
+// `getUserAgent()` and `detect()`. See PR #19282 (F1).
+const PM_OVERRIDE_ENV = "SUPERBLOCKS_TEMPLATE_PM";
+// Exported for unit testing — production callers should go through
+// TemplateRenderer.templateModulesDir.
+//
+// Detection happens in four layers, in order:
+//   0. `process.env.SUPERBLOCKS_TEMPLATE_PM` escape hatch.
+//   1. `getUserAgent()` reads `npm_config_user_agent`, which the surrounding
+//      package manager sets on every script invocation. cwd-independent —
+//      works even when the install spawns in a temp dir under os.tmpdir().
+//      This is the primary signal in any environment that ran csb-mock-server
+//      via `pnpm <script>` / `npm run <script>` / etc.
+//   2. `detect({ cwd })` walks up from `detectionCwd` looking for lockfiles /
+//      packageManager fields. Callers should pass the workspace root (e.g.
+//      AiService.appRootDirPath) — defaulting to `process.cwd()` is fragile
+//      in CI matrix shards and monorepo sub-invocations.
+//   3. Default to `npm`. npm ships with every Node install, so the bare
+//      ENOENT failure mode is impossible. In any pnpm-managed environment
+//      layer 1 or 2 will already have caught it; this fallback only fires
+//      when neither pnpm nor any other surrounding package manager is in
+//      sight, in which case there are no `catalog:` / `workspace:` env vars
+//      to leak into the spawned npm process either. See PR #19282 (F1).
+/**
+ * Pick the registry URL the template install will target from the most
+ * recent `maybeWriteNpmrcForDir` result. Mirrors the helper in
+ * `app-interface/shell.ts` so the metric stream reports the prefetch
+ * registry with the same semantics as the AppShell install paths.
+ */
+function templateRegistryUrlFromFetchResult(result) {
+    if (!result)
+        return undefined;
+    if (!result.config.configured)
+        return undefined;
+    return result.config.default?.url;
+}
+export async function installTemplateModules(dir, options) {
+    const installStart = Date.now();
+    let outcome = "success";
+    try {
+        return await installTemplateModulesImpl(dir, options);
+    }
+    catch (err) {
+        // Template prefetch never throws `NpmInstallBlocked` — it spawns the
+        // package manager directly with a generic exit-code error. Any failure
+        // here is therefore `other` on the metric outcome enum; granularity
+        // beyond that lives on the existing rejection message.
+        outcome = "other";
+        throw err;
+    }
+    finally {
+        recordNpmInstall({
+            runner: "prefetch",
+            registryHost: templateRegistryUrlFromFetchResult(options?.registryFetchResult),
+            outcome,
+            configured: options?.registryFetchResult?.config.configured ?? false,
+            durationMs: Date.now() - installStart,
+        });
+    }
+}
+async function installTemplateModulesImpl(dir, options) {
+    const detectionCwd = options?.detectionCwd ?? process.cwd();
+    const override = process.env[PM_OVERRIDE_ENV]?.trim();
+    // Cast: the env var is unvalidated user input. resolveCommand() will
+    // return null and throw an actionable error below if it isn't a known
+    // agent, so we don't need a stricter parse here.
+    const fromOverride = override ? override : null;
+    const isCloudMode = !fromOverride && process.env.SUPERBLOCKS_SERVER_ENV !== "local";
+    const fromCloud = isCloudMode ? "npm" : null;
+    const fromEnv = fromOverride || fromCloud ? null : getUserAgent();
+    const fromDetect = fromOverride || fromCloud || fromEnv
+        ? null
+        : ((await detect({
+            strategies: [
+                "packageManager-field",
+                "lockfile",
+                "install-metadata",
+                "devEngines-field",
+            ],
+            cwd: detectionCwd,
+        }))?.agent ?? null);
+    const fallback = "npm";
+    const agent = fromOverride ?? fromCloud ?? fromEnv ?? fromDetect ?? fallback;
+    const source = fromOverride
+        ? "override"
+        : fromCloud
+            ? "cloud"
+            : fromEnv
+                ? "env"
+                : fromDetect
+                    ? "detect"
+                    : "fallback";
+    getLogger().info(`[ai-service] template-install: resolved package manager '${agent}' via ${source}` +
+        (source === "fallback"
+            ? ` (no signal from $${PM_OVERRIDE_ENV}, $npm_config_user_agent, or detect(cwd=${detectionCwd}); set $${PM_OVERRIDE_ENV} to override)`
+            : ""));
+    const installCommand = resolveCommand(agent, "install", ["--ignore-scripts"]);
+    if (!installCommand) {
+        throw new Error(`Could not resolve install command for package manager "${agent}"`);
+    }
+    const { command, args } = installCommand;
+    return new Promise((resolve, reject) => {
+        // No `shell: true`: Node 22 deprecated passing an args array together with
+        // `shell: true` because the args are concatenated, not escaped. Spawning
+        // the package-manager binary directly with an args array sidesteps both
+        // the warning and the underlying shell-injection foot-gun. The CLI's
+        // other subprocess call sites (e.g. sync.mts `execFileAsync("git", [...])`)
+        // follow the same shell-less pattern.
+        const child = spawn(command, args, { cwd: dir });
+        let stdout = "";
+        let stderr = "";
+        child.stdout?.on("data", (chunk) => {
+            stdout += String(chunk);
+        });
+        child.stderr?.on("data", (chunk) => {
+            stderr += String(chunk);
+        });
+        child.on("error", (error) => {
+            reject(new Error(`${command} ${args.join(" ")} failed to start: ${error instanceof Error ? error.message : String(error)}`));
+        });
+        child.on("close", (code) => {
+            if (code !== 0) {
+                const stdoutTail = stdout.trim().split("\n").slice(-30).join("\n");
+                const stderrTail = stderr.trim().split("\n").slice(-30).join("\n");
+                reject(new Error([
+                    `${command} ${args.join(" ")} failed with code ${code}`,
+                    stdoutTail ? `stdout (tail):\n${stdoutTail}` : undefined,
+                    stderrTail ? `stderr (tail):\n${stderrTail}` : undefined,
+                ]
+                    .filter(Boolean)
+                    .join("\n\n")));
+                return;
+            }
+            resolve(path.join(dir, "node_modules"));
+        });
+    });
+}
+//# sourceMappingURL=template-renderer.js.map

package/dist/ai-service/template-renderer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"template-renderer.js","sourceRoot":"","sources":["../../src/ai-service/template-renderer.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAc,cAAc,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAEvE,OAAO,EACL,aAAa,GAEd,MAAM,sDAAsD,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,iDAAiD,CAAC;AAC5E,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAE9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EACL,qBAAqB,GAGtB,MAAM,iCAAiC,CAAC;AACzC,OAAO,MAAM,MAAM,mBAAmB,CAAC;AAIvC,MAAM,UAAU,GAAG,YAAY,CAAC;AAChC,MAAM,sBAAsB,GAAG;IAC7B,cAAc;IACd,qBAAqB;IACrB,gBAAgB;IAChB,mBAAmB;IACnB,WAAW;IACX,UAAU;IACV,WAAW;CACH,CAAC;IAEE,gBAAgB;;;iBAAhB,gBAAgB;;;0CAiE1B,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iCAAiC,EAAE,CAAC;YAC3E,6LAAM,cAAc,6DA+BnB;;;QAhGD,OAAO,GADI,mDAAgB,CACX;QAChB,cAAc,CAA+C;QAC5C,YAAY,CAAqB;QACjC,iBAAiB,CAAgC;QAElE,YAAY,EACV,mBAAmB,GAAG,KAAK,EAC3B,WAAW,EACX,YAAY,EACZ,iBAAiB,GAqBlB;YACC,kFAAkF;YAClF,4EAA4E;YAC5E,MAAM,UAAU,GAAG,WAAW,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAChD,MAAM,QAAQ,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;YACnD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,0BAA0B,QAAQ,EAAE,CAAC,CAAC;YAC5E,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;YACjC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;YAE3C,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;YAEzB,IAAI,mBAAmB,EAAE,CAAC;gBACxB,KAAK,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;oBACjE,SAAS,EAAE,CAAC,KAAK,CACf,4CAA4C,EAC5C,YAAY,CACb,CAAC;oBACF,uDAAuD;oBACvD,IAAI,CAAC,kBAAkB,CAAC,YAA4B,EAAE,QAAQ,CAAC,CAAC,KAAK,CACnE,CAAC,KAAK,EAAE,EAAE;wBACR,SAAS,EAAE,CAAC,KAAK,CACf,uCAAuC,YAAY,EAAE,EACrD,YAAY,CAAC,KAAK,CAAC,CACpB,CAAC;oBACJ,CAAC,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,KAAK,CAAC,oBAAoB;YACxB,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;QACxD,CAAC;QAGD,KAAK,CAAC,cAAc,CAAC,YAA0B;YAC7C,IAAI,CAAC,CAAC,YAAY,IAAI,SAAS,CAAC,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,YAAY,CAAC,CAAC;YACxD,CAAC;YAED,MAAM,QAAQ,GAAG,SAAS,CAAC,YAAY,CAAC,CAAC;YAEzC,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,kBAAkB,CACvD,YAAY,EACZ,QAAQ,CACT,CAAC;YAEF,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAC3B,IAAI,CAAC,OAAO,EACZ,YAAY,EACZ,OAAO,UAAU,EAAE,EAAE,CACtB,CAAC;YAEF,SAAS,EAAE,CAAC,KAAK,CACf,mCAAmC,YAAY,OAAO,WAAW,EAAE,CACpE,CAAC;YAEF,aAAa,CAAC,WAAW,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;YACpD,IAAI,mBAAmB,EAAE,CAAC;gBACxB,sDAAsD;gBACtD,MAAM,EAAE,CAAC,EAAE,CAAC,mBAAmB,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,EAAE;oBACvE,SAAS,EAAE,IAAI;iBAChB,CAAC,CAAC;YACL,CAAC;YAED,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,6CAA6C;QACrC,kBAAkB,CACxB,YAA0B,EAC1B,QAAiB;YAEjB,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YACjD,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,IAAI,CAAC,CAAC,cAAc,IAAI,QAAQ,CAAC,EAAE,CAAC;gBAClC,oCAAoC;gBACpC,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBAC1C,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,GAAG,WAAW,CAAC;gBAChD,OAAO,WAAW,CAAC;YACrB,CAAC;YAED,yEAAyE;YACzE,2EAA2E;YAC3E,oEAAoE;YACpE,0EAA0E;YAC1E,yEAAyE;YACzE,kEAAkE;YAClE,4CAA4C;YAC5C,EAAE;YACF,yEAAyE;YACzE,sEAAsE;YACtE,yEAAyE;YACzE,uEAAuE;YACvE,wEAAwE;YACxE,8DAA8D;YAC9D,EAAE;YACF,+DAA+D;YAC/D,qEAAqE;YACrE,sEAAsE;YACtE,sEAAsE;YACtE,mEAAmE;YACnE,qEAAqE;YACrE,iDAAiD;YACjD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;YAC9D,MAAM,UAAU,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC7B,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzC,MAAM,yBAAyB,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;gBAC/C,wEAAwE;gBACxE,uEAAuE;gBACvE,kEAAkE;gBAClE,sEAAsE;gBACtE,oEAAoE;gBACpE,qEAAqE;gBACrE,wBAAwB;gBACxB,MAAM,mBAAmB,GAAG,MAAM,qBAAqB,CACrD,GAAG,EACH,IAAI,CAAC,iBAAiB,CACvB,CAAC;gBACF,OAAO,sBAAsB,CAAC,GAAG,EAAE;oBACjC,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,mBAAmB;iBACpB,CAAC,CAAC;YACL,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,GAAG,UAAU,CAAC;YAC/C,OAAO,UAAU,CAAC;QACpB,CAAC;;;SAhKU,gBAAgB;AAmK7B,KAAK,UAAU,yBAAyB,CACtC,GAAW,EACX,QAAiB;IAEjB,MAAM,OAAO,CAAC,GAAG,CACf,sBAAsB,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;QAC1C,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnC,OAAO,OAAO,KAAK,SAAS;YAC1B,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CACH,CAAC;AACJ,CAAC;AAED,4EAA4E;AAC5E,6EAA6E;AAC7E,2EAA2E;AAC3E,uDAAuD;AACvD,MAAM,eAAe,GAAG,yBAAyB,CAAC;AAElD,mEAAmE;AACnE,uCAAuC;AACvC,EAAE;AACF,8CAA8C;AAC9C,2DAA2D;AAC3D,6EAA6E;AAC7E,0EAA0E;AAC1E,2EAA2E;AAC3E,8EAA8E;AAC9E,uDAAuD;AACvD,8EAA8E;AAC9E,2EAA2E;AAC3E,4EAA4E;AAC5E,yDAAyD;AACzD,wEAAwE;AACxE,0EAA0E;AAC1E,0EAA0E;AAC1E,yEAAyE;AACzE,4EAA4E;AAC5E,wEAAwE;AACxE;;;;;GAKG;AACH,SAAS,kCAAkC,CACzC,MAA0C;IAE1C,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU;QAAE,OAAO,SAAS,CAAC;IAChD,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,GAAG,CAAC;AACpC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,GAAW,EACX,OASC;IAED,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAChC,IAAI,OAAO,GAAG,SAAS,CAAC;IACxB,IAAI,CAAC;QACH,OAAO,MAAM,0BAA0B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,qEAAqE;QACrE,uEAAuE;QACvE,oEAAoE;QACpE,uDAAuD;QACvD,OAAO,GAAG,OAAO,CAAC;QAClB,MAAM,GAAG,CAAC;IACZ,CAAC;YAAS,CAAC;QACT,gBAAgB,CAAC;YACf,MAAM,EAAE,UAAU;YAClB,YAAY,EAAE,kCAAkC,CAC9C,OAAO,EAAE,mBAAmB,CAC7B;YACD,OAAO;YACP,UAAU,EAAE,OAAO,EAAE,mBAAmB,EAAE,MAAM,CAAC,UAAU,IAAI,KAAK;YACpE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY;SACtC,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,KAAK,UAAU,0BAA0B,CACvC,GAAW,EACX,OAAmC;IAEnC,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAC5D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,CAAC;IACtD,qEAAqE;IACrE,sEAAsE;IACtE,iDAAiD;IACjD,MAAM,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAE,QAAkB,CAAC,CAAC,CAAC,IAAI,CAAC;IAE3D,MAAM,WAAW,GACf,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,sBAAsB,KAAK,OAAO,CAAC;IAClE,MAAM,SAAS,GAAiB,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;IAE3D,MAAM,OAAO,GAAG,YAAY,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC;IAClE,MAAM,UAAU,GACd,YAAY,IAAI,SAAS,IAAI,OAAO;QAClC,CAAC,CAAC,IAAI;QACN,CAAC,CAAC,CAAC,CACC,MAAM,MAAM,CAAC;YACX,UAAU,EAAE;gBACV,sBAAsB;gBACtB,UAAU;gBACV,kBAAkB;gBAClB,kBAAkB;aACnB;YACD,GAAG,EAAE,YAAY;SAClB,CAAC,CACH,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC;IACxB,MAAM,QAAQ,GAAU,KAAK,CAAC;IAC9B,MAAM,KAAK,GACT,YAAY,IAAI,SAAS,IAAI,OAAO,IAAI,UAAU,IAAI,QAAQ,CAAC;IACjE,MAAM,MAAM,GACV,YAAY;QACV,CAAC,CAAC,UAAU;QACZ,CAAC,CAAC,SAAS;YACT,CAAC,CAAC,OAAO;YACT,CAAC,CAAC,OAAO;gBACP,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,UAAU;oBACV,CAAC,CAAC,QAAQ;oBACV,CAAC,CAAC,UAAU,CAAC;IACvB,SAAS,EAAE,CAAC,IAAI,CACd,4DAA4D,KAAK,SAAS,MAAM,EAAE;QAChF,CAAC,MAAM,KAAK,UAAU;YACpB,CAAC,CAAC,qBAAqB,eAAe,2CAA2C,YAAY,WAAW,eAAe,eAAe;YACtI,CAAC,CAAC,EAAE,CAAC,CACV,CAAC;IACF,MAAM,cAAc,GAAG,cAAc,CAAC,KAAK,EAAE,SAAS,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;IAC9E,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CACb,0DAA0D,KAAK,GAAG,CACnE,CAAC;IACJ,CAAC;IACD,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,cAAc,CAAC;IAEzC,OAAO,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC7C,2EAA2E;QAC3E,yEAAyE;QACzE,wEAAwE;QACxE,qEAAqE;QACrE,4EAA4E;QAC5E,sCAAsC;QACtC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;QACjD,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YACjC,MAAM,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YACjC,MAAM,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC1B,MAAM,CACJ,IAAI,KAAK,CACP,GAAG,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,qBAAqB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAC1G,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YACzB,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACf,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnE,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnE,MAAM,CACJ,IAAI,KAAK,CACP;oBACE,GAAG,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,qBAAqB,IAAI,EAAE;oBACvD,UAAU,CAAC,CAAC,CAAC,mBAAmB,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS;oBACxD,UAAU,CAAC,CAAC,CAAC,mBAAmB,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS;iBACzD;qBACE,MAAM,CAAC,OAAO,CAAC;qBACf,IAAI,CAAC,MAAM,CAAC,CAChB,CACF,CAAC;gBACF,OAAO;YACT,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/ai-service/test-utils/dd-trace-helpers.d.ts

@@ -0,0 +1,25 @@
+import type { Mock } from "vitest";
+type RegisteredSpan = {
+    span: {
+        id: number;
+        options: any;
+        parent?: RegisteredSpan["span"];
+        finish: Mock;
+        context: Mock;
+    };
+    options: any;
+    finishTime?: number;
+};
+export declare const mockLLMObs: {
+    trace: Mock;
+    wrap: Mock;
+    annotate: Mock;
+    exportSpan: Mock;
+    submitEvaluation: Mock;
+    _registerLLMObsSpan: Mock;
+    enabled: Mock;
+};
+export declare const __getRegisteredSpans: () => RegisteredSpan[];
+export declare const __resetDdTraceMock: () => void;
+export {};
+//# sourceMappingURL=dd-trace-helpers.d.ts.map

package/dist/ai-service/test-utils/dd-trace-helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dd-trace-helpers.d.ts","sourceRoot":"","sources":["../../../src/ai-service/test-utils/dd-trace-helpers.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAEnC,KAAK,cAAc,GAAG;IACpB,IAAI,EAAE;QACJ,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,EAAE,GAAG,CAAC;QACb,MAAM,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,EAAE,IAAI,CAAC;QACb,OAAO,EAAE,IAAI,CAAC;KACf,CAAC;IACF,OAAO,EAAE,GAAG,CAAC;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AAqBF,eAAO,MAAM,UAAU;WAjBZ,IAAI;UACL,IAAI;cACA,IAAI;gBACF,IAAI;sBACE,IAAI;yBACD,IAAI;aAChB,IAAI;CAW+B,CAAC;AACjD,eAAO,MAAM,oBAAoB,wBAA2C,CAAC;AAE7E,eAAO,MAAM,kBAAkB,YAAyC,CAAC"}

package/dist/ai-service/test-utils/dd-trace-helpers.js

@@ -0,0 +1,9 @@
+import ddTrace from "dd-trace";
+const ddTraceMock = ddTrace;
+export const mockLLMObs = ddTraceMock.mockLLMObs;
+export const __getRegisteredSpans = () => ddTraceMock.__getRegisteredSpans();
+const __resetRegisteredSpans = () => ddTraceMock.__resetRegisteredSpans();
+export const __resetDdTraceMock = () => ddTraceMock.__resetDdTraceMock();
+const __clearSpanListeners = () => ddTraceMock.__clearSpanListeners();
+const __addSpanListener = (listener) => ddTraceMock.__addSpanListener(listener);
+//# sourceMappingURL=dd-trace-helpers.js.map

package/dist/ai-service/test-utils/dd-trace-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dd-trace-helpers.js","sourceRoot":"","sources":["../../../src/ai-service/test-utils/dd-trace-helpers.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,UAAU,CAAC;AAgC/B,MAAM,WAAW,GAAG,OAAuC,CAAC;AAE5D,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC;AACjD,MAAM,CAAC,MAAM,oBAAoB,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,oBAAoB,EAAE,CAAC;AAC7E,MAAM,sBAAsB,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,sBAAsB,EAAE,CAAC;AAC1E,MAAM,CAAC,MAAM,kBAAkB,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,kBAAkB,EAAE,CAAC;AACzE,MAAM,oBAAoB,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,oBAAoB,EAAE,CAAC;AACtE,MAAM,iBAAiB,GAAG,CAAC,QAAwC,EAAE,EAAE,CACrE,WAAW,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC"}

package/dist/ai-service/test-utils/error-mocks.d.ts

@@ -0,0 +1,69 @@
+/**
+ * Error injection utilities for testing retry middleware behavior e2e.
+ *
+ * Provides pre-defined error scenarios that can be injected into mock streams
+ * to test how the retry middleware handles different provider errors.
+ */
+/**
+ * Error scenario configuration.
+ *
+ * Defines how errors should be injected into the stream and whether
+ * the stream should eventually succeed.
+ */
+export interface ErrorScenario {
+    /** Human-readable name for the scenario */
+    name: string;
+    /** Description of what this scenario tests */
+    description: string;
+    /** Errors to inject (one per retry attempt) */
+    errors: Array<{
+        type: string;
+        message: string;
+        /** Optional delay before this error (simulates network latency) */
+        delayMs?: number;
+        /** If true, throw the error instead of sending an error chunk (tests catch block) */
+        shouldThrow?: boolean;
+    }>;
+    /** Whether the stream should succeed after all errors */
+    eventuallySucceeds: boolean;
+    /** Mock content to return on success (if eventuallySucceeds is true) */
+    successContent?: string;
+}
+/**
+ * Pre-defined error scenarios for testing.
+ */
+export declare const ERROR_SCENARIOS: Record<string, ErrorScenario>;
+/**
+ * Generates an error chunk in the AI SDK format.
+ *
+ * @param errorType - Standardized error type (e.g., "rate_limit_error")
+ * @param message - Error message
+ * @returns Error chunk ready to be injected into a stream
+ */
+export declare function createErrorChunk(errorType: string, message: string): {
+    type: string;
+    error: {
+        statusCode?: number | undefined;
+        headers?: {
+            "retry-after": string;
+        } | undefined;
+        type: string;
+        message: string;
+    };
+};
+/**
+ * Generates mock chunks for an error scenario.
+ *
+ * Creates a sequence of chunks that includes:
+ * - Initial step-start
+ * - Text content (if partial success before error)
+ * - Error chunk
+ * - Success chunks (if eventuallySucceeds is true)
+ * - Final step-finish/finish chunks
+ *
+ * @param scenario - The error scenario to generate chunks for
+ * @param attemptNumber - Current retry attempt (0-indexed, 0 = first attempt)
+ * @returns Array of chunks ready to be streamed
+ */
+export declare function generateErrorScenarioChunks(scenario: ErrorScenario, attemptNumber: number): any[];
+//# sourceMappingURL=error-mocks.d.ts.map

package/dist/ai-service/test-utils/error-mocks.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"error-mocks.d.ts","sourceRoot":"","sources":["../../../src/ai-service/test-utils/error-mocks.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH;;;;;GAKG;AACH,MAAM,WAAW,aAAa;IAC5B,2CAA2C;IAC3C,IAAI,EAAE,MAAM,CAAC;IACb,8CAA8C;IAC9C,WAAW,EAAE,MAAM,CAAC;IACpB,+CAA+C;IAC/C,MAAM,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,mEAAmE;QACnE,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,qFAAqF;QACrF,WAAW,CAAC,EAAE,OAAO,CAAC;KACvB,CAAC,CAAC;IACH,yDAAyD;IACzD,kBAAkB,EAAE,OAAO,CAAC;IAC5B,wEAAwE;IACxE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,eAAO,MAAM,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CA6LzD,CAAC;AAEF;;;;;;GAMG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM;;;;;;;;;;EA2BlE;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,aAAa,EACvB,aAAa,EAAE,MAAM,GACpB,GAAG,EAAE,CAoHP"}