bleam 0.0.10 → 0.0.12

package/dist/ai.d.ts

@@ -1,6 +1,197 @@
-import { s as FileRef } from "./files-VrkQlKIT.js";
-import { h as StandardSchema, p as SchemaResult, s as InferOutput } from "./schema-DOOjfXvs.js";
+import "./files-VrkQlKIT.js";
 
+//#region src/ai/bonsai.d.ts
+type BonsaiTextModelId = 'ternary-bonsai-1.7b' | 'ternary-bonsai-4b' | 'ternary-bonsai-8b' | 'ternary-bonsai-27b';
+type BonsaiImageModelId = 'ternary-bonsai-image-4b';
+type BonsaiTextModel<Id extends BonsaiTextModelId = BonsaiTextModelId> = Readonly<{
+  provider: 'bonsai';
+  kind: 'text';
+  id: Id;
+}>;
+type BonsaiImageModel<Id extends BonsaiImageModelId = BonsaiImageModelId> = Readonly<{
+  provider: 'bonsai';
+  kind: 'image';
+  id: Id;
+}>;
+type BonsaiModel = BonsaiTextModel | BonsaiImageModel;
+type BonsaiTextModelMetadata = Readonly<{
+  id: BonsaiTextModelId;
+  repo: string;
+  revision: string;
+  estimatedDownloadBytes: number;
+  contextLength: number;
+}>;
+declare const bonsaiTextModelCatalog: readonly BonsaiTextModelMetadata[];
+declare function bonsai<const Id extends BonsaiTextModelId>(id: Id): BonsaiTextModel<Id>;
+declare function bonsai<const Id extends BonsaiImageModelId>(id: Id): BonsaiImageModel<Id>;
+//#endregion
+//#region src/ai/bonsai-text.d.ts
+declare const bonsaiTextModelFiles: readonly ["config.json", "tokenizer.json", "tokenizer_config.json", "chat_template.jinja", "model.safetensors", "model.safetensors.index.json"];
+type BonsaiTextModelFile = (typeof bonsaiTextModelFiles)[number];
+type PrepareBonsaiTextModelProgress = Readonly<{
+  model: BonsaiTextModelId;
+  file: BonsaiTextModelFile;
+  index: number;
+  total: number;
+  bytesWritten: number;
+  totalBytes?: number;
+  fraction?: number;
+  modelBytesWritten: number;
+  estimatedModelBytes: number;
+  modelFraction: number;
+}>;
+type PrepareBonsaiTextModelOptions = Readonly<{
+  accessToken?: string;
+  directory?: string;
+  signal?: AbortSignal;
+  onProgress?: (progress: PrepareBonsaiTextModelProgress) => void;
+}>;
+type PreparedBonsaiTextModel = Readonly<{
+  model: BonsaiTextModel;
+  path: string;
+}>;
+type BonsaiTextModelState = Readonly<{
+  model: BonsaiTextModel;
+  metadata: BonsaiTextModelMetadata;
+  path: string;
+  installed: boolean;
+  verified: boolean;
+  bytesPresent: number;
+  estimatedTotalBytes: number;
+  missingFiles: readonly string[];
+  invalidFiles: readonly string[];
+}>;
+type ModelInput = BonsaiTextModelId | BonsaiTextModel;
+declare function prepareBonsaiTextModel(model: ModelInput, options?: PrepareBonsaiTextModelOptions): Promise<PreparedBonsaiTextModel>;
+declare function verifyBonsaiTextModel(model: ModelInput, options?: Readonly<{
+  path?: string;
+}>): Promise<Readonly<{
+  model: BonsaiTextModel;
+  metadata: BonsaiTextModelMetadata;
+  path: string;
+  installed: boolean;
+  verified: boolean;
+  bytesPresent: number;
+  estimatedTotalBytes: number;
+  missingFiles: readonly string[];
+  invalidFiles: readonly string[];
+}>>;
+declare function listBonsaiTextModels(options?: Readonly<{
+  directory?: string;
+}>): Promise<Readonly<{
+  model: BonsaiTextModel;
+  metadata: BonsaiTextModelMetadata;
+  path: string;
+  installed: boolean;
+  verified: boolean;
+  bytesPresent: number;
+  estimatedTotalBytes: number;
+  missingFiles: readonly string[];
+  invalidFiles: readonly string[];
+}>[]>;
+declare function deleteBonsaiTextModel(model: ModelInput, options?: Readonly<{
+  path?: string;
+}>): Promise<void>;
+//#endregion
+//#region src/ai/chat-models.d.ts
+type ChatStatus = 'ready' | 'queued' | 'generating' | 'canceling' | 'failed' | 'canceled';
+type AssistantMessageStatus = 'queued' | 'generating' | 'canceling' | 'completed' | 'failed' | 'canceled';
+type ChatError = {
+  name: string;
+  message: string;
+  code?: string;
+  stage?: string;
+};
+type Chat = {
+  id: string;
+  title: string;
+  status: ChatStatus;
+  createdAt: string;
+  updatedAt: string;
+};
+type ChatSummary = Chat;
+type ChatMessageBase = {
+  id: string;
+  chatId: string;
+  content: string;
+  sequence: number;
+  attempt: number;
+  createdAt: string;
+  updatedAt: string;
+  startedAt?: string;
+  completedAt?: string;
+  error?: ChatError;
+};
+type UserChatMessage = ChatMessageBase & {
+  role: 'user';
+  status: 'completed';
+};
+type AssistantChatMessage = ChatMessageBase & {
+  role: 'assistant';
+  status: AssistantMessageStatus;
+};
+type ChatMessage = UserChatMessage | AssistantChatMessage;
+type FoundationChatGenerationOptions = {
+  model?: never;
+  path?: never;
+  system?: string;
+  temperature?: number;
+  maxTokens?: number;
+  seed?: never;
+};
+type BonsaiChatGenerationOptions = {
+  model: BonsaiTextModel;
+  path?: string;
+  system?: string;
+  temperature?: number;
+  maxTokens?: number;
+  seed?: number;
+};
+type ChatGenerationOptions = FoundationChatGenerationOptions | BonsaiChatGenerationOptions;
+type UseChatOptions = ChatGenerationOptions & {
+  onFinish?: (message: AssistantChatMessage) => void;
+  onError?: (error: ChatError) => void;
+};
+type ChatRunOptions = {
+  signal?: AbortSignal;
+};
+type ChatSendOptions = ChatGenerationOptions & ChatRunOptions;
+type SendChatResult = {
+  chatId: string;
+  userMessageId: string;
+  assistantMessageId: string;
+};
+type UseChatResult = {
+  chat: Chat | null;
+  messages: readonly ChatMessage[];
+  status: ChatStatus | 'not-found';
+  error: ChatError | null;
+  isGenerating: boolean;
+  canSend: boolean;
+  canCancel: boolean;
+  send(content: string, options?: ChatRunOptions): Promise<SendChatResult>;
+  cancel(): Promise<boolean>;
+  retry(messageId: string, options?: ChatRunOptions): Promise<SendChatResult>;
+  setTitle(title: string): Promise<Chat>;
+  remove(): Promise<boolean>;
+};
+//#endregion
+//#region src/ai/chat.d.ts
+declare const chat: {
+  create(input?: {
+    title?: string;
+  }): Promise<Chat>;
+  get(id: string): Promise<Chat | null>;
+  list(): Promise<readonly Chat[]>;
+  send(id: string, content: string, options?: ChatSendOptions): Promise<SendChatResult>;
+  cancel(id: string): Promise<boolean>;
+  retry(id: string, messageId: string, options?: ChatSendOptions): Promise<SendChatResult>;
+  setTitle(id: string, title: string): Promise<Chat>;
+  remove(id: string): Promise<boolean>;
+};
+declare function useChat(id: string, options?: UseChatOptions): UseChatResult;
+declare function useChatList(): readonly ChatSummary[];
+//#endregion
 //#region src/ai/errors.d.ts
 type AIErrorCode = 'unsupported_runtime' | 'unsupported_os' | 'model_unavailable' | 'generation_failed' | 'generation_canceled' | 'concurrent_request' | 'guardrail_violation' | 'unsupported_language_or_locale' | 'rate_limited' | 'tool_failed' | 'image_generation_failed' | 'invalid_message' | 'invalid_schema' | 'invalid_model' | 'validation_failed' | 'context_window_exceeded';
 type ImageGenerationStage = 'download' | 'tokenize' | 'encode' | 'denoise' | 'decode' | 'write';
@@ -19,19 +210,21 @@ declare class ImageGenerationError extends AIError {
 //#endregion
 //#region src/ai/image-models.d.ts
 type FluxImageModelId = 'flux2-klein-4b-8bit-abliterated';
+type ImageModelId = FluxImageModelId | 'ternary-bonsai-image-4b';
+type ImageModel = ImageModelId | BonsaiImageModel;
 type ImageModelCapability = 'text-to-image' | 'image-to-image' | 'inpainting' | 'outpainting' | 'references' | 'lora';
 type ImageModelMetadata = {
   displayName: string;
-  family: 'flux2-klein';
-  precision: '8-bit';
+  family: 'flux2-klein' | 'bonsai-image';
+  precision: '8-bit' | '2-bit';
   estimatedMemoryGB: number;
   quality: 'high';
   speed: 'fast';
   capabilities: ImageModelCapability[];
 };
-type FluxImageModelManifest = {
+type ImageModelManifest = {
   formatVersion: 1;
-  model: FluxImageModelId;
+  model: ImageModelId;
   directory: string;
   metadata: ImageModelMetadata;
   components: {
@@ -40,6 +233,9 @@ type FluxImageModelManifest = {
     vae: FluxModelComponentManifest;
   };
 };
+type FluxImageModelManifest = ImageModelManifest & {
+  model: FluxImageModelId;
+};
 type FluxModelComponentManifest = {
   repo: string;
   revision: string;
@@ -51,13 +247,13 @@ type FluxTokenizedPrompt = {
   attentionMask: number[];
 };
 type FluxImageModelPaths = {
-  modelRoot: string;
+  path: string;
   textEncoderPath: string;
   transformerPath: string;
   vaePath: string;
 };
 type PrepareImageModelProgress = {
-  model: FluxImageModelId;
+  model: ImageModelId;
   file: string;
   index: number;
   total: number;
@@ -76,7 +272,8 @@ type PrepareImageLoRAProgress = {
   fraction?: number;
 };
 type PrepareImageModelOptions = {
-  model?: FluxImageModelId;
+  model?: ImageModel;
+  directory?: string;
   accessToken?: string;
   concurrency?: number;
   onProgress?: (progress: PrepareImageModelProgress) => void;
@@ -98,7 +295,7 @@ type ImageLoRASource = {
 };
 type ImageLoRAPrecision = 'float16' | 'bfloat16' | 'float32';
 type ImageLoRAMetadata = {
-  baseModel?: FluxImageModelId;
+  baseModel?: ImageModelId;
   precision?: ImageLoRAPrecision;
   recommendedScale?: number;
   sigmaSchedule?: number[];
@@ -130,7 +327,7 @@ type ImageLoRAState = PreparedImageLoRA & {
   size?: number;
 };
 type PreparedImageModel = FluxImageModelPaths & {
-  model: FluxImageModelId;
+  model: ImageModelId;
   tokenizer: FluxTokenizer;
 };
 type FluxTokenizer = (input: {
@@ -175,8 +372,8 @@ type GenerateImageOptions = {
   guidance?: number;
   strength?: number;
   seed?: number;
-  model?: FluxImageModelId;
-  modelRoot?: string;
+  model?: ImageModel;
+  path?: string;
   tokenizer?: FluxTokenizer;
   accessToken?: string;
   concurrency?: number;
@@ -193,32 +390,44 @@ type GenerateImageResult = {
   uri: string;
   width: number;
   height: number;
-  model: FluxImageModelId;
+  model: ImageModelId;
   metadata: GenerateImageMetadata;
 };
 type GenerateImagesOptions = GenerateImageOptions & {
   count: number;
 };
-type ImageGenerationJobStatus = 'queued' | 'running' | 'completed' | 'failed' | 'canceled';
-type ImageGenerationJobSnapshot = {
+type QueueImageOptions = Omit<GenerateImageOptions, 'accessToken' | 'onModelProgress' | 'onProgress' | 'signal' | 'tokenizer'>;
+type ImageJobRequest = Omit<QueueImageOptions, 'model'> & {
+  model?: ImageModelId;
+  seed: number;
+};
+type JobStatus = 'queued' | 'running' | 'canceling' | 'paused' | 'completed' | 'failed' | 'canceled';
+type JobError = {
+  name: string;
+  message: string;
+  code?: string;
+  stage?: string;
+};
+type ImageJob = {
   id: string;
-  status: ImageGenerationJobStatus;
+  kind: 'image';
+  status: JobStatus;
+  request: ImageJobRequest;
+  attempt: number;
   createdAt: string;
+  updatedAt: string;
   startedAt?: string;
   completedAt?: string;
   progress?: ImagegenProgress;
   result?: GenerateImageResult;
-  error?: unknown;
+  error?: JobError;
 };
-type ImageGenerationJob = {
-  id: string;
-  promise: Promise<GenerateImageResult>;
-  cancel(): boolean;
-  getSnapshot(): ImageGenerationJobSnapshot;
-  subscribe(listener: (snapshot: ImageGenerationJobSnapshot) => void): () => void;
+type Job = ImageJob;
+type JobSummary = Pick<Job, 'attempt' | 'completedAt' | 'createdAt' | 'id' | 'kind' | 'progress' | 'startedAt' | 'status' | 'updatedAt'> & {
+  prompt: string;
 };
 type GenerateImageMetadata = {
-  model: FluxImageModelId;
+  model: ImageModelId;
   width: number;
   height: number;
   steps: number;
@@ -234,8 +443,8 @@ type GenerateImageMetadata = {
   sigmaSchedule?: number[];
 };
 type ImageModelState = {
-  model: FluxImageModelId;
-  modelRoot: string;
+  model: ImageModelId;
+  path: string;
   paths: FluxImageModelPaths;
   installed: boolean;
   verified: boolean;
@@ -260,9 +469,10 @@ type UpscaleImageResult = {
   scale: number;
 };
 declare const flux2Klein4B8BitAbliterated: FluxImageModelManifest;
-declare const imageModelManifests: readonly [FluxImageModelManifest];
-declare function imageModelManifest(model: FluxImageModelId): FluxImageModelManifest | undefined;
-declare function fluxImageModelPaths(modelRoot: string, manifest?: FluxImageModelManifest): FluxImageModelPaths;
+declare const ternaryBonsaiImage4B: ImageModelManifest;
+declare const imageModelManifests: readonly [FluxImageModelManifest, ImageModelManifest];
+declare function imageModelManifest(model: ImageModelId): ImageModelManifest | FluxImageModelManifest | undefined;
+declare function fluxImageModelPaths(path: string, manifest?: ImageModelManifest): FluxImageModelPaths;
 //#endregion
 //#region src/ai/image.d.ts
 declare const defaultImageLongSide = 1152;
@@ -270,15 +480,15 @@ declare function normalizeImageSize(size: ImageSize, maxLongSide?: number | fals
 declare function imageSizeForPreset(aspectRatio: ImageAspectRatio, maxLongSide?: number): ImageSize;
 declare function prepareImageModel(options?: PrepareImageModelOptions): Promise<PreparedImageModel>;
 declare function verifyImageModel(options?: {
-  model?: FluxImageModelId;
-  modelRoot?: string;
+  model?: ImageModel;
+  path?: string;
 }): Promise<ImageModelState>;
 declare function listImageModels(options?: {
-  modelRoot?: string;
+  directory?: string;
 }): Promise<ImageModelState[]>;
 declare function deleteImageModel(options?: {
-  model?: FluxImageModelId;
-  modelRoot?: string;
+  model?: ImageModel;
+  path?: string;
 }): Promise<void>;
 declare function imageMemoryTelemetry(): Promise<ImageMemoryTelemetry>;
 declare function unloadImageModel(): Promise<boolean>;
@@ -289,18 +499,23 @@ declare function listImageLoRAs(loras: ImageLoRADefinition[]): Promise<ImageLoRA
 declare function deleteImageLoRA(lora: string | ImageLoRADefinition): Promise<void>;
 declare function generateImage(options: GenerateImageOptions): Promise<GenerateImageResult>;
 declare function generateImages(options: GenerateImagesOptions): Promise<GenerateImageResult[]>;
-declare function scheduleImageGeneration(options: GenerateImageOptions): ImageGenerationJob;
-declare function getImageGenerationJob(id: string): ImageGenerationJob | undefined;
-declare function listImageGenerationJobs(): {
-  id: string;
-  status: ImageGenerationJobStatus;
-  createdAt: string;
-  startedAt?: string;
-  completedAt?: string;
-  progress?: ImagegenProgress;
-  result?: GenerateImageResult;
-  error?: unknown;
-}[];
+declare function queueImage(options: QueueImageOptions): Promise<ImageJob>;
+declare const image: Readonly<{
+  generate: typeof generateImage;
+  generateMany: typeof generateImages;
+  queue: typeof queueImage;
+  upscale: typeof upscaleImage;
+}>;
+declare const jobs: Readonly<{
+  get: (id: string) => Promise<ImageJob | undefined>;
+  list: () => Promise<JobSummary[]>;
+  cancel: (id: string) => Promise<boolean>;
+  retry: (id: string) => Promise<boolean>;
+  resume: (id: string) => Promise<boolean>;
+  remove: (id: string) => Promise<boolean>;
+}>;
+declare function useJob(id: string): ImageJob | undefined;
+declare function useJobList(): JobSummary[];
 //#endregion
 //#region src/ai/image-composer.d.ts
 type ImageComposerMode = 'text' | 'subject' | 'edit' | 'outpaint';
@@ -308,7 +523,9 @@ type ImageComposerSize = {
   width: number;
   height: number;
 };
-type ImageComposerImage = FileRef & {
+type ImageComposerImage = {
+  path: string;
+  uri: string;
   size: ImageComposerSize;
 };
 type ImageComposerRect = {
@@ -337,75 +554,6 @@ type ImageGenerationRequestInput = {
   source: ImageComposerImage | null;
   rect: ImageComposerRect | null;
 };
-type ImageComposerOptions = {
-  initialMode?: ImageComposerMode;
-  maxRefs?: number;
-  inputRoot?: string;
-};
-type TextComposerState = {
-  refs: ImageComposerImage[];
-  isPicking: boolean;
-  error: Error | null;
-};
-type SubjectComposerState = TextComposerState & {
-  subject: ImageComposerImage | null;
-};
-type EditComposerState = TextComposerState & {
-  source: ImageComposerImage | null;
-};
-type OutpaintComposerState = {
-  source: ImageComposerImage | null;
-  rect: ImageComposerRect | null;
-  isPicking: boolean;
-  error: Error | null;
-};
-type TextImageComposer = TextComposerState & {
-  mode: 'text';
-  pickRefs(): Promise<void>;
-  removeRef(index: number): void;
-  request(prompt: string): ImageGenerationRequestInput;
-  clear(): void;
-};
-type SubjectImageComposer = SubjectComposerState & {
-  mode: 'subject';
-  pickSubject(): Promise<void>;
-  removeSubject(): void;
-  pickRefs(): Promise<void>;
-  removeRef(index: number): void;
-  request(prompt: string): ImageGenerationRequestInput;
-  clear(): void;
-};
-type EditImageComposer = EditComposerState & {
-  mode: 'edit';
-  pickSource(): Promise<void>;
-  removeSource(): void;
-  pickRefs(): Promise<void>;
-  removeRef(index: number): void;
-  request(prompt: string): ImageGenerationRequestInput;
-  clear(): void;
-};
-type OutpaintImageComposer = OutpaintComposerState & {
-  mode: 'outpaint';
-  pickSource(): Promise<void>;
-  removeSource(): void;
-  setRect(rect: ImageComposerRect | null): void;
-  request(prompt: string): ImageGenerationRequestInput;
-  clear(): void;
-};
-type ImageComposer = TextImageComposer | SubjectImageComposer | EditImageComposer | OutpaintImageComposer;
-type ImageComposerController = {
-  mode: ImageComposerMode;
-  setMode(mode: ImageComposerMode): void;
-  active: ImageComposer;
-  text: TextImageComposer;
-  subject: SubjectImageComposer;
-  edit: EditImageComposer;
-  outpaint: OutpaintImageComposer;
-  isPicking: boolean;
-  error: Error | null;
-  request(prompt: string): ImageGenerationRequestInput;
-  clear(): void;
-};
 declare function roundImageDimension(value: number): number;
 declare function imageSizeForAspectRatio(size: ImageComposerSize, maxSide?: number): ImageComposerSize;
 declare function roundedImageSize(size: ImageComposerSize): ImageComposerSize;
@@ -415,121 +563,6 @@ declare function rectForSourceScale(sourceSize: ImageComposerSize, scale: number
   y: number;
 }): ImageComposerRect;
 declare function initialOutpaintRect(sourceSize: ImageComposerSize): ImageComposerRect;
-declare function buildImageGenerationRequest(input: ImageGenerationRequestInput, options?: Pick<GenerateImageOptions, 'signal' | 'onModelProgress' | 'onProgress' | 'model' | 'modelRoot' | 'tokenizer' | 'accessToken' | 'concurrency' | 'loras'>): GenerateImageOptions;
-declare function useImageComposer(options?: ImageComposerOptions): ImageComposerController;
-declare function useImageComposer(mode: 'text', options?: ImageComposerOptions): TextImageComposer;
-declare function useImageComposer(mode: 'subject', options?: ImageComposerOptions): SubjectImageComposer;
-declare function useImageComposer(mode: 'edit', options?: ImageComposerOptions): EditImageComposer;
-declare function useImageComposer(mode: 'outpaint', options?: ImageComposerOptions): OutpaintImageComposer;
-declare function useImageGeneration(): {
-  run: (input: ImageGenerationRequestInput, options?: Pick<GenerateImageOptions, "loras">) => Promise<GenerateImageResult | null>;
-  cancel: () => void;
-  clear: () => void;
-  imageUri: string | null;
-  result: GenerateImageResult | null;
-  progress: number;
-  status: string;
-  isGenerating: boolean;
-  error: Error | null;
-};
-declare function useImageModel(): {
-  installed: boolean;
-  verified: boolean;
-  missingFiles: string[];
-  bytesPresent: number;
-  totalBytes: number;
-  check: () => Promise<ImageModelState | null>;
-  checking: boolean;
-  state: ImageModelState | null;
-  error: Error | null;
-};
-//#endregion
-//#region src/ai/shape.d.ts
-type PrimitiveToken = 'string' | 'number' | 'boolean';
-type ShapeArray = readonly [ShapeDefinition];
-type ShapeObject = {
-  readonly [key: string]: ShapeDefinition;
-};
-type ShapeDefinition = PrimitiveToken | ShapeObject | ShapeArray;
-type InferShapeValue<Definition extends ShapeDefinition> = Definition extends 'string' ? string : Definition extends 'number' ? number : Definition extends 'boolean' ? boolean : Definition extends readonly [infer Item] ? Item extends ShapeDefinition ? Array<InferShapeValue<Item>> : never : Definition extends ShapeObject ? { [Key in keyof Definition]: InferShapeValue<Definition[Key]> } : never;
-type InferShape<Definition extends ShapeObject> = { [Key in keyof Definition]: InferShapeValue<Definition[Key]> };
-type BridgeSchema = {
-  type: PrimitiveToken;
-} | {
-  type: 'array';
-  items: BridgeSchema;
-} | {
-  type: 'object';
-  properties: Record<string, BridgeSchema>;
-  required: string[];
-};
-type StructuredInput = {
-  value: unknown;
-  shape: BleamAIShape<unknown, unknown> | StandardSchema;
-};
-type StructuredSchema = BleamAIShape<unknown, unknown> | StandardSchema;
-type BridgeStructuredSchema = StandardSchema & {
-  toJSONSchema(): BridgeSchema;
-};
-interface BleamAIShape<Input, Output$1> extends StandardSchema<Input, Output$1> {
-  parse(value: unknown): Output$1;
-  safeParse(value: unknown): SchemaResult<Output$1>;
-  toJSONSchema(): BridgeSchema;
-}
-type InferShapeOutput<Value extends BleamAIShape<unknown, unknown>> = Value extends BleamAIShape<unknown, infer Output> ? Output : never;
-declare function shape<Definition extends ShapeObject>(definition: Definition): BleamAIShape<InferShape<Definition>, InferShape<Definition>>;
-type GenerateResult<Output$1 extends StructuredSchema | undefined> = Output$1 extends BleamAIShape<unknown, infer ShapeOutput> ? ShapeOutput : Output$1 extends StandardSchema ? InferOutput<Output$1> : string;
-//#endregion
-//#region src/ai/native.d.ts
-type ModelAvailabilityResult = {
-  available: boolean;
-  reason?: 'device_not_eligible' | 'apple_intelligence_not_enabled' | 'model_not_ready' | 'unknown';
-};
-//#endregion
-//#region src/ai/text.d.ts
-type AITool<Input extends BridgeStructuredSchema = BridgeStructuredSchema> = {
-  name: string;
-  description: string;
-  input: Input;
-  execute(input: InferToolInput<Input>): unknown | Promise<unknown>;
-};
-type InferToolInput<Input extends BridgeStructuredSchema> = Input extends {
-  readonly '~standard': {
-    readonly types?: {
-      readonly output: infer Output;
-    };
-  };
-} ? Output : unknown;
-type GenerateOptions<Output$1 extends BridgeStructuredSchema | undefined = undefined> = {
-  system?: string;
-  prompt?: string;
-  input?: string | StructuredInput;
-  output?: Output$1;
-  temperature?: number;
-  maxTokens?: number;
-  tools?: readonly AITool[];
-  signal?: AbortSignal;
-};
-type StreamOptions = Omit<GenerateOptions<undefined>, 'output'>;
-type ModelAvailability = ModelAvailabilityResult;
-declare function isAvailable(): Promise<boolean>;
-declare function availability(): Promise<ModelAvailability>;
-declare function generate<Output$1 extends BridgeStructuredSchema | undefined = undefined>(options: GenerateOptions<Output$1>): Promise<GenerateResult<Output$1>>;
-declare function stream(options: StreamOptions): AsyncGenerator<string>;
-//#endregion
-//#region src/ai/text-hook.d.ts
-type TextGenerationState = {
-  text: string;
-  isGenerating: boolean;
-  error: Error | null;
-};
-declare function useTextGeneration(): {
-  run: (options: StreamOptions) => Promise<string | null>;
-  cancel: () => void;
-  clear: () => void;
-  text: string;
-  isGenerating: boolean;
-  error: Error | null;
-};
+declare function buildImageGenerationRequest(input: ImageGenerationRequestInput, options?: Pick<GenerateImageOptions, 'signal' | 'onModelProgress' | 'onProgress' | 'model' | 'path' | 'tokenizer' | 'accessToken' | 'concurrency' | 'loras'>): GenerateImageOptions;
 //#endregion
-export { AIError, AIErrorCode, AITool, BleamAIShape, EditImageComposer, FluxImageModelId, FluxImageModelManifest, FluxImageModelPaths, FluxModelComponentManifest, FluxTokenizedPrompt, FluxTokenizer, GenerateImageMetadata, GenerateImageOptions, GenerateImageResult, GenerateImagesOptions, GenerateOptions, ImageAspectRatio, ImageComposer, ImageComposerController, ImageComposerImage, ImageComposerMode, ImageComposerRect, ImageComposerSize, ImageGenerationError, ImageGenerationErrorDetails, ImageGenerationJob, ImageGenerationJobSnapshot, ImageGenerationJobStatus, ImageGenerationRequestInput, ImageGenerationStage, ImageLoRADefinition, ImageLoRAMetadata, ImageLoRAPrecision, ImageLoRASource, ImageLoRAState, ImageMaskMode, ImageMemoryTelemetry, ImageModelCapability, ImageModelMetadata, ImageModelState, ImageOutpaintOptions, ImageOutpaintSourceRect, ImageResizeMode, ImageSize, ImagegenProgress, InferShapeOutput, ModelAvailability, OutpaintImageComposer, PrepareImageLoRAOptions, PrepareImageLoRAProgress, PrepareImageModelOptions, PrepareImageModelProgress, PreparedImageLoRA, PreparedImageModel, StreamOptions, SubjectImageComposer, TextGenerationState, TextImageComposer, UpscaleImageOptions, UpscaleImageResult, availability, buildImageGenerationRequest, clampOutpaintRect, defaultImageLongSide, deleteImageLoRA, deleteImageModel, flux2Klein4B8BitAbliterated, fluxImageModelPaths, generate, generateImage, generateImages, getImageGenerationJob, imageMemoryTelemetry, imageModelManifest, imageModelManifests, imageSizeForAspectRatio, imageSizeForPreset, initialOutpaintRect, isAvailable, listImageGenerationJobs, listImageLoRAs, listImageModels, normalizeImageSize, prepareImageLoRA, prepareImageModel, rectForSourceScale, roundImageDimension, roundedImageSize, scheduleImageGeneration, shape, stream, unloadImageModel, upscaleImage, useImageComposer, useImageGeneration, useImageModel, useTextGeneration, verifyImageLoRA, verifyImageModel };
+export { AIError, AIErrorCode, type AssistantChatMessage, type AssistantMessageStatus, BonsaiImageModel, BonsaiImageModelId, BonsaiModel, BonsaiTextModel, BonsaiTextModelId, BonsaiTextModelMetadata, type Chat, type ChatError, type ChatGenerationOptions, type ChatMessage, type ChatRunOptions, type ChatSendOptions, type ChatStatus, type ChatSummary, FluxImageModelId, FluxImageModelManifest, FluxImageModelPaths, FluxModelComponentManifest, FluxTokenizedPrompt, FluxTokenizer, GenerateImageMetadata, GenerateImageOptions, GenerateImageResult, GenerateImagesOptions, ImageAspectRatio, ImageComposerImage, ImageComposerMode, ImageComposerRect, ImageComposerSize, ImageGenerationError, ImageGenerationErrorDetails, ImageGenerationRequestInput, ImageGenerationStage, ImageJob, ImageJobRequest, ImageLoRADefinition, ImageLoRAMetadata, ImageLoRAPrecision, ImageLoRASource, ImageLoRAState, ImageMaskMode, ImageMemoryTelemetry, ImageModel, ImageModelCapability, ImageModelId, ImageModelManifest, ImageModelMetadata, ImageModelState, ImageOutpaintOptions, ImageOutpaintSourceRect, ImageResizeMode, ImageSize, ImagegenProgress, Job, JobError, JobStatus, JobSummary, PrepareImageLoRAOptions, PrepareImageLoRAProgress, PrepareImageModelOptions, PrepareImageModelProgress, PrepareBonsaiTextModelOptions as PrepareTextModelOptions, PrepareBonsaiTextModelProgress as PrepareTextModelProgress, PreparedImageLoRA, PreparedImageModel, PreparedBonsaiTextModel as PreparedTextModel, QueueImageOptions, type SendChatResult, BonsaiTextModelState as TextModelState, UpscaleImageOptions, UpscaleImageResult, type UseChatOptions, type UseChatResult, type UserChatMessage, bonsai, bonsaiTextModelCatalog, buildImageGenerationRequest, chat, clampOutpaintRect, defaultImageLongSide, deleteImageLoRA, deleteImageModel, deleteBonsaiTextModel as deleteTextModel, flux2Klein4B8BitAbliterated, fluxImageModelPaths, image, imageMemoryTelemetry, imageModelManifest, imageModelManifests, imageSizeForAspectRatio, imageSizeForPreset, initialOutpaintRect, jobs, listImageLoRAs, listImageModels, listBonsaiTextModels as listTextModels, normalizeImageSize, prepareImageLoRA, prepareImageModel, prepareBonsaiTextModel as prepareTextModel, rectForSourceScale, roundImageDimension, roundedImageSize, ternaryBonsaiImage4B, unloadImageModel, useChat, useChatList, useJob, useJobList, verifyImageLoRA, verifyImageModel, verifyBonsaiTextModel as verifyTextModel };