@avocadostudio-ai/orchestrator-core 0.1.0

package/dist/chat/variation-pipeline.d.ts

@@ -0,0 +1,135 @@
+import type { Logger } from "../logger.ts";
+import { z } from "zod";
+import { type BlockType, type PageDoc } from "@avocadostudio-ai/shared";
+import { type AIProvider, type ModelKey } from "../state/session-state.ts";
+import { resolveUnsplashImage } from "../image/image-helpers.ts";
+import type { UnsplashImage } from "../variation-images.ts";
+export declare const variationRequestBodySchema: z.ZodObject<{
+    session: z.ZodOptional<z.ZodString>;
+    siteId: z.ZodOptional<z.ZodString>;
+    sitePurpose: z.ZodOptional<z.ZodString>;
+    siteHosting: z.ZodOptional<z.ZodString>;
+    businessContext: z.ZodOptional<z.ZodUnion<readonly [z.ZodObject<{
+        purpose: z.ZodOptional<z.ZodString>;
+        tone: z.ZodOptional<z.ZodString>;
+        constraints: z.ZodOptional<z.ZodArray<z.ZodString>>;
+    }, z.core.$strip>, z.ZodString]>>;
+    siteContext: z.ZodOptional<z.ZodUnion<readonly [z.ZodObject<{
+        siteId: z.ZodOptional<z.ZodString>;
+        siteName: z.ZodOptional<z.ZodString>;
+        purpose: z.ZodOptional<z.ZodString>;
+        hosting: z.ZodOptional<z.ZodString>;
+        tone: z.ZodOptional<z.ZodString>;
+        constraints: z.ZodOptional<z.ZodArray<z.ZodString>>;
+    }, z.core.$strip>, z.ZodString]>>;
+    slug: z.ZodOptional<z.ZodString>;
+    message: z.ZodOptional<z.ZodString>;
+    modelKey: z.ZodOptional<z.ZodEnum<{
+        fast: "fast";
+        balanced: "balanced";
+        reasoning: "reasoning";
+        codex: "codex";
+    }>>;
+    provider: z.ZodOptional<z.ZodEnum<{
+        openai: "openai";
+        anthropic: "anthropic";
+        gemini: "gemini";
+    }>>;
+    activeBlockId: z.ZodOptional<z.ZodString>;
+    activeBlockType: z.ZodOptional<z.ZodString>;
+    activeEditablePath: z.ZodOptional<z.ZodString>;
+    locale: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export type VariationRequestBody = z.infer<typeof variationRequestBodySchema>;
+export type VariationOption = {
+    id: string;
+    title: string;
+    summary: string;
+    patch: Record<string, unknown>;
+    changedKeys: string[];
+};
+export type VariationResult = {
+    status: "ok";
+    summary: string;
+    blockId: string;
+    blockType: BlockType;
+    pageSlug: string;
+    baseProps: Record<string, unknown>;
+    variations: VariationOption[];
+    suggestions?: string[];
+    plannerSource: "openai" | "anthropic" | "gemini" | "demo";
+    modelUsed: string;
+    modelKey: ModelKey;
+    usage?: {
+        inputTokens: number;
+        outputTokens: number;
+        totalTokens: number;
+        cacheCreationInputTokens?: number;
+        cacheReadInputTokens?: number;
+        estimatedUsd: number | null;
+    };
+};
+export type VariationIntent = "generate" | "regenerate" | "show";
+export declare function getCachedVariations(session: string, blockId: string): VariationResult | null;
+export type VariationPipelineContext = {
+    log: Logger;
+    modelLookup: Record<AIProvider, Record<ModelKey, string>>;
+    availableProviders: AIProvider[];
+};
+export declare function setResolveUnsplashImageForTests(fn?: typeof resolveUnsplashImage): void;
+type GenerateVariationImageFn = (args: {
+    prompt: string;
+    altText: string;
+}) => Promise<UnsplashImage | null>;
+export declare function setGenerateVariationImageForTests(fn?: GenerateVariationImageFn): void;
+export declare function parseJsonMaybe(text: string): unknown;
+export declare function requestedVariationCount(message: string): number;
+export declare function withDefaultImageVariations(args: {
+    block: PageDoc["blocks"][number];
+    message: string;
+    variations: VariationOption[];
+    page?: PageDoc;
+    log?: Logger;
+}): Promise<VariationOption[]>;
+export type VariationImageUpdate = {
+    variationId: string;
+    index: number;
+    patch: Record<string, unknown>;
+    changedKeys: string[];
+    imageUrl?: string;
+    imageAlt?: string;
+};
+export type StreamingVariationsPlan = {
+    initial: VariationOption[];
+    run: (onProgress: (update: VariationImageUpdate) => void) => Promise<VariationOption[]>;
+};
+export declare function planDefaultImageVariations(args: {
+    block: PageDoc["blocks"][number];
+    message: string;
+    variations: VariationOption[];
+    page?: PageDoc;
+    log?: Logger;
+}): StreamingVariationsPlan;
+export declare function runVariationPipeline(ctx: VariationPipelineContext, body: VariationRequestBody, options?: {
+    intent?: VariationIntent;
+}): Promise<{
+    code: number;
+    payload: VariationResult | {
+        error: string;
+    };
+}>;
+export type VariationStreamPipelineHooks = {
+    onInitial: (result: VariationResult & {
+        imagesPending: boolean;
+    }) => void;
+    onImageResolved: (update: VariationImageUpdate) => void;
+};
+export declare function runVariationPipelineStreaming(ctx: VariationPipelineContext, body: VariationRequestBody, hooks: VariationStreamPipelineHooks, options?: {
+    intent?: VariationIntent;
+}): Promise<{
+    code: number;
+    payload: VariationResult | {
+        error: string;
+    };
+}>;
+export {};