@office-ai/aioncli-core 0.18.7 → 0.24.0-preview.1

package/dist/src/agents/types.d.ts

@@ -32,6 +32,12 @@ export interface OutputObject {
  * Used primarily for templating the system prompt. (Replaces ContextState)
  */
 export type AgentInputs = Record<string, unknown>;
+/**
+ * Simplified input structure for Remote Agents, which consumes a single string query.
+ */
+export type RemoteAgentInputs = {
+    query: string;
+};
 /**
  * Structured events emitted during subagent execution for user observability.
  */
@@ -42,20 +48,23 @@ export interface SubagentActivityEvent {
     data: Record<string, unknown>;
 }
 /**
- * The definition for an agent.
+ * The base definition for an agent.
  * @template TOutput The specific Zod schema for the agent's final output object.
  */
-export interface AgentDefinition<TOutput extends z.ZodTypeAny = z.ZodUnknown> {
+export interface BaseAgentDefinition<TOutput extends z.ZodTypeAny = z.ZodUnknown> {
     /** Unique identifier for the agent. */
     name: string;
     displayName?: string;
     description: string;
+    inputConfig: InputConfig;
+    outputConfig?: OutputConfig<TOutput>;
+}
+export interface LocalAgentDefinition<TOutput extends z.ZodTypeAny = z.ZodUnknown> extends BaseAgentDefinition<TOutput> {
+    kind: 'local';
     promptConfig: PromptConfig;
     modelConfig: ModelConfig;
     runConfig: RunConfig;
     toolConfig?: ToolConfig;
-    outputConfig?: OutputConfig<TOutput>;
-    inputConfig: InputConfig;
     /**
      * An optional function to process the raw output from the agent's final tool
      * call into a string format.
@@ -65,6 +74,11 @@ export interface AgentDefinition<TOutput extends z.ZodTypeAny = z.ZodUnknown> {
      */
     processOutput?: (output: z.infer<TOutput>) => string;
 }
+export interface RemoteAgentDefinition<TOutput extends z.ZodTypeAny = z.ZodUnknown> extends BaseAgentDefinition<TOutput> {
+    kind: 'remote';
+    agentCardUrl: string;
+}
+export type AgentDefinition<TOutput extends z.ZodTypeAny = z.ZodUnknown> = LocalAgentDefinition<TOutput> | RemoteAgentDefinition<TOutput>;
 /**
  * Configures the initial prompt for the agent.
  */

package/dist/src/availability/errorClassification.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { FailureKind } from './modelPolicy.js';
+export declare function classifyFailureKind(error: unknown): FailureKind;

package/dist/src/availability/errorClassification.js

@@ -0,0 +1,20 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { TerminalQuotaError, RetryableQuotaError, } from '../utils/googleQuotaErrors.js';
+import { ModelNotFoundError } from '../utils/httpErrors.js';
+export function classifyFailureKind(error) {
+    if (error instanceof TerminalQuotaError) {
+        return 'terminal';
+    }
+    if (error instanceof RetryableQuotaError) {
+        return 'transient';
+    }
+    if (error instanceof ModelNotFoundError) {
+        return 'not_found';
+    }
+    return 'unknown';
+}
+//# sourceMappingURL=errorClassification.js.map

package/dist/src/availability/errorClassification.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errorClassification.js","sourceRoot":"","sources":["../../../src/availability/errorClassification.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAG5D,MAAM,UAAU,mBAAmB,CAAC,KAAc;IAChD,IAAI,KAAK,YAAY,kBAAkB,EAAE,CAAC;QACxC,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,IAAI,KAAK,YAAY,mBAAmB,EAAE,CAAC;QACzC,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,IAAI,KAAK,YAAY,kBAAkB,EAAE,CAAC;QACxC,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/src/availability/modelAvailabilityService.d.ts

@@ -0,0 +1,36 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export type ModelId = string;
+type TerminalUnavailabilityReason = 'quota' | 'capacity';
+export type TurnUnavailabilityReason = 'retry_once_per_turn';
+export type UnavailabilityReason = TerminalUnavailabilityReason | TurnUnavailabilityReason | 'unknown';
+export type ModelHealthStatus = 'terminal' | 'sticky_retry';
+export interface ModelAvailabilitySnapshot {
+    available: boolean;
+    reason?: UnavailabilityReason;
+}
+export interface ModelSelectionResult {
+    selectedModel: ModelId | null;
+    attempts?: number;
+    skipped: Array<{
+        model: ModelId;
+        reason: UnavailabilityReason;
+    }>;
+}
+export declare class ModelAvailabilityService {
+    private readonly health;
+    markTerminal(model: ModelId, reason: TerminalUnavailabilityReason): void;
+    markHealthy(model: ModelId): void;
+    markRetryOncePerTurn(model: ModelId): void;
+    consumeStickyAttempt(model: ModelId): void;
+    snapshot(model: ModelId): ModelAvailabilitySnapshot;
+    selectFirstAvailable(models: ModelId[]): ModelSelectionResult;
+    resetTurn(): void;
+    reset(): void;
+    private setState;
+    private clearState;
+}
+export {};

package/dist/src/availability/modelAvailabilityService.js

@@ -0,0 +1,87 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export class ModelAvailabilityService {
+    health = new Map();
+    markTerminal(model, reason) {
+        this.setState(model, {
+            status: 'terminal',
+            reason,
+        });
+    }
+    markHealthy(model) {
+        this.clearState(model);
+    }
+    markRetryOncePerTurn(model) {
+        const currentState = this.health.get(model);
+        // Do not override a terminal failure with a transient one.
+        if (currentState?.status === 'terminal') {
+            return;
+        }
+        // Only reset consumption if we are not already in the sticky_retry state.
+        // This prevents infinite loops if the model fails repeatedly in the same turn.
+        let consumed = false;
+        if (currentState?.status === 'sticky_retry') {
+            consumed = currentState.consumed;
+        }
+        this.setState(model, {
+            status: 'sticky_retry',
+            reason: 'retry_once_per_turn',
+            consumed,
+        });
+    }
+    consumeStickyAttempt(model) {
+        const state = this.health.get(model);
+        if (state?.status === 'sticky_retry') {
+            this.setState(model, { ...state, consumed: true });
+        }
+    }
+    snapshot(model) {
+        const state = this.health.get(model);
+        if (!state) {
+            return { available: true };
+        }
+        if (state.status === 'terminal') {
+            return { available: false, reason: state.reason };
+        }
+        if (state.status === 'sticky_retry' && state.consumed) {
+            return { available: false, reason: state.reason };
+        }
+        return { available: true };
+    }
+    selectFirstAvailable(models) {
+        const skipped = [];
+        for (const model of models) {
+            const snapshot = this.snapshot(model);
+            if (snapshot.available) {
+                const state = this.health.get(model);
+                // A sticky model is being attempted, so note that.
+                const attempts = state?.status === 'sticky_retry' ? 1 : undefined;
+                return { selectedModel: model, skipped, attempts };
+            }
+            else {
+                skipped.push({ model, reason: snapshot.reason ?? 'unknown' });
+            }
+        }
+        return { selectedModel: null, skipped };
+    }
+    resetTurn() {
+        for (const [model, state] of this.health.entries()) {
+            if (state.status === 'sticky_retry') {
+                this.setState(model, { ...state, consumed: false });
+            }
+        }
+    }
+    reset() {
+        this.health.clear();
+    }
+    setState(model, nextState) {
+        this.health.set(model, nextState);
+    }
+    clearState(model) {
+        this.health.delete(model);
+    }
+}
+//# sourceMappingURL=modelAvailabilityService.js.map

package/dist/src/availability/modelAvailabilityService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelAvailabilityService.js","sourceRoot":"","sources":["../../../src/availability/modelAvailabilityService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAoCH,MAAM,OAAO,wBAAwB;IAClB,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;IAE1D,YAAY,CAAC,KAAc,EAAE,MAAoC;QAC/D,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE;YACnB,MAAM,EAAE,UAAU;YAClB,MAAM;SACP,CAAC,CAAC;IACL,CAAC;IAED,WAAW,CAAC,KAAc;QACxB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,oBAAoB,CAAC,KAAc;QACjC,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5C,2DAA2D;QAC3D,IAAI,YAAY,EAAE,MAAM,KAAK,UAAU,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QAED,0EAA0E;QAC1E,+EAA+E;QAC/E,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,YAAY,EAAE,MAAM,KAAK,cAAc,EAAE,CAAC;YAC5C,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE;YACnB,MAAM,EAAE,cAAc;YACtB,MAAM,EAAE,qBAAqB;YAC7B,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAAC,KAAc;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,KAAK,EAAE,MAAM,KAAK,cAAc,EAAE,CAAC;YACrC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAAc;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAErC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YAChC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;QACpD,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,cAAc,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACtD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;QACpD,CAAC;QAED,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,oBAAoB,CAAC,MAAiB;QACpC,MAAM,OAAO,GAAoC,EAAE,CAAC;QAEpD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACvB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACrC,mDAAmD;gBACnD,MAAM,QAAQ,GAAG,KAAK,EAAE,MAAM,KAAK,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAClE,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;YACrD,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QACD,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IAC1C,CAAC;IAED,SAAS;QACP,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,IAAI,KAAK,CAAC,MAAM,KAAK,cAAc,EAAE,CAAC;gBACpC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAEO,QAAQ,CAAC,KAAc,EAAE,SAAsB;QACrD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC;IAEO,UAAU,CAAC,KAAc;QAC/B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;CACF"}

package/dist/src/availability/modelAvailabilityService.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/availability/modelAvailabilityService.test.js

@@ -0,0 +1,140 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, expect, it, vi, beforeEach } from 'vitest';
+import { ModelAvailabilityService } from './modelAvailabilityService.js';
+describe('ModelAvailabilityService', () => {
+    let service;
+    const model = 'test-model';
+    beforeEach(() => {
+        service = new ModelAvailabilityService();
+        vi.useRealTimers();
+    });
+    it('returns available snapshot when no state recorded', () => {
+        expect(service.snapshot(model)).toEqual({ available: true });
+    });
+    it('tracks retry-once-per-turn failures', () => {
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model)).toEqual({ available: true });
+        service.consumeStickyAttempt(model);
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'retry_once_per_turn',
+        });
+        service.resetTurn();
+        expect(service.snapshot(model)).toEqual({ available: true });
+    });
+    it('tracks terminal failures', () => {
+        service.markTerminal(model, 'quota');
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'quota',
+        });
+    });
+    it('does not override terminal failure with sticky failure', () => {
+        service.markTerminal(model, 'quota');
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'quota',
+        });
+    });
+    it('selects models respecting terminal and sticky states', () => {
+        const stickyModel = 'stick-model';
+        const healthyModel = 'healthy-model';
+        service.markTerminal(model, 'capacity');
+        service.markRetryOncePerTurn(stickyModel);
+        const first = service.selectFirstAvailable([
+            model,
+            stickyModel,
+            healthyModel,
+        ]);
+        expect(first).toEqual({
+            selectedModel: stickyModel,
+            attempts: 1,
+            skipped: [
+                {
+                    model,
+                    reason: 'capacity',
+                },
+            ],
+        });
+        service.consumeStickyAttempt(stickyModel);
+        const second = service.selectFirstAvailable([
+            model,
+            stickyModel,
+            healthyModel,
+        ]);
+        expect(second).toEqual({
+            selectedModel: healthyModel,
+            skipped: [
+                {
+                    model,
+                    reason: 'capacity',
+                },
+                {
+                    model: stickyModel,
+                    reason: 'retry_once_per_turn',
+                },
+            ],
+        });
+        service.resetTurn();
+        const third = service.selectFirstAvailable([
+            model,
+            stickyModel,
+            healthyModel,
+        ]);
+        expect(third).toEqual({
+            selectedModel: stickyModel,
+            attempts: 1,
+            skipped: [
+                {
+                    model,
+                    reason: 'capacity',
+                },
+            ],
+        });
+    });
+    it('preserves consumed state when marking retry-once-per-turn again', () => {
+        service.markRetryOncePerTurn(model);
+        service.consumeStickyAttempt(model);
+        // It is currently consumed
+        expect(service.snapshot(model).available).toBe(false);
+        // Marking it again should not reset the consumed flag
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model).available).toBe(false);
+    });
+    it('clears consumed state when marked healthy', () => {
+        service.markRetryOncePerTurn(model);
+        service.consumeStickyAttempt(model);
+        expect(service.snapshot(model).available).toBe(false);
+        service.markHealthy(model);
+        expect(service.snapshot(model).available).toBe(true);
+        // If we mark it sticky again, it should be fresh (not consumed)
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model).available).toBe(true);
+    });
+    it('resetTurn resets consumed state for multiple sticky models', () => {
+        const model2 = 'model-2';
+        service.markRetryOncePerTurn(model);
+        service.markRetryOncePerTurn(model2);
+        service.consumeStickyAttempt(model);
+        service.consumeStickyAttempt(model2);
+        expect(service.snapshot(model).available).toBe(false);
+        expect(service.snapshot(model2).available).toBe(false);
+        service.resetTurn();
+        expect(service.snapshot(model).available).toBe(true);
+        expect(service.snapshot(model2).available).toBe(true);
+    });
+    it('resetTurn does not affect terminal models', () => {
+        service.markTerminal(model, 'quota');
+        service.resetTurn();
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'quota',
+        });
+    });
+});
+//# sourceMappingURL=modelAvailabilityService.test.js.map

package/dist/src/availability/modelAvailabilityService.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelAvailabilityService.test.js","sourceRoot":"","sources":["../../../src/availability/modelAvailabilityService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AAEzE,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,OAAiC,CAAC;IACtC,MAAM,KAAK,GAAG,YAAY,CAAC;IAE3B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,wBAAwB,EAAE,CAAC;QACzC,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE7D,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,qBAAqB;SAC9B,CAAC,CAAC;QAEH,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACrC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,WAAW,GAAG,aAAa,CAAC;QAClC,MAAM,YAAY,GAAG,eAAe,CAAC;QAErC,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACxC,OAAO,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAE1C,MAAM,KAAK,GAAG,OAAO,CAAC,oBAAoB,CAAC;YACzC,KAAK;YACL,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,aAAa,EAAE,WAAW;YAC1B,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE;gBACP;oBACE,KAAK;oBACL,MAAM,EAAE,UAAU;iBACnB;aACF;SACF,CAAC,CAAC;QAEH,OAAO,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC;YAC1C,KAAK;YACL,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,aAAa,EAAE,YAAY;YAC3B,OAAO,EAAE;gBACP;oBACE,KAAK;oBACL,MAAM,EAAE,UAAU;iBACnB;gBACD;oBACE,KAAK,EAAE,WAAW;oBAClB,MAAM,EAAE,qBAAqB;iBAC9B;aACF;SACF,CAAC,CAAC;QAEH,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,MAAM,KAAK,GAAG,OAAO,CAAC,oBAAoB,CAAC;YACzC,KAAK;YACL,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,aAAa,EAAE,WAAW;YAC1B,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE;gBACP;oBACE,KAAK;oBACL,MAAM,EAAE,UAAU;iBACnB;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEpC,2BAA2B;QAC3B,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEtD,sDAAsD;QACtD,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEtD,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC3B,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAErD,gEAAgE;QAChE,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,MAAM,GAAG,SAAS,CAAC;QACzB,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAErC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAErC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvD,OAAO,CAAC,SAAS,EAAE,CAAC;QAEpB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACrC,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/availability/modelPolicy.d.ts

@@ -0,0 +1,49 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ModelAvailabilityService, ModelHealthStatus, ModelId } from './modelAvailabilityService.js';
+/**
+ * Whether to prompt the user or fallback silently on a model API failure.
+ */
+export type FallbackAction = 'silent' | 'prompt';
+/**
+ * Type of possible errors from model API failures.
+ */
+export type FailureKind = 'terminal' | 'transient' | 'not_found' | 'unknown';
+/**
+ * Map from model API failure reason to user interaction.
+ */
+export type ModelPolicyActionMap = Partial<Record<FailureKind, FallbackAction>>;
+/**
+ * What state (e.g. Terminal, Sticky Retry) to set a model after failed API call.
+ */
+export type ModelPolicyStateMap = Partial<Record<FailureKind, ModelHealthStatus>>;
+/**
+ * Defines the policy for a single model in the availability chain.
+ *
+ * This includes:
+ * - Which model this policy applies to.
+ * - What actions to take (prompt vs silent fallback) for different failure kinds.
+ * - How the model's health status should transition upon failure.
+ * - Whether this model is considered a "last resort" (i.e. use if all models are unavailable).
+ */
+export interface ModelPolicy {
+    model: ModelId;
+    actions: ModelPolicyActionMap;
+    stateTransitions: ModelPolicyStateMap;
+    isLastResort?: boolean;
+}
+/**
+ * A chain of model policies defining the priority and fallback behavior.
+ * The first model in the chain is the primary model.
+ */
+export type ModelPolicyChain = ModelPolicy[];
+/**
+ * Context required by retry logic to apply availability policies on failure.
+ */
+export interface RetryAvailabilityContext {
+    service: ModelAvailabilityService;
+    policy: ModelPolicy;
+}

package/dist/src/availability/modelPolicy.js

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};
+//# sourceMappingURL=modelPolicy.js.map

package/dist/src/availability/modelPolicy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelPolicy.js","sourceRoot":"","sources":["../../../src/availability/modelPolicy.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}

package/dist/src/availability/policyCatalog.d.ts

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ModelPolicy, ModelPolicyChain } from './modelPolicy.js';
+import type { UserTierId } from '../code_assist/types.js';
+export interface ModelPolicyOptions {
+    previewEnabled: boolean;
+    userTier?: UserTierId;
+}
+/**
+ * Returns the default ordered model policy chain for the user.
+ */
+export declare function getModelPolicyChain(options: ModelPolicyOptions): ModelPolicyChain;
+export declare function createSingleModelChain(model: string): ModelPolicyChain;
+/**
+ * Provides a default policy scaffold for models not present in the catalog.
+ */
+export declare function createDefaultPolicy(model: string, options?: {
+    isLastResort?: boolean;
+}): ModelPolicy;
+export declare function validateModelPolicyChain(chain: ModelPolicyChain): void;

package/dist/src/availability/policyCatalog.js

@@ -0,0 +1,82 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_FLASH_MODEL, PREVIEW_GEMINI_MODEL, } from '../config/models.js';
+const DEFAULT_ACTIONS = {
+    terminal: 'prompt',
+    transient: 'prompt',
+    not_found: 'prompt',
+    unknown: 'prompt',
+};
+const DEFAULT_STATE = {
+    terminal: 'terminal',
+    transient: 'terminal',
+    not_found: 'terminal',
+    unknown: 'terminal',
+};
+const DEFAULT_CHAIN = [
+    definePolicy({ model: DEFAULT_GEMINI_MODEL }),
+    definePolicy({ model: DEFAULT_GEMINI_FLASH_MODEL, isLastResort: true }),
+];
+const PREVIEW_CHAIN = [
+    definePolicy({ model: PREVIEW_GEMINI_MODEL }),
+    definePolicy({ model: PREVIEW_GEMINI_FLASH_MODEL, isLastResort: true }),
+];
+/**
+ * Returns the default ordered model policy chain for the user.
+ */
+export function getModelPolicyChain(options) {
+    if (options.previewEnabled) {
+        return cloneChain(PREVIEW_CHAIN);
+    }
+    return cloneChain(DEFAULT_CHAIN);
+}
+export function createSingleModelChain(model) {
+    return [definePolicy({ model, isLastResort: true })];
+}
+/**
+ * Provides a default policy scaffold for models not present in the catalog.
+ */
+export function createDefaultPolicy(model, options) {
+    return definePolicy({ model, isLastResort: options?.isLastResort });
+}
+export function validateModelPolicyChain(chain) {
+    if (chain.length === 0) {
+        throw new Error('Model policy chain must include at least one model.');
+    }
+    const lastResortCount = chain.filter((policy) => policy.isLastResort).length;
+    if (lastResortCount === 0) {
+        throw new Error('Model policy chain must include an `isLastResort` model.');
+    }
+    if (lastResortCount > 1) {
+        throw new Error('Model policy chain must only have one `isLastResort`.');
+    }
+}
+/**
+ * Helper to define a ModelPolicy with default actions and state transitions.
+ * Ensures every policy is a fresh instance to avoid shared state.
+ */
+function definePolicy(config) {
+    return {
+        model: config.model,
+        isLastResort: config.isLastResort,
+        actions: { ...DEFAULT_ACTIONS, ...(config.actions ?? {}) },
+        stateTransitions: {
+            ...DEFAULT_STATE,
+            ...(config.stateTransitions ?? {}),
+        },
+    };
+}
+function clonePolicy(policy) {
+    return {
+        ...policy,
+        actions: { ...policy.actions },
+        stateTransitions: { ...policy.stateTransitions },
+    };
+}
+function cloneChain(chain) {
+    return chain.map(clonePolicy);
+}
+//# sourceMappingURL=policyCatalog.js.map

package/dist/src/availability/policyCatalog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policyCatalog.js","sourceRoot":"","sources":["../../../src/availability/policyCatalog.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,OAAO,EACL,0BAA0B,EAC1B,oBAAoB,EACpB,0BAA0B,EAC1B,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAc7B,MAAM,eAAe,GAAyB;IAC5C,QAAQ,EAAE,QAAQ;IAClB,SAAS,EAAE,QAAQ;IACnB,SAAS,EAAE,QAAQ;IACnB,OAAO,EAAE,QAAQ;CAClB,CAAC;AAEF,MAAM,aAAa,GAAwB;IACzC,QAAQ,EAAE,UAAU;IACpB,SAAS,EAAE,UAAU;IACrB,SAAS,EAAE,UAAU;IACrB,OAAO,EAAE,UAAU;CACpB,CAAC;AAEF,MAAM,aAAa,GAAqB;IACtC,YAAY,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC;IAC7C,YAAY,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;CACxE,CAAC;AAEF,MAAM,aAAa,GAAqB;IACtC,YAAY,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC;IAC7C,YAAY,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;CACxE,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,OAA2B;IAE3B,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;QAC3B,OAAO,UAAU,CAAC,aAAa,CAAC,CAAC;IACnC,CAAC;IAED,OAAO,UAAU,CAAC,aAAa,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,KAAa;IAClD,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AACvD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,KAAa,EACb,OAAoC;IAEpC,OAAO,YAAY,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;AACtE,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,KAAuB;IAC9D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IACD,MAAM,eAAe,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC;IAC7E,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;IAC3E,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,YAAY,CAAC,MAAoB;IACxC,OAAO;QACL,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,OAAO,EAAE,EAAE,GAAG,eAAe,EAAE,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE;QAC1D,gBAAgB,EAAE;YAChB,GAAG,aAAa;YAChB,GAAG,CAAC,MAAM,CAAC,gBAAgB,IAAI,EAAE,CAAC;SACnC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,MAAmB;IACtC,OAAO;QACL,GAAG,MAAM;QACT,OAAO,EAAE,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE;QAC9B,gBAAgB,EAAE,EAAE,GAAG,MAAM,CAAC,gBAAgB,EAAE;KACjD,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,KAAuB;IACzC,OAAO,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;AAChC,CAAC"}

package/dist/src/availability/policyCatalog.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/availability/policyCatalog.test.js

@@ -0,0 +1,70 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { createDefaultPolicy, getModelPolicyChain, validateModelPolicyChain, } from './policyCatalog.js';
+import { DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_MODEL, } from '../config/models.js';
+describe('policyCatalog', () => {
+    it('returns preview chain when preview enabled', () => {
+        const chain = getModelPolicyChain({ previewEnabled: true });
+        expect(chain[0]?.model).toBe(PREVIEW_GEMINI_MODEL);
+        expect(chain).toHaveLength(2);
+    });
+    it('returns default chain when preview disabled', () => {
+        const chain = getModelPolicyChain({ previewEnabled: false });
+        expect(chain[0]?.model).toBe(DEFAULT_GEMINI_MODEL);
+        expect(chain).toHaveLength(2);
+    });
+    it('marks preview transients as sticky retries', () => {
+        const [previewPolicy] = getModelPolicyChain({ previewEnabled: true });
+        expect(previewPolicy.model).toBe(PREVIEW_GEMINI_MODEL);
+        expect(previewPolicy.stateTransitions.transient).toBe('terminal');
+    });
+    it('applies default actions and state transitions for unspecified kinds', () => {
+        const [previewPolicy] = getModelPolicyChain({ previewEnabled: true });
+        expect(previewPolicy.stateTransitions.not_found).toBe('terminal');
+        expect(previewPolicy.stateTransitions.unknown).toBe('terminal');
+        expect(previewPolicy.actions.unknown).toBe('prompt');
+    });
+    it('clones policy maps so edits do not leak between calls', () => {
+        const firstCall = getModelPolicyChain({ previewEnabled: false });
+        firstCall[0].actions.terminal = 'silent';
+        const secondCall = getModelPolicyChain({ previewEnabled: false });
+        expect(secondCall[0].actions.terminal).toBe('prompt');
+    });
+    it('passes when there is exactly one last-resort policy', () => {
+        const validChain = [
+            createDefaultPolicy('test-model'),
+            { ...createDefaultPolicy('last-resort'), isLastResort: true },
+        ];
+        expect(() => validateModelPolicyChain(validChain)).not.toThrow();
+    });
+    it('fails when no policies are marked last-resort', () => {
+        const chain = [
+            createDefaultPolicy('model-a'),
+            createDefaultPolicy('model-b'),
+        ];
+        expect(() => validateModelPolicyChain(chain)).toThrow('must include an `isLastResort`');
+    });
+    it('fails when a single-model chain is not last-resort', () => {
+        const chain = [createDefaultPolicy('lonely-model')];
+        expect(() => validateModelPolicyChain(chain)).toThrow('must include an `isLastResort`');
+    });
+    it('fails when multiple policies are marked last-resort', () => {
+        const chain = [
+            { ...createDefaultPolicy('model-a'), isLastResort: true },
+            { ...createDefaultPolicy('model-b'), isLastResort: true },
+        ];
+        expect(() => validateModelPolicyChain(chain)).toThrow('must only have one `isLastResort`');
+    });
+    it('createDefaultPolicy seeds default actions and states', () => {
+        const policy = createDefaultPolicy('custom');
+        expect(policy.actions.terminal).toBe('prompt');
+        expect(policy.actions.unknown).toBe('prompt');
+        expect(policy.stateTransitions.terminal).toBe('terminal');
+        expect(policy.stateTransitions.unknown).toBe('terminal');
+    });
+});
+//# sourceMappingURL=policyCatalog.test.js.map

package/dist/src/availability/policyCatalog.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policyCatalog.test.js","sourceRoot":"","sources":["../../../src/availability/policyCatalog.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,mBAAmB,EACnB,mBAAmB,EACnB,wBAAwB,GACzB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,oBAAoB,EACpB,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAE7B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,KAAK,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,KAAK,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,aAAa,CAAC,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACvD,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,CAAC,aAAa,CAAC,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClE,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAChE,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,SAAS,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzC,MAAM,UAAU,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,UAAU,GAAG;YACjB,mBAAmB,CAAC,YAAY,CAAC;YACjC,EAAE,GAAG,mBAAmB,CAAC,aAAa,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE;SAC9D,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,KAAK,GAAG;YACZ,mBAAmB,CAAC,SAAS,CAAC;YAC9B,mBAAmB,CAAC,SAAS,CAAC;SAC/B,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACnD,gCAAgC,CACjC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,KAAK,GAAG,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACnD,gCAAgC,CACjC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG;YACZ,EAAE,GAAG,mBAAmB,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE;YACzD,EAAE,GAAG,mBAAmB,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE;SAC1D,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACnD,mCAAmC,CACpC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}