runbios-mcp 0.2.8-dev.210 → 0.2.8-dev.213

package/README.md

@@ -360,11 +360,14 @@ deployed, not a roadmap item:
   `/.well-known/openid-configuration` (each also under `/mcp`).
 - Consent is granted in the console (`CONSOLE_BASE_URL`); every request
   re-validates the bearer token and re-checks scopes and workspace membership
-  through the internal api-gateway.
-- `chat_with_inference` is intentionally unavailable on the hosted connection:
-  a deployment's serving key is shown exactly once at create time and is never
-  retrievable, so the tool fails with that explanation instead of a
-  local-environment hint that does not apply.
+  through the internal api-gateway. `tools/list` is filtered to that grant and
+  the organization’s active release entitlements, so unavailable capabilities
+  are absent rather than advertised and rejected later.
+- `chat_with_inference` and `upload_dataset` are intentionally unavailable on
+  the hosted connection. A deployment's serving key is shown exactly once and
+  is never retrievable, while a hosted process cannot read a file from the
+  client machine. Use the OpenAI-compatible inference API for chat and the
+  console’s direct upload flow for local datasets.
 
 Any documentation still describing the hosted connector as "planned" is stale.
 

package/dist/assistant/approval.d.ts

@@ -0,0 +1,28 @@
+import type { McpStore } from "../http/store.js";
+export type AssistantToolRisk = "write" | "spend" | "destructive";
+export interface ApprovedToolCall {
+    id: string;
+    name: string;
+    arguments: Record<string, unknown>;
+    risk: AssistantToolRisk;
+}
+export interface ApprovalPayload {
+    version: 1;
+    nonce: string;
+    userId: string;
+    workspaceId: string;
+    orgId: string;
+    expiresAt: number;
+    calls: ApprovedToolCall[];
+}
+export declare function issueApprovalToken(store: McpStore, key: string, input: Omit<ApprovalPayload, "version" | "nonce" | "expiresAt">, now?: () => number): Promise<{
+    token: string;
+    expiresAt: number;
+}>;
+export declare function verifyApprovalToken(token: string, key: string, identity: {
+    userId: string;
+    workspaceId: string;
+    orgId: string;
+}, now?: () => number): ApprovalPayload | null;
+export declare function consumeApprovalToken(store: McpStore, payload: ApprovalPayload, decision: "approve" | "reject"): Promise<boolean>;
+//# sourceMappingURL=approval.d.ts.map

package/dist/assistant/approval.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"approval.d.ts","sourceRoot":"","sources":["../../src/assistant/approval.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAEjD,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,OAAO,GAAG,aAAa,CAAC;AAElE,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,IAAI,EAAE,iBAAiB,CAAC;CACzB;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,CAAC,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,gBAAgB,EAAE,CAAC;CAC3B;AA2BD,wBAAsB,kBAAkB,CACtC,KAAK,EAAE,QAAQ,EACf,GAAG,EAAE,MAAM,EACX,KAAK,EAAE,IAAI,CAAC,eAAe,EAAE,SAAS,GAAG,OAAO,GAAG,WAAW,CAAC,EAC/D,GAAG,GAAE,MAAM,MAAiB,GAC3B,OAAO,CAAC;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAAC,CAiB/C;AAED,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,MAAM,EACb,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,EAChE,GAAG,GAAE,MAAM,MAAiB,GAC3B,eAAe,GAAG,IAAI,CAmCxB;AAED,wBAAsB,oBAAoB,CACxC,KAAK,EAAE,QAAQ,EACf,OAAO,EAAE,eAAe,EACxB,QAAQ,EAAE,SAAS,GAAG,QAAQ,GAC7B,OAAO,CAAC,OAAO,CAAC,CAOlB"}

package/dist/assistant/approval.js

@@ -0,0 +1,85 @@
+import { createHash, createHmac, randomUUID, timingSafeEqual } from "node:crypto";
+const APPROVAL_TTL_MS = 10 * 60_000;
+const MAX_TOKEN_BYTES = 128 * 1024;
+function digest(value) {
+    return createHash("sha256").update(value).digest("hex");
+}
+function signature(encoded, key) {
+    return createHmac("sha256", key).update(encoded).digest();
+}
+function validCall(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const call = value;
+    return typeof call.id === "string"
+        && call.id.length > 0
+        && call.id.length <= 256
+        && typeof call.name === "string"
+        && /^[A-Za-z0-9_-]{1,128}$/.test(call.name)
+        && !!call.arguments
+        && typeof call.arguments === "object"
+        && !Array.isArray(call.arguments)
+        && (call.risk === "write" || call.risk === "spend" || call.risk === "destructive");
+}
+export async function issueApprovalToken(store, key, input, now = Date.now) {
+    if (!key || input.calls.length === 0 || input.calls.length > 16 || !input.calls.every(validCall)) {
+        throw new Error("Approval request is invalid");
+    }
+    const nonce = randomUUID();
+    const expiresAt = now() + APPROVAL_TTL_MS;
+    const payload = { version: 1, nonce, expiresAt, ...input };
+    const encoded = Buffer.from(JSON.stringify(payload), "utf8").toString("base64url");
+    const token = `${encoded}.${signature(encoded, key).toString("base64url")}`;
+    await store.insertAssistantApproval({
+        id_hash: digest(nonce),
+        user_id: input.userId,
+        workspace_id: input.workspaceId,
+        tool_names: input.calls.map((call) => call.name),
+        expires_at: new Date(expiresAt),
+    });
+    return { token, expiresAt };
+}
+export function verifyApprovalToken(token, key, identity, now = Date.now) {
+    if (!key || !token || Buffer.byteLength(token, "utf8") > MAX_TOKEN_BYTES)
+        return null;
+    const parts = token.split(".");
+    if (parts.length !== 2)
+        return null;
+    const [encoded, supplied] = parts;
+    let suppliedSignature;
+    try {
+        suppliedSignature = Buffer.from(supplied, "base64url");
+    }
+    catch {
+        return null;
+    }
+    const expected = signature(encoded, key);
+    if (suppliedSignature.length !== expected.length || !timingSafeEqual(suppliedSignature, expected))
+        return null;
+    let payload;
+    try {
+        payload = JSON.parse(Buffer.from(encoded, "base64url").toString("utf8"));
+    }
+    catch {
+        return null;
+    }
+    if (payload.version !== 1
+        || typeof payload.nonce !== "string"
+        || typeof payload.expiresAt !== "number"
+        || payload.expiresAt <= now()
+        || payload.expiresAt > now() + APPROVAL_TTL_MS
+        || payload.userId !== identity.userId
+        || payload.workspaceId !== identity.workspaceId
+        || payload.orgId !== identity.orgId
+        || !Array.isArray(payload.calls)
+        || payload.calls.length === 0
+        || payload.calls.length > 16
+        || !payload.calls.every(validCall)) {
+        return null;
+    }
+    return payload;
+}
+export async function consumeApprovalToken(store, payload, decision) {
+    return store.consumeAssistantApproval(digest(payload.nonce), payload.userId, payload.workspaceId, decision);
+}
+//# sourceMappingURL=approval.js.map

package/dist/assistant/approval.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"approval.js","sourceRoot":"","sources":["../../src/assistant/approval.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAsBlF,MAAM,eAAe,GAAG,EAAE,GAAG,MAAM,CAAC;AACpC,MAAM,eAAe,GAAG,GAAG,GAAG,IAAI,CAAC;AAEnC,SAAS,MAAM,CAAC,KAAa;IAC3B,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,SAAS,CAAC,OAAe,EAAE,GAAW;IAC7C,OAAO,UAAU,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;AAC5D,CAAC;AAED,SAAS,SAAS,CAAC,KAAc;IAC/B,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,IAAI,GAAG,KAAkC,CAAC;IAChD,OAAO,OAAO,IAAI,CAAC,EAAE,KAAK,QAAQ;WAC7B,IAAI,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC;WAClB,IAAI,CAAC,EAAE,CAAC,MAAM,IAAI,GAAG;WACrB,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ;WAC7B,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;WACxC,CAAC,CAAC,IAAI,CAAC,SAAS;WAChB,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ;WAClC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;WAC9B,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;AACvF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,KAAe,EACf,GAAW,EACX,KAA+D,EAC/D,MAAoB,IAAI,CAAC,GAAG;IAE5B,IAAI,CAAC,GAAG,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;QACjG,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;IACD,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;IAC3B,MAAM,SAAS,GAAG,GAAG,EAAE,GAAG,eAAe,CAAC;IAC1C,MAAM,OAAO,GAAoB,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,KAAK,EAAE,CAAC;IAC5E,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IACnF,MAAM,KAAK,GAAG,GAAG,OAAO,IAAI,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;IAC5E,MAAM,KAAK,CAAC,uBAAuB,CAAC;QAClC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC;QACtB,OAAO,EAAE,KAAK,CAAC,MAAM;QACrB,YAAY,EAAE,KAAK,CAAC,WAAW;QAC/B,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;QAChD,UAAU,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC;KAChC,CAAC,CAAC;IACH,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,KAAa,EACb,GAAW,EACX,QAAgE,EAChE,MAAoB,IAAI,CAAC,GAAG;IAE5B,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,IAAI,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,eAAe;QAAE,OAAO,IAAI,CAAC;IACtF,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACpC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,GAAG,KAAK,CAAC;IAClC,IAAI,iBAAyB,CAAC;IAC9B,IAAI,CAAC;QACH,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACzD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACzC,IAAI,iBAAiB,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,QAAQ,CAAC;QAAE,OAAO,IAAI,CAAC;IAE/G,IAAI,OAAwB,CAAC;IAC7B,IAAI,CAAC;QACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAoB,CAAC;IAC9F,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,CAAC;WACpB,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ;WACjC,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ;WACrC,OAAO,CAAC,SAAS,IAAI,GAAG,EAAE;WAC1B,OAAO,CAAC,SAAS,GAAG,GAAG,EAAE,GAAG,eAAe;WAC3C,OAAO,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM;WAClC,OAAO,CAAC,WAAW,KAAK,QAAQ,CAAC,WAAW;WAC5C,OAAO,CAAC,KAAK,KAAK,QAAQ,CAAC,KAAK;WAChC,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;WAC7B,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;WAC1B,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,EAAE;WACzB,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,KAAe,EACf,OAAwB,EACxB,QAA8B;IAE9B,OAAO,KAAK,CAAC,wBAAwB,CACnC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EACrB,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,WAAW,EACnB,QAAQ,CACT,CAAC;AACJ,CAAC"}

package/dist/assistant/model.d.ts

@@ -0,0 +1,63 @@
+export type AssistantContentPart = {
+    type: "text";
+    text: string;
+} | {
+    type: "image_url";
+    image_url: {
+        url: string;
+    };
+} | {
+    type: "video_url";
+    video_url: {
+        url: string;
+    };
+};
+export interface AssistantModelToolCall {
+    id: string;
+    type: "function";
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+export interface AssistantModelMessage {
+    role: "system" | "user" | "assistant" | "tool";
+    content: string | AssistantContentPart[] | null;
+    tool_calls?: AssistantModelToolCall[];
+    tool_call_id?: string;
+}
+export interface AssistantUsage {
+    prompt_tokens?: number;
+    completion_tokens?: number;
+    total_tokens?: number;
+    prompt_tokens_details?: Record<string, number> | null;
+}
+export interface AssistantModelResult {
+    message: AssistantModelMessage;
+    finishReason: string | null;
+    usage: AssistantUsage | null;
+    contextTransform: Record<string, unknown> | null;
+}
+export declare class AssistantUpstreamError extends Error {
+    readonly status: number;
+    readonly code: string;
+    constructor(status: number, code: string, message: string);
+}
+interface ModelStreamInput {
+    baseUrl: string;
+    authorization: string;
+    orgId: string;
+    workspaceId: string;
+    model: string;
+    effort?: string;
+    maxTokens?: number;
+    messages: AssistantModelMessage[];
+    tools: Record<string, unknown>[];
+    signal: AbortSignal;
+    fetchImpl?: typeof fetch;
+    onTextDelta?: (text: string) => void;
+    onContextTransform?: (value: Record<string, unknown>) => void;
+}
+export declare function streamAssistantModel(input: ModelStreamInput): Promise<AssistantModelResult>;
+export {};
+//# sourceMappingURL=model.d.ts.map

package/dist/assistant/model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/assistant/model.ts"],"names":[],"mappings":"AAGA,MAAM,MAAM,oBAAoB,GAC5B;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,SAAS,EAAE;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GACjD;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,SAAS,EAAE;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAAC;AAEtD,MAAM,WAAW,sBAAsB;IACrC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;CAC/C;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,GAAG,oBAAoB,EAAE,GAAG,IAAI,CAAC;IAChD,UAAU,CAAC,EAAE,sBAAsB,EAAE,CAAC;IACtC,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAC7B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,qBAAqB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;CACvD;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,qBAAqB,CAAC;IAC/B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,KAAK,EAAE,cAAc,GAAG,IAAI,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAClD;AAED,qBAAa,sBAAuB,SAAQ,KAAK;IAE7C,QAAQ,CAAC,MAAM,EAAE,MAAM;IACvB,QAAQ,CAAC,IAAI,EAAE,MAAM;gBADZ,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,MAAM,EACrB,OAAO,EAAE,MAAM;CAKlB;AAED,UAAU,gBAAgB;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,qBAAqB,EAAE,CAAC;IAClC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;IACjC,MAAM,EAAE,WAAW,CAAC;IACpB,SAAS,CAAC,EAAE,OAAO,KAAK,CAAC;IACzB,WAAW,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IACrC,kBAAkB,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;CAC/D;AAyHD,wBAAsB,oBAAoB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CA+FjG"}

package/dist/assistant/model.js

@@ -0,0 +1,238 @@
+import { randomUUID } from "node:crypto";
+import { redactBuildIdentity } from "../redaction.js";
+export class AssistantUpstreamError extends Error {
+    status;
+    code;
+    constructor(status, code, message) {
+        super(message);
+        this.status = status;
+        this.code = code;
+        this.name = "AssistantUpstreamError";
+    }
+}
+function upstreamError(status, raw) {
+    let message = "The assistant model could not complete this request.";
+    let code = "assistant_model_error";
+    try {
+        const parsed = JSON.parse(raw);
+        const nested = parsed.error && typeof parsed.error === "object"
+            ? parsed.error
+            : {};
+        const candidate = nested.message ?? parsed.message ?? parsed.detail;
+        const candidateCode = nested.code ?? nested.type ?? parsed.code;
+        if (typeof candidate === "string" && candidate.trim())
+            message = redactBuildIdentity(candidate.trim());
+        if (typeof candidateCode === "string" && candidateCode.trim())
+            code = candidateCode.trim();
+    }
+    catch {
+        if (raw.trim() && raw.length <= 500 && !raw.trim().startsWith("<"))
+            message = redactBuildIdentity(raw.trim());
+    }
+    if (status === 402) {
+        code = "insufficient_funds";
+        if (!message || message === "The assistant model could not complete this request.") {
+            message = "Your wallet does not have enough available credit for this request.";
+        }
+    }
+    return new AssistantUpstreamError(status, code, message);
+}
+function mergeToolCall(calls, indexMap, raw) {
+    const fn = raw.function && typeof raw.function === "object"
+        ? raw.function
+        : {};
+    const wireIndex = typeof raw.index === "number" ? raw.index : undefined;
+    const id = typeof raw.id === "string" ? raw.id : "";
+    const known = id ? calls.findIndex((call) => call.id === id) : -1;
+    let index;
+    if (known >= 0)
+        index = known;
+    else if (id && wireIndex !== undefined && calls[indexMap.get(wireIndex) ?? wireIndex]?.id)
+        index = calls.length;
+    else if (wireIndex !== undefined)
+        index = indexMap.get(wireIndex) ?? wireIndex;
+    else if (id || typeof fn.name === "string")
+        index = calls.length;
+    else
+        index = Math.max(0, calls.length - 1);
+    if (wireIndex !== undefined)
+        indexMap.set(wireIndex, index);
+    if (!calls[index]) {
+        calls[index] = { id: "", type: "function", function: { name: "", arguments: "" } };
+    }
+    if (id)
+        calls[index].id = id;
+    if (typeof fn.name === "string")
+        calls[index].function.name += fn.name;
+    if (typeof fn.arguments === "string")
+        calls[index].function.arguments += fn.arguments;
+}
+function consumeCompletionObject(object, state, input) {
+    if (object.error) {
+        const error = object.error;
+        const message = typeof error === "string" ? error : String(error.message ?? "The model stream failed.");
+        const code = typeof error === "object" && typeof error.code === "string" ? error.code : "assistant_model_error";
+        throw new AssistantUpstreamError(502, code, redactBuildIdentity(message));
+    }
+    if (object.usage && typeof object.usage === "object")
+        state.usage = object.usage;
+    if (object.bios_context_transform && typeof object.bios_context_transform === "object") {
+        state.contextTransform = object.bios_context_transform;
+        input.onContextTransform?.(state.contextTransform);
+    }
+    const choices = Array.isArray(object.choices) ? object.choices : [];
+    for (const rawChoice of choices) {
+        if (!rawChoice || typeof rawChoice !== "object")
+            continue;
+        const choice = rawChoice;
+        if (typeof choice.finish_reason === "string")
+            state.finishReason = choice.finish_reason;
+        const payload = choice.delta && typeof choice.delta === "object"
+            ? choice.delta
+            : choice.message && typeof choice.message === "object"
+                ? choice.message
+                : {};
+        if (typeof payload.content === "string" && payload.content) {
+            state.content += payload.content;
+            input.onTextDelta?.(payload.content);
+        }
+        const toolCalls = Array.isArray(payload.tool_calls) ? payload.tool_calls : [];
+        for (const call of toolCalls) {
+            if (call && typeof call === "object")
+                mergeToolCall(state.calls, state.indexMap, call);
+        }
+    }
+}
+function emitFrame(frame, consume) {
+    const data = [];
+    for (const line of frame.replace(/\r\n/g, "\n").replace(/\r/g, "\n").split("\n")) {
+        if (!line || line.startsWith(":"))
+            continue;
+        const colon = line.indexOf(":");
+        const field = colon < 0 ? line : line.slice(0, colon);
+        let value = colon < 0 ? "" : line.slice(colon + 1);
+        if (value.startsWith(" "))
+            value = value.slice(1);
+        if (field === "data")
+            data.push(value);
+    }
+    if (data.length === 0)
+        return false;
+    const payload = data.join("\n").trim();
+    if (payload === "[DONE]")
+        return true;
+    try {
+        const parsed = JSON.parse(payload);
+        if (parsed && typeof parsed === "object")
+            consume(parsed);
+    }
+    catch {
+        return false;
+    }
+    return false;
+}
+export async function streamAssistantModel(input) {
+    const fetchImpl = input.fetchImpl ?? fetch;
+    const body = {
+        model: input.model,
+        messages: input.messages,
+        stream: true,
+        stream_options: { include_usage: true },
+    };
+    if (input.effort)
+        body.reasoning_effort = input.effort;
+    if (input.maxTokens && input.maxTokens > 0)
+        body.max_tokens = input.maxTokens;
+    if (input.tools.length > 0) {
+        body.tools = input.tools;
+        body.tool_choice = "auto";
+    }
+    const response = await fetchImpl(`${input.baseUrl.replace(/\/+$/, "")}/v1/chat/completions`, {
+        method: "POST",
+        headers: {
+            Accept: "text/event-stream",
+            Authorization: input.authorization,
+            "Content-Type": "application/json",
+            "User-Agent": "bios-assistant/1.0",
+            "X-Org-ID": input.orgId,
+            "X-Workspace-ID": input.workspaceId,
+            "X-Request-ID": randomUUID(),
+        },
+        body: JSON.stringify(body),
+        signal: input.signal,
+    });
+    if (!response.ok)
+        throw upstreamError(response.status, await response.text().catch(() => ""));
+    if (!response.body)
+        throw new AssistantUpstreamError(502, "empty_model_stream", "The assistant model returned no response stream.");
+    const state = {
+        content: "",
+        finishReason: null,
+        usage: null,
+        contextTransform: null,
+        calls: [],
+        indexMap: new Map(),
+    };
+    const contentType = response.headers.get("content-type") ?? "";
+    if (!contentType.includes("text/event-stream")) {
+        const text = await response.text();
+        let parsed;
+        try {
+            parsed = JSON.parse(text);
+        }
+        catch {
+            throw new AssistantUpstreamError(502, "invalid_model_response", "The assistant model returned an unreadable response.");
+        }
+        if (!parsed || typeof parsed !== "object") {
+            throw new AssistantUpstreamError(502, "invalid_model_response", "The assistant model returned an empty response.");
+        }
+        consumeCompletionObject(parsed, state, input);
+    }
+    else {
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = "";
+        let finished = false;
+        try {
+            while (!finished) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                buffer += decoder.decode(value, { stream: true });
+                for (;;) {
+                    const match = /\r\n\r\n|\n\n|\r\r/.exec(buffer);
+                    if (!match || match.index === undefined)
+                        break;
+                    const frame = buffer.slice(0, match.index);
+                    buffer = buffer.slice(match.index + match[0].length);
+                    if (emitFrame(frame, (object) => consumeCompletionObject(object, state, input))) {
+                        finished = true;
+                        break;
+                    }
+                }
+            }
+            buffer += decoder.decode();
+            if (!finished && buffer.trim())
+                emitFrame(buffer, (object) => consumeCompletionObject(object, state, input));
+        }
+        finally {
+            await reader.cancel().catch(() => undefined);
+            reader.releaseLock();
+        }
+    }
+    const calls = state.calls.filter((call) => call?.function.name).map((call, index) => ({
+        ...call,
+        id: call.id || `call_${index}_${randomUUID().slice(0, 8)}`,
+    }));
+    return {
+        message: {
+            role: "assistant",
+            content: state.content || null,
+            ...(calls.length > 0 ? { tool_calls: calls } : {}),
+        },
+        finishReason: state.finishReason,
+        usage: state.usage,
+        contextTransform: state.contextTransform,
+    };
+}
+//# sourceMappingURL=model.js.map

package/dist/assistant/model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/assistant/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAkCtD,MAAM,OAAO,sBAAuB,SAAQ,KAAK;IAEpC;IACA;IAFX,YACW,MAAc,EACd,IAAY,EACrB,OAAe;QAEf,KAAK,CAAC,OAAO,CAAC,CAAC;QAJN,WAAM,GAAN,MAAM,CAAQ;QACd,SAAI,GAAJ,IAAI,CAAQ;QAIrB,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;IACvC,CAAC;CACF;AAkBD,SAAS,aAAa,CAAC,MAAc,EAAE,GAAW;IAChD,IAAI,OAAO,GAAG,sDAAsD,CAAC;IACrE,IAAI,IAAI,GAAG,uBAAuB,CAAC;IACnC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4B,CAAC;QAC1D,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;YAC7D,CAAC,CAAC,MAAM,CAAC,KAAgC;YACzC,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC;QACpE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;QAChE,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE;YAAE,OAAO,GAAG,mBAAmB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACvG,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE;YAAE,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC;IAC7F,CAAC;IAAC,MAAM,CAAC;QACP,IAAI,GAAG,CAAC,IAAI,EAAE,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,OAAO,GAAG,mBAAmB,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAChH,CAAC;IACD,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;QACnB,IAAI,GAAG,oBAAoB,CAAC;QAC5B,IAAI,CAAC,OAAO,IAAI,OAAO,KAAK,sDAAsD,EAAE,CAAC;YACnF,OAAO,GAAG,qEAAqE,CAAC;QAClF,CAAC;IACH,CAAC;IACD,OAAO,IAAI,sBAAsB,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AAC3D,CAAC;AAED,SAAS,aAAa,CACpB,KAA+B,EAC/B,QAA6B,EAC7B,GAA4B;IAE5B,MAAM,EAAE,GAAG,GAAG,CAAC,QAAQ,IAAI,OAAO,GAAG,CAAC,QAAQ,KAAK,QAAQ;QACzD,CAAC,CAAC,GAAG,CAAC,QAAmC;QACzC,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,SAAS,GAAG,OAAO,GAAG,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IACxE,MAAM,EAAE,GAAG,OAAO,GAAG,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACpD,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAClE,IAAI,KAAa,CAAC;IAClB,IAAI,KAAK,IAAI,CAAC;QAAE,KAAK,GAAG,KAAK,CAAC;SACzB,IAAI,EAAE,IAAI,SAAS,KAAK,SAAS,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,EAAE,EAAE;QAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC;SAC3G,IAAI,SAAS,KAAK,SAAS;QAAE,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC;SAC1E,IAAI,EAAE,IAAI,OAAO,EAAE,CAAC,IAAI,KAAK,QAAQ;QAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC;;QAC5D,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC3C,IAAI,SAAS,KAAK,SAAS;QAAE,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC5D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;QAClB,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,CAAC;IACrF,CAAC;IACD,IAAI,EAAE;QAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC;IAC7B,IAAI,OAAO,EAAE,CAAC,IAAI,KAAK,QAAQ;QAAE,KAAK,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,IAAI,CAAC;IACvE,IAAI,OAAO,EAAE,CAAC,SAAS,KAAK,QAAQ;QAAE,KAAK,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC,SAAS,CAAC;AACxF,CAAC;AAED,SAAS,uBAAuB,CAC9B,MAA+B,EAC/B,KAOC,EACD,KAAuB;IAEvB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAyC,CAAC;QAC/D,MAAM,OAAO,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,0BAA0B,CAAC,CAAC;QACxG,MAAM,IAAI,GAAG,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,uBAAuB,CAAC;QAChH,MAAM,IAAI,sBAAsB,CAAC,GAAG,EAAE,IAAI,EAAE,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC5E,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;QAAE,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,KAAuB,CAAC;IACnG,IAAI,MAAM,CAAC,sBAAsB,IAAI,OAAO,MAAM,CAAC,sBAAsB,KAAK,QAAQ,EAAE,CAAC;QACvF,KAAK,CAAC,gBAAgB,GAAG,MAAM,CAAC,sBAAiD,CAAC;QAClF,KAAK,CAAC,kBAAkB,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACrD,CAAC;IACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IACpE,KAAK,MAAM,SAAS,IAAI,OAAO,EAAE,CAAC;QAChC,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ;YAAE,SAAS;QAC1D,MAAM,MAAM,GAAG,SAAoC,CAAC;QACpD,IAAI,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ;YAAE,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;QACxF,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;YAC9D,CAAC,CAAC,MAAM,CAAC,KAAgC;YACzC,CAAC,CAAC,MAAM,CAAC,OAAO,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ;gBACpD,CAAC,CAAC,MAAM,CAAC,OAAkC;gBAC3C,CAAC,CAAC,EAAE,CAAC;QACT,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAC3D,KAAK,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;YACjC,KAAK,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QACD,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9E,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;gBAAE,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,IAA+B,CAAC,CAAC;QACpH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAChB,KAAa,EACb,OAAkD;IAElD,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACjF,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,SAAS;QAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAChC,MAAM,KAAK,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACtD,IAAI,KAAK,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACnD,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClD,IAAI,KAAK,KAAK,MAAM;YAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IACD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACpC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;IACvC,IAAI,OAAO,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACtC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAY,CAAC;QAC9C,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,CAAC,MAAiC,CAAC,CAAC;IACvF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,KAAuB;IAChE,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC;IAC3C,MAAM,IAAI,GAA4B;QACpC,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,MAAM,EAAE,IAAI;QACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;KACxC,CAAC;IACF,IAAI,KAAK,CAAC,MAAM;QAAE,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,MAAM,CAAC;IACvD,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC;QAAE,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,SAAS,CAAC;IAC9E,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;IAC5B,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,sBAAsB,EAAE;QAC3F,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,MAAM,EAAE,mBAAmB;YAC3B,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,cAAc,EAAE,kBAAkB;YAClC,YAAY,EAAE,oBAAoB;YAClC,UAAU,EAAE,KAAK,CAAC,KAAK;YACvB,gBAAgB,EAAE,KAAK,CAAC,WAAW;YACnC,cAAc,EAAE,UAAU,EAAE;SAC7B;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;QAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ,CAAC,EAAE;QAAE,MAAM,aAAa,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9F,IAAI,CAAC,QAAQ,CAAC,IAAI;QAAE,MAAM,IAAI,sBAAsB,CAAC,GAAG,EAAE,oBAAoB,EAAE,kDAAkD,CAAC,CAAC;IAEpI,MAAM,KAAK,GAAG;QACZ,OAAO,EAAE,EAAE;QACX,YAAY,EAAE,IAAqB;QACnC,KAAK,EAAE,IAA6B;QACpC,gBAAgB,EAAE,IAAsC;QACxD,KAAK,EAAE,EAA8B;QACrC,QAAQ,EAAE,IAAI,GAAG,EAAkB;KACpC,CAAC;IACF,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAC/D,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,sBAAsB,CAAC,GAAG,EAAE,wBAAwB,EAAE,sDAAsD,CAAC,CAAC;QAC1H,CAAC;QACD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC1C,MAAM,IAAI,sBAAsB,CAAC,GAAG,EAAE,wBAAwB,EAAE,iDAAiD,CAAC,CAAC;QACrH,CAAC;QACD,uBAAuB,CAAC,MAAiC,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;IAC3E,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,CAAC;YACH,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACjB,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAChB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,SAAS,CAAC;oBACR,MAAM,KAAK,GAAG,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAChD,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;wBAAE,MAAM;oBAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC3C,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;oBACrD,IAAI,SAAS,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC;wBAChF,QAAQ,GAAG,IAAI,CAAC;wBAChB,MAAM;oBACR,CAAC;gBACH,CAAC;YACH,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE;gBAAE,SAAS,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;QAC/G,CAAC;gBAAS,CAAC;YACT,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC7C,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QACpF,GAAG,IAAI;QACP,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,QAAQ,KAAK,IAAI,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;KAC3D,CAAC,CAAC,CAAC;IACJ,OAAO;QACL,OAAO,EAAE;YACP,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI;YAC9B,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACnD;QACD,YAAY,EAAE,KAAK,CAAC,YAAY;QAChC,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;KACzC,CAAC;AACJ,CAAC"}

package/dist/assistant/tool-runtime.d.ts

@@ -0,0 +1,28 @@
+import type { BiosClient } from "../api-client.js";
+import type { LaunchGates } from "../config.js";
+import { type ServerHooks } from "../server.js";
+import type { AssistantToolRisk } from "./approval.js";
+export interface AssistantToolDefinition {
+    name: string;
+    description: string;
+    inputSchema: Record<string, unknown>;
+}
+export interface AssistantToolResult {
+    ok: boolean;
+    text: string;
+}
+export interface AssistantToolRuntime {
+    tools: AssistantToolDefinition[];
+    call(name: string, args: Record<string, unknown>): Promise<AssistantToolResult>;
+    close(): Promise<void>;
+}
+export declare function toolRisk(name: string): "read" | AssistantToolRisk;
+export declare function rankTools(tools: AssistantToolDefinition[], query: string, limit?: number): AssistantToolDefinition[];
+export declare function openAITool(tool: AssistantToolDefinition): Record<string, unknown>;
+export declare function createAssistantToolRuntime(input: {
+    client: BiosClient;
+    launchGates: LaunchGates;
+    allowedTools: ReadonlySet<string>;
+    hooks?: ServerHooks;
+}): Promise<AssistantToolRuntime>;
+//# sourceMappingURL=tool-runtime.d.ts.map

package/dist/assistant/tool-runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-runtime.d.ts","sourceRoot":"","sources":["../../src/assistant/tool-runtime.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAChD,OAAO,EAAuB,KAAK,WAAW,EAAE,MAAM,cAAc,CAAC;AACrE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAEvD,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,OAAO,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,uBAAuB,EAAE,CAAC;IACjC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAChF,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAqBD,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,iBAAiB,CAKjE;AA4BD,wBAAgB,SAAS,CACvB,KAAK,EAAE,uBAAuB,EAAE,EAChC,KAAK,EAAE,MAAM,EACb,KAAK,SAAK,GACT,uBAAuB,EAAE,CA6B3B;AAED,wBAAgB,UAAU,CAAC,IAAI,EAAE,uBAAuB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CASjF;AAED,wBAAsB,0BAA0B,CAAC,KAAK,EAAE;IACtD,MAAM,EAAE,UAAU,CAAC;IACnB,WAAW,EAAE,WAAW,CAAC;IACzB,YAAY,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAClC,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAyChC"}

package/dist/assistant/tool-runtime.js

@@ -0,0 +1,140 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
+import { createBiosMcpServer } from "../server.js";
+const SPEND_TOOLS = new Set([
+    "create_inference",
+    "create_training_job",
+    "resume_inference",
+    "restart_inference",
+    "resume_training_job",
+    "update_inference_policy",
+    "loop_consent_training_rule",
+    "loop_run_training_rule",
+    "loop_promote_training_run",
+    "loop_enable_agent",
+    "loop_create_sample_run",
+    "loop_start_judge_run",
+    "loop_update_agent_settings",
+]);
+const READ_PREFIXES = ["get_", "list_", "search_", "preview_", "preflight_", "recommend_", "introspect_"];
+const DESTRUCTIVE_PREFIXES = ["delete_", "remove_", "stop_", "cancel_", "reject_", "rollback_", "abort_", "disconnect_"];
+export function toolRisk(name) {
+    if (SPEND_TOOLS.has(name))
+        return "spend";
+    if (DESTRUCTIVE_PREFIXES.some((prefix) => name.startsWith(prefix)))
+        return "destructive";
+    if (READ_PREFIXES.some((prefix) => name.startsWith(prefix)))
+        return "read";
+    return "write";
+}
+function words(value) {
+    return value.toLowerCase().split(/[^a-z0-9]+/).filter((word) => word.length > 1);
+}
+function expandedQuery(query) {
+    const tokens = new Set(words(query));
+    const expansions = {
+        deploy: ["deployment", "inference"],
+        deployment: ["deploy", "inference"],
+        train: ["training", "job"],
+        training: ["train", "job"],
+        data: ["dataset"],
+        dataset: ["data"],
+        cost: ["wallet", "pricing", "billing"],
+        price: ["wallet", "pricing", "billing"],
+        model: ["models", "catalog"],
+        endpoint: ["inference", "deployment"],
+        checkpoint: ["training", "checkpoint"],
+        monitor: ["status", "metrics", "logs"],
+    };
+    for (const token of [...tokens]) {
+        for (const expansion of expansions[token] ?? [])
+            tokens.add(expansion);
+    }
+    return tokens;
+}
+export function rankTools(tools, query, limit = 24) {
+    const queryText = query.toLowerCase();
+    const queryWords = expandedQuery(query);
+    const scored = tools.map((tool, index) => {
+        const name = tool.name.toLowerCase();
+        const description = tool.description.toLowerCase();
+        const toolWords = new Set([...words(name), ...words(description)]);
+        let score = 0;
+        if (queryText.includes(name.replaceAll("_", " ")))
+            score += 30;
+        for (const word of queryWords) {
+            if (name.split("_").includes(word))
+                score += 8;
+            else if (name.includes(word))
+                score += 4;
+            if (toolWords.has(word))
+                score += 2;
+            else if (description.includes(word))
+                score += 1;
+        }
+        if (name === "introspect_api_key")
+            score += 1;
+        if (name === "get_wallet_balance" && /cost|price|spend|wallet|balance|train|deploy/.test(queryText))
+            score += 6;
+        return { tool, score, index };
+    });
+    scored.sort((a, b) => b.score - a.score || a.index - b.index);
+    const positive = scored.filter((entry) => entry.score > 0).slice(0, limit).map((entry) => entry.tool);
+    if (positive.length >= Math.min(6, tools.length))
+        return positive;
+    const selected = new Set(positive.map((tool) => tool.name));
+    for (const entry of scored) {
+        if (positive.length >= Math.min(limit, 8) || selected.has(entry.tool.name))
+            continue;
+        positive.push(entry.tool);
+        selected.add(entry.tool.name);
+    }
+    return positive;
+}
+export function openAITool(tool) {
+    return {
+        type: "function",
+        function: {
+            name: tool.name,
+            description: tool.description,
+            parameters: tool.inputSchema,
+        },
+    };
+}
+export async function createAssistantToolRuntime(input) {
+    const server = createBiosMcpServer(input.client, input.hooks, undefined, input.launchGates, input.allowedTools);
+    const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
+    const client = new Client({ name: "runbios-assistant", version: "1.0.0" });
+    try {
+        await Promise.all([server.connect(serverTransport), client.connect(clientTransport)]);
+        const listed = await client.listTools();
+        const tools = listed.tools.map((tool) => ({
+            name: tool.name,
+            description: tool.description ?? "",
+            inputSchema: (tool.inputSchema ?? { type: "object", properties: {} }),
+        }));
+        return {
+            tools,
+            async call(name, args) {
+                const response = await client.callTool({ name, arguments: args });
+                const content = Array.isArray(response.content) ? response.content : [];
+                const text = content
+                    .filter((part) => part?.type === "text" && typeof part.text === "string")
+                    .map((part) => part.text)
+                    .join("\n");
+                const fallback = response.structuredContent == null ? "" : JSON.stringify(response.structuredContent);
+                return { ok: response.isError !== true, text: text || fallback || "{}" };
+            },
+            async close() {
+                await client.close().catch(() => undefined);
+                await server.close().catch(() => undefined);
+            },
+        };
+    }
+    catch (error) {
+        await client.close().catch(() => undefined);
+        await server.close().catch(() => undefined);
+        throw error;
+    }
+}
+//# sourceMappingURL=tool-runtime.js.map