experimental-ash 0.26.0 → 0.28.0

package/dist/src/public/channels/ash.js

@@ -1,346 +1,2 @@
-import {} from "ai";
-import { parseSessionCallback } from "#channel/session-callback.js";
-import { ASH_MESSAGE_STREAM_CONTENT_TYPE, ASH_MESSAGE_STREAM_FORMAT, ASH_MESSAGE_STREAM_VERSION, ASH_SESSION_ID_HEADER, ASH_STREAM_FORMAT_HEADER, ASH_STREAM_VERSION_HEADER, } from "#protocol/message.js";
-import { isInputResponse } from "#runtime/input/types.js";
-import { createUnauthorizedResponse } from "#public/channels/auth.js";
-import { collectUploadPolicyViolations, formatUploadPolicyViolation, mergeUploadPolicy, } from "#public/channels/upload-policy.js";
-import { defineChannel, POST, GET } from "#public/definitions/defineChannel.js";
-import { parseJsonObject } from "#shared/json.js";
-export function ashChannel(input) {
-    const uploadPolicy = mergeUploadPolicy(input.uploadPolicy);
-    return defineChannel({
-        routes: [
-            POST("/ash/v1/session", async (req, { send }) => {
-                const authResult = await runRouteAuth(input.auth, req);
-                if (authResult instanceof Response)
-                    return authResult;
-                const sessionAuth = authResult;
-                let payload;
-                try {
-                    payload = await req.json();
-                }
-                catch {
-                    return Response.json({ error: "Invalid JSON body.", ok: false }, { status: 400 });
-                }
-                if (payload === null || typeof payload !== "object") {
-                    return Response.json({ error: "Expected a JSON object.", ok: false }, { status: 400 });
-                }
-                const body = parseCreateBody(payload);
-                if (body instanceof Response)
-                    return body;
-                const policyRejection = checkUploadPolicy(body, uploadPolicy);
-                if (policyRejection !== null)
-                    return policyRejection;
-                const token = `ash:${crypto.randomUUID()}`;
-                const session = await send(createSendPayload(body), {
-                    auth: sessionAuth,
-                    callback: body.callback,
-                    continuationToken: token,
-                    mode: body.mode,
-                });
-                return Response.json({
-                    continuationToken: session.continuationToken,
-                    ok: true,
-                    sessionId: session.id,
-                }, {
-                    headers: {
-                        "cache-control": "no-store",
-                        [ASH_SESSION_ID_HEADER]: session.id,
-                    },
-                    status: 202,
-                });
-            }),
-            POST("/ash/v1/session/:sessionId", async (req, { send, getSession, params }) => {
-                const authResult = await runRouteAuth(input.auth, req);
-                if (authResult instanceof Response)
-                    return authResult;
-                const sessionAuth = authResult;
-                const sessionId = params.sessionId;
-                if (!sessionId) {
-                    return Response.json({ error: "Missing session id.", ok: false }, { status: 400 });
-                }
-                try {
-                    getSession(sessionId);
-                }
-                catch {
-                    return Response.json({ error: "Session not found.", ok: false }, { status: 404 });
-                }
-                let payload;
-                try {
-                    payload = await req.json();
-                }
-                catch {
-                    return Response.json({ error: "Invalid JSON body.", ok: false }, { status: 400 });
-                }
-                if (payload === null || typeof payload !== "object") {
-                    return Response.json({ error: "Expected a JSON object.", ok: false }, { status: 400 });
-                }
-                const body = parseContinueBody(payload);
-                if (body instanceof Response)
-                    return body;
-                const policyRejection = checkUploadPolicy(body, uploadPolicy);
-                if (policyRejection !== null)
-                    return policyRejection;
-                const session = await send({
-                    inputResponses: body.inputResponses,
-                    message: body.message,
-                    modelContext: body.modelContext,
-                }, {
-                    auth: sessionAuth,
-                    continuationToken: body.continuationToken,
-                });
-                return Response.json({
-                    ok: true,
-                    sessionId: session.id,
-                }, {
-                    headers: {
-                        "cache-control": "no-store",
-                        [ASH_SESSION_ID_HEADER]: session.id,
-                    },
-                    status: 200,
-                });
-            }),
-            GET("/ash/v1/session/:sessionId/stream", async (req, { getSession, params }) => {
-                const authResult = await runRouteAuth(input.auth, req);
-                if (authResult instanceof Response)
-                    return authResult;
-                const sessionId = params.sessionId;
-                if (!sessionId) {
-                    return Response.json({ error: "Missing session id.", ok: false }, { status: 400 });
-                }
-                const startIndex = parseStartIndex(req);
-                if (startIndex instanceof Response)
-                    return startIndex;
-                try {
-                    const session = getSession(sessionId);
-                    const events = await session.getEventStream({ startIndex });
-                    const ndjson = serializeAsNdjson(events);
-                    return new Response(ndjson, {
-                        headers: {
-                            "cache-control": "no-store",
-                            "content-type": ASH_MESSAGE_STREAM_CONTENT_TYPE,
-                            [ASH_SESSION_ID_HEADER]: sessionId,
-                            [ASH_STREAM_FORMAT_HEADER]: ASH_MESSAGE_STREAM_FORMAT,
-                            [ASH_STREAM_VERSION_HEADER]: ASH_MESSAGE_STREAM_VERSION,
-                        },
-                    });
-                }
-                catch {
-                    return Response.json({ error: "Session not found.", ok: false }, { status: 404 });
-                }
-            }),
-        ],
-    });
-}
-async function runRouteAuth(auth, request) {
-    const result = await auth(request);
-    if (result === null) {
-        return createUnauthorizedResponse({ challenges: [{ scheme: "Bearer" }] });
-    }
-    return result ?? null;
-}
-function parseCreateBody(payload) {
-    const message = parseMessageField(payload.message);
-    if (message instanceof Response)
-        return message;
-    const modelContext = parseClientContextField(payload.clientContext);
-    if (modelContext instanceof Response)
-        return modelContext;
-    const callback = parseCallbackField(payload.callback);
-    if (callback instanceof Response)
-        return callback;
-    const mode = parseModeField(payload.mode);
-    if (mode instanceof Response)
-        return mode;
-    if (message === undefined) {
-        return Response.json({ error: "Missing or empty 'message' field.", ok: false }, { status: 400 });
-    }
-    return { callback, message, mode, modelContext };
-}
-function parseContinueBody(payload) {
-    const continuationToken = typeof payload.continuationToken === "string" && payload.continuationToken.length > 0
-        ? payload.continuationToken
-        : undefined;
-    if (continuationToken === undefined) {
-        return Response.json({ error: "Missing or empty 'continuationToken' field.", ok: false }, { status: 400 });
-    }
-    const message = parseMessageField(payload.message);
-    if (message instanceof Response)
-        return message;
-    const inputResponses = parseInputResponses(payload.inputResponses);
-    if (inputResponses instanceof Response)
-        return inputResponses;
-    const modelContext = parseClientContextField(payload.clientContext);
-    if (modelContext instanceof Response)
-        return modelContext;
-    if (message === undefined && inputResponses === undefined) {
-        return Response.json({
-            error: "Expected a non-empty 'message', a non-empty 'inputResponses' array, or both.",
-            ok: false,
-        }, { status: 400 });
-    }
-    return { message, continuationToken, inputResponses, modelContext };
-}
-function createSendPayload(body) {
-    if (body.modelContext === undefined) {
-        return body.message;
-    }
-    return { message: body.message, modelContext: body.modelContext };
-}
-function parseCallbackField(value) {
-    if (value === undefined)
-        return undefined;
-    const parsed = parseSessionCallback(value);
-    if (parsed.ok)
-        return parsed.callback;
-    return Response.json({ error: parsed.message, ok: false }, { status: 400 });
-}
-function parseModeField(value) {
-    if (value === undefined)
-        return undefined;
-    if (value === "conversation" || value === "task")
-        return value;
-    return Response.json({ error: "Expected 'mode' to be either 'conversation' or 'task'.", ok: false }, { status: 400 });
-}
-function parseMessageField(value) {
-    if (value === undefined)
-        return undefined;
-    if (typeof value === "string")
-        return value.length > 0 ? value : undefined;
-    if (!Array.isArray(value)) {
-        return Response.json({ error: "Expected 'message' to be a string or an array of text/file parts.", ok: false }, { status: 400 });
-    }
-    if (value.length === 0)
-        return undefined;
-    const parts = [];
-    for (const raw of value) {
-        const parsed = parseMessagePart(raw);
-        if (parsed instanceof Response)
-            return parsed;
-        parts.push(parsed);
-    }
-    return parts;
-}
-function parseMessagePart(raw) {
-    if (raw === null || typeof raw !== "object") {
-        return Response.json({ error: "Expected each message part to be an object.", ok: false }, { status: 400 });
-    }
-    const part = raw;
-    if (part.type === "text") {
-        if (typeof part.text !== "string" || part.text.length === 0) {
-            return Response.json({ error: "Text parts require a non-empty 'text' string.", ok: false }, { status: 400 });
-        }
-        return { type: "text", text: part.text };
-    }
-    if (part.type === "file") {
-        if (typeof part.mediaType !== "string" || part.mediaType.length === 0) {
-            return Response.json({ error: "File parts require a non-empty 'mediaType' string.", ok: false }, { status: 400 });
-        }
-        if (typeof part.data !== "string") {
-            return Response.json({ error: "File parts require a 'data' string (base64, data URL, or URL).", ok: false }, { status: 400 });
-        }
-        const filePart = { type: "file", mediaType: part.mediaType, data: part.data };
-        if (typeof part.filename === "string" && part.filename.length > 0) {
-            filePart.filename = part.filename;
-        }
-        return filePart;
-    }
-    return Response.json({
-        error: `Unsupported message part type "${String(part.type)}". Use 'text' or 'file'.`,
-        ok: false,
-    }, { status: 400 });
-}
-function checkUploadPolicy(body, policy) {
-    if (!body.message)
-        return null;
-    const violations = collectUploadPolicyViolations(body.message, policy);
-    if (violations.length === 0)
-        return null;
-    const [first] = violations;
-    if (!first)
-        return null;
-    const status = first.kind === "too-large" ? 413 : 415;
-    return Response.json({
-        error: formatUploadPolicyViolation(first),
-        ok: false,
-        violations: violations.map((v) => v.kind === "too-large"
-            ? {
-                byteLength: v.byteLength,
-                filename: v.filename,
-                kind: v.kind,
-                limit: v.limit,
-                mediaType: v.mediaType,
-            }
-            : {
-                allowedMediaTypes: v.allowedMediaTypes,
-                filename: v.filename,
-                kind: v.kind,
-                mediaType: v.mediaType,
-            }),
-    }, { status });
-}
-function parseInputResponses(value) {
-    if (value === undefined)
-        return undefined;
-    if (!Array.isArray(value) || value.length === 0) {
-        return Response.json({ error: "Expected 'inputResponses' to be a non-empty array.", ok: false }, { status: 400 });
-    }
-    const inputResponses = value.filter(isInputResponse);
-    if (inputResponses.length !== value.length) {
-        return Response.json({
-            error: "Expected every 'inputResponses' entry to match the HITL response schema.",
-            ok: false,
-        }, { status: 400 });
-    }
-    return inputResponses;
-}
-const CLIENT_CONTEXT_PREFIX = "Ephemeral client context:\n";
-function parseClientContextField(value) {
-    if (value === undefined)
-        return undefined;
-    if (typeof value === "string") {
-        return value.length > 0 ? [toClientContextMessage(value)] : undefined;
-    }
-    if (Array.isArray(value)) {
-        if (value.length === 0)
-            return undefined;
-        if (!value.every((entry) => typeof entry === "string" && entry.length > 0)) {
-            return Response.json({ error: "Expected 'clientContext' array entries to be non-empty strings.", ok: false }, { status: 400 });
-        }
-        return value.map((entry) => toClientContextMessage(entry));
-    }
-    if (value === null || typeof value !== "object") {
-        return Response.json({
-            error: "Expected 'clientContext' to be a string, string array, or JSON object.",
-            ok: false,
-        }, { status: 400 });
-    }
-    try {
-        const json = parseJsonObject(value);
-        return [toClientContextMessage(JSON.stringify(json))];
-    }
-    catch {
-        return Response.json({ error: "Expected 'clientContext' to be a JSON-serializable object.", ok: false }, { status: 400 });
-    }
-}
-function toClientContextMessage(content) {
-    return { role: "user", content: `${CLIENT_CONTEXT_PREFIX}${content}` };
-}
-function parseStartIndex(request) {
-    const raw = new URL(request.url).searchParams.get("startIndex");
-    if (raw === null)
-        return undefined;
-    const parsed = Number.parseInt(raw, 10);
-    if (!Number.isSafeInteger(parsed) || parsed < 0) {
-        return Response.json({ error: "Expected startIndex to be a non-negative integer.", ok: false }, { status: 400 });
-    }
-    return parsed;
-}
-function serializeAsNdjson(events) {
-    const encoder = new TextEncoder();
-    return events.pipeThrough(new TransformStream({
-        transform(event, controller) {
-            controller.enqueue(encoder.encode(`${JSON.stringify(event)}\n`));
-        },
-    }));
-}
+import{ASH_MESSAGE_STREAM_CONTENT_TYPE,ASH_MESSAGE_STREAM_FORMAT,ASH_MESSAGE_STREAM_VERSION,ASH_SESSION_ID_HEADER,ASH_STREAM_FORMAT_HEADER,ASH_STREAM_VERSION_HEADER}from"#protocol/message.js";import{isInputResponse}from"#runtime/input/types.js";import{parseJsonObject}from"#shared/json.js";import"ai";import{parseSessionCallback}from"#channel/session-callback.js";import{createUnauthorizedResponse}from"#public/channels/auth.js";import{collectUploadPolicyViolations,formatUploadPolicyViolation,mergeUploadPolicy}from"#public/channels/upload-policy.js";import{GET,POST,defineChannel}from"#public/definitions/defineChannel.js";function ashChannel(o){let s=mergeUploadPolicy(o.uploadPolicy);return defineChannel({routes:[POST(`/ash/v1/session`,async(e,{send:t})=>{let n=await runRouteAuth(o.auth,e);if(n instanceof Response)return n;let i=n,a;try{a=await e.json()}catch{return Response.json({error:`Invalid JSON body.`,ok:!1},{status:400})}if(typeof a!=`object`||!a)return Response.json({error:`Expected a JSON object.`,ok:!1},{status:400});let c=parseCreateBody(a);if(c instanceof Response)return c;let l=checkUploadPolicy(c,s);if(l!==null)return l;let u=`ash:${crypto.randomUUID()}`,d=await t(createSendPayload(c),{auth:i,callback:c.callback,continuationToken:u,mode:c.mode});return Response.json({continuationToken:d.continuationToken,ok:!0,sessionId:d.id},{headers:{"cache-control":`no-store`,[ASH_SESSION_ID_HEADER]:d.id},status:202})}),POST(`/ash/v1/session/:sessionId`,async(e,{send:t,getSession:n,params:i})=>{let a=await runRouteAuth(o.auth,e);if(a instanceof Response)return a;let c=a,l=i.sessionId;if(!l)return Response.json({error:`Missing session id.`,ok:!1},{status:400});try{n(l)}catch{return Response.json({error:`Session not found.`,ok:!1},{status:404})}let u;try{u=await e.json()}catch{return Response.json({error:`Invalid JSON body.`,ok:!1},{status:400})}if(typeof u!=`object`||!u)return Response.json({error:`Expected a JSON object.`,ok:!1},{status:400});let d=parseContinueBody(u);if(d instanceof Response)return d;let f=checkUploadPolicy(d,s);if(f!==null)return f;let p=await t({inputResponses:d.inputResponses,message:d.message,modelContext:d.modelContext},{auth:c,continuationToken:d.continuationToken});return Response.json({ok:!0,sessionId:p.id},{headers:{"cache-control":`no-store`,[ASH_SESSION_ID_HEADER]:p.id},status:200})}),GET(`/ash/v1/session/:sessionId/stream`,async(s,{getSession:c,params:l})=>{let u=await runRouteAuth(o.auth,s);if(u instanceof Response)return u;let d=l.sessionId;if(!d)return Response.json({error:`Missing session id.`,ok:!1},{status:400});let f=parseStartIndex(s);if(f instanceof Response)return f;try{let o=serializeAsNdjson(await c(d).getEventStream({startIndex:f}));return new Response(o,{headers:{"cache-control":`no-store`,"content-type":ASH_MESSAGE_STREAM_CONTENT_TYPE,[ASH_SESSION_ID_HEADER]:d,[ASH_STREAM_FORMAT_HEADER]:ASH_MESSAGE_STREAM_FORMAT,[ASH_STREAM_VERSION_HEADER]:ASH_MESSAGE_STREAM_VERSION}})}catch{return Response.json({error:`Session not found.`,ok:!1},{status:404})}})]})}async function runRouteAuth(e,t){let n=await e(t);return n===null?createUnauthorizedResponse({challenges:[{scheme:`Bearer`}]}):n??null}function parseCreateBody(e){let t=parseMessageField(e.message);if(t instanceof Response)return t;let n=parseClientContextField(e.clientContext);if(n instanceof Response)return n;let r=parseCallbackField(e.callback);if(r instanceof Response)return r;let i=parseModeField(e.mode);return i instanceof Response?i:t===void 0?Response.json({error:`Missing or empty 'message' field.`,ok:!1},{status:400}):{callback:r,message:t,mode:i,modelContext:n}}function parseContinueBody(e){let t=typeof e.continuationToken==`string`&&e.continuationToken.length>0?e.continuationToken:void 0;if(t===void 0)return Response.json({error:`Missing or empty 'continuationToken' field.`,ok:!1},{status:400});let n=parseMessageField(e.message);if(n instanceof Response)return n;let r=parseInputResponses(e.inputResponses);if(r instanceof Response)return r;let i=parseClientContextField(e.clientContext);return i instanceof Response?i:n===void 0&&r===void 0?Response.json({error:`Expected a non-empty 'message', a non-empty 'inputResponses' array, or both.`,ok:!1},{status:400}):{message:n,continuationToken:t,inputResponses:r,modelContext:i}}function createSendPayload(e){return e.modelContext===void 0?e.message:{message:e.message,modelContext:e.modelContext}}function parseCallbackField(e){if(e===void 0)return;let t=parseSessionCallback(e);return t.ok?t.callback:Response.json({error:t.message,ok:!1},{status:400})}function parseModeField(e){if(e!==void 0)return e===`conversation`||e===`task`?e:Response.json({error:`Expected 'mode' to be either 'conversation' or 'task'.`,ok:!1},{status:400})}function parseMessageField(e){if(e===void 0)return;if(typeof e==`string`)return e.length>0?e:void 0;if(!Array.isArray(e))return Response.json({error:`Expected 'message' to be a string or an array of text/file parts.`,ok:!1},{status:400});if(e.length===0)return;let t=[];for(let n of e){let e=parseMessagePart(n);if(e instanceof Response)return e;t.push(e)}return t}function parseMessagePart(e){if(typeof e!=`object`||!e)return Response.json({error:`Expected each message part to be an object.`,ok:!1},{status:400});let t=e;if(t.type===`text`)return typeof t.text!=`string`||t.text.length===0?Response.json({error:`Text parts require a non-empty 'text' string.`,ok:!1},{status:400}):{type:`text`,text:t.text};if(t.type===`file`){if(typeof t.mediaType!=`string`||t.mediaType.length===0)return Response.json({error:`File parts require a non-empty 'mediaType' string.`,ok:!1},{status:400});if(typeof t.data!=`string`)return Response.json({error:`File parts require a 'data' string (base64, data URL, or URL).`,ok:!1},{status:400});let e={type:`file`,mediaType:t.mediaType,data:t.data};return typeof t.filename==`string`&&t.filename.length>0&&(e.filename=t.filename),e}return Response.json({error:`Unsupported message part type "${String(t.type)}". Use 'text' or 'file'.`,ok:!1},{status:400})}function checkUploadPolicy(e,t){if(!e.message)return null;let n=collectUploadPolicyViolations(e.message,t);if(n.length===0)return null;let[r]=n;if(!r)return null;let i=r.kind===`too-large`?413:415;return Response.json({error:formatUploadPolicyViolation(r),ok:!1,violations:n.map(e=>e.kind===`too-large`?{byteLength:e.byteLength,filename:e.filename,kind:e.kind,limit:e.limit,mediaType:e.mediaType}:{allowedMediaTypes:e.allowedMediaTypes,filename:e.filename,kind:e.kind,mediaType:e.mediaType})},{status:i})}function parseInputResponses(e){if(e===void 0)return;if(!Array.isArray(e)||e.length===0)return Response.json({error:`Expected 'inputResponses' to be a non-empty array.`,ok:!1},{status:400});let t=e.filter(isInputResponse);return t.length===e.length?t:Response.json({error:`Expected every 'inputResponses' entry to match the HITL response schema.`,ok:!1},{status:400})}var CLIENT_CONTEXT_PREFIX=`Ephemeral client context:
+`;function parseClientContextField(e){if(e!==void 0){if(typeof e==`string`)return e.length>0?[toClientContextMessage(e)]:void 0;if(Array.isArray(e))return e.length===0?void 0:e.every(e=>typeof e==`string`&&e.length>0)?e.map(e=>toClientContextMessage(e)):Response.json({error:`Expected 'clientContext' array entries to be non-empty strings.`,ok:!1},{status:400});if(typeof e!=`object`||!e)return Response.json({error:`Expected 'clientContext' to be a string, string array, or JSON object.`,ok:!1},{status:400});try{let t=parseJsonObject(e);return[toClientContextMessage(JSON.stringify(t))]}catch{return Response.json({error:`Expected 'clientContext' to be a JSON-serializable object.`,ok:!1},{status:400})}}}function toClientContextMessage(e){return{role:`user`,content:`${CLIENT_CONTEXT_PREFIX}${e}`}}function parseStartIndex(e){let t=new URL(e.url).searchParams.get(`startIndex`);if(t===null)return;let n=Number.parseInt(t,10);return!Number.isSafeInteger(n)||n<0?Response.json({error:`Expected startIndex to be a non-negative integer.`,ok:!1},{status:400}):n}function serializeAsNdjson(e){let t=new TextEncoder;return e.pipeThrough(new TransformStream({transform(e,n){n.enqueue(t.encode(`${JSON.stringify(e)}\n`))}}))}export{ashChannel};