@nocobase/app-plugin-agents 0.1.0-beta.9 → 1.0.0-beta.16

package/dist/server/online/agent/messages/validation.js

@@ -0,0 +1,159 @@
+import { jsonSchema, } from 'ai';
+import { invalid } from '../../../kernel/errors.js';
+import { jsonSafe } from './json.js';
+import { Ajv } from 'ajv';
+import addFormats from 'ajv-formats';
+import { RE2JS } from 're2js';
+/** Normalize trusted background user input to the SDK's JSON-persistable UI parts. */
+export function nativeInputParts(message) {
+    if (message.role !== 'user')
+        throw invalid('Input must contain user messages.');
+    if (typeof message.content === 'string')
+        return [{ type: 'text', text: message.content }];
+    return message.content.map((part) => {
+        const providerMetadata = part.providerOptions
+            ? jsonSafe(part.providerOptions)
+            : undefined;
+        if (part.type === 'text')
+            return {
+                type: 'text',
+                text: part.text,
+                ...(providerMetadata ? { providerMetadata } : {}),
+            };
+        let data = part.type === 'image' ? part.image : part.data;
+        const mediaType = part.mediaType ?? 'image';
+        let url;
+        let providerReference;
+        const stringDataUrl = (value) => URL.canParse(value) ? value : `data:${mediaType};base64,${value}`;
+        if (data instanceof URL) {
+            url = data.href;
+        }
+        else if (typeof data === 'object' &&
+            data !== null &&
+            !(data instanceof Uint8Array) &&
+            !(data instanceof ArrayBuffer)) {
+            if ('type' in data && data.type === 'url')
+                url = data.url.href;
+            else if ('type' in data && data.type === 'text')
+                url = `data:${mediaType};base64,${Buffer.from(data.text).toString('base64')}`;
+            else if ('type' in data && data.type === 'data') {
+                data = data.data;
+                if (typeof data === 'string' && data.startsWith('data:'))
+                    throw invalid('Data URLs must use a URL input, not tagged inline data.');
+                url =
+                    typeof data === 'string'
+                        ? `data:${mediaType};base64,${data}`
+                        : stringDataUrl(Buffer.from(data instanceof ArrayBuffer ? new Uint8Array(data) : data).toString('base64'));
+            }
+            else {
+                providerReference =
+                    'type' in data && data.type === 'reference' ? data.reference : data;
+                url = '';
+            }
+        }
+        else {
+            url =
+                typeof data === 'string'
+                    ? stringDataUrl(data)
+                    : stringDataUrl(Buffer.from(data instanceof ArrayBuffer ? new Uint8Array(data) : data).toString('base64'));
+        }
+        return {
+            type: 'file',
+            mediaType,
+            url,
+            ...(part.type === 'file' && part.filename
+                ? { filename: part.filename }
+                : {}),
+            ...(providerReference ? { providerReference } : {}),
+            ...(providerMetadata ? { providerMetadata } : {}),
+        };
+    });
+}
+function invalidSchema(message) {
+    return invalid(message, { field: 'schema' });
+}
+const draft = 'http://json-schema.org/draft-07/schema#';
+const linearRegExp = Object.assign((pattern, flags) => RE2JS.compile(RE2JS.translateRegExp(new RegExp(pattern, flags))), { code: 'linearRegExp' });
+/** Compile one call's JSON Schema without retaining user schemas in a shared cache. */
+export function createOutputSchema(schema) {
+    let nodes = 0;
+    const checkSize = (value, depth) => {
+        if (++nodes > 4096 || depth > 32)
+            throw invalidSchema('The output JSON Schema is too complex.');
+        if (value !== null && typeof value === 'object')
+            for (const child of Object.values(value))
+                checkSize(child, depth + 1);
+    };
+    checkSize(schema, 0);
+    const snapshot = jsonSafe(schema);
+    if (!snapshot || typeof snapshot !== 'object' || Array.isArray(snapshot))
+        throw invalidSchema('An output JSON Schema object is required.');
+    if (Buffer.byteLength(JSON.stringify(snapshot)) > 65536)
+        throw invalidSchema('The output JSON Schema exceeds 64 KiB.');
+    if (snapshot.$schema !== undefined && snapshot.$schema !== draft)
+        throw invalidSchema('The output JSON Schema must use Draft-07.');
+    if ('$async' in snapshot)
+        throw invalidSchema('Asynchronous output JSON Schemas are not supported.');
+    const checkReferences = (node) => {
+        if (typeof node !== 'object' || node === null)
+            return;
+        if ('nullable' in node)
+            throw invalidSchema('Use a JSON Schema null type instead of nullable.');
+        if (node.$ref !== undefined && !node.$ref.startsWith('#'))
+            throw invalidSchema('Output JSON Schema references must be local.');
+        for (const map of [
+            node.properties,
+            node.patternProperties,
+            node.definitions,
+            node.dependencies,
+        ])
+            for (const child of Object.values(map ?? {}))
+                if (!Array.isArray(child))
+                    checkReferences(child);
+        for (const child of [
+            node.additionalProperties,
+            node.additionalItems,
+            node.contains,
+            node.propertyNames,
+            node.not,
+            node.if,
+            node.then,
+            node.else,
+            ...(Array.isArray(node.items) ? node.items : [node.items]),
+            ...(node.allOf ?? []),
+            ...(node.anyOf ?? []),
+            ...(node.oneOf ?? []),
+        ])
+            if (child !== undefined)
+                checkReferences(child);
+    };
+    try {
+        const ajv = new Ajv({
+            strictSchema: true,
+            strictTypes: false,
+            strictTuples: false,
+            ownProperties: true,
+            coerceTypes: false,
+            useDefaults: false,
+            removeAdditional: false,
+            code: { regExp: linearRegExp },
+        });
+        addFormats.default(ajv);
+        if (!ajv.validateSchema(snapshot))
+            throw new Error(ajv.errorsText());
+        checkReferences(snapshot);
+        const validate = ajv.compile(snapshot);
+        return jsonSchema(snapshot, {
+            validate: (value) => validate(value)
+                ? { success: true, value }
+                : {
+                    success: false,
+                    error: new Error('The model output does not match the JSON Schema.'),
+                },
+        });
+    }
+    catch (error) {
+        throw invalidSchema(`Invalid output JSON Schema: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}
+//# sourceMappingURL=validation.js.map

package/dist/server/online/agent/messages/validation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validation.js","sourceRoot":"","sources":["../../../../../server/online/agent/messages/validation.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,UAAU,GAGX,MAAM,IAAI,CAAC;AACZ,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACrC,OAAO,EAAE,GAAG,EAAoB,MAAM,KAAK,CAAC;AAC5C,OAAO,UAAU,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,KAAK,EAAE,MAAM,OAAO,CAAC;AAG9B,sFAAsF;AACtF,MAAM,UAAU,gBAAgB,CAAC,OAAqB;IACpD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM;QACzB,MAAM,OAAO,CAAC,mCAAmC,CAAC,CAAC;IACrD,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;QACrC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IACnD,OAAO,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAA8B,EAAE;QAC9D,MAAM,gBAAgB,GAAG,IAAI,CAAC,eAAe;YAC3C,CAAC,CAAE,QAAQ,CAAC,IAAI,CAAC,eAAe,CAE5B;YACJ,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YACtB,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAClD,CAAC;QACJ,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QAC1D,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC;QAC5C,IAAI,GAAW,CAAC;QAChB,IAAI,iBAGkB,CAAC;QACvB,MAAM,aAAa,GAAG,CAAC,KAAa,EAAU,EAAE,CAC9C,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,SAAS,WAAW,KAAK,EAAE,CAAC;QACpE,IAAI,IAAI,YAAY,GAAG,EAAE,CAAC;YACxB,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC;QAClB,CAAC;aAAM,IACL,OAAO,IAAI,KAAK,QAAQ;YACxB,IAAI,KAAK,IAAI;YACb,CAAC,CAAC,IAAI,YAAY,UAAU,CAAC;YAC7B,CAAC,CAAC,IAAI,YAAY,WAAW,CAAC,EAC9B,CAAC;YACD,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK;gBAAE,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;iBAC1D,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;gBAC7C,GAAG,GAAG,QAAQ,SAAS,WAAW,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;iBAC3E,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAChD,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACjB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;oBACtD,MAAM,OAAO,CACX,yDAAyD,CAC1D,CAAC;gBACJ,GAAG;oBACD,OAAO,IAAI,KAAK,QAAQ;wBACtB,CAAC,CAAC,QAAQ,SAAS,WAAW,IAAI,EAAE;wBACpC,CAAC,CAAC,aAAa,CACX,MAAM,CAAC,IAAI,CACT,IAAI,YAAY,WAAW,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAC1D,CAAC,QAAQ,CAAC,QAAQ,CAAC,CACrB,CAAC;YACV,CAAC;iBAAM,CAAC;gBACN,iBAAiB;oBACf,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;gBACtE,GAAG,GAAG,EAAE,CAAC;YACX,CAAC;QACH,CAAC;aAAM,CAAC;YACN,GAAG;gBACD,OAAO,IAAI,KAAK,QAAQ;oBACtB,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC;oBACrB,CAAC,CAAC,aAAa,CACX,MAAM,CAAC,IAAI,CACT,IAAI,YAAY,WAAW,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAC1D,CAAC,QAAQ,CAAC,QAAQ,CAAC,CACrB,CAAC;QACV,CAAC;QACD,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,SAAS;YACT,GAAG;YACH,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,QAAQ;gBACvC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE;gBAC7B,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAClD,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AACD,SAAS,aAAa,CAAC,OAAe;IACpC,OAAO,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,KAAK,GAAG,yCAAyC,CAAC;AACxD,MAAM,YAAY,GAAuC,MAAM,CAAC,MAAM,CACpE,CAAC,OAAe,EAAE,KAAa,EAAE,EAAE,CACjC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,EAClE,EAAE,IAAI,EAAE,cAAc,EAAE,CACzB,CAAC;AAEF,uFAAuF;AACvF,MAAM,UAAU,kBAAkB,CAChC,MAAmB;IAEnB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,MAAM,SAAS,GAAG,CAAC,KAAc,EAAE,KAAa,EAAQ,EAAE;QACxD,IAAI,EAAE,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE;YAC9B,MAAM,aAAa,CAAC,wCAAwC,CAAC,CAAC;QAChE,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;YAC7C,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;gBAAE,SAAS,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC;IACF,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACrB,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAgB,CAAC;IACjD,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;QACtE,MAAM,aAAa,CAAC,2CAA2C,CAAC,CAAC;IACnE,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,GAAG,KAAK;QACrD,MAAM,aAAa,CAAC,wCAAwC,CAAC,CAAC;IAChE,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,IAAI,QAAQ,CAAC,OAAO,KAAK,KAAK;QAC9D,MAAM,aAAa,CAAC,2CAA2C,CAAC,CAAC;IACnE,IAAI,QAAQ,IAAI,QAAQ;QACtB,MAAM,aAAa,CAAC,qDAAqD,CAAC,CAAC;IAE7E,MAAM,eAAe,GAAG,CAAC,IAA2B,EAAQ,EAAE;QAC5D,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI;YAAE,OAAO;QACtD,IAAI,UAAU,IAAI,IAAI;YACpB,MAAM,aAAa,CAAC,kDAAkD,CAAC,CAAC;QAC1E,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YACvD,MAAM,aAAa,CAAC,8CAA8C,CAAC,CAAC;QACtE,KAAK,MAAM,GAAG,IAAI;YAChB,IAAI,CAAC,UAAU;YACf,IAAI,CAAC,iBAAiB;YACtB,IAAI,CAAC,WAAW;YAChB,IAAI,CAAC,YAAY;SAClB;YACC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC;gBAC1C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;oBAAE,eAAe,CAAC,KAAK,CAAC,CAAC;QACtD,KAAK,MAAM,KAAK,IAAI;YAClB,IAAI,CAAC,oBAAoB;YACzB,IAAI,CAAC,eAAe;YACpB,IAAI,CAAC,QAAQ;YACb,IAAI,CAAC,aAAa;YAClB,IAAI,CAAC,GAAG;YACR,IAAI,CAAC,EAAE;YACP,IAAI,CAAC,IAAI;YACT,IAAI,CAAC,IAAI;YACT,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1D,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YACrB,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YACrB,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;SACtB;YACC,IAAI,KAAK,KAAK,SAAS;gBAAE,eAAe,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC;YAClB,YAAY,EAAE,IAAI;YAClB,WAAW,EAAE,KAAK;YAClB,YAAY,EAAE,KAAK;YACnB,aAAa,EAAE,IAAI;YACnB,WAAW,EAAE,KAAK;YAClB,WAAW,EAAE,KAAK;YAClB,gBAAgB,EAAE,KAAK;YACvB,IAAI,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE;SAC/B,CAAC,CAAC;QACH,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,QAAQ,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;QACrE,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC1B,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAkB,QAAQ,CAAC,CAAC;QACxD,OAAO,UAAU,CAAkB,QAAQ,EAAE;YAC3C,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAClB,QAAQ,CAAC,KAAK,CAAC;gBACb,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE;gBAC1B,CAAC,CAAC;oBACE,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,IAAI,KAAK,CACd,kDAAkD,CACnD;iBACF;SACR,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,aAAa,CACjB,+BAA+B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACxF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/server/online/agent/native/backend.d.ts

@@ -0,0 +1,60 @@
+import { NativeRealtimeOnlineAgent, NativeBackgroundOnlineAgent } from './session.js';
+import type { OnlineAgentBackend, OnlineAgentSessionManager } from '../contracts.js';
+import type { OnlineAgentUIStreams } from '../transport/streams.js';
+import type { OnlineAgentRealtimeOptions, OnlineAgentBoundary, OnlineAgentInitialConfiguration, OnlineAgentConfiguration, OnlineAgentContext, OnlineAgentConfig, OnlineAgentOptions } from '../types.js';
+import type { LanguageModel } from 'ai';
+import type { ServiceResolver } from '@nocobase/service-provider';
+import type { IdSource } from '../../../kernel/ids.js';
+import type { ModelRef } from '../../../../shared/models.js';
+import type { OnlineAgentConfigurator } from '../configuration.js';
+export declare class NativeOnlineAgentBackend implements OnlineAgentBackend {
+    private deps;
+    readonly kind: "native";
+    readonly configurator: OnlineAgentConfigurator;
+    private stopping;
+    private active;
+    private gates;
+    private native;
+    readonly uiStreams: OnlineAgentUIStreams;
+    constructor(deps: {
+        sessions: OnlineAgentSessionManager;
+        configurator: OnlineAgentConfigurator;
+        ids: IdSource;
+        resolver?: ServiceResolver;
+        model(ref: ModelRef, sessionId: string): Promise<LanguageModel>;
+        defaultModel(): Promise<ModelRef | null>;
+        uiStreams: OnlineAgentUIStreams;
+        onError(error: unknown): void;
+    });
+    shutdown(): Promise<void>;
+    /** @internal Native process occupancy; no session lookup or authorization bypass. */
+    hasLocalExecution(sessionId: string): boolean;
+    private gate;
+    private available;
+    createRealtime(context: OnlineAgentContext, initial: OnlineAgentInitialConfiguration, options: OnlineAgentRealtimeOptions, boundary?: OnlineAgentBoundary): Promise<NativeRealtimeOnlineAgent>;
+    restoreRealtime(context: OnlineAgentContext, options: OnlineAgentRealtimeOptions & {
+        sessionId: string;
+    }, boundary?: OnlineAgentBoundary): Promise<NativeRealtimeOnlineAgent>;
+    createBackground(context: OnlineAgentContext, initial: OnlineAgentInitialConfiguration, options: OnlineAgentRealtimeOptions, boundary?: OnlineAgentBoundary): Promise<NativeBackgroundOnlineAgent>;
+    restoreBackground(context: OnlineAgentContext, options: OnlineAgentRealtimeOptions & {
+        sessionId: string;
+    }, boundary?: OnlineAgentBoundary): Promise<NativeBackgroundOnlineAgent>;
+    private create;
+    private load;
+    private retryFinalization;
+    private abort;
+    private host;
+    configure(context: OnlineAgentContext, id: string, options: OnlineAgentOptions, selectedModel?: ModelRef, boundary?: 'http'): Promise<OnlineAgentConfig & {
+        selectedModel: ModelRef | null;
+    }>;
+    configuration(context: OnlineAgentContext, id: string): Promise<OnlineAgentConfiguration>;
+    private firstModel;
+    private validateModel;
+    private notifyObservers;
+    private observe;
+    private result;
+    private finish;
+    private approve;
+    private start;
+}
+//# sourceMappingURL=backend.d.ts.map

package/dist/server/online/agent/native/backend.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"backend.d.ts","sourceRoot":"","sources":["../../../../../server/online/agent/native/backend.ts"],"names":[],"mappings":"AACA,OAAO,EACL,yBAAyB,EACzB,2BAA2B,EAC5B,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EAEV,kBAAkB,EAClB,yBAAyB,EAC1B,MAAM,iBAAiB,CAAC;AACzB,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAGpE,OAAO,KAAK,EAEV,0BAA0B,EAC1B,mBAAmB,EACnB,+BAA+B,EAC/B,wBAAwB,EAExB,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EAGnB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAU,aAAa,EAAgB,MAAM,IAAI,CAAC;AAC9D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAElE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAoBnE,qBAAa,wBAAyB,YAAW,kBAAkB;IAS/D,OAAO,CAAC,IAAI;IARd,QAAQ,CAAC,IAAI,EAAG,QAAQ,CAAU;IAClC,QAAQ,CAAC,YAAY,EAAE,uBAAuB,CAAC;IAC/C,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,MAAM,CAA2C;IACzD,OAAO,CAAC,KAAK,CAAuC;IACpD,OAAO,CAAC,MAAM,CAAoB;IAClC,QAAQ,CAAC,SAAS,EAAE,oBAAoB,CAAC;gBAE/B,IAAI,EAAE;QACZ,QAAQ,EAAE,yBAAyB,CAAC;QACpC,YAAY,EAAE,uBAAuB,CAAC;QACtC,GAAG,EAAE,QAAQ,CAAC;QACd,QAAQ,CAAC,EAAE,eAAe,CAAC;QAC3B,KAAK,CAAC,GAAG,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;QAChE,YAAY,IAAI,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,CAAC;QACzC,SAAS,EAAE,oBAAoB,CAAC;QAChC,OAAO,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC;KAC/B;IAqCG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB/B,qFAAqF;IACrF,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;YAI/B,IAAI;IAUlB,OAAO,CAAC,SAAS;IAGX,cAAc,CAClB,OAAO,EAAE,kBAAkB,EAC3B,OAAO,EAAE,+BAA+B,EACxC,OAAO,EAAE,0BAA0B,EACnC,QAAQ,CAAC,EAAE,mBAAmB,GAC7B,OAAO,CAAC,yBAAyB,CAAC;IAI/B,eAAe,CACnB,OAAO,EAAE,kBAAkB,EAC3B,OAAO,EAAE,0BAA0B,GAAG;QAAE,SAAS,EAAE,MAAM,CAAA;KAAE,EAC3D,QAAQ,CAAC,EAAE,mBAAmB,GAC7B,OAAO,CAAC,yBAAyB,CAAC;IAU/B,gBAAgB,CACpB,OAAO,EAAE,kBAAkB,EAC3B,OAAO,EAAE,+BAA+B,EACxC,OAAO,EAAE,0BAA0B,EACnC,QAAQ,CAAC,EAAE,mBAAmB,GAC7B,OAAO,CAAC,2BAA2B,CAAC;IAIjC,iBAAiB,CACrB,OAAO,EAAE,kBAAkB,EAC3B,OAAO,EAAE,0BAA0B,GAAG;QAAE,SAAS,EAAE,MAAM,CAAA;KAAE,EAC3D,QAAQ,CAAC,EAAE,mBAAmB,GAC7B,OAAO,CAAC,2BAA2B,CAAC;YAUzB,MAAM;YA4BN,IAAI;YAmBJ,iBAAiB;YAgBjB,KAAK;IAgDnB,OAAO,CAAC,IAAI;IAaN,SAAS,CACb,OAAO,EAAE,kBAAkB,EAC3B,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,kBAAkB,EAC3B,aAAa,CAAC,EAAE,QAAQ,EACxB,QAAQ,CAAC,EAAE,MAAM,GAChB,OAAO,CAAC,iBAAiB,GAAG;QAAE,aAAa,EAAE,QAAQ,GAAG,IAAI,CAAA;KAAE,CAAC;IA4B5D,aAAa,CACjB,OAAO,EAAE,kBAAkB,EAC3B,EAAE,EAAE,MAAM,GACT,OAAO,CAAC,wBAAwB,CAAC;IAKpC,OAAO,CAAC,UAAU;IAUlB,OAAO,CAAC,aAAa;IAIrB,OAAO,CAAC,eAAe;YAQT,OAAO;IAmErB,OAAO,CAAC,MAAM;IA6Bd,OAAO,CAAC,MAAM;YAmBA,OAAO;YAgCP,KAAK;CAkEpB"}

package/dist/server/online/agent/native/backend.js

@@ -0,0 +1,411 @@
+import { nativeInputParts } from '../messages/validation.js';
+import { NativeRealtimeOnlineAgent, NativeBackgroundOnlineAgent, } from './session.js';
+import { NativeMessages } from '../messages/projection.js';
+import { NativeOnlineAgent } from './execution.js';
+import { conflict, invalid } from '../../../kernel/errors.js';
+import { jsonSafe } from '../messages/json.js';
+function acceptance() {
+    let resolve;
+    let reject;
+    const promise = new Promise((done, failed) => {
+        resolve = done;
+        reject = failed;
+    });
+    return { promise, resolve, reject };
+}
+export class NativeOnlineAgentBackend {
+    deps;
+    kind = 'native';
+    configurator;
+    stopping = false;
+    active = new Map();
+    gates = new Map();
+    native;
+    uiStreams;
+    constructor(deps) {
+        this.deps = deps;
+        this.configurator = deps.configurator;
+        this.uiStreams = deps.uiStreams;
+        deps.sessions.native.messages = new NativeMessages(this.configurator, deps.resolver);
+        this.native = new NativeOnlineAgent({
+            ...deps,
+            configurator: this.configurator,
+            finish: (execution) => {
+                const current = [...this.active.values()].find((value) => value === execution);
+                return current ? this.finish(current) : undefined;
+            },
+            complete: (id, execution, result) => {
+                const current = this.active.get(id);
+                if (current !== execution)
+                    return;
+                this.active.delete(id);
+                if (result && current) {
+                    try {
+                        this.notifyObservers(current, {
+                            result: this.result(result.outcome, result.data, result.outputInvalid),
+                        });
+                    }
+                    catch (error) {
+                        this.notifyObservers(current, { error });
+                    }
+                }
+            },
+        });
+    }
+    async shutdown() {
+        this.stopping = true;
+        const executions = [...this.active.values()];
+        executions.forEach((execution) => execution.abort.abort());
+        const results = await Promise.allSettled(executions.map(async (execution) => {
+            await execution.task;
+            await this.finish(execution);
+        }));
+        const failures = results.flatMap((result) => result.status === 'rejected' ? [result.reason] : []);
+        if (failures.length)
+            console.warn('OnlineAgent finalization failed during shutdown.', new AggregateError(failures));
+    }
+    /** @internal Native process occupancy; no session lookup or authorization bypass. */
+    hasLocalExecution(sessionId) {
+        return this.active.has(sessionId);
+    }
+    async gate(id, work) {
+        const previous = this.gates.get(id) ?? Promise.resolve();
+        const next = previous.catch(() => { }).then(work);
+        this.gates.set(id, next);
+        try {
+            return await next;
+        }
+        finally {
+            if (this.gates.get(id) === next)
+                this.gates.delete(id);
+        }
+    }
+    available() {
+        if (this.stopping)
+            throw conflict('OnlineAgent is stopping.');
+    }
+    async createRealtime(context, initial, options, boundary) {
+        const id = await this.create(context, 'realtime', initial, options);
+        return this.restoreRealtime(context, { sessionId: id }, boundary);
+    }
+    async restoreRealtime(context, options, boundary) {
+        const loaded = await this.load(context, 'realtime', options);
+        return new NativeRealtimeOnlineAgent(this.host(), context, loaded, options, boundary);
+    }
+    async createBackground(context, initial, options, boundary) {
+        const id = await this.create(context, 'background', initial, options);
+        return this.restoreBackground(context, { sessionId: id }, boundary);
+    }
+    async restoreBackground(context, options, boundary) {
+        const loaded = await this.load(context, 'background', options);
+        return new NativeBackgroundOnlineAgent(this.host(), context, loaded, options, boundary);
+    }
+    async create(context, kind, initial, options) {
+        this.available();
+        const config = structuredClone(initial.config);
+        if (config.agentId && config.agentId !== 'online')
+            throw invalid('This configuration belongs to a Runner backend.');
+        if (!initial.hasModel) {
+            const fallback = options.selectedModel ??
+                initial.defaultModel ??
+                (await this.deps.defaultModel());
+            if (fallback)
+                config.model = [`${fallback.modelService}/${fallback.model}`];
+        }
+        const model = options.selectedModel ?? this.firstModel(config.model);
+        this.validateModel(config.model, model);
+        await this.deps.model(model, '');
+        this.available();
+        return this.deps.sessions.create(context.user.id, kind, config, model, {
+            source: options.source,
+            title: options.title,
+            showInChatList: options.showInChatList,
+        });
+    }
+    async load(context, kind, options) {
+        this.available();
+        const loaded = await this.deps.sessions.load(context.user.id, options.sessionId);
+        if (loaded.kind !== kind)
+            throw invalid('This session has a different execution kind.');
+        if (options.agentId &&
+            options.agentId !== (loaded.config.agentId ?? 'online'))
+            throw conflict('Start a new conversation to choose another agent.');
+        return loaded;
+    }
+    async retryFinalization(context, id) {
+        const current = await this.deps.sessions.load(context.user.id, id);
+        const execution = this.active.get(id);
+        if (execution) {
+            await execution.task;
+            await this.finish(execution);
+        }
+        else if (current.call?.state === 'running' || current.call?.closing) {
+            throw conflict('Finalization requires the live producer that owns its snapshot; reconcile history and execution.');
+        }
+    }
+    async abort(context, id, kind) {
+        if (kind === 'background') {
+            await this.deps.sessions.load(context.user.id, id);
+            const execution = await this.gate(id, async () => {
+                const current = this.active.get(id);
+                current?.abort.abort();
+                return current;
+            });
+            if (execution) {
+                await execution.task;
+                await this.finish(execution);
+            }
+            return;
+        }
+        // The facade was authorized when opened. Signal before asynchronous reads.
+        const execution = this.active.get(id);
+        if (execution) {
+            execution.abort.abort();
+            await execution.task;
+            await this.finish(execution);
+            return;
+        }
+        const current = await this.deps.sessions.load(context.user.id, id);
+        if (current.call?.state === 'waitingApproval') {
+            await this.deps.sessions.native.stop(context.user.id, id);
+            return;
+        }
+        if (current.call?.state === 'running' && !this.active.has(id)) {
+            if (current.call.executionId &&
+                (await this.uiStreams.requestStop(context.user.id, id, current.call.executionId)))
+                return;
+            throw conflict('The producer is unavailable; stopping has not been confirmed.');
+        }
+        await this.deps.sessions.load(context.user.id, id);
+    }
+    host() {
+        return {
+            sessions: this.deps.sessions,
+            native: this.native,
+            uiStreams: this.uiStreams,
+            isActive: (id) => this.hasLocalExecution(id),
+            abort: (...args) => this.abort(...args),
+            retryFinalization: (...args) => this.retryFinalization(...args),
+            start: (...args) => this.start(...args),
+            approve: (...args) => this.approve(...args),
+            observe: (...args) => this.observe(...args),
+        };
+    }
+    async configure(context, id, options, selectedModel, boundary) {
+        this.available();
+        return this.gate(id, async () => {
+            this.available();
+            if (this.active.has(id))
+                throw conflict('This conversation is busy.');
+            const session = await this.deps.sessions.load(context.user.id, id);
+            const config = await this.configurator.resolve(context, options, session.config, [], boundary === 'http');
+            if ((config.agentId ?? 'online') !== (session.config.agentId ?? 'online'))
+                throw conflict('Start a new conversation to choose another agent.');
+            const model = selectedModel ??
+                (options.model !== undefined ||
+                    JSON.stringify(config.model) !== JSON.stringify(session.config.model)
+                    ? this.firstModel(config.model)
+                    : session.model);
+            this.validateModel(config.model, model);
+            await this.deps.model(model, id);
+            this.available();
+            await this.deps.sessions.configure(context.user.id, id, config, model);
+            return { ...config, selectedModel: model };
+        });
+    }
+    async configuration(context, id) {
+        this.available();
+        const session = await this.deps.sessions.load(context.user.id, id);
+        return { ...session.config, selectedModel: session.model };
+    }
+    firstModel(models) {
+        const value = models[0];
+        const separator = value?.indexOf('/') ?? -1;
+        if (separator < 1 || separator === value.length - 1)
+            throw invalid('The selected model is outside this configuration.');
+        return {
+            modelService: value.slice(0, separator),
+            model: value.slice(separator + 1),
+        };
+    }
+    validateModel(allowed, model) {
+        if (!allowed.includes(`${model.modelService}/${model.model}`))
+            throw invalid('The selected model is outside this configuration.');
+    }
+    notifyObservers(execution, value) {
+        execution.observation = value;
+        for (const observer of execution.observers)
+            observer(value);
+        execution.observers.clear();
+    }
+    async observe(userId, id, wait) {
+        // Capture before the ownership read: an earlier waiter must never follow a later turn.
+        const execution = this.active.get(id);
+        const admitted = execution?.admitted;
+        const started = performance.now();
+        const row = await (execution
+            ? this.deps.sessions.load(userId, id)
+            : this.gate(id, async () => {
+                if (this.active.has(id))
+                    throw conflict('The execution changed while reading its result. Read again.');
+                return this.deps.sessions.load(userId, id);
+            }));
+        let observed;
+        if (execution) {
+            if (!admitted)
+                throw conflict('No accepted execution is available.');
+            if (execution.observation)
+                observed = execution.observation;
+            else {
+                const remaining = wait === undefined
+                    ? undefined
+                    : Math.max(0, wait - (performance.now() - started));
+                if (remaining === 0)
+                    return { ok: false, reason: 'pending' };
+                observed = await new Promise((resolve) => {
+                    let timer;
+                    const done = (value) => {
+                        clearTimeout(timer);
+                        execution.observers.delete(done);
+                        resolve(value);
+                    };
+                    execution.observers.add(done);
+                    if (remaining !== undefined) {
+                        // Avoid Node's overflow-to-1ms behavior for valid large budgets.
+                        const deadline = performance.now() + remaining;
+                        const expire = () => {
+                            const left = deadline - performance.now();
+                            if (left <= 0)
+                                done({ result: { ok: false, reason: 'pending' } });
+                            else
+                                timer = setTimeout(expire, Math.min(left, 2_147_483_647));
+                        };
+                        timer = setTimeout(expire, Math.min(remaining, 2_147_483_647));
+                    }
+                });
+            }
+        }
+        else {
+            if (!row.call)
+                throw conflict('No accepted execution is available.');
+            if (row.call.state === 'running' || row.call.closing)
+                throw conflict('This execution is not owned by the current server process.');
+            observed = { result: this.result(row.call.outcome, row.call.result) };
+        }
+        if ('error' in observed)
+            throw observed.error;
+        // Each observer owns its returned value; mutations cannot affect other waiters.
+        return observed.result.ok
+            ? { ok: true, data: jsonSafe(observed.result.data) }
+            : {
+                ...observed.result,
+                ...(observed.result.reason === 'pending'
+                    ? {}
+                    : { error: { ...observed.result.error } }),
+            };
+    }
+    result(outcome, data, outputInvalid = false) {
+        if (outcome === 'completed' && data !== undefined)
+            return { ok: true, data };
+        if (outcome === 'cancelled')
+            return {
+                ok: false,
+                reason: 'cancelled',
+                error: {
+                    code: 'EXECUTION_CANCELLED',
+                    message: 'The execution was cancelled.',
+                },
+            };
+        if (outcome === 'failed')
+            return {
+                ok: false,
+                reason: 'failed',
+                error: {
+                    code: outputInvalid ? 'OUTPUT_INVALID' : 'EXECUTION_FAILED',
+                    message: outputInvalid
+                        ? 'The model output does not match the JSON Schema.'
+                        : 'The execution failed.',
+                },
+            };
+        throw conflict('No successful structured result is available.');
+    }
+    finish(execution) {
+        if (!execution.finalizing && execution.finalize) {
+            execution.observation = undefined;
+            execution.finalizing = execution
+                .finalize()
+                .catch((error) => {
+                this.notifyObservers(execution, {
+                    error: conflict('The execution could not be finalized. Retry finalization before reading its result.'),
+                });
+                throw error;
+            })
+                .finally(() => {
+                execution.finalizing = undefined;
+            });
+        }
+        return execution.finalizing;
+    }
+    async approve(context, id, input, onExecution) {
+        this.available();
+        const accepted = acceptance();
+        const execution = {
+            abort: new AbortController(),
+            observers: new Set(),
+        };
+        await this.gate(id, async () => {
+            if (this.stopping)
+                throw conflict('OnlineAgent is stopping.');
+            if (this.active.size >= 1024)
+                throw conflict('OnlineAgent execution limit exceeded.');
+            if (this.active.has(id))
+                throw conflict('This conversation is busy.');
+            this.active.set(id, execution);
+            execution.task = this.native.run(context, id, [], undefined, execution, accepted, input);
+        });
+        const seq = await accepted.promise;
+        onExecution?.(execution);
+        return seq;
+    }
+    async start(context, id, messages, input, schema, onExecution) {
+        if (!messages.length || messages.some((message) => message.role !== 'user'))
+            throw invalid('Input must contain user messages.');
+        // Normalize URL/binary SDK inputs before reserving occupancy; unsupported inputs leave no live producer.
+        const inputs = structuredClone(messages.map((message, index) => ({
+            id: index === 0 && input?.messageId
+                ? input.messageId
+                : this.deps.ids.next(),
+            parts: nativeInputParts(message),
+            attachments: input?.attachmentIds?.[index] ?? [],
+            clientId: input?.clientId,
+            workContext: input?.workContext,
+            ...(message.providerOptions
+                ? {
+                    providerOptions: jsonSafe(message.providerOptions),
+                }
+                : {}),
+        })));
+        await this.deps.sessions.load(context.user.id, id);
+        if (this.stopping)
+            throw conflict('OnlineAgent is stopping.');
+        const execution = {
+            abort: new AbortController(),
+            observers: new Set(),
+        };
+        const accepted = acceptance();
+        await this.gate(id, async () => {
+            if (this.stopping)
+                throw conflict('OnlineAgent is stopping.');
+            if (this.active.has(id))
+                throw conflict('This conversation is busy.');
+            if (this.active.size >= 1024)
+                throw conflict('OnlineAgent execution limit exceeded.');
+            this.active.set(id, execution);
+            execution.task = this.native.run(context, id, inputs, input, execution, accepted, undefined, schema);
+        });
+        const seq = await accepted.promise;
+        onExecution?.(execution);
+        return seq;
+    }
+}
+//# sourceMappingURL=backend.js.map