@m5kdev/backend 0.8.6 → 0.8.8

package/dist/src/modules/ai/ai.service.cjs

@@ -0,0 +1,312 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+require("../../../_virtual/_rolldown/runtime.cjs");
+const require_src_modules_base_base_service = require("../base/base.service.cjs");
+const require_src_modules_ai_ai_prompts = require("./ai.prompts.cjs");
+let neverthrow = require("neverthrow");
+let _m5kdev_commons_modules_ai_ai_constants = require("@m5kdev/commons/modules/ai/ai.constants");
+let _m5kdev_commons_modules_ai_ai_utils = require("@m5kdev/commons/modules/ai/ai.utils");
+let _mastra_core_request_context = require("@mastra/core/request-context");
+let _mastra_rag = require("@mastra/rag");
+let ai = require("ai");
+let jsonrepair = require("jsonrepair");
+//#region src/modules/ai/ai.service.ts
+var AIService = class extends require_src_modules_base_base_service.BaseService {
+	helpers = { arrayToPseudoXML: _m5kdev_commons_modules_ai_ai_utils.arrayToPseudoXML };
+	mastra;
+	openrouter;
+	replicate;
+	options;
+	constructor(repositories, services, libs, options) {
+		super(repositories, services);
+		this.mastra = libs.mastra;
+		this.openrouter = libs.openrouter;
+		this.replicate = libs.replicate;
+		this.options = options;
+	}
+	getMastra() {
+		if (!this.mastra) throw new Error("Mastra is not available");
+		return this.mastra;
+	}
+	prepareModel(model) {
+		if (!this.openrouter) throw new Error("OpenRouter is not configured");
+		return this.openrouter.chat(model, { usage: { include: true } });
+	}
+	prepareEmbeddingModel(model) {
+		if (!this.openrouter) throw new Error("OpenRouter is not configured");
+		const openrouter = this.openrouter;
+		return openrouter.embeddingModel?.(model) ?? openrouter.textEmbeddingModel(model);
+	}
+	async agentUse(agent, options, ctx) {
+		return this.throwableAsync(async () => {
+			this.logger.info("AGENT USE");
+			const { prompt, messages, ...rest } = options;
+			const payload = messages || prompt;
+			if (!payload) return this.error("BAD_REQUEST", "No prompt or messages provided");
+			const requestContext = options.requestContext ?? new _mastra_core_request_context.RequestContext();
+			if (ctx?.actor) requestContext.set("userId", ctx.actor.userId);
+			if (ctx?.model) requestContext.set("model", ctx.model);
+			const result = await this.getMastra().getAgent(agent).generate(payload, {
+				...rest,
+				requestContext: rest.requestContext ?? requestContext
+			});
+			this.logger.info("AGENT USE DONE");
+			if (this.repository.aiUsage) {
+				const createUsageResult = await this.repository.aiUsage.create({
+					userId: ctx?.actor?.userId,
+					model: ctx?.model ?? "unknown",
+					provider: "openrouter",
+					feature: agent,
+					traceId: result.traceId,
+					inputTokens: result.usage.inputTokens,
+					outputTokens: result.usage.outputTokens,
+					totalTokens: result.usage.totalTokens,
+					cost: (result?.providerMetadata?.openrouter?.usage)?.cost ?? 0
+				});
+				if (createUsageResult.isErr()) return (0, neverthrow.err)(createUsageResult.error);
+			}
+			this.logger.info("AGENT USE CREATED USAGE");
+			return (0, neverthrow.ok)(result);
+		});
+	}
+	async agentText(agent, options, ctx) {
+		const result = await this.agentUse(agent, options, ctx);
+		if (result.isErr()) return this.error("SERVICE_UNAVAILABLE", "AI: Agent text failed", { cause: result.error });
+		return (0, neverthrow.ok)(result.value.text);
+	}
+	async agentTextResult(agent, options, ctx) {
+		const result = await this.agentUse(agent, options, ctx);
+		if (result.isErr()) return (0, neverthrow.err)(result.error);
+		return (0, neverthrow.ok)(result.value);
+	}
+	async agentObject(agent, options, ctx) {
+		const { schema, ...rest } = options;
+		const result = await this.agentUse(agent, {
+			...rest,
+			structuredOutput: { schema }
+		}, ctx);
+		if (result.isErr()) return this.error("SERVICE_UNAVAILABLE", "AI: Agent object failed", { cause: result.error });
+		return (0, neverthrow.ok)(result.value.object);
+	}
+	async agentObjectResult(agent, options, ctx) {
+		this.logger.info("AGENT OBJECT RESULT");
+		const { schema, ...rest } = options;
+		const result = await this.agentUse(agent, {
+			...rest,
+			structuredOutput: { schema }
+		}, ctx);
+		if (result.isErr()) return (0, neverthrow.err)(result.error);
+		this.logger.info("AGENT OBJECT RESULT DONE");
+		return (0, neverthrow.ok)({
+			...result.value,
+			object: result.value.object
+		});
+	}
+	async embedDocument(value, options, type = "text", model = _m5kdev_commons_modules_ai_ai_constants.OPENAI_TEXT_EMBEDDING_3_SMALL) {
+		return this.throwableAsync(async () => {
+			if (type === "text") {
+				const chunks = await _mastra_rag.MDocument.fromText(value).chunk(options ?? {
+					strategy: "recursive",
+					maxSize: 512,
+					overlap: 50,
+					separators: ["\n"]
+				});
+				const embeddings = await this.embedMany(chunks, model);
+				if (embeddings.isErr()) return (0, neverthrow.err)(embeddings.error);
+				return (0, neverthrow.ok)({
+					embeddings: embeddings.value.embeddings,
+					chunks
+				});
+			}
+			return this.error("BAD_REQUEST", "Unsupported document type");
+		});
+	}
+	async embed(text, model = _m5kdev_commons_modules_ai_ai_constants.OPENAI_TEXT_EMBEDDING_3_SMALL) {
+		return this.throwableAsync(async () => {
+			return (0, neverthrow.ok)(await (0, ai.embed)({
+				model: this.prepareEmbeddingModel(model),
+				value: text
+			}));
+		});
+	}
+	async embedMany(chunks, model = _m5kdev_commons_modules_ai_ai_constants.OPENAI_TEXT_EMBEDDING_3_SMALL) {
+		return this.throwableAsync(async () => {
+			return (0, neverthrow.ok)(await (0, ai.embedMany)({
+				model: this.prepareEmbeddingModel(model),
+				values: chunks.map((chunk) => chunk.text)
+			}));
+		});
+	}
+	async generateText(params) {
+		return this.throwableAsync(async () => {
+			const { removeMDash = this.options?.removeMDash ?? true, model, prompt, messages, ctx, retryAttempts = this.options?.retryAttempts ?? 0, retryModels = this.options?.retryModels ?? [], ...rest } = params;
+			const request = messages ? {
+				...rest,
+				model: this.prepareModel(model),
+				messages
+			} : {
+				...rest,
+				model: this.prepareModel(model),
+				prompt
+			};
+			try {
+				const result = await (0, ai.generateText)(request);
+				if (this.repository.aiUsage) {
+					const createUsageResult = await this.repository.aiUsage.create({
+						userId: ctx?.actor?.userId,
+						model,
+						provider: "openrouter",
+						feature: "generateText",
+						traceId: result.providerMetadata?.openrouter?.traceId?.toString(),
+						inputTokens: result.usage.inputTokens,
+						outputTokens: result.usage.outputTokens,
+						totalTokens: result.usage.totalTokens,
+						cost: (result?.providerMetadata?.openrouter?.usage)?.cost ?? 0
+					});
+					if (createUsageResult.isErr()) return (0, neverthrow.err)(createUsageResult.error);
+				}
+				return (0, neverthrow.ok)(removeMDash ? result.text.replace(/\u2013|\u2014/g, "-") : result.text);
+			} catch (error) {
+				if (retryAttempts <= 0) throw error;
+				this.logger.warn(`generateText failed, retrying (${retryAttempts} attempts left)`, {
+					model,
+					error
+				});
+				const delay = Math.min(1e3 * 2 ** ((this.options?.retryAttempts ?? 3) - retryAttempts), 1e4);
+				await new Promise((resolve) => setTimeout(resolve, delay));
+				const nextModel = retryModels?.[0] ?? model;
+				const nextRetryModels = retryModels ? [...retryModels.slice(1), model] : void 0;
+				return this.generateText({
+					...rest,
+					...messages ? { messages } : { prompt },
+					model: nextModel,
+					removeMDash,
+					ctx,
+					retryAttempts: retryAttempts - 1,
+					retryModels: nextRetryModels
+				});
+			}
+		});
+	}
+	async generateObject(params) {
+		const { model, schema, prompt, messages, repairAttempts = this.options?.repairAttempts ?? 0, repairModel = this.options?.repairModel ?? model, ctx, retryAttempts = this.options?.retryAttempts ?? 0, retryModels = this.options?.retryModels ?? [], ...rest } = params;
+		const request = messages ? {
+			...rest,
+			model: this.prepareModel(model),
+			messages,
+			output: ai.Output.object({ schema })
+		} : {
+			...rest,
+			model: this.prepareModel(model),
+			prompt,
+			output: ai.Output.object({ schema })
+		};
+		try {
+			const result = await (0, ai.generateText)(request);
+			if (this.repository.aiUsage) {
+				const createUsageResult = await this.repository.aiUsage.create({
+					userId: ctx?.actor?.userId,
+					model,
+					provider: "openrouter",
+					feature: "generateObject",
+					traceId: result.providerMetadata?.openrouter?.traceId?.toString(),
+					inputTokens: result.usage.inputTokens,
+					outputTokens: result.usage.outputTokens,
+					totalTokens: result.usage.totalTokens,
+					cost: (result?.providerMetadata?.openrouter?.usage)?.cost ?? 0
+				});
+				if (createUsageResult.isErr()) return (0, neverthrow.err)(createUsageResult.error);
+			}
+			return (0, neverthrow.ok)(result.output);
+		} catch (error) {
+			if (ai.NoObjectGeneratedError.isInstance(error)) {
+				if (this.repository.aiUsage) {
+					const createUsageResult = await this.repository.aiUsage.create({
+						userId: ctx?.actor?.userId,
+						model,
+						provider: "openrouter",
+						feature: "generateObject",
+						traceId: null,
+						inputTokens: error?.usage?.inputTokens,
+						outputTokens: error?.usage?.outputTokens,
+						totalTokens: error?.usage?.totalTokens,
+						cost: 0
+					});
+					if (createUsageResult.isErr()) return (0, neverthrow.err)(createUsageResult.error);
+				}
+				if (error.text) {
+					const repairedText = (0, jsonrepair.jsonrepair)(error.text);
+					const parsed = schema.safeParse(repairedText);
+					if (parsed.success) return (0, neverthrow.ok)(parsed.data);
+					if (repairAttempts === 0) return this.error("PARSE_ERROR", "AI: Agent object failed", { cause: error });
+					return this.generateObject({
+						...rest,
+						prompt: require_src_modules_ai_ai_prompts.repairJsonPrompt.compile({
+							text: error.text,
+							error: JSON.stringify(error.cause ?? "Unknown error")
+						}),
+						repairAttempts: repairAttempts - 1,
+						model: repairModel ?? model,
+						schema,
+						ctx
+					});
+				}
+				return this.error("PARSE_ERROR", "AI: Agent object failed without text", { cause: error });
+			}
+			if (retryAttempts <= 0) return this.error("BAD_REQUEST", "AI: Provider failed to generate object", { cause: error });
+			this.logger.warn(`generateObject failed, retrying (${retryAttempts} attempts left)`, {
+				model,
+				error
+			});
+			const delay = Math.min(1e3 * 2 ** ((this.options?.retryAttempts ?? 3) - retryAttempts), 1e4);
+			await new Promise((resolve) => setTimeout(resolve, delay));
+			const nextModel = retryModels?.[0] ?? model;
+			const nextRetryModels = retryModels ? [...retryModels.slice(1), model] : void 0;
+			return this.generateObject({
+				...rest,
+				...messages ? { messages } : { prompt },
+				model: nextModel,
+				schema,
+				repairAttempts,
+				repairModel,
+				ctx,
+				retryAttempts: retryAttempts - 1,
+				retryModels: nextRetryModels
+			});
+		}
+	}
+	async generateReplicate(model, options) {
+		return this.throwableAsync(async () => {
+			if (!this.replicate) return this.error("INTERNAL_SERVER_ERROR", "Replicate is not configured");
+			try {
+				return (0, neverthrow.ok)(await this.replicate.run(model, options));
+			} catch (error) {
+				return this.error("INTERNAL_SERVER_ERROR", void 0, { cause: error });
+			}
+		});
+	}
+	async generateTranscript(file_url) {
+		const output = await this.generateReplicate("thomasmol/whisper-diarization:1495a9cddc83b2203b0d8d3516e38b80fd1572ebc4bc5700ac1da56a9b3ed886", { input: { file_url } });
+		if (output.isErr()) return (0, neverthrow.err)(output.error);
+		try {
+			const { segments } = output.value;
+			return (0, neverthrow.ok)({
+				text: segments.map((segment) => segment.text).join(""),
+				metadata: segments
+			});
+		} catch (error) {
+			return this.error("INTERNAL_SERVER_ERROR", void 0, { cause: error });
+		}
+	}
+	async generateIdeogram(input) {
+		if (!this.service.ideogram) return this.error("INTERNAL_SERVER_ERROR", "Ideogram service is not available");
+		return this.service.ideogram.generate(input);
+	}
+	async getUsage(userId) {
+		if (!this.repository.aiUsage) return this.error("INTERNAL_SERVER_ERROR", "AI usage repository is not available");
+		return this.repository.aiUsage.getUsage(userId);
+	}
+};
+//#endregion
+exports.AIService = AIService;
+
+//# sourceMappingURL=ai.service.cjs.map

package/dist/src/modules/ai/ai.service.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.service.cjs","names":["BaseService","RequestContext","OPENAI_TEXT_EMBEDDING_3_SMALL","MDocument","Output","NoObjectGeneratedError","repairJsonPrompt"],"sources":["../../../../src/modules/ai/ai.service.ts"],"sourcesContent":["import { OPENAI_TEXT_EMBEDDING_3_SMALL } from \"@m5kdev/commons/modules/ai/ai.constants\";\r\nimport { arrayToPseudoXML } from \"@m5kdev/commons/modules/ai/ai.utils\";\r\nimport type { Mastra } from \"@mastra/core\";\r\nimport { RequestContext } from \"@mastra/core/request-context\";\r\nimport type { FullOutput, MastraModelOutput } from \"@mastra/core/stream\";\r\nimport { MDocument } from \"@mastra/rag\";\r\nimport type { OpenRouterProvider } from \"@openrouter/ai-sdk-provider\";\r\nimport {\r\n  embed,\r\n  embedMany,\r\n  generateText,\r\n  type ModelMessage,\r\n  NoObjectGeneratedError,\r\n  Output,\r\n} from \"ai\";\r\nimport { jsonrepair } from \"jsonrepair\";\r\nimport { err, ok } from \"neverthrow\";\r\nimport type Replicate from \"replicate\";\r\nimport type { ZodType, z } from \"zod\";\r\nimport type { RequiredServiceActor } from \"../base/base.actor\";\r\nimport type { ServerResultAsync } from \"../base/base.dto\";\r\nimport { BaseService } from \"../base/base.service\";\r\nimport { repairJsonPrompt } from \"./ai.prompts\";\r\nimport type { AiUsageRepository, AiUsageRow } from \"./ai.repository\";\r\nimport type { IdeogramV3GenerateInput, IdeogramV3GenerateOutput } from \"./ideogram/ideogram.dto\";\r\nimport type { IdeogramService } from \"./ideogram/ideogram.service\";\r\n\r\ntype MastraAgent = ReturnType<Mastra[\"getAgent\"]>;\r\ntype MastraAgentGenerateOptions = Parameters<MastraAgent[\"generate\"]>[1];\r\ntype MessageListInput = { role: \"user\" | \"assistant\" | \"system\"; content: string }[];\r\ntype GenerateTextParams = Parameters<typeof generateText>[0];\r\ntype GenerateTextInput =\r\n  | { prompt: string | ModelMessage[]; messages?: never }\r\n  | { messages: ModelMessage[]; prompt?: never };\r\ntype AIServiceActorContext = { actor: RequiredServiceActor<\"user\"> };\r\ntype AIServiceGenerateTextParams = Omit<GenerateTextParams, \"model\" | \"prompt\" | \"messages\"> &\r\n  GenerateTextInput & {\r\n    model: string;\r\n    removeMDash?: boolean;\r\n    ctx?: AIServiceActorContext;\r\n    retryAttempts?: number;\r\n    retryModels?: string[];\r\n  };\r\ntype AIServiceGenerateObjectParams<T extends ZodType> = Omit<\r\n  GenerateTextParams,\r\n  \"model\" | \"prompt\" | \"messages\" | \"output\"\r\n> &\r\n  GenerateTextInput & {\r\n    model: string;\r\n    schema: T;\r\n    repairAttempts?: number;\r\n    repairModel?: string;\r\n    ctx?: AIServiceActorContext;\r\n    retryAttempts?: number;\r\n    retryModels?: string[];\r\n  };\r\n\r\ntype AIServiceOptions = {\r\n  retryAttempts?: number;\r\n  retryModels?: string[];\r\n  repairAttempts?: number;\r\n  repairModel?: string;\r\n  removeMDash?: boolean;\r\n};\r\n\r\nexport class AIService<MastraInstance extends Mastra> extends BaseService<\r\n  { aiUsage?: AiUsageRepository },\r\n  { ideogram?: IdeogramService }\r\n> {\r\n  helpers = {\r\n    arrayToPseudoXML,\r\n  };\r\n\r\n  mastra?: MastraInstance;\r\n  openrouter?: OpenRouterProvider;\r\n  replicate?: Replicate;\r\n  options?: AIServiceOptions;\r\n\r\n  constructor(\r\n    repositories: { aiUsage?: AiUsageRepository },\r\n    services: { ideogram?: IdeogramService },\r\n    libs: { mastra?: MastraInstance; openrouter?: OpenRouterProvider; replicate?: Replicate },\r\n    options?: AIServiceOptions\r\n  ) {\r\n    super(repositories, services);\r\n    this.mastra = libs.mastra;\r\n    this.openrouter = libs.openrouter;\r\n    this.replicate = libs.replicate;\r\n    this.options = options;\r\n  }\r\n\r\n  getMastra(): MastraInstance {\r\n    if (!this.mastra) {\r\n      throw new Error(\"Mastra is not available\");\r\n    }\r\n    return this.mastra;\r\n  }\r\n\r\n  prepareModel(model: string): ReturnType<OpenRouterProvider[\"chat\"]> {\r\n    if (!this.openrouter) {\r\n      throw new Error(\"OpenRouter is not configured\");\r\n    }\r\n    return this.openrouter.chat(model, {\r\n      usage: {\r\n        include: true,\r\n      },\r\n    });\r\n  }\r\n\r\n  prepareEmbeddingModel(model: string): ReturnType<OpenRouterProvider[\"textEmbeddingModel\"]> {\r\n    if (!this.openrouter) {\r\n      throw new Error(\"OpenRouter is not configured\");\r\n    }\r\n    const openrouter = this.openrouter as OpenRouterProvider & {\r\n      embeddingModel?: (modelId: string) => unknown;\r\n    };\r\n    return (openrouter.embeddingModel?.(model) ??\r\n      openrouter.textEmbeddingModel(model)) as ReturnType<OpenRouterProvider[\"textEmbeddingModel\"]>;\r\n  }\r\n\r\n  async agentUse(\r\n    agent: string,\r\n    options: MastraAgentGenerateOptions & { prompt?: string; messages?: MessageListInput },\r\n    ctx?: AIServiceActorContext & { model?: string }\r\n  ): ServerResultAsync<Awaited<ReturnType<MastraModelOutput<any>[\"getFullOutput\"]>>> {\r\n    return this.throwableAsync(async () => {\r\n      this.logger.info(\"AGENT USE\");\r\n      const { prompt, messages, ...rest } = options;\r\n      const payload = messages || prompt;\r\n      if (!payload) return this.error(\"BAD_REQUEST\", \"No prompt or messages provided\");\r\n      const requestContext = options.requestContext ?? new RequestContext();\r\n\r\n      if (ctx?.actor) {\r\n        requestContext.set(\"userId\", ctx.actor.userId);\r\n      }\r\n      if (ctx?.model) {\r\n        requestContext.set(\"model\", ctx.model);\r\n      }\r\n      const mAgent = this.getMastra().getAgent(agent);\r\n\r\n      const result = await mAgent.generate(payload as any, {\r\n        ...rest,\r\n        requestContext: rest.requestContext ?? requestContext,\r\n      });\r\n      this.logger.info(\"AGENT USE DONE\");\r\n      if (this.repository.aiUsage) {\r\n        const createUsageResult = await this.repository.aiUsage.create({\r\n          userId: ctx?.actor?.userId,\r\n          model: ctx?.model ?? \"unknown\",\r\n          provider: \"openrouter\",\r\n          feature: agent,\r\n          traceId: result.traceId,\r\n          inputTokens: result.usage.inputTokens,\r\n          outputTokens: result.usage.outputTokens,\r\n          totalTokens: result.usage.totalTokens,\r\n          cost: (result?.providerMetadata?.openrouter?.usage as any)?.cost ?? 0,\r\n        });\r\n        if (createUsageResult.isErr()) return err(createUsageResult.error);\r\n      }\r\n      this.logger.info(\"AGENT USE CREATED USAGE\");\r\n      return ok(result);\r\n    });\r\n  }\r\n\r\n  async agentText(\r\n    agent: string,\r\n    options: MastraAgentGenerateOptions & { prompt?: string; messages?: MessageListInput },\r\n    ctx?: AIServiceActorContext & { model?: string }\r\n  ): ServerResultAsync<string> {\r\n    const result = await this.agentUse(agent, options, ctx);\r\n    if (result.isErr())\r\n      return this.error(\"SERVICE_UNAVAILABLE\", \"AI: Agent text failed\", { cause: result.error });\r\n    return ok(result.value.text);\r\n  }\r\n\r\n  async agentTextResult(\r\n    agent: string,\r\n    options: MastraAgentGenerateOptions & { prompt?: string; messages?: MessageListInput },\r\n    ctx?: AIServiceActorContext & { model?: string }\r\n  ): ServerResultAsync<FullOutput<any>> {\r\n    const result = await this.agentUse(agent, options, ctx);\r\n    if (result.isErr()) return err(result.error);\r\n    return ok(result.value);\r\n  }\r\n\r\n  async agentObject<T extends ZodType<any>>(\r\n    agent: string,\r\n    options: MastraAgentGenerateOptions & {\r\n      schema: T;\r\n      prompt?: string;\r\n      messages?: MessageListInput;\r\n    },\r\n    ctx?: AIServiceActorContext & { model?: string }\r\n  ): ServerResultAsync<z.infer<T>> {\r\n    const { schema, ...rest } = options;\r\n    const result = await this.agentUse(agent, { ...rest, structuredOutput: { schema } }, ctx);\r\n    if (result.isErr())\r\n      return this.error(\"SERVICE_UNAVAILABLE\", \"AI: Agent object failed\", { cause: result.error });\r\n    return ok(result.value.object as z.infer<T>);\r\n  }\r\n\r\n  async agentObjectResult<T extends ZodType<any>>(\r\n    agent: string,\r\n    options: MastraAgentGenerateOptions & {\r\n      schema: T;\r\n      prompt?: string;\r\n      messages?: MessageListInput;\r\n    },\r\n    ctx?: AIServiceActorContext & { model?: string }\r\n  ): ServerResultAsync<FullOutput<any> & { object: z.infer<T> }> {\r\n    this.logger.info(\"AGENT OBJECT RESULT\");\r\n    const { schema, ...rest } = options;\r\n    const result = await this.agentUse(agent, { ...rest, structuredOutput: { schema } }, ctx);\r\n    if (result.isErr()) return err(result.error);\r\n    this.logger.info(\"AGENT OBJECT RESULT DONE\");\r\n    return ok({ ...result.value, object: result.value.object as z.infer<T> });\r\n  }\r\n\r\n  async embedDocument(\r\n    value: string,\r\n    options?: Parameters<ReturnType<typeof MDocument.fromText>[\"chunk\"]>[0],\r\n    type: \"text\" | \"markdown\" | \"html\" | \"json\" = \"text\",\r\n    model: string = OPENAI_TEXT_EMBEDDING_3_SMALL\r\n  ): ServerResultAsync<{ embeddings: number[][]; chunks: { text: string }[] }> {\r\n    return this.throwableAsync(async () => {\r\n      if (type === \"text\") {\r\n        const doc = MDocument.fromText(value);\r\n        const chunks = await doc.chunk(\r\n          options ?? {\r\n            strategy: \"recursive\",\r\n            maxSize: 512,\r\n            overlap: 50,\r\n            separators: [\"\\n\"],\r\n          }\r\n        );\r\n        const embeddings = await this.embedMany(chunks, model);\r\n        if (embeddings.isErr()) return err(embeddings.error);\r\n        return ok({ embeddings: embeddings.value.embeddings, chunks });\r\n      }\r\n      return this.error(\"BAD_REQUEST\", \"Unsupported document type\");\r\n    });\r\n  }\r\n\r\n  async embed(\r\n    text: string,\r\n    model: string = OPENAI_TEXT_EMBEDDING_3_SMALL\r\n  ): ServerResultAsync<{ embedding: number[] }> {\r\n    return this.throwableAsync(async () => {\r\n      const result = await embed({\r\n        model: this.prepareEmbeddingModel(model),\r\n        value: text,\r\n      });\r\n      return ok(result);\r\n    });\r\n  }\r\n\r\n  async embedMany(\r\n    chunks: { text: string }[],\r\n    model: string = OPENAI_TEXT_EMBEDDING_3_SMALL\r\n  ): ServerResultAsync<{ embeddings: number[][] }> {\r\n    return this.throwableAsync(async () => {\r\n      const result = await embedMany({\r\n        model: this.prepareEmbeddingModel(model),\r\n        values: chunks.map((chunk) => chunk.text),\r\n      });\r\n      return ok(result);\r\n    });\r\n  }\r\n\r\n  async generateText(params: AIServiceGenerateTextParams): ServerResultAsync<string> {\r\n    return this.throwableAsync(async () => {\r\n      const {\r\n        removeMDash = this.options?.removeMDash ?? true,\r\n        model,\r\n        prompt,\r\n        messages,\r\n        ctx,\r\n        retryAttempts = this.options?.retryAttempts ?? 0,\r\n        retryModels = this.options?.retryModels ?? [],\r\n        ...rest\r\n      } = params;\r\n      const request = messages\r\n        ? { ...rest, model: this.prepareModel(model), messages }\r\n        : { ...rest, model: this.prepareModel(model), prompt };\r\n      try {\r\n        const result = await generateText(request);\r\n        if (this.repository.aiUsage) {\r\n          const createUsageResult = await this.repository.aiUsage.create({\r\n            userId: ctx?.actor?.userId,\r\n            model,\r\n            provider: \"openrouter\",\r\n            feature: \"generateText\",\r\n            traceId: result.providerMetadata?.openrouter?.traceId?.toString(),\r\n            inputTokens: result.usage.inputTokens,\r\n            outputTokens: result.usage.outputTokens,\r\n            totalTokens: result.usage.totalTokens,\r\n            cost: (result?.providerMetadata?.openrouter?.usage as any)?.cost ?? 0,\r\n          });\r\n          if (createUsageResult.isErr()) return err(createUsageResult.error);\r\n        }\r\n        return ok(removeMDash ? result.text.replace(/\\u2013|\\u2014/g, \"-\") : result.text);\r\n      } catch (error) {\r\n        if (retryAttempts <= 0) throw error;\r\n        this.logger.warn(`generateText failed, retrying (${retryAttempts} attempts left)`, {\r\n          model,\r\n          error,\r\n        });\r\n        // Exponential backoff: wait before retrying\r\n        const delay = Math.min(\r\n          1000 * 2 ** ((this.options?.retryAttempts ?? 3) - retryAttempts),\r\n          10000\r\n        );\r\n        await new Promise<void>((resolve) => setTimeout(resolve, delay));\r\n        const nextModel = retryModels?.[0] ?? model;\r\n        const nextRetryModels = retryModels ? [...retryModels.slice(1), model] : undefined;\r\n        return this.generateText({\r\n          ...rest,\r\n          ...(messages ? { messages } : { prompt: prompt! }),\r\n          model: nextModel,\r\n          removeMDash,\r\n          ctx,\r\n          retryAttempts: retryAttempts - 1,\r\n          retryModels: nextRetryModels,\r\n        } as AIServiceGenerateTextParams);\r\n      }\r\n    });\r\n  }\r\n\r\n  async generateObject<T extends ZodType>(\r\n    params: AIServiceGenerateObjectParams<T>\r\n  ): ServerResultAsync<z.infer<T>> {\r\n    const {\r\n      model,\r\n      schema,\r\n      prompt,\r\n      messages,\r\n      repairAttempts = this.options?.repairAttempts ?? 0,\r\n      repairModel = this.options?.repairModel ?? model,\r\n      ctx,\r\n      retryAttempts = this.options?.retryAttempts ?? 0,\r\n      retryModels = this.options?.retryModels ?? [],\r\n      ...rest\r\n    } = params;\r\n    const request = messages\r\n      ? {\r\n          ...rest,\r\n          model: this.prepareModel(model),\r\n          messages,\r\n          output: Output.object({ schema }),\r\n        }\r\n      : {\r\n          ...rest,\r\n          model: this.prepareModel(model),\r\n          prompt,\r\n          output: Output.object({ schema }),\r\n        };\r\n    try {\r\n      const result = await generateText(request);\r\n      if (this.repository.aiUsage) {\r\n        const createUsageResult = await this.repository.aiUsage.create({\r\n          userId: ctx?.actor?.userId,\r\n          model,\r\n          provider: \"openrouter\",\r\n          feature: \"generateObject\",\r\n          traceId: result.providerMetadata?.openrouter?.traceId?.toString(),\r\n          inputTokens: result.usage.inputTokens,\r\n          outputTokens: result.usage.outputTokens,\r\n          totalTokens: result.usage.totalTokens,\r\n          cost: (result?.providerMetadata?.openrouter?.usage as any)?.cost ?? 0,\r\n        });\r\n        if (createUsageResult.isErr()) return err(createUsageResult.error);\r\n      }\r\n      return ok(result.output as z.infer<T>);\r\n    } catch (error) {\r\n      if (NoObjectGeneratedError.isInstance(error)) {\r\n        if (this.repository.aiUsage) {\r\n          const createUsageResult = await this.repository.aiUsage.create({\r\n            userId: ctx?.actor?.userId,\r\n            model,\r\n            provider: \"openrouter\",\r\n            feature: \"generateObject\",\r\n            traceId: null,\r\n            inputTokens: error?.usage?.inputTokens,\r\n            outputTokens: error?.usage?.outputTokens,\r\n            totalTokens: error?.usage?.totalTokens,\r\n            cost: 0,\r\n          });\r\n          if (createUsageResult.isErr()) return err(createUsageResult.error);\r\n        }\r\n        if (error.text) {\r\n          const repairedText = jsonrepair(error.text);\r\n          const parsed = schema.safeParse(repairedText);\r\n          if (parsed.success) return ok(parsed.data);\r\n\r\n          if (repairAttempts === 0)\r\n            return this.error(\"PARSE_ERROR\", \"AI: Agent object failed\", { cause: error });\r\n\r\n          return this.generateObject({\r\n            ...rest,\r\n            prompt: repairJsonPrompt.compile({\r\n              text: error.text,\r\n              error: JSON.stringify(error.cause ?? \"Unknown error\"),\r\n            }),\r\n            repairAttempts: repairAttempts - 1,\r\n            model: repairModel ?? model,\r\n            schema,\r\n            ctx,\r\n          });\r\n        }\r\n        return this.error(\"PARSE_ERROR\", \"AI: Agent object failed without text\", {\r\n          cause: error,\r\n        });\r\n      }\r\n      if (retryAttempts <= 0)\r\n        return this.error(\"BAD_REQUEST\", \"AI: Provider failed to generate object\", {\r\n          cause: error,\r\n        });\r\n      this.logger.warn(`generateObject failed, retrying (${retryAttempts} attempts left)`, {\r\n        model,\r\n        error,\r\n      });\r\n      // Exponential backoff: wait before retrying\r\n      const delay = Math.min(\r\n        1000 * 2 ** ((this.options?.retryAttempts ?? 3) - retryAttempts),\r\n        10000\r\n      );\r\n      await new Promise<void>((resolve) => setTimeout(resolve, delay));\r\n      const nextModel = retryModels?.[0] ?? model;\r\n      const nextRetryModels = retryModels ? [...retryModels.slice(1), model] : undefined;\r\n      return this.generateObject({\r\n        ...rest,\r\n        ...(messages ? { messages } : { prompt: prompt! }),\r\n        model: nextModel,\r\n        schema,\r\n        repairAttempts,\r\n        repairModel,\r\n        ctx,\r\n        retryAttempts: retryAttempts - 1,\r\n        retryModels: nextRetryModels,\r\n      } as AIServiceGenerateObjectParams<T>);\r\n    }\r\n  }\r\n\r\n  async generateReplicate(\r\n    model: Parameters<Replicate[\"run\"]>[0],\r\n    options: Parameters<Replicate[\"run\"]>[1]\r\n  ): ServerResultAsync<object> {\r\n    return this.throwableAsync(async () => {\r\n      if (!this.replicate) {\r\n        return this.error(\"INTERNAL_SERVER_ERROR\", \"Replicate is not configured\");\r\n      }\r\n      try {\r\n        return ok(await this.replicate.run(model, options));\r\n      } catch (error) {\r\n        return this.error(\"INTERNAL_SERVER_ERROR\", undefined, { cause: error });\r\n      }\r\n    });\r\n  }\r\n\r\n  async generateTranscript(\r\n    file_url: string\r\n  ): ServerResultAsync<{ text: string; metadata: unknown }> {\r\n    const output = await this.generateReplicate(\r\n      \"thomasmol/whisper-diarization:1495a9cddc83b2203b0d8d3516e38b80fd1572ebc4bc5700ac1da56a9b3ed886\",\r\n      {\r\n        input: {\r\n          file_url,\r\n        },\r\n      }\r\n    );\r\n\r\n    if (output.isErr()) return err(output.error);\r\n\r\n    try {\r\n      const { segments } = output.value as { segments: { text: string }[] };\r\n      return ok({ text: segments.map((segment) => segment.text).join(\"\"), metadata: segments });\r\n    } catch (error) {\r\n      return this.error(\"INTERNAL_SERVER_ERROR\", undefined, { cause: error });\r\n    }\r\n  }\r\n\r\n  async generateIdeogram(\r\n    input: IdeogramV3GenerateInput\r\n  ): ServerResultAsync<IdeogramV3GenerateOutput> {\r\n    if (!this.service.ideogram) {\r\n      return this.error(\"INTERNAL_SERVER_ERROR\", \"Ideogram service is not available\");\r\n    }\r\n    return this.service.ideogram.generate(input);\r\n  }\r\n\r\n  async getUsage(\r\n    userId: string\r\n  ): ServerResultAsync<Pick<AiUsageRow, \"inputTokens\" | \"outputTokens\" | \"totalTokens\" | \"cost\">> {\r\n    if (!this.repository.aiUsage) {\r\n      return this.error(\"INTERNAL_SERVER_ERROR\", \"AI usage repository is not available\");\r\n    }\r\n    return this.repository.aiUsage.getUsage(userId);\r\n  }\r\n}\r\n"],"mappings":";;;;;;;;;;;;AAiEA,IAAa,YAAb,cAA8DA,sCAAAA,YAG5D;CACA,UAAU,EACR,kBAAA,oCAAA,kBACD;CAED;CACA;CACA;CACA;CAEA,YACE,cACA,UACA,MACA,SACA;AACA,QAAM,cAAc,SAAS;AAC7B,OAAK,SAAS,KAAK;AACnB,OAAK,aAAa,KAAK;AACvB,OAAK,YAAY,KAAK;AACtB,OAAK,UAAU;;CAGjB,YAA4B;AAC1B,MAAI,CAAC,KAAK,OACR,OAAM,IAAI,MAAM,0BAA0B;AAE5C,SAAO,KAAK;;CAGd,aAAa,OAAuD;AAClE,MAAI,CAAC,KAAK,WACR,OAAM,IAAI,MAAM,+BAA+B;AAEjD,SAAO,KAAK,WAAW,KAAK,OAAO,EACjC,OAAO,EACL,SAAS,MACV,EACF,CAAC;;CAGJ,sBAAsB,OAAqE;AACzF,MAAI,CAAC,KAAK,WACR,OAAM,IAAI,MAAM,+BAA+B;EAEjD,MAAM,aAAa,KAAK;AAGxB,SAAQ,WAAW,iBAAiB,MAAM,IACxC,WAAW,mBAAmB,MAAM;;CAGxC,MAAM,SACJ,OACA,SACA,KACiF;AACjF,SAAO,KAAK,eAAe,YAAY;AACrC,QAAK,OAAO,KAAK,YAAY;GAC7B,MAAM,EAAE,QAAQ,UAAU,GAAG,SAAS;GACtC,MAAM,UAAU,YAAY;AAC5B,OAAI,CAAC,QAAS,QAAO,KAAK,MAAM,eAAe,iCAAiC;GAChF,MAAM,iBAAiB,QAAQ,kBAAkB,IAAIC,6BAAAA,gBAAgB;AAErE,OAAI,KAAK,MACP,gBAAe,IAAI,UAAU,IAAI,MAAM,OAAO;AAEhD,OAAI,KAAK,MACP,gBAAe,IAAI,SAAS,IAAI,MAAM;GAIxC,MAAM,SAAS,MAFA,KAAK,WAAW,CAAC,SAAS,MAAM,CAEnB,SAAS,SAAgB;IACnD,GAAG;IACH,gBAAgB,KAAK,kBAAkB;IACxC,CAAC;AACF,QAAK,OAAO,KAAK,iBAAiB;AAClC,OAAI,KAAK,WAAW,SAAS;IAC3B,MAAM,oBAAoB,MAAM,KAAK,WAAW,QAAQ,OAAO;KAC7D,QAAQ,KAAK,OAAO;KACpB,OAAO,KAAK,SAAS;KACrB,UAAU;KACV,SAAS;KACT,SAAS,OAAO;KAChB,aAAa,OAAO,MAAM;KAC1B,cAAc,OAAO,MAAM;KAC3B,aAAa,OAAO,MAAM;KAC1B,OAAO,QAAQ,kBAAkB,YAAY,QAAe,QAAQ;KACrE,CAAC;AACF,QAAI,kBAAkB,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,kBAAkB,MAAM;;AAEpE,QAAK,OAAO,KAAK,0BAA0B;AAC3C,WAAA,GAAA,WAAA,IAAU,OAAO;IACjB;;CAGJ,MAAM,UACJ,OACA,SACA,KAC2B;EAC3B,MAAM,SAAS,MAAM,KAAK,SAAS,OAAO,SAAS,IAAI;AACvD,MAAI,OAAO,OAAO,CAChB,QAAO,KAAK,MAAM,uBAAuB,yBAAyB,EAAE,OAAO,OAAO,OAAO,CAAC;AAC5F,UAAA,GAAA,WAAA,IAAU,OAAO,MAAM,KAAK;;CAG9B,MAAM,gBACJ,OACA,SACA,KACoC;EACpC,MAAM,SAAS,MAAM,KAAK,SAAS,OAAO,SAAS,IAAI;AACvD,MAAI,OAAO,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,OAAO,MAAM;AAC5C,UAAA,GAAA,WAAA,IAAU,OAAO,MAAM;;CAGzB,MAAM,YACJ,OACA,SAKA,KAC+B;EAC/B,MAAM,EAAE,QAAQ,GAAG,SAAS;EAC5B,MAAM,SAAS,MAAM,KAAK,SAAS,OAAO;GAAE,GAAG;GAAM,kBAAkB,EAAE,QAAQ;GAAE,EAAE,IAAI;AACzF,MAAI,OAAO,OAAO,CAChB,QAAO,KAAK,MAAM,uBAAuB,2BAA2B,EAAE,OAAO,OAAO,OAAO,CAAC;AAC9F,UAAA,GAAA,WAAA,IAAU,OAAO,MAAM,OAAqB;;CAG9C,MAAM,kBACJ,OACA,SAKA,KAC6D;AAC7D,OAAK,OAAO,KAAK,sBAAsB;EACvC,MAAM,EAAE,QAAQ,GAAG,SAAS;EAC5B,MAAM,SAAS,MAAM,KAAK,SAAS,OAAO;GAAE,GAAG;GAAM,kBAAkB,EAAE,QAAQ;GAAE,EAAE,IAAI;AACzF,MAAI,OAAO,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,OAAO,MAAM;AAC5C,OAAK,OAAO,KAAK,2BAA2B;AAC5C,UAAA,GAAA,WAAA,IAAU;GAAE,GAAG,OAAO;GAAO,QAAQ,OAAO,MAAM;GAAsB,CAAC;;CAG3E,MAAM,cACJ,OACA,SACA,OAA8C,QAC9C,QAAgBC,wCAAAA,+BAC2D;AAC3E,SAAO,KAAK,eAAe,YAAY;AACrC,OAAI,SAAS,QAAQ;IAEnB,MAAM,SAAS,MADHC,YAAAA,UAAU,SAAS,MAAM,CACZ,MACvB,WAAW;KACT,UAAU;KACV,SAAS;KACT,SAAS;KACT,YAAY,CAAC,KAAK;KACnB,CACF;IACD,MAAM,aAAa,MAAM,KAAK,UAAU,QAAQ,MAAM;AACtD,QAAI,WAAW,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,WAAW,MAAM;AACpD,YAAA,GAAA,WAAA,IAAU;KAAE,YAAY,WAAW,MAAM;KAAY;KAAQ,CAAC;;AAEhE,UAAO,KAAK,MAAM,eAAe,4BAA4B;IAC7D;;CAGJ,MAAM,MACJ,MACA,QAAgBD,wCAAAA,+BAC4B;AAC5C,SAAO,KAAK,eAAe,YAAY;AAKrC,WAAA,GAAA,WAAA,IAJe,OAAA,GAAA,GAAA,OAAY;IACzB,OAAO,KAAK,sBAAsB,MAAM;IACxC,OAAO;IACR,CAAC,CACe;IACjB;;CAGJ,MAAM,UACJ,QACA,QAAgBA,wCAAAA,+BAC+B;AAC/C,SAAO,KAAK,eAAe,YAAY;AAKrC,WAAA,GAAA,WAAA,IAJe,OAAA,GAAA,GAAA,WAAgB;IAC7B,OAAO,KAAK,sBAAsB,MAAM;IACxC,QAAQ,OAAO,KAAK,UAAU,MAAM,KAAK;IAC1C,CAAC,CACe;IACjB;;CAGJ,MAAM,aAAa,QAAgE;AACjF,SAAO,KAAK,eAAe,YAAY;GACrC,MAAM,EACJ,cAAc,KAAK,SAAS,eAAe,MAC3C,OACA,QACA,UACA,KACA,gBAAgB,KAAK,SAAS,iBAAiB,GAC/C,cAAc,KAAK,SAAS,eAAe,EAAE,EAC7C,GAAG,SACD;GACJ,MAAM,UAAU,WACZ;IAAE,GAAG;IAAM,OAAO,KAAK,aAAa,MAAM;IAAE;IAAU,GACtD;IAAE,GAAG;IAAM,OAAO,KAAK,aAAa,MAAM;IAAE;IAAQ;AACxD,OAAI;IACF,MAAM,SAAS,OAAA,GAAA,GAAA,cAAmB,QAAQ;AAC1C,QAAI,KAAK,WAAW,SAAS;KAC3B,MAAM,oBAAoB,MAAM,KAAK,WAAW,QAAQ,OAAO;MAC7D,QAAQ,KAAK,OAAO;MACpB;MACA,UAAU;MACV,SAAS;MACT,SAAS,OAAO,kBAAkB,YAAY,SAAS,UAAU;MACjE,aAAa,OAAO,MAAM;MAC1B,cAAc,OAAO,MAAM;MAC3B,aAAa,OAAO,MAAM;MAC1B,OAAO,QAAQ,kBAAkB,YAAY,QAAe,QAAQ;MACrE,CAAC;AACF,SAAI,kBAAkB,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,kBAAkB,MAAM;;AAEpE,YAAA,GAAA,WAAA,IAAU,cAAc,OAAO,KAAK,QAAQ,kBAAkB,IAAI,GAAG,OAAO,KAAK;YAC1E,OAAO;AACd,QAAI,iBAAiB,EAAG,OAAM;AAC9B,SAAK,OAAO,KAAK,kCAAkC,cAAc,kBAAkB;KACjF;KACA;KACD,CAAC;IAEF,MAAM,QAAQ,KAAK,IACjB,MAAO,OAAO,KAAK,SAAS,iBAAiB,KAAK,gBAClD,IACD;AACD,UAAM,IAAI,SAAe,YAAY,WAAW,SAAS,MAAM,CAAC;IAChE,MAAM,YAAY,cAAc,MAAM;IACtC,MAAM,kBAAkB,cAAc,CAAC,GAAG,YAAY,MAAM,EAAE,EAAE,MAAM,GAAG,KAAA;AACzE,WAAO,KAAK,aAAa;KACvB,GAAG;KACH,GAAI,WAAW,EAAE,UAAU,GAAG,EAAU,QAAS;KACjD,OAAO;KACP;KACA;KACA,eAAe,gBAAgB;KAC/B,aAAa;KACd,CAAgC;;IAEnC;;CAGJ,MAAM,eACJ,QAC+B;EAC/B,MAAM,EACJ,OACA,QACA,QACA,UACA,iBAAiB,KAAK,SAAS,kBAAkB,GACjD,cAAc,KAAK,SAAS,eAAe,OAC3C,KACA,gBAAgB,KAAK,SAAS,iBAAiB,GAC/C,cAAc,KAAK,SAAS,eAAe,EAAE,EAC7C,GAAG,SACD;EACJ,MAAM,UAAU,WACZ;GACE,GAAG;GACH,OAAO,KAAK,aAAa,MAAM;GAC/B;GACA,QAAQE,GAAAA,OAAO,OAAO,EAAE,QAAQ,CAAC;GAClC,GACD;GACE,GAAG;GACH,OAAO,KAAK,aAAa,MAAM;GAC/B;GACA,QAAQA,GAAAA,OAAO,OAAO,EAAE,QAAQ,CAAC;GAClC;AACL,MAAI;GACF,MAAM,SAAS,OAAA,GAAA,GAAA,cAAmB,QAAQ;AAC1C,OAAI,KAAK,WAAW,SAAS;IAC3B,MAAM,oBAAoB,MAAM,KAAK,WAAW,QAAQ,OAAO;KAC7D,QAAQ,KAAK,OAAO;KACpB;KACA,UAAU;KACV,SAAS;KACT,SAAS,OAAO,kBAAkB,YAAY,SAAS,UAAU;KACjE,aAAa,OAAO,MAAM;KAC1B,cAAc,OAAO,MAAM;KAC3B,aAAa,OAAO,MAAM;KAC1B,OAAO,QAAQ,kBAAkB,YAAY,QAAe,QAAQ;KACrE,CAAC;AACF,QAAI,kBAAkB,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,kBAAkB,MAAM;;AAEpE,WAAA,GAAA,WAAA,IAAU,OAAO,OAAqB;WAC/B,OAAO;AACd,OAAIC,GAAAA,uBAAuB,WAAW,MAAM,EAAE;AAC5C,QAAI,KAAK,WAAW,SAAS;KAC3B,MAAM,oBAAoB,MAAM,KAAK,WAAW,QAAQ,OAAO;MAC7D,QAAQ,KAAK,OAAO;MACpB;MACA,UAAU;MACV,SAAS;MACT,SAAS;MACT,aAAa,OAAO,OAAO;MAC3B,cAAc,OAAO,OAAO;MAC5B,aAAa,OAAO,OAAO;MAC3B,MAAM;MACP,CAAC;AACF,SAAI,kBAAkB,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,kBAAkB,MAAM;;AAEpE,QAAI,MAAM,MAAM;KACd,MAAM,gBAAA,GAAA,WAAA,YAA0B,MAAM,KAAK;KAC3C,MAAM,SAAS,OAAO,UAAU,aAAa;AAC7C,SAAI,OAAO,QAAS,SAAA,GAAA,WAAA,IAAU,OAAO,KAAK;AAE1C,SAAI,mBAAmB,EACrB,QAAO,KAAK,MAAM,eAAe,2BAA2B,EAAE,OAAO,OAAO,CAAC;AAE/E,YAAO,KAAK,eAAe;MACzB,GAAG;MACH,QAAQC,kCAAAA,iBAAiB,QAAQ;OAC/B,MAAM,MAAM;OACZ,OAAO,KAAK,UAAU,MAAM,SAAS,gBAAgB;OACtD,CAAC;MACF,gBAAgB,iBAAiB;MACjC,OAAO,eAAe;MACtB;MACA;MACD,CAAC;;AAEJ,WAAO,KAAK,MAAM,eAAe,wCAAwC,EACvE,OAAO,OACR,CAAC;;AAEJ,OAAI,iBAAiB,EACnB,QAAO,KAAK,MAAM,eAAe,0CAA0C,EACzE,OAAO,OACR,CAAC;AACJ,QAAK,OAAO,KAAK,oCAAoC,cAAc,kBAAkB;IACnF;IACA;IACD,CAAC;GAEF,MAAM,QAAQ,KAAK,IACjB,MAAO,OAAO,KAAK,SAAS,iBAAiB,KAAK,gBAClD,IACD;AACD,SAAM,IAAI,SAAe,YAAY,WAAW,SAAS,MAAM,CAAC;GAChE,MAAM,YAAY,cAAc,MAAM;GACtC,MAAM,kBAAkB,cAAc,CAAC,GAAG,YAAY,MAAM,EAAE,EAAE,MAAM,GAAG,KAAA;AACzE,UAAO,KAAK,eAAe;IACzB,GAAG;IACH,GAAI,WAAW,EAAE,UAAU,GAAG,EAAU,QAAS;IACjD,OAAO;IACP;IACA;IACA;IACA;IACA,eAAe,gBAAgB;IAC/B,aAAa;IACd,CAAqC;;;CAI1C,MAAM,kBACJ,OACA,SAC2B;AAC3B,SAAO,KAAK,eAAe,YAAY;AACrC,OAAI,CAAC,KAAK,UACR,QAAO,KAAK,MAAM,yBAAyB,8BAA8B;AAE3E,OAAI;AACF,YAAA,GAAA,WAAA,IAAU,MAAM,KAAK,UAAU,IAAI,OAAO,QAAQ,CAAC;YAC5C,OAAO;AACd,WAAO,KAAK,MAAM,yBAAyB,KAAA,GAAW,EAAE,OAAO,OAAO,CAAC;;IAEzE;;CAGJ,MAAM,mBACJ,UACwD;EACxD,MAAM,SAAS,MAAM,KAAK,kBACxB,kGACA,EACE,OAAO,EACL,UACD,EACF,CACF;AAED,MAAI,OAAO,OAAO,CAAE,SAAA,GAAA,WAAA,KAAW,OAAO,MAAM;AAE5C,MAAI;GACF,MAAM,EAAE,aAAa,OAAO;AAC5B,WAAA,GAAA,WAAA,IAAU;IAAE,MAAM,SAAS,KAAK,YAAY,QAAQ,KAAK,CAAC,KAAK,GAAG;IAAE,UAAU;IAAU,CAAC;WAClF,OAAO;AACd,UAAO,KAAK,MAAM,yBAAyB,KAAA,GAAW,EAAE,OAAO,OAAO,CAAC;;;CAI3E,MAAM,iBACJ,OAC6C;AAC7C,MAAI,CAAC,KAAK,QAAQ,SAChB,QAAO,KAAK,MAAM,yBAAyB,oCAAoC;AAEjF,SAAO,KAAK,QAAQ,SAAS,SAAS,MAAM;;CAG9C,MAAM,SACJ,QAC8F;AAC9F,MAAI,CAAC,KAAK,WAAW,QACnB,QAAO,KAAK,MAAM,yBAAyB,uCAAuC;AAEpF,SAAO,KAAK,WAAW,QAAQ,SAAS,OAAO"}

package/dist/src/modules/ai/ai.service.d.cts

@@ -0,0 +1,141 @@
+import { ServerResultAsync } from "../base/base.dto.cjs";
+import { RequiredServiceActor } from "../base/base.actor.cjs";
+import { BaseService } from "../base/base.service.cjs";
+import { AiUsageRepository, AiUsageRow } from "./ai.repository.cjs";
+import { IdeogramV3GenerateInput, IdeogramV3GenerateOutput } from "./ideogram/ideogram.dto.cjs";
+import { IdeogramService } from "./ideogram/ideogram.service.cjs";
+import { ZodType, z } from "zod";
+import { arrayToPseudoXML } from "@m5kdev/commons/modules/ai/ai.utils";
+import { Mastra } from "@mastra/core";
+import { FullOutput, MastraModelOutput } from "@mastra/core/stream";
+import { MDocument } from "@mastra/rag";
+import { OpenRouterProvider } from "@openrouter/ai-sdk-provider";
+import { ModelMessage, generateText } from "ai";
+import Replicate from "replicate";
+
+//#region src/modules/ai/ai.service.d.ts
+type MastraAgent = ReturnType<Mastra["getAgent"]>;
+type MastraAgentGenerateOptions = Parameters<MastraAgent["generate"]>[1];
+type MessageListInput = {
+  role: "user" | "assistant" | "system";
+  content: string;
+}[];
+type GenerateTextParams = Parameters<typeof generateText>[0];
+type GenerateTextInput = {
+  prompt: string | ModelMessage[];
+  messages?: never;
+} | {
+  messages: ModelMessage[];
+  prompt?: never;
+};
+type AIServiceActorContext = {
+  actor: RequiredServiceActor<"user">;
+};
+type AIServiceGenerateTextParams = Omit<GenerateTextParams, "model" | "prompt" | "messages"> & GenerateTextInput & {
+  model: string;
+  removeMDash?: boolean;
+  ctx?: AIServiceActorContext;
+  retryAttempts?: number;
+  retryModels?: string[];
+};
+type AIServiceGenerateObjectParams<T extends ZodType> = Omit<GenerateTextParams, "model" | "prompt" | "messages" | "output"> & GenerateTextInput & {
+  model: string;
+  schema: T;
+  repairAttempts?: number;
+  repairModel?: string;
+  ctx?: AIServiceActorContext;
+  retryAttempts?: number;
+  retryModels?: string[];
+};
+type AIServiceOptions = {
+  retryAttempts?: number;
+  retryModels?: string[];
+  repairAttempts?: number;
+  repairModel?: string;
+  removeMDash?: boolean;
+};
+declare class AIService<MastraInstance extends Mastra> extends BaseService<{
+  aiUsage?: AiUsageRepository;
+}, {
+  ideogram?: IdeogramService;
+}> {
+  helpers: {
+    arrayToPseudoXML: typeof arrayToPseudoXML;
+  };
+  mastra?: MastraInstance;
+  openrouter?: OpenRouterProvider;
+  replicate?: Replicate;
+  options?: AIServiceOptions;
+  constructor(repositories: {
+    aiUsage?: AiUsageRepository;
+  }, services: {
+    ideogram?: IdeogramService;
+  }, libs: {
+    mastra?: MastraInstance;
+    openrouter?: OpenRouterProvider;
+    replicate?: Replicate;
+  }, options?: AIServiceOptions);
+  getMastra(): MastraInstance;
+  prepareModel(model: string): ReturnType<OpenRouterProvider["chat"]>;
+  prepareEmbeddingModel(model: string): ReturnType<OpenRouterProvider["textEmbeddingModel"]>;
+  agentUse(agent: string, options: MastraAgentGenerateOptions & {
+    prompt?: string;
+    messages?: MessageListInput;
+  }, ctx?: AIServiceActorContext & {
+    model?: string;
+  }): ServerResultAsync<Awaited<ReturnType<MastraModelOutput<any>["getFullOutput"]>>>;
+  agentText(agent: string, options: MastraAgentGenerateOptions & {
+    prompt?: string;
+    messages?: MessageListInput;
+  }, ctx?: AIServiceActorContext & {
+    model?: string;
+  }): ServerResultAsync<string>;
+  agentTextResult(agent: string, options: MastraAgentGenerateOptions & {
+    prompt?: string;
+    messages?: MessageListInput;
+  }, ctx?: AIServiceActorContext & {
+    model?: string;
+  }): ServerResultAsync<FullOutput<any>>;
+  agentObject<T extends ZodType<any>>(agent: string, options: MastraAgentGenerateOptions & {
+    schema: T;
+    prompt?: string;
+    messages?: MessageListInput;
+  }, ctx?: AIServiceActorContext & {
+    model?: string;
+  }): ServerResultAsync<z.infer<T>>;
+  agentObjectResult<T extends ZodType<any>>(agent: string, options: MastraAgentGenerateOptions & {
+    schema: T;
+    prompt?: string;
+    messages?: MessageListInput;
+  }, ctx?: AIServiceActorContext & {
+    model?: string;
+  }): ServerResultAsync<FullOutput<any> & {
+    object: z.infer<T>;
+  }>;
+  embedDocument(value: string, options?: Parameters<ReturnType<typeof MDocument.fromText>["chunk"]>[0], type?: "text" | "markdown" | "html" | "json", model?: string): ServerResultAsync<{
+    embeddings: number[][];
+    chunks: {
+      text: string;
+    }[];
+  }>;
+  embed(text: string, model?: string): ServerResultAsync<{
+    embedding: number[];
+  }>;
+  embedMany(chunks: {
+    text: string;
+  }[], model?: string): ServerResultAsync<{
+    embeddings: number[][];
+  }>;
+  generateText(params: AIServiceGenerateTextParams): ServerResultAsync<string>;
+  generateObject<T extends ZodType>(params: AIServiceGenerateObjectParams<T>): ServerResultAsync<z.infer<T>>;
+  generateReplicate(model: Parameters<Replicate["run"]>[0], options: Parameters<Replicate["run"]>[1]): ServerResultAsync<object>;
+  generateTranscript(file_url: string): ServerResultAsync<{
+    text: string;
+    metadata: unknown;
+  }>;
+  generateIdeogram(input: IdeogramV3GenerateInput): ServerResultAsync<IdeogramV3GenerateOutput>;
+  getUsage(userId: string): ServerResultAsync<Pick<AiUsageRow, "inputTokens" | "outputTokens" | "totalTokens" | "cost">>;
+}
+//#endregion
+export { AIService };
+//# sourceMappingURL=ai.service.d.cts.map

package/dist/src/modules/ai/ai.trpc.cjs

@@ -0,0 +1,19 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+require("../../../_virtual/_rolldown/runtime.cjs");
+const require_src_utils_trpc = require("../../utils/trpc.cjs");
+let zod = require("zod");
+//#region src/modules/ai/ai.trpc.ts
+function createAITRPC({ router, adminProcedure }, aiService) {
+	return router({ getUserUsage: adminProcedure.input(zod.z.object({ userId: zod.z.string() })).output(zod.z.object({
+		inputTokens: zod.z.number().nullable(),
+		outputTokens: zod.z.number().nullable(),
+		totalTokens: zod.z.number().nullable(),
+		cost: zod.z.number().nullable()
+	})).query(async ({ input }) => {
+		return require_src_utils_trpc.handleTRPCResult(await aiService.getUsage(input.userId));
+	}) });
+}
+//#endregion
+exports.createAITRPC = createAITRPC;
+
+//# sourceMappingURL=ai.trpc.cjs.map

package/dist/src/modules/ai/ai.trpc.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.trpc.cjs","names":["z","handleTRPCResult"],"sources":["../../../../src/modules/ai/ai.trpc.ts"],"sourcesContent":["import type { Mastra } from \"@mastra/core\";\r\nimport { z } from \"zod\";\r\nimport type { AIService } from \"./ai.service\";\r\nimport { handleTRPCResult, type TRPCMethods } from \"../../utils/trpc\";\r\n\r\nexport function createAITRPC<MastraInstance extends Mastra>(\r\n  { router, adminProcedure }: TRPCMethods,\r\n  aiService: AIService<MastraInstance>\r\n) {\r\n  return router({\r\n    getUserUsage: adminProcedure\r\n      .input(z.object({ userId: z.string() }))\r\n      .output(\r\n        z.object({\r\n          inputTokens: z.number().nullable(),\r\n          outputTokens: z.number().nullable(),\r\n          totalTokens: z.number().nullable(),\r\n          cost: z.number().nullable(),\r\n        })\r\n      )\r\n      .query(async ({ input }) => {\r\n        return handleTRPCResult(await aiService.getUsage(input.userId));\r\n      }),\r\n  });\r\n}\r\n"],"mappings":";;;;;AAKA,SAAgB,aACd,EAAE,QAAQ,kBACV,WACA;AACA,QAAO,OAAO,EACZ,cAAc,eACX,MAAMA,IAAAA,EAAE,OAAO,EAAE,QAAQA,IAAAA,EAAE,QAAQ,EAAE,CAAC,CAAC,CACvC,OACCA,IAAAA,EAAE,OAAO;EACP,aAAaA,IAAAA,EAAE,QAAQ,CAAC,UAAU;EAClC,cAAcA,IAAAA,EAAE,QAAQ,CAAC,UAAU;EACnC,aAAaA,IAAAA,EAAE,QAAQ,CAAC,UAAU;EAClC,MAAMA,IAAAA,EAAE,QAAQ,CAAC,UAAU;EAC5B,CAAC,CACH,CACA,MAAM,OAAO,EAAE,YAAY;AAC1B,SAAOC,uBAAAA,iBAAiB,MAAM,UAAU,SAAS,MAAM,OAAO,CAAC;GAC/D,EACL,CAAC"}

package/dist/src/modules/ai/ai.trpc.d.cts

@@ -0,0 +1,31 @@
+import { Context, TRPCMethods } from "../../utils/trpc.cjs";
+import { AIService } from "./ai.service.cjs";
+import * as _$_trpc_server0 from "@trpc/server";
+import { Mastra } from "@mastra/core";
+
+//#region src/modules/ai/ai.trpc.d.ts
+declare function createAITRPC<MastraInstance extends Mastra>({
+  router,
+  adminProcedure
+}: TRPCMethods, aiService: AIService<MastraInstance>): _$_trpc_server0.TRPCBuiltRouter<{
+  ctx: Context;
+  meta: any;
+  errorShape: _$_trpc_server0.TRPCDefaultErrorShape;
+  transformer: true;
+}, _$_trpc_server0.TRPCDecorateCreateRouterOptions<{
+  getUserUsage: _$_trpc_server0.TRPCQueryProcedure<{
+    input: {
+      userId: string;
+    };
+    output: {
+      inputTokens: number | null;
+      outputTokens: number | null;
+      totalTokens: number | null;
+      cost: number | null;
+    };
+    meta: any;
+  }>;
+}>>;
+//#endregion
+export { createAITRPC };
+//# sourceMappingURL=ai.trpc.d.cts.map