const FRONTEND_MODULE = "https://esm.town/v/dglazkov/bbaasfe"; export type NodeDescriberResult = { title?: string; description?: string; metadata?: unknown; inputSchema: { properties: Record; required?: string[]; }; outputSchema: { properties: Record; required?: string[]; }; }; export type InputValues = Record; export type OutputValues = Record; // Copied from https://github.com/breadboard-ai/breadboard/blob/main/packages/breadboard/src/types.ts export type BehaviorSchema = /** * This port is deprecated and is only there because there might be graphs * that still use it. Don't show it in the UI unless there are existing * incoming wires. */ | "deprecated" /** * Indicates that this particular input port value should not be cached by * the input bubbling machinery. * Use this when you'd like to continually ask the user for the same input, * rather that re-using cached answer (default behavior). */ | "transient" /** * Indicates that the output node should bubble up to the invoking runner, * if any. * This is useful for sending outputs to the user from inside of the nested * graphs. */ | "bubble" /** * Indicates that the input our output port is a `BoardCapability`. * @see [BoardCapability] */ | "board" /** * Indicates that the input or output port is a `StreamCapability`. * @see [StreamCapability] */ | "stream" /** * Indicates that the input or output port is an `ErrorCapability`. * @see [ErrorCapability] */ | "error" /** * Indicates that the input port is usually configured, rather than wired in. */ | "config" /** * Indicates that the input or output port represents base structured * datatype containing multi-part content of a message, generated by an LLM. * See [Content](https://ai.google.dev/api/rest/v1beta/Content) for details * on the datatype. */ | "llm-content" /** * Indicates that the input or output port represents a JSON schema. */ | "json-schema" /** * Indicates that the input or output port represents a JSON schema that * describes an input or output port. */ | "ports-spec" /** * Indicates that the input or output port represents an image. The image can * be a URL or a base64 encoded image. */ | "image" /** * Indicates that the input or output represents some sort of code */ | "code"; export type Schema = { title?: string; description?: string; type?: string | string[]; properties?: Record; required?: string[]; format?: string; /** * Can be used to provide additional hints to the UI or to other parts of * the system about behavior of this particular input/output or input/output * port. */ behavior?: BehaviorSchema[]; transient?: boolean; enum?: string[]; /** * The default value of the schema. The UI can use this to pre-populate a * field with a value, if there is no `examples` present. */ default?: string; additionalProperties?: boolean | Schema; items?: Schema | Schema[]; minItems?: number; /** * Can be used by UI to pre-populate a field with a value that could be * used as an example. */ examples?: string[]; }; export type InvokeFunction = (inputs: InputValues) => Promise; export type DescribeFunction = ( inputs?: InputValues, ) => Promise | NodeDescriberResult; export type ServiceOptions = { private?: boolean; }; export const boardFromDescriberResult = ( $service: string, describe: NodeDescriberResult, options?: ServiceOptions, ) => { const nodes: any[] = [ { id: "input", type: "input", configuration: { schema: describe.inputSchema }, }, { id: "output", type: "output", configuration: { schema: describe.outputSchema }, }, { id: "service", type: "service", configuration: { $service }, metadata: { title: describe.title }, }, ]; const edges: any[] = []; Object.entries(describe.inputSchema.properties).forEach(([key, value]) => { edges.push({ from: "input", out: key, to: "service", in: key }); }); Object.entries(describe.outputSchema.properties).forEach(([key, value]) => { edges.push({ from: "service", out: key, to: "output", in: key }); }); if (options?.private) { nodes.push({ id: "secret", type: "secrets", configuration: { keys: ["BB_SERVICE_KEY"], }, }); edges.push({ from: "secret", out: "BB_SERVICE_KEY", to: "service", in: "$key", }); } return { title: describe.title, description: describe.description, nodes, edges, }; }; export const addKeyInput = (describeResult: NodeDescriberResult) => { const inputSchema = describeResult.inputSchema; const properties = inputSchema.properties; if (properties) { properties.$key = { type: "string", title: "Service Key", description: "The key to access the service", }; } inputSchema.required ??= []; inputSchema.required.push("$key"); }; export const service = ( describe: DescribeFunction, invoke: InvokeFunction, options?: ServiceOptions, ) => { return async function(req: Request): Promise { const url = new URL(req.url); const path = url.pathname; const method = req.method; if (method === "GET") { if (path === "/bgl") { const serviceURL = new URL(url); url.pathname = ""; return Response.json( boardFromDescriberResult(url.href, await describe(), options), ); } return new Response( ``, { headers: { "Content-Type": "text/html", }, }, ); } if (method !== "POST") { return new Response("Method Not Allowed", { status: 405 }); } if (path === "/describe") { let inputs; try { inputs = await req.json(); } catch { // eat the error } const describeResult = await describe(inputs); if (options?.private) { addKeyInput(describeResult); } return Response.json(describeResult); } if (path === "/invoke") { const inputs = await req.json(); if (options?.private) { if (!inputs.$key || inputs.$key != Deno.env.get("BB_SERVICE_KEY")) { return Response.json({ $error: "Must provide an API key to access the service.", }); } } delete inputs.$key; return Response.json(await invoke(inputs)); } }; };