@m5kdev/backend 0.1.0 → 0.1.2

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

@@ -0,0 +1,115 @@
+import { type AiEmbeddingModel, type AiModel } from "@m5kdev/commons/modules/ai/ai.constants";
+import { arrayToPseudoXML } from "@m5kdev/commons/modules/ai/ai.utils";
+import type { Mastra } from "@mastra/core";
+import type { FullOutput, MastraModelOutput } from "@mastra/core/stream";
+import { MDocument } from "@mastra/rag";
+import type { OpenRouterProvider } from "@openrouter/ai-sdk-provider";
+import { generateObject, generateText } from "ai";
+import type Replicate from "replicate";
+import type { ZodType, z } from "zod";
+import type { AiUsageRepository, AiUsageRow } from "#modules/ai/ai.repository";
+import type { IdeogramV3GenerateInput, IdeogramV3GenerateOutput } from "#modules/ai/ideogram/ideogram.dto";
+import type { IdeogramService } from "#modules/ai/ideogram/ideogram.service";
+import type { User } from "#modules/auth/auth.lib";
+import type { ServerResultAsync } from "#modules/base/base.dto";
+import { BaseService } from "#modules/base/base.service";
+type MastraAgent = ReturnType<Mastra["getAgent"]>;
+type MastraAgentGenerateOptions = Parameters<MastraAgent["generate"]>[1];
+type MessageListInput = {
+    role: "user" | "assistant" | "system";
+    content: string;
+}[];
+export declare class AIService<MastraInstance extends Mastra> extends BaseService<{
+    aiUsage?: AiUsageRepository;
+}, {
+    ideogram?: IdeogramService;
+}> {
+    helpers: {
+        arrayToPseudoXML: typeof arrayToPseudoXML;
+    };
+    mastra?: MastraInstance;
+    openrouter?: OpenRouterProvider;
+    replicate?: Replicate;
+    constructor(repositories: {
+        aiUsage?: AiUsageRepository;
+    }, services: {
+        ideogram?: IdeogramService;
+    }, libs: {
+        mastra?: MastraInstance;
+        openrouter?: OpenRouterProvider;
+        replicate?: Replicate;
+    });
+    getMastra(): MastraInstance;
+    prepareModel(model: AiModel): any;
+    prepareEmbeddingModel(model: AiEmbeddingModel): any;
+    agentUse(agent: string, options: MastraAgentGenerateOptions & {
+        prompt?: string;
+        messages?: MessageListInput;
+    }, ctx?: {
+        user: User;
+        model?: string;
+    }): ServerResultAsync<Awaited<ReturnType<MastraModelOutput<any>["getFullOutput"]>>>;
+    agentText(agent: string, options: MastraAgentGenerateOptions & {
+        prompt?: string;
+        messages?: MessageListInput;
+    }, ctx?: {
+        user: User;
+        model?: string;
+    }): ServerResultAsync<string>;
+    agentTextResult(agent: string, options: MastraAgentGenerateOptions & {
+        prompt?: string;
+        messages?: MessageListInput;
+    }, ctx?: {
+        user: User;
+        model?: string;
+    }): ServerResultAsync<FullOutput<any>>;
+    agentObject<T extends ZodType<any>>(agent: string, options: MastraAgentGenerateOptions & {
+        schema: T;
+        prompt?: string;
+        messages?: MessageListInput;
+    }, ctx?: {
+        user: User;
+        model?: string;
+    }): ServerResultAsync<z.infer<T>>;
+    agentObjectResult<T extends ZodType<any>>(agent: string, options: MastraAgentGenerateOptions & {
+        schema: T;
+        prompt?: string;
+        messages?: MessageListInput;
+    }, ctx?: {
+        user: User;
+        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?: AiEmbeddingModel): ServerResultAsync<{
+        embeddings: number[][];
+        chunks: {
+            text: string;
+        }[];
+    }>;
+    embed(text: string, model?: AiEmbeddingModel): ServerResultAsync<{
+        embedding: number[];
+    }>;
+    embedMany(chunks: {
+        text: string;
+    }[], model?: AiEmbeddingModel): ServerResultAsync<{
+        embeddings: number[][];
+    }>;
+    generateText(params: Omit<Parameters<typeof generateText>[0], "model"> & {
+        model: AiModel;
+        removeMDash?: boolean;
+    }): ServerResultAsync<string>;
+    generateObject<T extends ZodType>(params: Omit<Parameters<typeof generateObject<T>>[0], "model" | "schema"> & {
+        model: AiModel;
+        schema: 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">>;
+}
+export {};
+//# sourceMappingURL=ai.service.d.ts.map

package/dist/src/modules/ai/ai.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.service.d.ts","sourceRoot":"","sources":["../../../../src/modules/ai/ai.service.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,gBAAgB,EACrB,KAAK,OAAO,EAEb,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AACvE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAE3C,OAAO,KAAK,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AACtE,OAAO,EAAoB,cAAc,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAEpE,OAAO,KAAK,SAAS,MAAM,WAAW,CAAC;AACvC,OAAO,KAAK,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACtC,OAAO,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAC/E,OAAO,KAAK,EACV,uBAAuB,EACvB,wBAAwB,EACzB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AAC7E,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD,KAAK,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;AAClD,KAAK,0BAA0B,GAAG,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACzE,KAAK,gBAAgB,GAAG;IAAE,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,EAAE,CAAC;AAErF,qBAAa,SAAS,CAAC,cAAc,SAAS,MAAM,CAAE,SAAQ,WAAW,CACvE;IAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;CAAE,EAC/B;IAAE,QAAQ,CAAC,EAAE,eAAe,CAAA;CAAE,CAC/B;IACC,OAAO;;MAEL;IAEF,MAAM,CAAC,EAAE,cAAc,CAAC;IACxB,UAAU,CAAC,EAAE,kBAAkB,CAAC;IAChC,SAAS,CAAC,EAAE,SAAS,CAAC;gBAGpB,YAAY,EAAE;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAC7C,QAAQ,EAAE;QAAE,QAAQ,CAAC,EAAE,eAAe,CAAA;KAAE,EACxC,IAAI,EAAE;QAAE,MAAM,CAAC,EAAE,cAAc,CAAC;QAAC,UAAU,CAAC,EAAE,kBAAkB,CAAC;QAAC,SAAS,CAAC,EAAE,SAAS,CAAA;KAAE;IAQ3F,SAAS,IAAI,cAAc;IAO3B,YAAY,CAAC,KAAK,EAAE,OAAO,GAAG,GAAG;IAQjC,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,GAAG,GAAG;IAQ7C,QAAQ,CACZ,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,0BAA0B,GAAG;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,gBAAgB,CAAA;KAAE,EACtF,GAAG,CAAC,EAAE;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACnC,iBAAiB,CAAC,OAAO,CAAC,UAAU,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAwC5E,SAAS,CACb,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,0BAA0B,GAAG;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,gBAAgB,CAAA;KAAE,EACtF,GAAG,CAAC,EAAE;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACnC,iBAAiB,CAAC,MAAM,CAAC;IAOtB,eAAe,CACnB,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,0BAA0B,GAAG;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,gBAAgB,CAAA;KAAE,EACtF,GAAG,CAAC,EAAE;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACnC,iBAAiB,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IAM/B,WAAW,CAAC,CAAC,SAAS,OAAO,CAAC,GAAG,CAAC,EACtC,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,0BAA0B,GAAG;QACpC,MAAM,EAAE,CAAC,CAAC;QACV,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,gBAAgB,CAAC;KAC7B,EACD,GAAG,CAAC,EAAE;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACnC,iBAAiB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAQ1B,iBAAiB,CAAC,CAAC,SAAS,OAAO,CAAC,GAAG,CAAC,EAC5C,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,0BAA0B,GAAG;QACpC,MAAM,EAAE,CAAC,CAAC;QACV,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,gBAAgB,CAAC;KAC7B,EACD,GAAG,CAAC,EAAE;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACnC,iBAAiB,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG;QAAE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;KAAE,CAAC;IASxD,aAAa,CACjB,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,UAAU,CAAC,UAAU,CAAC,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EACvE,IAAI,GAAE,MAAM,GAAG,UAAU,GAAG,MAAM,GAAG,MAAe,EACpD,KAAK,GAAE,gBAAgD,GACtD,iBAAiB,CAAC;QAAE,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC;QAAC,MAAM,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,EAAE,CAAA;KAAE,CAAC;IAoBtE,KAAK,CACT,IAAI,EAAE,MAAM,EACZ,KAAK,GAAE,gBAAgD,GACtD,iBAAiB,CAAC;QAAE,SAAS,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAUvC,SAAS,CACb,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,EAC1B,KAAK,GAAE,gBAAgD,GACtD,iBAAiB,CAAC;QAAE,UAAU,EAAE,MAAM,EAAE,EAAE,CAAA;KAAE,CAAC;IAU1C,YAAY,CAChB,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG;QAC1D,KAAK,EAAE,OAAO,CAAC;QACf,WAAW,CAAC,EAAE,OAAO,CAAC;KACvB,GACA,iBAAiB,CAAC,MAAM,CAAC;IAQtB,cAAc,CAAC,CAAC,SAAS,OAAO,EACpC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,GAAG,QAAQ,CAAC,GAAG;QAC1E,KAAK,EAAE,OAAO,CAAC;QACf,MAAM,EAAE,CAAC,CAAC;KACX,GACA,iBAAiB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAY1B,iBAAiB,CACrB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EACtC,OAAO,EAAE,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GACvC,iBAAiB,CAAC,MAAM,CAAC;IAatB,kBAAkB,CACtB,QAAQ,EAAE,MAAM,GACf,iBAAiB,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,OAAO,CAAA;KAAE,CAAC;IAoBnD,gBAAgB,CACpB,KAAK,EAAE,uBAAuB,GAC7B,iBAAiB,CAAC,wBAAwB,CAAC;IAOxC,QAAQ,CACZ,MAAM,EAAE,MAAM,GACb,iBAAiB,CAAC,IAAI,CAAC,UAAU,EAAE,aAAa,GAAG,cAAc,GAAG,aAAa,GAAG,MAAM,CAAC,CAAC;CAMhG"}

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

@@ -0,0 +1,207 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIService = void 0;
+const ai_constants_1 = require("@m5kdev/commons/modules/ai/ai.constants");
+const ai_utils_1 = require("@m5kdev/commons/modules/ai/ai.utils");
+const request_context_1 = require("@mastra/core/request-context");
+const rag_1 = require("@mastra/rag");
+const ai_1 = require("ai");
+const neverthrow_1 = require("neverthrow");
+const base_service_1 = require("#modules/base/base.service");
+class AIService extends base_service_1.BaseService {
+    helpers = {
+        arrayToPseudoXML: ai_utils_1.arrayToPseudoXML,
+    };
+    mastra;
+    openrouter;
+    replicate;
+    constructor(repositories, services, libs) {
+        super(repositories, services);
+        this.mastra = libs.mastra;
+        this.openrouter = libs.openrouter;
+        this.replicate = libs.replicate;
+    }
+    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");
+        }
+        const openrouterModel = this.openrouter.chat(model);
+        return openrouterModel;
+    }
+    prepareEmbeddingModel(model) {
+        if (!this.openrouter) {
+            throw new Error("OpenRouter is not configured");
+        }
+        const openrouterModel = this.openrouter.textEmbeddingModel(model);
+        return openrouterModel;
+    }
+    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 request_context_1.RequestContext();
+            if (ctx?.user) {
+                requestContext.set("userId", ctx.user.id);
+            }
+            if (ctx?.model) {
+                requestContext.set("model", ctx.model);
+            }
+            const mAgent = this.getMastra().getAgent(agent);
+            const result = await mAgent.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?.user?.id,
+                    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_1.err)(createUsageResult.error);
+            }
+            this.logger.info("AGENT USE CREATED USAGE");
+            return (0, neverthrow_1.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_1.ok)(result.value.text);
+    }
+    async agentTextResult(agent, options, ctx) {
+        const result = await this.agentUse(agent, options, ctx);
+        if (result.isErr())
+            return (0, neverthrow_1.err)(result.error);
+        return (0, neverthrow_1.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_1.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_1.err)(result.error);
+        this.logger.info("AGENT OBJECT RESULT DONE");
+        return (0, neverthrow_1.ok)({ ...result.value, object: result.value.object });
+    }
+    async embedDocument(value, options, type = "text", model = ai_constants_1.OPENAI_TEXT_EMBEDDING_3_SMALL) {
+        return this.throwableAsync(async () => {
+            if (type === "text") {
+                const doc = rag_1.MDocument.fromText(value);
+                const chunks = await doc.chunk(options ?? {
+                    strategy: "recursive",
+                    maxSize: 512,
+                    overlap: 50,
+                    separators: ["\n"],
+                });
+                const embeddings = await this.embedMany(chunks, model);
+                if (embeddings.isErr())
+                    return (0, neverthrow_1.err)(embeddings.error);
+                return (0, neverthrow_1.ok)({ embeddings: embeddings.value.embeddings, chunks });
+            }
+            return this.error("BAD_REQUEST", "Unsupported document type");
+        });
+    }
+    async embed(text, model = ai_constants_1.OPENAI_TEXT_EMBEDDING_3_SMALL) {
+        return this.throwableAsync(async () => {
+            const result = await (0, ai_1.embed)({
+                model: this.prepareEmbeddingModel(model),
+                value: text,
+            });
+            return (0, neverthrow_1.ok)(result);
+        });
+    }
+    async embedMany(chunks, model = ai_constants_1.OPENAI_TEXT_EMBEDDING_3_SMALL) {
+        return this.throwableAsync(async () => {
+            const result = await (0, ai_1.embedMany)({
+                model: this.prepareEmbeddingModel(model),
+                values: chunks.map((chunk) => chunk.text),
+            });
+            return (0, neverthrow_1.ok)(result);
+        });
+    }
+    async generateText(params) {
+        return this.throwableAsync(async () => {
+            const { removeMDash = true, model, ...rest } = params;
+            const result = await (0, ai_1.generateText)({ ...rest, model: this.prepareModel(model) });
+            return (0, neverthrow_1.ok)(removeMDash ? result.text.replace(/\u2013|\u2014/g, "-") : result.text);
+        });
+    }
+    async generateObject(params) {
+        return this.throwableAsync(async () => {
+            const model = this.prepareModel(params.model);
+            const result = await (0, ai_1.generateObject)({
+                ...params,
+                model,
+                schema: params.schema,
+            });
+            return (0, neverthrow_1.ok)(result.object);
+        });
+    }
+    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_1.ok)(await this.replicate.run(model, options));
+            }
+            catch (error) {
+                return this.error("INTERNAL_SERVER_ERROR", undefined, { cause: error });
+            }
+        });
+    }
+    async generateTranscript(file_url) {
+        const output = await this.generateReplicate("thomasmol/whisper-diarization:1495a9cddc83b2203b0d8d3516e38b80fd1572ebc4bc5700ac1da56a9b3ed886", {
+            input: {
+                file_url,
+            },
+        });
+        if (output.isErr())
+            return (0, neverthrow_1.err)(output.error);
+        try {
+            const { segments } = output.value;
+            return (0, neverthrow_1.ok)({ text: segments.map((segment) => segment.text).join(""), metadata: segments });
+        }
+        catch (error) {
+            return this.error("INTERNAL_SERVER_ERROR", undefined, { 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);
+    }
+}
+exports.AIService = AIService;

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

@@ -0,0 +1,59 @@
+import type { Mastra } from "@mastra/core";
+import type { AIService } from "#modules/ai/ai.service";
+export declare function createAITRPC<MastraInstance extends Mastra>(aiService: AIService<MastraInstance>): import("@trpc/server").TRPCBuiltRouter<{
+    ctx: {
+        session: {
+            id: string;
+            userId: string;
+            updatedAt: Date;
+            createdAt: Date;
+            expiresAt: Date;
+            token: string;
+            ipAddress: string | null;
+            userAgent: string | null;
+            impersonatedBy: string | null;
+            activeOrganizationId: string | null;
+            activeOrganizationRole: string | null;
+            activeTeamId: string | null;
+            activeTeamRole: string | null;
+        };
+        user: {
+            name: string;
+            id: string;
+            updatedAt: Date;
+            email: string;
+            emailVerified: boolean;
+            image: string | null;
+            createdAt: Date;
+            role: string | null;
+            banned: boolean | null;
+            banReason: string | null;
+            banExpires: Date | null;
+            stripeCustomerId: string | null;
+            paymentCustomerId: string | null;
+            paymentPlanTier: string | null;
+            paymentPlanExpiresAt: Date | null;
+            preferences: string | null;
+            metadata: Record<string, unknown>;
+            onboarding: number | null;
+            flags: string | null;
+        };
+    };
+    meta: import("trpc-to-openapi").OpenApiMeta;
+    errorShape: import("@trpc/server").TRPCDefaultErrorShape;
+    transformer: true;
+}, import("@trpc/server").TRPCDecorateCreateRouterOptions<{
+    getUserUsage: import("@trpc/server").TRPCQueryProcedure<{
+        input: {
+            userId: string;
+        };
+        output: {
+            inputTokens: number | null;
+            outputTokens: number | null;
+            totalTokens: number | null;
+            cost: number | null;
+        };
+        meta: import("trpc-to-openapi").OpenApiMeta;
+    }>;
+}>>;
+//# sourceMappingURL=ai.trpc.d.ts.map

package/dist/src/modules/ai/ai.trpc.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.trpc.d.ts","sourceRoot":"","sources":["../../../../src/modules/ai/ai.trpc.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAE3C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAGxD,wBAAgB,YAAY,CAAC,cAAc,SAAS,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,cAAc,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAgB/F"}

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

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createAITRPC = createAITRPC;
+const zod_1 = require("zod");
+const _trpc_1 = require("#trpc");
+function createAITRPC(aiService) {
+    return (0, _trpc_1.router)({
+        getUserUsage: _trpc_1.adminProcedure
+            .input(zod_1.z.object({ userId: zod_1.z.string() }))
+            .output(zod_1.z.object({
+            inputTokens: zod_1.z.number().nullable(),
+            outputTokens: zod_1.z.number().nullable(),
+            totalTokens: zod_1.z.number().nullable(),
+            cost: zod_1.z.number().nullable(),
+        }))
+            .query(async ({ input }) => {
+            return (0, _trpc_1.handleTRPCResult)(await aiService.getUsage(input.userId));
+        }),
+    });
+}

package/dist/src/modules/ai/ideogram/ideogram.constants.d.ts

@@ -0,0 +1,8 @@
+export declare const ideogramStyleTypes: readonly ["AUTO", "GENERAL", "REALISTIC", "DESIGN", "FICTION"];
+export declare const ideogramStylePresets: readonly ["80S_ILLUSTRATION", "90S_NOSTALGIA", "ABSTRACT_ORGANIC", "ANALOG_NOSTALGIA", "ART_BRUT", "ART_DECO", "ART_POSTER", "AURA", "AVANT_GARDE", "BAUHAUS", "BLUEPRINT", "BLURRY_MOTION", "BRIGHT_ART", "C4D_CARTOON", "CHILDRENS_BOOK", "COLLAGE", "COLORING_BOOK_I", "COLORING_BOOK_II", "CUBISM", "DARK_AURA", "DOODLE", "DOUBLE_EXPOSURE", "DRAMATIC_CINEMA", "EDITORIAL", "EMOTIONAL_MINIMAL", "ETHEREAL_PARTY", "EXPIRED_FILM", "FLAT_ART", "FLAT_VECTOR", "FOREST_REVERIE", "GEO_MINIMALIST", "GLASS_PRISM", "GOLDEN_HOUR", "GRAFFITI_I", "GRAFFITI_II", "HALFTONE_PRINT", "HIGH_CONTRAST", "HIPPIE_ERA", "ICONIC", "JAPANDI_FUSION", "JAZZY", "LONG_EXPOSURE", "MAGAZINE_EDITORIAL", "MINIMAL_ILLUSTRATION", "MIXED_MEDIA", "MONOCHROME", "NIGHTLIFE", "OIL_PAINTING", "OLD_CARTOONS", "PAINT_GESTURE", "POP_ART", "RETRO_ETCHING", "RIVIERA_POP", "SPOTLIGHT_80S", "STYLIZED_RED", "SURREAL_COLLAGE", "TRAVEL_POSTER", "VINTAGE_GEO", "VINTAGE_POSTER"];
+export declare const ideogramResolutions: readonly ["512x1536", "576x1408", "576x1472", "576x1536", "640x1344", "640x1408", "640x1472", "640x1536", "704x1152", "704x1216", "704x1280", "704x1344", "704x1408", "704x1472", "736x1312", "768x1088", "768x1216", "768x1280", "768x1344", "800x1280", "832x960", "832x1024", "832x1088", "832x1152", "832x1216", "832x1248", "864x1152", "896x960", "896x1024", "896x1088", "896x1120", "896x1152", "960x832", "960x896", "960x1024", "960x1088", "960x1120", "960x1152", "1024x832", "1024x896", "1024x960", "1024x1024", "1088x768", "1088x832", "1088x896", "1088x960", "1120x896", "1152x704", "1152x832", "1152x864", "1152x896", "1216x704", "1216x768", "1216x832", "1248x832", "1280x704", "1280x768", "1280x800", "1312x736", "1344x640", "1344x704", "1344x768", "1408x576", "1408x640", "1408x704", "1472x576", "1472x640", "1472x704", "1536x512", "1536x576", "1536x640"];
+export declare const ideogramAspectRatios: readonly ["1x3", "3x1", "1x2", "2x1", "9x16", "16x9", "10x16", "16x10", "2x3", "3x2", "3x4", "4x3", "4x5", "5x4", "1x1"];
+export declare const ideogramRenderingSpeed: readonly ["FLASH", "TURBO", "DEFAULT", "QUALITY"];
+export declare const ideogramMagicPrompt: readonly ["AUTO", "ON", "OFF"];
+export declare const ideogramColorPalettePreset: readonly ["EMBER", "FRESH", "JUNGLE", "MAGIC", "MELON", "MOSAIC", "PASTEL", "ULTRAMARINE"];
+//# sourceMappingURL=ideogram.constants.d.ts.map

package/dist/src/modules/ai/ideogram/ideogram.constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ideogram.constants.d.ts","sourceRoot":"","sources":["../../../../../src/modules/ai/ideogram/ideogram.constants.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,kBAAkB,gEAAiE,CAAC;AAEjG,eAAO,MAAM,oBAAoB,q6BA4DvB,CAAC;AAEX,eAAO,MAAM,mBAAmB,41BAwEtB,CAAC;AAEX,eAAO,MAAM,oBAAoB,0HAgBvB,CAAC;AAEX,eAAO,MAAM,sBAAsB,mDAAoD,CAAC;AAExF,eAAO,MAAM,mBAAmB,gCAAiC,CAAC;AAElE,eAAO,MAAM,0BAA0B,4FAS7B,CAAC"}

package/dist/src/modules/ai/ideogram/ideogram.constants.js

@@ -0,0 +1,167 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ideogramColorPalettePreset = exports.ideogramMagicPrompt = exports.ideogramRenderingSpeed = exports.ideogramAspectRatios = exports.ideogramResolutions = exports.ideogramStylePresets = exports.ideogramStyleTypes = void 0;
+exports.ideogramStyleTypes = ["AUTO", "GENERAL", "REALISTIC", "DESIGN", "FICTION"];
+exports.ideogramStylePresets = [
+    "80S_ILLUSTRATION",
+    "90S_NOSTALGIA",
+    "ABSTRACT_ORGANIC",
+    "ANALOG_NOSTALGIA",
+    "ART_BRUT",
+    "ART_DECO",
+    "ART_POSTER",
+    "AURA",
+    "AVANT_GARDE",
+    "BAUHAUS",
+    "BLUEPRINT",
+    "BLURRY_MOTION",
+    "BRIGHT_ART",
+    "C4D_CARTOON",
+    "CHILDRENS_BOOK",
+    "COLLAGE",
+    "COLORING_BOOK_I",
+    "COLORING_BOOK_II",
+    "CUBISM",
+    "DARK_AURA",
+    "DOODLE",
+    "DOUBLE_EXPOSURE",
+    "DRAMATIC_CINEMA",
+    "EDITORIAL",
+    "EMOTIONAL_MINIMAL",
+    "ETHEREAL_PARTY",
+    "EXPIRED_FILM",
+    "FLAT_ART",
+    "FLAT_VECTOR",
+    "FOREST_REVERIE",
+    "GEO_MINIMALIST",
+    "GLASS_PRISM",
+    "GOLDEN_HOUR",
+    "GRAFFITI_I",
+    "GRAFFITI_II",
+    "HALFTONE_PRINT",
+    "HIGH_CONTRAST",
+    "HIPPIE_ERA",
+    "ICONIC",
+    "JAPANDI_FUSION",
+    "JAZZY",
+    "LONG_EXPOSURE",
+    "MAGAZINE_EDITORIAL",
+    "MINIMAL_ILLUSTRATION",
+    "MIXED_MEDIA",
+    "MONOCHROME",
+    "NIGHTLIFE",
+    "OIL_PAINTING",
+    "OLD_CARTOONS",
+    "PAINT_GESTURE",
+    "POP_ART",
+    "RETRO_ETCHING",
+    "RIVIERA_POP",
+    "SPOTLIGHT_80S",
+    "STYLIZED_RED",
+    "SURREAL_COLLAGE",
+    "TRAVEL_POSTER",
+    "VINTAGE_GEO",
+    "VINTAGE_POSTER",
+];
+exports.ideogramResolutions = [
+    "512x1536",
+    "576x1408",
+    "576x1472",
+    "576x1536",
+    "640x1344",
+    "640x1408",
+    "640x1472",
+    "640x1536",
+    "704x1152",
+    "704x1216",
+    "704x1280",
+    "704x1344",
+    "704x1408",
+    "704x1472",
+    "736x1312",
+    "768x1088",
+    "768x1216",
+    "768x1280",
+    "768x1344",
+    "800x1280",
+    "832x960",
+    "832x1024",
+    "832x1088",
+    "832x1152",
+    "832x1216",
+    "832x1248",
+    "864x1152",
+    "896x960",
+    "896x1024",
+    "896x1088",
+    "896x1120",
+    "896x1152",
+    "960x832",
+    "960x896",
+    "960x1024",
+    "960x1088",
+    "960x1120",
+    "960x1152",
+    "1024x832",
+    "1024x896",
+    "1024x960",
+    "1024x1024",
+    "1088x768",
+    "1088x832",
+    "1088x896",
+    "1088x960",
+    "1120x896",
+    "1152x704",
+    "1152x832",
+    "1152x864",
+    "1152x896",
+    "1216x704",
+    "1216x768",
+    "1216x832",
+    "1248x832",
+    "1280x704",
+    "1280x768",
+    "1280x800",
+    "1312x736",
+    "1344x640",
+    "1344x704",
+    "1344x768",
+    "1408x576",
+    "1408x640",
+    "1408x704",
+    "1472x576",
+    "1472x640",
+    "1472x704",
+    "1536x512",
+    "1536x576",
+    "1536x640",
+];
+exports.ideogramAspectRatios = [
+    "1x3",
+    "3x1",
+    "1x2",
+    "2x1",
+    "9x16",
+    "16x9",
+    "10x16",
+    "16x10",
+    "2x3",
+    "3x2",
+    "3x4",
+    "4x3",
+    "4x5",
+    "5x4",
+    "1x1",
+];
+exports.ideogramRenderingSpeed = ["FLASH", "TURBO", "DEFAULT", "QUALITY"];
+exports.ideogramMagicPrompt = ["AUTO", "ON", "OFF"];
+exports.ideogramColorPalettePreset = [
+    "EMBER",
+    "FRESH",
+    "JUNGLE",
+    "MAGIC",
+    "MELON",
+    "MOSAIC",
+    "PASTEL",
+    "ULTRAMARINE",
+];