diff --git a/src/client/AiplClients.ts b/src/client/AiplClients.ts index 1495f6d..bdfbed8 100644 --- a/src/client/AiplClients.ts +++ b/src/client/AiplClients.ts @@ -1,11 +1,12 @@ import { isUndefined, type TypeInfo } from "@mjtdev/engine"; // import { connectWs } from "./connectWs"; -import { type AppMessageMap } from "ai-worker-common"; +import { type AppMessageMap, type SdApiTxt2ImgRequest } from "ai-worker-common"; import { getBackendUser } from "../backend/user/getBackendUser"; import { startPublicAccessPoint } from "../startPublicAccessPoint"; import { AppMessagesState } from "../state/ws/AppMessagesState"; import { findFirstPapId } from "../ui/overlay/findFirstPapId"; import { log } from "./log"; +import { askForGeneratedImages } from "../ai/askForGeneratedImages"; export type AiplClient = ReturnType; @@ -33,6 +34,12 @@ export const createAiplClient = ( systemMessage, }); }, + askForGeneratedImages: async ( + request: Partial = {}, + options: Partial<{ signal: AbortSignal }> = {} + ) => { + return askForGeneratedImages(request, options); + }, listAgents: async () => { const user = await getBackendUser(); if (isUndefined(user)) {