@nodetool-ai/runtime 0.8.3 → 0.8.4-a13.1

package/dist/providers/antigravity-provider.js

@@ -0,0 +1,527 @@
+/**
+ * AntigravityProvider — models reached through Google's Antigravity CLI
+ * (`agy`) in headless mode, on the machine's signed-in Antigravity account.
+ *
+ * There is no API key: `agy` authenticates with credentials cached by an
+ * earlier interactive `agy` session. Each turn spawns
+ * `agy --input-format stream-json --output-format stream-json`, writes the
+ * prompt as one `user` event on stdin, closes stdin, and reads the NDJSON event
+ * stream (`init`, `step_update`, `result`). See
+ * https://antigravity.google/docs/cli/headless/.
+ *
+ * The CLI runs its own agent with its own tools, so NodeTool tools are not
+ * offered to it. It runs in a dedicated empty scratch directory, never the
+ * server's working directory, because headless runs auto-allow file reads and
+ * writes inside the active workspace. Shell commands stay soft-denied: the
+ * provider never passes `--dangerously-skip-permissions`.
+ */
+import { createLogger, importNodeBuiltin } from "@nodetool-ai/config";
+import { PROVIDER_IDS, isString } from "@nodetool-ai/protocol";
+import { z } from "zod";
+import { BaseProvider } from "./base-provider.js";
+import { hashSystemPrompt } from "./provider-session.js";
+import { isProviderMessageEvent, isProviderSessionUpdate, isProviderStop } from "./types.js";
+const log = createLogger("nodetool.runtime.providers.antigravity");
+/** The executable name the official installer puts on PATH. */
+const DEFAULT_COMMAND = "agy";
+/** Where the official installer places `agy` on macOS and Linux. */
+const INSTALLER_BIN_DIR = [".local", "bin"];
+/** How long a finished run may take to exit before it is stopped. */
+const RESULT_EXIT_GRACE_MS = 30_000;
+const INSTALL_HINT = "Install the Antigravity CLI (agy), then run `agy` once in a terminal to sign in.";
+/** A field the CLI may omit or change shape on. Unexpected values read as absent. */
+const lenient = (schema) => schema.optional().catch(undefined);
+/** The fields of a `stream-json` event this provider reads. Others are ignored. */
+const cliEventSchema = z.object({
+    event: lenient(z.string()),
+    conversation_id: lenient(z.string()),
+    step_type: lenient(z.string()),
+    text_delta: lenient(z.string()),
+    status: lenient(z.string()),
+    response: lenient(z.string()),
+    error: lenient(z.union([z.string(), z.object({ message: lenient(z.string()) })])),
+    structured_output: lenient(z.record(z.string(), z.unknown())),
+    usage: lenient(z.object({
+        input_tokens: lenient(z.number()),
+        output_tokens: lenient(z.number()),
+        thinking_tokens: lenient(z.number()),
+        cache_read_tokens: lenient(z.number())
+    }))
+});
+/** A failed `agy` run. Its message names the remedy. */
+class AntigravityCliError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "AntigravityCliError";
+    }
+}
+export class AntigravityProvider extends BaseProvider {
+    static requiredSecrets() {
+        // Auth lives in the CLI's own credential cache, not in a NodeTool secret.
+        return [];
+    }
+    runCli;
+    /**
+     * Within-process cache of the active conversation per thread. The persisted
+     * `provider_session` column is the source of truth; this only covers callers
+     * that do not thread a session token in (the CLI chat).
+     */
+    sessions = new Map();
+    modelsCache = null;
+    constructor(_secrets = {}, options = {}) {
+        super(PROVIDER_IDS.ANTIGRAVITY);
+        this.runCli = options.runCli ?? spawnAgy;
+    }
+    /** The CLI's credentials never leave this process. */
+    getContainerEnv() {
+        return {};
+    }
+    /** `agy` runs its own tools. NodeTool tools are not forwarded to it. */
+    async hasToolSupport() {
+        return false;
+    }
+    async unavailableReason() {
+        try {
+            const run = await this.runCli(["--version"], { stdin: "", cwd: null });
+            await readLines(run);
+            const { code, stderr } = await run.exit;
+            if (code === 0)
+                return null;
+            return `The Antigravity CLI failed to start (exit ${code}): ${stderr.trim() || "no output"}`;
+        }
+        catch (err) {
+            return err instanceof Error ? err.message : String(err);
+        }
+    }
+    /**
+     * The slugs `agy models` prints. Effort is part of the slug (for example
+     * `gemini-3.1-pro-high`). Empty when the CLI is missing or signed out.
+     */
+    async getAvailableLanguageModels() {
+        this.modelsCache ??= this.listModels().catch((err) => {
+            log.warn("Listing Antigravity models failed", { error: err.message });
+            this.modelsCache = null;
+            return [];
+        });
+        return this.modelsCache;
+    }
+    async listModels() {
+        const run = await this.runCli(["models"], { stdin: "", cwd: null });
+        const lines = await readLines(run);
+        const { code, stderr } = await run.exit;
+        if (code !== 0) {
+            throw new AntigravityCliError(`agy models exited ${code}: ${stderr.trim() || "no output"}`);
+        }
+        return parseModelSlugs(lines).map((id) => ({
+            id,
+            name: modelDisplayName(id),
+            provider: PROVIDER_IDS.ANTIGRAVITY
+        }));
+    }
+    async *generateMessages(args) {
+        // A single forced tool is how callers ask for structured output. `agy`
+        // enforces a JSON schema natively, so the result comes back as that call.
+        const resultTool = args.tools?.length === 1 && args.toolChoice === args.tools[0].name
+            ? args.tools[0]
+            : undefined;
+        const systemPrompt = extractSystemPrompt(args.messages);
+        const systemHash = hashSystemPrompt(systemPrompt);
+        const threadId = args.threadId ?? null;
+        const prior = args.providerSession ??
+            (threadId ? (this.sessions.get(threadId) ?? null) : null);
+        const canResume = prior != null &&
+            prior.providerId === this.provider &&
+            prior.model === args.model &&
+            (prior.systemHash == null || prior.systemHash === systemHash) &&
+            args.messages.length > prior.checkpoint;
+        const turn = {
+            model: args.model,
+            checkpoint: args.messages.length,
+            systemHash,
+            threadId,
+            cacheSession: args.providerSession == null,
+            effort: args.effort,
+            resultTool,
+            signal: args.signal
+        };
+        // Resuming is best-effort: the CLI may have pruned the conversation. A
+        // resume that fails before any content reached the consumer falls back to
+        // a fresh conversation primed with the full history.
+        const emitted = { content: false };
+        if (canResume && prior) {
+            try {
+                yield* this.runTurn({
+                    ...turn,
+                    prompt: buildResumeDelta(args.messages, prior.checkpoint),
+                    conversation: prior.token,
+                    emitted
+                });
+                return;
+            }
+            catch (err) {
+                if (emitted.content || args.signal?.aborted)
+                    throw err;
+                log.warn("Antigravity conversation resume failed; starting fresh", {
+                    error: err instanceof Error ? err.message : String(err)
+                });
+            }
+        }
+        const history = args.loadFullHistory
+            ? await args.loadFullHistory()
+            : args.messages;
+        yield* this.runTurn({
+            ...turn,
+            prompt: buildFreshPrompt(history, systemPrompt),
+            conversation: null,
+            emitted
+        });
+    }
+    /** Run one `agy` turn and translate its event stream into stream items. */
+    async *runTurn(plan) {
+        if (plan.signal?.aborted)
+            return;
+        const cliArgs = buildCliArgs(plan);
+        this.recordRequestPayload({ args: cliArgs, prompt: plan.prompt });
+        const cwd = await scratchWorkspace();
+        const run = await this.runCli(cliArgs, {
+            stdin: `${JSON.stringify({ event: "user", message: { content: plan.prompt } })}\n`,
+            cwd
+        });
+        const onAbort = () => run.kill();
+        plan.signal?.addEventListener("abort", onAbort, { once: true });
+        let sessionSent = false;
+        let result = null;
+        try {
+            for await (const line of run.lines) {
+                const event = parseEvent(line);
+                if (!event)
+                    continue;
+                const conversationId = event.conversation_id;
+                if (!sessionSent && conversationId) {
+                    sessionSent = true;
+                    if (plan.threadId) {
+                        const session = {
+                            providerId: this.provider,
+                            model: plan.model,
+                            token: conversationId,
+                            checkpoint: plan.checkpoint,
+                            systemHash: plan.systemHash
+                        };
+                        if (plan.cacheSession)
+                            this.sessions.set(plan.threadId, session);
+                        yield { type: "session", session };
+                    }
+                }
+                if (event.event === "step_update") {
+                    const delta = event.text_delta;
+                    if (event.step_type === "agent_response" && delta && !plan.resultTool) {
+                        plan.emitted.content = true;
+                        yield { type: "chunk", content: delta, done: false };
+                    }
+                    continue;
+                }
+                if (event.event === "result") {
+                    // The provider sends one prompt and closed stdin, so the first
+                    // result ends the turn and `agy` exits on its own.
+                    result = event;
+                    break;
+                }
+            }
+        }
+        finally {
+            plan.signal?.removeEventListener("abort", onAbort);
+            // Stop the process only when the turn did not finish: the consumer
+            // stopped early, the run was aborted, or a parse step threw. A finished
+            // run is left to exit so it can save the conversation for a resume.
+            if (result) {
+                // A run that never exits after its result would leak a process.
+                setTimeout(() => run.kill(), RESULT_EXIT_GRACE_MS).unref?.();
+            }
+            else {
+                run.kill();
+            }
+        }
+        if (plan.signal?.aborted)
+            return;
+        if (!result) {
+            const { code, stderr } = await run.exit;
+            throw cliFailure(`agy exited (${code ?? "signal"}) without a result`, stderr);
+        }
+        this.trackResultUsage(plan.model, result.usage);
+        if (result.status !== "SUCCESS") {
+            const error = result.error;
+            const detail = isString(error) ? error : (error?.message ?? "");
+            throw cliFailure(`agy returned ${result.status ?? "no status"}`, detail);
+        }
+        if (plan.resultTool) {
+            if (!result.structured_output) {
+                throw new AntigravityCliError(`agy returned no structured output for ${plan.resultTool.name}`);
+            }
+            yield {
+                id: `result_${result.conversation_id ?? Date.now()}`,
+                name: plan.resultTool.name,
+                args: result.structured_output
+            };
+        }
+        else if (!plan.emitted.content && result.response) {
+            // No deltas streamed (for example an older CLI): use the final text.
+            plan.emitted.content = true;
+            yield { type: "chunk", content: result.response, done: false };
+        }
+        yield { type: "chunk", content: "", done: true };
+    }
+    trackResultUsage(model, usage) {
+        if (!usage)
+            return;
+        this.trackUsage(model, {
+            inputTokens: usage.input_tokens ?? 0,
+            outputTokens: (usage.output_tokens ?? 0) + (usage.thinking_tokens ?? 0),
+            cachedTokens: usage.cache_read_tokens ?? 0
+        });
+    }
+    async generateMessage(args) {
+        let content = "";
+        const toolCalls = [];
+        for await (const item of this.generateMessages(args)) {
+            if (isProviderSessionUpdate(item) ||
+                isProviderMessageEvent(item) ||
+                isProviderStop(item)) {
+                continue;
+            }
+            if ("args" in item) {
+                toolCalls.push(item);
+            }
+            else if (!item.thinking && isString(item.content)) {
+                content += item.content;
+            }
+        }
+        return {
+            role: "assistant",
+            content: content || null,
+            toolCalls: toolCalls.length ? toolCalls : null
+        };
+    }
+}
+/** The `agy` flags for one headless turn. */
+export function buildCliArgs(plan) {
+    const args = [
+        "--input-format",
+        "stream-json",
+        "--output-format",
+        "stream-json"
+    ];
+    if (plan.model)
+        args.push("--model", plan.model);
+    if (plan.conversation)
+        args.push("--conversation", plan.conversation);
+    if (plan.effort)
+        args.push("--effort", cliEffort(plan.effort));
+    if (plan.resultTool) {
+        args.push("--json-schema", JSON.stringify(plan.resultTool.inputSchema ?? { type: "object" }));
+    }
+    return args;
+}
+/** Every stdout line of a short-lived `agy` command. */
+async function readLines(run) {
+    const lines = [];
+    for await (const line of run.lines)
+        lines.push(line);
+    return lines;
+}
+/** `agy --effort` accepts only low, medium and high. */
+function cliEffort(effort) {
+    return effort === "xhigh" || effort === "max" ? "high" : effort;
+}
+/**
+ * Model slugs from `agy models` output. The CLI prints one model per line,
+ * possibly with a bullet or a trailing description, so this keeps the first
+ * token of each line that looks like a slug. Every slug carries a version
+ * number, which separates slugs from header words.
+ */
+export function parseModelSlugs(lines) {
+    const slugs = [];
+    for (const raw of lines) {
+        const token = raw.trim().replace(/^[-*•]\s*/, "").split(/\s+/)[0] ?? "";
+        if (!/^[a-z][a-z0-9._-]*$/i.test(token) || !/\d/.test(token))
+            continue;
+        if (!slugs.includes(token))
+            slugs.push(token);
+    }
+    return slugs;
+}
+/** `gemini-3.1-pro-high` → `Gemini 3.1 Pro High`. */
+function modelDisplayName(slug) {
+    return slug
+        .split("-")
+        .map((part) => (part ? part[0].toUpperCase() + part.slice(1) : part))
+        .join(" ");
+}
+/** One stdout line as an event, or null for a line that is not one. */
+function parseEvent(line) {
+    const trimmed = line.trim();
+    if (!trimmed.startsWith("{"))
+        return null;
+    let json;
+    try {
+        json = JSON.parse(trimmed);
+    }
+    catch {
+        log.debug("Skipping non-JSON agy output line");
+        return null;
+    }
+    const parsed = cliEventSchema.safeParse(json);
+    return parsed.success ? parsed.data : null;
+}
+/**
+ * An Error for a failed run. It carries no HTTP status on purpose: the generic
+ * 401 remedy points at Settings, and `agy` is signed in from a terminal.
+ */
+function cliFailure(summary, detail) {
+    const text = detail.trim();
+    if (/auth|sign[ -]?in|log[ -]?in|credential/i.test(text)) {
+        return new AntigravityCliError(`Antigravity sign-in required: ${text}. Run \`agy\` once in a terminal to sign in.`);
+    }
+    return new AntigravityCliError(text ? `${summary}: ${text}` : summary);
+}
+/** Flatten a message's content to plain text (text blocks only). */
+function textOf(content) {
+    if (content == null)
+        return "";
+    if (isString(content))
+        return content;
+    return content
+        .filter((c) => c.type === "text")
+        .map((c) => c.text)
+        .join("");
+}
+/** `agy` accepts only text prompts. Refuse images rather than drop them. */
+function assertTextOnly(messages) {
+    const withMedia = messages.some((m) => Array.isArray(m.content) && m.content.some((c) => c.type !== "text"));
+    if (withMedia) {
+        throw new AntigravityCliError("The Antigravity CLI accepts text prompts only. Remove the attached images or media, or pick another provider.");
+    }
+}
+function extractSystemPrompt(messages) {
+    return messages
+        .filter((m) => m.role === "system")
+        .map((m) => textOf(m.content))
+        .filter(Boolean)
+        .join("\n\n");
+}
+/**
+ * The user messages added since the checkpoint. Assistant turns in the slice
+ * came from the resumed conversation itself, so they are not replayed.
+ */
+export function buildResumeDelta(messages, checkpoint) {
+    const delta = messages.slice(checkpoint).filter((m) => m.role === "user");
+    assertTextOnly(delta);
+    return delta
+        .map((m) => textOf(m.content))
+        .filter(Boolean)
+        .join("\n\n");
+}
+/**
+ * The prompt for a fresh conversation. `agy` has no system-prompt flag and
+ * cannot import earlier turns, so instructions and prior history are sent as
+ * delimited blocks ahead of the new user turn.
+ */
+export function buildFreshPrompt(messages, systemPrompt) {
+    const convo = messages.filter((m) => m.role === "user" || m.role === "assistant");
+    assertTextOnly(convo);
+    const last = convo[convo.length - 1];
+    const newTurn = last?.role === "user" ? textOf(last.content) : "";
+    const prior = last?.role === "user" ? convo.slice(0, -1) : convo;
+    const transcript = prior
+        .map((m) => {
+        const text = textOf(m.content);
+        return text ? `${m.role === "assistant" ? "Assistant" : "User"}: ${text}` : "";
+    })
+        .filter(Boolean)
+        .join("\n\n");
+    return [
+        systemPrompt ? `<instructions>\n${systemPrompt}\n</instructions>` : "",
+        transcript ? `<conversation_so_far>\n${transcript}\n</conversation_so_far>` : "",
+        newTurn
+    ]
+        .filter(Boolean)
+        .join("\n\n");
+}
+/**
+ * The empty directory `agy` runs in. Headless runs may read and write inside
+ * their workspace, so it must never be the server's own working directory.
+ */
+async function scratchWorkspace() {
+    const [fs, os, path] = await Promise.all([
+        importNodeBuiltin("node:fs/promises"),
+        importNodeBuiltin("node:os"),
+        importNodeBuiltin("node:path")
+    ]);
+    if (!fs || !os || !path)
+        return null;
+    const dir = path.join(os.tmpdir(), "nodetool-antigravity");
+    await fs.mkdir(dir, { recursive: true });
+    return dir;
+}
+/** The real runner: spawns `agy` with `~/.local/bin` appended to PATH. */
+const spawnAgy = async (args, options) => {
+    const [cp, os, path, readline] = await Promise.all([
+        importNodeBuiltin("node:child_process"),
+        importNodeBuiltin("node:os"),
+        importNodeBuiltin("node:path"),
+        importNodeBuiltin("node:readline")
+    ]);
+    if (!cp || !os || !path || !readline) {
+        throw new AntigravityCliError("The Antigravity provider requires Node.js");
+    }
+    // A desktop app launched outside a shell often lacks the installer's bin
+    // directory on PATH.
+    const installerBin = path.join(os.homedir(), ...INSTALLER_BIN_DIR);
+    const env = {
+        ...process.env,
+        PATH: [process.env["PATH"], installerBin].filter(Boolean).join(path.delimiter)
+    };
+    const child = cp.spawn(DEFAULT_COMMAND, args, {
+        cwd: options.cwd ?? undefined,
+        env,
+        stdio: ["pipe", "pipe", "pipe"],
+        windowsHide: true
+    });
+    let stderr = "";
+    child.stderr.setEncoding("utf8");
+    child.stderr.on("data", (data) => {
+        // Keep the tail. It holds the failure reason.
+        stderr = (stderr + data).slice(-8192);
+    });
+    const exit = new Promise((resolve, reject) => {
+        child.once("error", (err) => {
+            reject(err.code === "ENOENT"
+                ? new AntigravityCliError(`Antigravity CLI (agy) not found. ${INSTALL_HINT}`)
+                : err);
+        });
+        child.once("close", (code) => resolve({ code, stderr }));
+    });
+    // A caller that stops early never awaits `exit`. Mark it handled.
+    exit.catch(() => undefined);
+    // Surface a spawn failure here, before the caller waits on stdout.
+    await Promise.race([
+        exit.then(() => undefined),
+        new Promise((resolve) => child.once("spawn", () => resolve()))
+    ]);
+    child.stdin.on("error", () => {
+        // The process may exit before reading stdin; its exit code reports why.
+    });
+    child.stdin.end(options.stdin);
+    const lines = readline.createInterface({
+        input: child.stdout,
+        crlfDelay: Infinity
+    });
+    return {
+        lines,
+        exit,
+        kill: () => {
+            if (child.exitCode === null && child.signalCode === null)
+                child.kill();
+        }
+    };
+};
+//# sourceMappingURL=antigravity-provider.js.map

package/dist/providers/antigravity-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"antigravity-provider.js","sourceRoot":"","sources":["../../src/providers/antigravity-provider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EACL,YAAY,EACZ,iBAAiB,EAClB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EACL,sBAAsB,EACtB,uBAAuB,EACvB,cAAc,EASf,MAAM,YAAY,CAAC;AAEpB,MAAM,GAAG,GAAG,YAAY,CAAC,wCAAwC,CAAC,CAAC;AAEnE,+DAA+D;AAC/D,MAAM,eAAe,GAAG,KAAK,CAAC;AAE9B,oEAAoE;AACpE,MAAM,iBAAiB,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;AAE5C,qEAAqE;AACrE,MAAM,oBAAoB,GAAG,MAAM,CAAC;AAEpC,MAAM,YAAY,GAChB,kFAAkF,CAAC;AAErF,qFAAqF;AACrF,MAAM,OAAO,GAAG,CAAsB,MAAS,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;AAEvF,mFAAmF;AACnF,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC1B,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACpC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC9B,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC/B,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC3B,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC7B,KAAK,EAAE,OAAO,CACZ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAClE;IACD,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IAC7D,KAAK,EAAE,OAAO,CACZ,CAAC,CAAC,MAAM,CAAC;QACP,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QACjC,aAAa,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAClC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QACpC,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KACvC,CAAC,CACH;CACF,CAAC,CAAC;AAwBH,wDAAwD;AACxD,MAAM,mBAAoB,SAAQ,KAAK;IACrC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,qBAAqB,CAAC;IACpC,CAAC;CACF;AAED,MAAM,OAAO,mBAAoB,SAAQ,YAAY;IACnD,MAAM,CAAC,eAAe;QACpB,0EAA0E;QAC1E,OAAO,EAAE,CAAC;IACZ,CAAC;IAEgB,MAAM,CAAuB;IAC9C;;;;OAIG;IACc,QAAQ,GAAG,IAAI,GAAG,EAA2B,CAAC;IACvD,WAAW,GAAoC,IAAI,CAAC;IAE5D,YACE,QAAQ,GAA4B,EAAE,EACtC,OAAO,GAA+B,EAAE;QAExC,KAAK,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,QAAQ,CAAC;IAC3C,CAAC;IAED,sDAAsD;IAC7C,eAAe;QACtB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,wEAAwE;IAC/D,KAAK,CAAC,cAAc;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IAEQ,KAAK,CAAC,iBAAiB;QAC9B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;YACvE,MAAM,SAAS,CAAC,GAAG,CAAC,CAAC;YACrB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC;YACxC,IAAI,IAAI,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YAC5B,OAAO,6CAA6C,IAAI,MAAM,MAAM,CAAC,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC;QAC/F,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED;;;OAGG;IACM,KAAK,CAAC,0BAA0B;QACvC,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE;YAC1D,GAAG,CAAC,IAAI,CAAC,mCAAmC,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACtE,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;QACpE,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC;QACxC,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACf,MAAM,IAAI,mBAAmB,CAC3B,qBAAqB,IAAI,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,WAAW,EAAE,CAC7D,CAAC;QACJ,CAAC;QACD,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YACzC,EAAE;YACF,IAAI,EAAE,gBAAgB,CAAC,EAAE,CAAC;YAC1B,QAAQ,EAAE,YAAY,CAAC,WAAW;SACnC,CAAC,CAAC,CAAC;IACN,CAAC;IAEQ,KAAK,CAAC,CAAC,gBAAgB,CAAC,IAUhC;QACC,uEAAuE;QACvE,0EAA0E;QAC1E,MAAM,UAAU,GACd,IAAI,CAAC,KAAK,EAAE,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI;YAChE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACf,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,YAAY,GAAG,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,UAAU,GAAG,gBAAgB,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC;QACvC,MAAM,KAAK,GACT,IAAI,CAAC,eAAe;YACpB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC5D,MAAM,SAAS,GACb,KAAK,IAAI,IAAI;YACb,KAAK,CAAC,UAAU,KAAK,IAAI,CAAC,QAAQ;YAClC,KAAK,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK;YAC1B,CAAC,KAAK,CAAC,UAAU,IAAI,IAAI,IAAI,KAAK,CAAC,UAAU,KAAK,UAAU,CAAC;YAC7D,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC;QAC1C,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM;YAChC,UAAU;YACV,QAAQ;YACR,YAAY,EAAE,IAAI,CAAC,eAAe,IAAI,IAAI;YAC1C,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,UAAU;YACV,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;QAEF,uEAAuE;QACvE,0EAA0E;QAC1E,qDAAqD;QACrD,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QACnC,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;oBAClB,GAAG,IAAI;oBACP,MAAM,EAAE,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC;oBACzD,YAAY,EAAE,KAAK,CAAC,KAAK;oBACzB,OAAO;iBACR,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO;oBAAE,MAAM,GAAG,CAAC;gBACvD,GAAG,CAAC,IAAI,CAAC,wDAAwD,EAAE;oBACjE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;iBACxD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe;YAClC,CAAC,CAAC,MAAM,IAAI,CAAC,eAAe,EAAE;YAC9B,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;QAClB,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC;YAClB,GAAG,IAAI;YACP,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,YAAY,CAAC;YAC/C,YAAY,EAAE,IAAI;YAClB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED,2EAA2E;IACnE,KAAK,CAAC,CAAC,OAAO,CAAC,IAYtB;QACC,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO;YAAE,OAAO;QACjC,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAClE,MAAM,GAAG,GAAG,MAAM,gBAAgB,EAAE,CAAC;QACrC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;YACrC,KAAK,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI;YAClF,GAAG;SACJ,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QACjC,IAAI,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAEhE,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,MAAM,GAAoB,IAAI,CAAC;QACnC,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACnC,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;gBAC/B,IAAI,CAAC,KAAK;oBAAE,SAAS;gBAErB,MAAM,cAAc,GAAG,KAAK,CAAC,eAAe,CAAC;gBAC7C,IAAI,CAAC,WAAW,IAAI,cAAc,EAAE,CAAC;oBACnC,WAAW,GAAG,IAAI,CAAC;oBACnB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;wBAClB,MAAM,OAAO,GAAoB;4BAC/B,UAAU,EAAE,IAAI,CAAC,QAAQ;4BACzB,KAAK,EAAE,IAAI,CAAC,KAAK;4BACjB,KAAK,EAAE,cAAc;4BACrB,UAAU,EAAE,IAAI,CAAC,UAAU;4BAC3B,UAAU,EAAE,IAAI,CAAC,UAAU;yBAC5B,CAAC;wBACF,IAAI,IAAI,CAAC,YAAY;4BAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;wBACjE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;oBACrC,CAAC;gBACH,CAAC;gBAED,IAAI,KAAK,CAAC,KAAK,KAAK,aAAa,EAAE,CAAC;oBAClC,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC;oBAC/B,IAAI,KAAK,CAAC,SAAS,KAAK,gBAAgB,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;wBACtE,IAAI,CAAC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;wBAC5B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;oBACvD,CAAC;oBACD,SAAS;gBACX,CAAC;gBAED,IAAI,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAC7B,+DAA+D;oBAC/D,mDAAmD;oBACnD,MAAM,GAAG,KAAK,CAAC;oBACf,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACnD,mEAAmE;YACnE,wEAAwE;YACxE,oEAAoE;YACpE,IAAI,MAAM,EAAE,CAAC;gBACX,gEAAgE;gBAChE,UAAU,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;YAC/D,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,IAAI,EAAE,CAAC;YACb,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO;YAAE,OAAO;QACjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC;YACxC,MAAM,UAAU,CACd,eAAe,IAAI,IAAI,QAAQ,oBAAoB,EACnD,MAAM,CACP,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAChD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAChC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YAC3B,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;YAChE,MAAM,UAAU,CAAC,gBAAgB,MAAM,CAAC,MAAM,IAAI,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;gBAC9B,MAAM,IAAI,mBAAmB,CAC3B,yCAAyC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAChE,CAAC;YACJ,CAAC;YACD,MAAM;gBACJ,EAAE,EAAE,UAAU,MAAM,CAAC,eAAe,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE;gBACpD,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;gBAC1B,IAAI,EAAE,MAAM,CAAC,iBAAiB;aAC/B,CAAC;QACJ,CAAC;aAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpD,qEAAqE;YACrE,IAAI,CAAC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;YAC5B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACjE,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACnD,CAAC;IAEO,gBAAgB,CAAC,KAAa,EAAE,KAAwB;QAC9D,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE;YACrB,WAAW,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;YACpC,YAAY,EAAE,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC;YACvE,YAAY,EAAE,KAAK,CAAC,iBAAiB,IAAI,CAAC;SAC3C,CAAC,CAAC;IACL,CAAC;IAEQ,KAAK,CAAC,eAAe,CAAC,IAS9B;QACC,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YACrD,IACE,uBAAuB,CAAC,IAAI,CAAC;gBAC7B,sBAAsB,CAAC,IAAI,CAAC;gBAC5B,cAAc,CAAC,IAAI,CAAC,EACpB,CAAC;gBACD,SAAS;YACX,CAAC;YACD,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;gBACnB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;iBAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpD,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC;YAC1B,CAAC;QACH,CAAC;QACD,OAAO;YACL,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,OAAO,IAAI,IAAI;YACxB,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI;SAC/C,CAAC;IACJ,CAAC;CACF;AAED,6CAA6C;AAC7C,MAAM,UAAU,YAAY,CAAC,IAK5B;IACC,MAAM,IAAI,GAAG;QACX,gBAAgB;QAChB,aAAa;QACb,iBAAiB;QACjB,aAAa;KACd,CAAC;IACF,IAAI,IAAI,CAAC,KAAK;QAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IACjD,IAAI,IAAI,CAAC,YAAY;QAAE,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IACtE,IAAI,IAAI,CAAC,MAAM;QAAE,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IAC/D,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,CACP,eAAe,EACf,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAClE,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,wDAAwD;AACxD,KAAK,UAAU,SAAS,CAAC,GAAsB;IAC7C,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK;QAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,wDAAwD;AACxD,SAAS,SAAS,CAAC,MAAsB;IACvC,OAAO,MAAM,KAAK,OAAO,IAAI,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;AAClE,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,eAAe,CAAC,KAAwB;IACtD,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;QACxB,MAAM,KAAK,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACxE,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;YAAE,SAAS;QACvE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,qDAAqD;AACrD,SAAS,gBAAgB,CAAC,IAAY;IACpC,OAAO,IAAI;SACR,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;SACpE,IAAI,CAAC,GAAG,CAAC,CAAC;AACf,CAAC;AAED,uEAAuE;AACvE,SAAS,UAAU,CAAC,IAAY;IAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAC1C,IAAI,IAAa,CAAC;IAClB,IAAI,CAAC;QACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,GAAG,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,MAAM,GAAG,cAAc,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC9C,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;AAC7C,CAAC;AAED;;;GAGG;AACH,SAAS,UAAU,CAAC,OAAe,EAAE,MAAc;IACjD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IAC3B,IAAI,yCAAyC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACzD,OAAO,IAAI,mBAAmB,CAC5B,iCAAiC,IAAI,8CAA8C,CACpF,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;AACzE,CAAC;AAED,oEAAoE;AACpE,SAAS,MAAM,CAAC,OAA2B;IACzC,IAAI,OAAO,IAAI,IAAI;QAAE,OAAO,EAAE,CAAC;IAC/B,IAAI,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,OAAO,CAAC;IACtC,OAAO,OAAO;SACX,MAAM,CAAC,CAAC,CAAC,EAA2B,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;SACzD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAED,4EAA4E;AAC5E,SAAS,cAAc,CAAC,QAA4B;IAClD,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAC5E,CAAC;IACF,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,IAAI,mBAAmB,CAC3B,+GAA+G,CAChH,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,QAA4B;IACvD,OAAO,QAAQ;SACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;SAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;SAC7B,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,MAAM,CAAC,CAAC;AAClB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,QAA4B,EAC5B,UAAkB;IAElB,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IAC1E,cAAc,CAAC,KAAK,CAAC,CAAC;IACtB,OAAO,KAAK;SACT,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;SAC7B,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,MAAM,CAAC,CAAC;AAClB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAC9B,QAA4B,EAC5B,YAAoB;IAEpB,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;IAClF,cAAc,CAAC,KAAK,CAAC,CAAC;IACtB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAClE,MAAM,KAAK,GAAG,IAAI,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IACjE,MAAM,UAAU,GAAG,KAAK;SACrB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACT,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAC/B,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACjF,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,MAAM,CAAC,CAAC;IAChB,OAAO;QACL,YAAY,CAAC,CAAC,CAAC,mBAAmB,YAAY,mBAAmB,CAAC,CAAC,CAAC,EAAE;QACtE,UAAU,CAAC,CAAC,CAAC,0BAA0B,UAAU,0BAA0B,CAAC,CAAC,CAAC,EAAE;QAChF,OAAO;KACR;SACE,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,MAAM,CAAC,CAAC;AAClB,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,gBAAgB;IAC7B,MAAM,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QACvC,iBAAiB,CAAoC,kBAAkB,CAAC;QACxE,iBAAiB,CAA2B,SAAS,CAAC;QACtD,iBAAiB,CAA6B,WAAW,CAAC;KAC3D,CAAC,CAAC;IACH,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACrC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC;IAC3D,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACzC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,0EAA0E;AAC1E,MAAM,QAAQ,GAAyB,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE;IAC7D,MAAM,CAAC,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QACjD,iBAAiB,CAAsC,oBAAoB,CAAC;QAC5E,iBAAiB,CAA2B,SAAS,CAAC;QACtD,iBAAiB,CAA6B,WAAW,CAAC;QAC1D,iBAAiB,CAAiC,eAAe,CAAC;KACnE,CAAC,CAAC;IACH,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QACrC,MAAM,IAAI,mBAAmB,CAAC,2CAA2C,CAAC,CAAC;IAC7E,CAAC;IACD,yEAAyE;IACzE,qBAAqB;IACrB,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,GAAG,iBAAiB,CAAC,CAAC;IACnE,MAAM,GAAG,GAAG;QACV,GAAG,OAAO,CAAC,GAAG;QACd,IAAI,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,YAAY,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;KAC/E,CAAC;IACF,MAAM,KAAK,GAAG,EAAE,CAAC,KAAK,CAAC,eAAe,EAAE,IAAI,EAAE;QAC5C,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,SAAS;QAC7B,GAAG;QACH,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;QAC/B,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACjC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;QACvC,8CAA8C;QAC9C,MAAM,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,IAAI,OAAO,CACtB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClB,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,GAA0B,EAAE,EAAE;YACjD,MAAM,CACJ,GAAG,CAAC,IAAI,KAAK,QAAQ;gBACnB,CAAC,CAAC,IAAI,mBAAmB,CAAC,oCAAoC,YAAY,EAAE,CAAC;gBAC7E,CAAC,CAAC,GAAG,CACR,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC,CACF,CAAC;IACF,kEAAkE;IAClE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC5B,mEAAmE;IACnE,MAAM,OAAO,CAAC,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC1B,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;KACrE,CAAC,CAAC;IAEH,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QAC3B,wEAAwE;IAC1E,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC/B,MAAM,KAAK,GAAG,QAAQ,CAAC,eAAe,CAAC;QACrC,KAAK,EAAE,KAAK,CAAC,MAAM;QACnB,SAAS,EAAE,QAAQ;KACpB,CAAC,CAAC;IACH,OAAO;QACL,KAAK;QACL,IAAI;QACJ,IAAI,EAAE,GAAG,EAAE;YACT,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI;gBAAE,KAAK,CAAC,IAAI,EAAE,CAAC;QACzE,CAAC;KACF,CAAC;AACJ,CAAC,CAAC"}

package/dist/providers/azure-openai-provider.d.ts

@@ -0,0 +1,69 @@
+import OpenAI from "openai";
+import type { ProviderCapability } from "./base-provider.js";
+import { OpenAIProvider } from "./openai-provider.js";
+import type { ASRModel, EmbeddingModel, ImageModel, LanguageModel, TTSModel } from "./types.js";
+/** What a deployment serves. Azure does not report it, so it comes from config or the name. */
+export type AzureDeploymentKind = "chat" | "image" | "tts" | "asr" | "embedding";
+export interface AzureDeployment {
+    id: string;
+    kind: AzureDeploymentKind;
+    model?: string;
+}
+/**
+ * The kind a deployment name implies. Azure deployments default to the name of
+ * the model they serve, so `gpt-image-2`, `tts-hd`, `whisper` and
+ * `text-embedding-3-large` classify themselves. Anything else is a chat model.
+ */
+export declare function inferAzureDeploymentKind(name: string): AzureDeploymentKind;
+/**
+ * Parse `AZURE_OPENAI_DEPLOYMENTS`: a comma-separated list of deployment
+ * names. A name may carry an explicit kind after a colon
+ * (`my-painter:image`) when the name does not reveal it. Append `=model`
+ * to resolve capabilities independently of the deployment routing name.
+ */
+export declare function parseAzureDeployments(raw: string | undefined): AzureDeployment[];
+/**
+ * The resource root of an Azure OpenAI or Microsoft Foundry endpoint. Accepts
+ * the forms the Azure portal shows: with or without a trailing `/openai` or
+ * `/openai/v1`.
+ */
+export declare function azureOpenAIResourceRoot(endpoint: string): string;
+interface AzureOpenAIProviderOptions {
+    client?: OpenAI;
+    fetchFn?: typeof fetch;
+}
+/**
+ * Azure OpenAI Service and Microsoft Foundry. The wire format is OpenAI's, so
+ * chat, image, speech, transcription and embedding calls reuse
+ * {@link OpenAIProvider}. Models are the deployments the user lists, because
+ * Azure addresses a model by its deployment name.
+ *
+ * Without `AZURE_OPENAI_API_VERSION` (or with `v1`) requests go to the v1 API
+ * at `<resource>/openai/v1`. With a dated version they go to the classic
+ * `<resource>/openai/deployments/<name>?api-version=<version>` routes.
+ */
+export declare class AzureOpenAIProvider extends OpenAIProvider {
+    static requiredSecrets(): string[];
+    readonly endpoint: string;
+    readonly apiVersion: string;
+    readonly deployments: AzureDeployment[];
+    constructor(secrets: {
+        AZURE_OPENAI_API_KEY?: string;
+        AZURE_OPENAI_ENDPOINT?: string;
+        AZURE_OPENAI_DEPLOYMENTS?: string;
+        AZURE_OPENAI_API_VERSION?: string;
+    }, options?: AzureOpenAIProviderOptions);
+    protected mediaModelIdentity(model: string): string;
+    getContainerEnv(): Record<string, string>;
+    hasToolSupport(_model: string): Promise<boolean>;
+    private deploymentsOf;
+    protected declaredCapabilities(): readonly ProviderCapability[];
+    getAvailableLanguageModels(): Promise<LanguageModel[]>;
+    getAvailableImageModels(): Promise<ImageModel[]>;
+    getAvailableTTSModels(): Promise<TTSModel[]>;
+    getAvailableASRModels(): Promise<ASRModel[]>;
+    getAvailableEmbeddingModels(): Promise<EmbeddingModel[]>;
+    getAvailableVideoModels(): Promise<never[]>;
+}
+export {};
+//# sourceMappingURL=azure-openai-provider.d.ts.map

package/dist/providers/azure-openai-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"azure-openai-provider.d.ts","sourceRoot":"","sources":["../../src/providers/azure-openai-provider.ts"],"names":[],"mappings":"AAAA,OAAO,MAAuB,MAAM,QAAQ,CAAC;AAO7C,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,OAAO,KAAK,EACV,QAAQ,EACR,cAAc,EACd,UAAU,EACV,aAAa,EACb,QAAQ,EACT,MAAM,YAAY,CAAC;AAEpB,+FAA+F;AAC/F,MAAM,MAAM,mBAAmB,GAAG,MAAM,GAAG,OAAO,GAAG,KAAK,GAAG,KAAK,GAAG,WAAW,CAAC;AAEjF,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,mBAAmB,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAwBD;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,MAAM,GAAG,mBAAmB,CAO1E;AAED;;;;;GAKG;AACH,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,GAAG,eAAe,EAAE,CAuBhF;AAED;;;;GAIG;AACH,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAYhE;AAUD,UAAU,0BAA0B;IAClC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,KAAK,CAAC;CACxB;AAED;;;;;;;;;GASG;AACH,qBAAa,mBAAoB,SAAQ,cAAc;IACrD,OAAgB,eAAe,IAAI,MAAM,EAAE,CAE1C;IAED,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,WAAW,EAAE,eAAe,EAAE,CAAC;IAExC,YACE,OAAO,EAAE;QACP,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;QAC/B,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,wBAAwB,CAAC,EAAE,MAAM,CAAC;KACnC,EACD,OAAO,GAAE,0BAA+B,EAyDzC;IAED,UAAmB,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAG3D;IAEQ,eAAe,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAKjD;IAEc,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAE9D;IAED,OAAO,CAAC,aAAa;IAIrB,UAAmB,oBAAoB,IAAI,SAAS,kBAAkB,EAAE,CAavE;IAEc,0BAA0B,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC,CAMpE;IAEc,uBAAuB,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC,CAO9D;IAEc,qBAAqB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC,CAO1D;IAEc,qBAAqB,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC,CAM1D;IAEc,2BAA2B,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAMtE;IAEc,uBAAuB,qBAErC;CACF"}