@drax/ai-back 3.16.0 → 3.17.1

package/dist/controllers/AIController.js

@@ -0,0 +1,150 @@
+import { z } from "zod";
+import { CommonController } from "@drax/common-back";
+import AiProviderFactory from "../factory/AiProviderFactory.js";
+import AIPermissions from "../permissions/AIPermissions.js";
+const CrudAiFieldSchema = z.lazy(() => z.object({
+    name: z.string(),
+    type: z.string(),
+    label: z.string(),
+    hint: z.string().nullable().default(null),
+    placeholder: z.string().nullable().default(null),
+    readonly: z.boolean().nullable().default(null),
+    default: z.any().nullable().default(null),
+    enum: z.array(z.string()).nullable().default(null),
+    items: z.array(z.object({
+        title: z.string().nullable().default(null),
+        value: z.any().nullable().default(null),
+    })).nullable().default(null),
+    ref: z.string().nullable().default(null),
+    refDisplay: z.string().nullable().default(null),
+    objectFields: z.array(CrudAiFieldSchema).nullable().default(null),
+}));
+const PromptRequestSchema = z.object({
+    prompt: z.string().min(1),
+    operation: z.enum(["create", "edit"]).default("create"),
+    entity: z.object({
+        name: z.string(),
+        identifier: z.string().optional().nullable(),
+    }),
+    currentValues: z.record(z.string(), z.any()).default({}),
+    fields: z.array(CrudAiFieldSchema).min(1),
+});
+class AIController extends CommonController {
+    buildFieldValueSchema(field) {
+        switch (field.type) {
+            case "number":
+                return z.number().nullable().default(null);
+            case "boolean":
+                return z.boolean().nullable().default(null);
+            case "array.string":
+            case "array.enum":
+            case "array.ref":
+                return z.array(z.string()).nullable().default(null);
+            case "array.number":
+                return z.array(z.number()).nullable().default(null);
+            case "object":
+                if (field.objectFields && Array.isArray(field.objectFields) && field.objectFields.length > 0) {
+                    return z.object(this.buildFieldShape(field.objectFields)).nullable().default(null);
+                }
+                return z.string().nullable().default(null);
+            case "record":
+            case "array.object":
+            case "array.record":
+            case "array.fullFile":
+            case "file":
+            case "fullFile":
+                return z.string().nullable().default(null);
+            case "id":
+            case "string":
+            case "longString":
+            case "date":
+            case "ref":
+            case "enum":
+            case "select":
+            case "password":
+            default:
+                return z.string().nullable().default(null);
+        }
+    }
+    buildFieldShape(fields) {
+        return fields.reduce((acc, field) => {
+            acc[field.name] = this.buildFieldValueSchema(field);
+            return acc;
+        }, {});
+    }
+    buildSystemPrompt(input) {
+        return [
+            "Sos un asistente de formularios para un sistema CRUD.",
+            "Tu tarea es proponer valores JSON para completar o editar una entidad.",
+            "Debes respetar exactamente los nombres de campo entregados.",
+            "No inventes campos adicionales.",
+            "Si un campo no tiene una propuesta razonable, omitilo.",
+            "Usa tipos compatibles con JSON.",
+            "Para campos enum o select, elegí solamente valores válidos de la lista provista.",
+            "Para campos record, object sin estructura fija, array.object o array.record, devolve un string JSON serializado valido.",
+            "La respuesta debe describir sugerencias concretas, breves y aplicables.",
+            `Operacion actual: ${input.operation}.`,
+            `Entidad actual: ${input.entity.name}.`,
+        ].join("\n");
+    }
+    buildUserInput(input) {
+        return JSON.stringify({
+            task: input.prompt,
+            operation: input.operation,
+            entity: input.entity,
+            currentValues: input.currentValues,
+            fields: input.fields,
+            expectedResponse: {
+                message: "string",
+                suggestions: "object with proposed values indexed by field name",
+                warnings: ["optional string warnings"],
+            }
+        });
+    }
+    async prompt(req, rep) {
+        try {
+            req.rbac.assertPermission(AIPermissions.PromptCrud);
+            const input = PromptRequestSchema.parse(req.body ?? {});
+            const aiProvider = AiProviderFactory.instance();
+            const responseSchema = z.object({
+                message: z.string().nullable().default(null),
+                suggestions: z.object(this.buildFieldShape(input.fields)).strict().default({}),
+                warnings: z.array(z.string()).default([]),
+            });
+            const response = await aiProvider.prompt({
+                systemPrompt: this.buildSystemPrompt(input),
+                userInput: this.buildUserInput(input),
+                zodSchema: responseSchema,
+                operationTitle: `crud-${input.operation}-assistant`,
+                operationGroup: "crud-form-assistant",
+                ip: req.ip,
+                userAgent: req.headers["user-agent"],
+                tenant: req.rbac?.tenantId ?? null,
+                user: req.rbac?.userId ?? null,
+            });
+            const parsedOutput = responseSchema.parse(typeof response.output === "string"
+                ? JSON.parse(response.output)
+                : response.output);
+            return rep.send({
+                message: parsedOutput.message,
+                suggestions: parsedOutput.suggestions,
+                warnings: parsedOutput.warnings,
+                meta: {
+                    tokens: response.tokens,
+                    inputTokens: response.inputTokens,
+                    outputTokens: response.outputTokens,
+                    time: response.time,
+                }
+            });
+        }
+        catch (e) {
+            console.error("AIController.prompt error", e);
+            const statusCode = e?.name === "ZodError" ? 400 : 500;
+            return rep.status(statusCode).send({
+                message: e?.message || "AI prompt error",
+            });
+        }
+    }
+}
+export default AIController;
+export { AIController };

package/dist/controllers/AICrudController.js

@@ -0,0 +1,150 @@
+import { z } from "zod";
+import { CommonController } from "@drax/common-back";
+import AiProviderFactory from "../factory/AiProviderFactory.js";
+import AIPermissions from "../permissions/AIPermissions.js";
+const CrudAiFieldSchema = z.lazy(() => z.object({
+    name: z.string(),
+    type: z.string(),
+    label: z.string(),
+    hint: z.string().nullable().default(null),
+    placeholder: z.string().nullable().default(null),
+    readonly: z.boolean().nullable().default(null),
+    default: z.any().nullable().default(null),
+    enum: z.array(z.string()).nullable().default(null),
+    items: z.array(z.object({
+        title: z.string().nullable().default(null),
+        value: z.any().nullable().default(null),
+    })).nullable().default(null),
+    ref: z.string().nullable().default(null),
+    refDisplay: z.string().nullable().default(null),
+    objectFields: z.array(CrudAiFieldSchema).nullable().default(null),
+}));
+const PromptRequestSchema = z.object({
+    prompt: z.string().min(1),
+    operation: z.enum(["create", "edit"]).default("create"),
+    entity: z.object({
+        name: z.string(),
+        identifier: z.string().optional().nullable(),
+    }),
+    currentValues: z.record(z.string(), z.any()).default({}),
+    fields: z.array(CrudAiFieldSchema).min(1),
+});
+class AICrudController extends CommonController {
+    buildFieldValueSchema(field) {
+        switch (field.type) {
+            case "number":
+                return z.number().nullable().default(null);
+            case "boolean":
+                return z.boolean().nullable().default(null);
+            case "array.string":
+            case "array.enum":
+            case "array.ref":
+                return z.array(z.string()).nullable().default(null);
+            case "array.number":
+                return z.array(z.number()).nullable().default(null);
+            case "object":
+                if (field.objectFields && Array.isArray(field.objectFields) && field.objectFields.length > 0) {
+                    return z.object(this.buildFieldShape(field.objectFields)).nullable().default(null);
+                }
+                return z.string().nullable().default(null);
+            case "record":
+            case "array.object":
+            case "array.record":
+            case "array.fullFile":
+            case "file":
+            case "fullFile":
+                return z.string().nullable().default(null);
+            case "id":
+            case "string":
+            case "longString":
+            case "date":
+            case "ref":
+            case "enum":
+            case "select":
+            case "password":
+            default:
+                return z.string().nullable().default(null);
+        }
+    }
+    buildFieldShape(fields) {
+        return fields.reduce((acc, field) => {
+            acc[field.name] = this.buildFieldValueSchema(field);
+            return acc;
+        }, {});
+    }
+    buildSystemPrompt(input) {
+        return [
+            "Sos un asistente de formularios para un sistema CRUD.",
+            "Tu tarea es proponer valores JSON para completar o editar una entidad.",
+            "Debes respetar exactamente los nombres de campo entregados.",
+            "No inventes campos adicionales.",
+            "Si un campo no tiene una propuesta razonable, omitilo.",
+            "Usa tipos compatibles con JSON.",
+            "Para campos enum o select, elegí solamente valores válidos de la lista provista.",
+            "Para campos record, object sin estructura fija, array.object o array.record, devolve un string JSON serializado valido.",
+            "La respuesta debe describir sugerencias concretas, breves y aplicables.",
+            `Operacion actual: ${input.operation}.`,
+            `Entidad actual: ${input.entity.name}.`,
+        ].join("\n");
+    }
+    buildUserInput(input) {
+        return JSON.stringify({
+            task: input.prompt,
+            operation: input.operation,
+            entity: input.entity,
+            currentValues: input.currentValues,
+            fields: input.fields,
+            expectedResponse: {
+                message: "string",
+                suggestions: "object with proposed values indexed by field name",
+                warnings: ["optional string warnings"],
+            }
+        });
+    }
+    async prompt(req, rep) {
+        try {
+            req.rbac.assertPermission(AIPermissions.PromptCrud);
+            const input = PromptRequestSchema.parse(req.body ?? {});
+            const aiProvider = AiProviderFactory.instance();
+            const responseSchema = z.object({
+                message: z.string().nullable().default(null),
+                suggestions: z.object(this.buildFieldShape(input.fields)).strict().default({}),
+                warnings: z.array(z.string()).default([]),
+            });
+            const response = await aiProvider.prompt({
+                systemPrompt: this.buildSystemPrompt(input),
+                userInput: this.buildUserInput(input),
+                zodSchema: responseSchema,
+                operationTitle: `crud-${input.operation}-assistant`,
+                operationGroup: "crud-form-assistant",
+                ip: req.ip,
+                userAgent: req.headers["user-agent"],
+                tenant: req.rbac?.tenantId ?? null,
+                user: req.rbac?.userId ?? null,
+            });
+            const parsedOutput = responseSchema.parse(typeof response.output === "string"
+                ? JSON.parse(response.output)
+                : response.output);
+            return rep.send({
+                message: parsedOutput.message,
+                suggestions: parsedOutput.suggestions,
+                warnings: parsedOutput.warnings,
+                meta: {
+                    tokens: response.tokens,
+                    inputTokens: response.inputTokens,
+                    outputTokens: response.outputTokens,
+                    time: response.time,
+                }
+            });
+        }
+        catch (e) {
+            console.error("AIController.prompt error", e);
+            const statusCode = e?.name === "ZodError" ? 400 : 500;
+            return rep.status(statusCode).send({
+                message: e?.message || "AI prompt error",
+            });
+        }
+    }
+}
+export default AICrudController;
+export { AICrudController };

package/dist/controllers/AIGenericController.js

@@ -0,0 +1,83 @@
+import { z } from "zod";
+import { CommonController } from "@drax/common-back";
+import AiProviderFactory from "../factory/AiProviderFactory.js";
+import AIPermissions from "../permissions/AIPermissions.js";
+const PromptImageSchema = z.object({
+    url: z.string().min(1),
+    detail: z.enum(["auto", "low", "high"]).optional(),
+});
+const PromptContentPartSchema = z.discriminatedUnion("type", [
+    z.object({
+        type: z.literal("text"),
+        text: z.string(),
+    }),
+    z.object({
+        type: z.literal("image"),
+        imageUrl: z.string().min(1),
+        detail: z.enum(["auto", "low", "high"]).optional(),
+    }),
+]);
+const PromptMessageSchema = z.object({
+    role: z.enum(["user", "assistant", "system"]),
+    content: z.union([
+        z.string(),
+        z.array(PromptContentPartSchema),
+    ]),
+});
+const PromptMemorySchema = z.object({
+    key: z.string().min(1),
+    value: z.string(),
+});
+const PromptInputFileSchema = z.object({
+    filename: z.string().optional(),
+    filepath: z.string().optional(),
+    size: z.number().nullable().optional(),
+    mimetype: z.string().optional(),
+    url: z.string().optional(),
+});
+const GenericPromptRequestSchema = z.object({
+    systemPrompt: z.string().min(1),
+    userInput: z.string().optional(),
+    userImages: z.array(PromptImageSchema).optional(),
+    inputFiles: z.array(PromptInputFileSchema).optional(),
+    userContent: z.array(PromptContentPartSchema).optional(),
+    history: z.array(PromptMessageSchema).optional(),
+    memory: z.array(PromptMemorySchema).optional(),
+    memoryHeader: z.string().optional(),
+    knowledgeBase: z.array(z.string()).optional(),
+    knowledgeBaseHeader: z.string().optional(),
+    jsonSchema: z.record(z.string(), z.any()).or(z.array(z.any())).optional(),
+    model: z.string().optional(),
+    operationTitle: z.string().optional(),
+    operationGroup: z.string().optional(),
+});
+class AIGenericController extends CommonController {
+    async prompt(request, reply) {
+        try {
+            request.rbac.assertPermission(AIPermissions.Prompt);
+            const input = GenericPromptRequestSchema.parse(request.body ?? {});
+            const aiProvider = AiProviderFactory.instance();
+            const promptInput = {
+                ...input,
+                operationTitle: input.operationTitle ?? "generic-ai-prompt",
+                operationGroup: input.operationGroup ?? "generic-ai-prompt",
+                ip: request.ip,
+                userAgent: request.headers["user-agent"],
+                tenant: request.rbac?.tenantId ?? null,
+                user: request.rbac?.userId ?? null,
+            };
+            const response = await aiProvider.prompt(promptInput);
+            return reply.send(response);
+        }
+        catch (e) {
+            if (e?.name === "ZodError") {
+                return reply.status(400).send({
+                    message: e?.message || "AI prompt validation error",
+                });
+            }
+            this.handleError(e, reply);
+        }
+    }
+}
+export default AIGenericController;
+export { AIGenericController };

package/dist/factory/AiProviderFactory.js

@@ -1,6 +1,6 @@
 import OpenAiProviderFactory from "./OpenAiProviderFactory.js";
 class AiProviderFactory {
-    static instance(provider) {
+    static instance(provider = 'OpenAi') {
         if (!AiProviderFactory.singleton) {
             switch (provider) {
                 case 'OpenAi':

package/dist/index.js

@@ -4,11 +4,16 @@ import AILogModel from "./models/AILogModel.js";
 import AILogMongoRepository from "./repository/mongo/AILogMongoRepository.js";
 import AILogSqliteRepository from "./repository/sqlite/AILogSqliteRepository.js";
 import { OpenAiProviderFactory } from "./factory/OpenAiProviderFactory.js";
+import { AiProviderFactory } from "./factory/AiProviderFactory.js";
 import AILogServiceFactory from "./factory/services/AILogServiceFactory.js";
 import { OpenAiProvider } from "./providers/OpenAiProvider.js";
 import { KnowledgeService } from "./services/KnowledgeService.js";
 import { AILogService } from "./services/AILogService.js";
 import AILogPermissions from "./permissions/AILogPermissions.js";
+import AIPermissions from "./permissions/AIPermissions.js";
 import AILogController from "./controllers/AILogController.js";
+import AICrudController from "./controllers/AICrudController.js";
+import AIGenericController from "./controllers/AIGenericController.js";
 import AILogRoutes from "./routes/AILogRoutes.js";
-export { OpenAiConfig, AILogSchema, AILogBaseSchema, AILogModel, AILogMongoRepository, AILogSqliteRepository, OpenAiProviderFactory, AILogServiceFactory, OpenAiProvider, KnowledgeService, AILogService, AILogPermissions, AILogController, AILogRoutes, };
+import AIRoutes from "./routes/AIRoutes.js";
+export { OpenAiConfig, AILogSchema, AILogBaseSchema, AILogModel, AILogMongoRepository, AILogSqliteRepository, OpenAiProviderFactory, AiProviderFactory, AILogServiceFactory, OpenAiProvider, KnowledgeService, AILogService, AILogPermissions, AIPermissions, AILogController, AICrudController, AIGenericController, AILogRoutes, AIRoutes };

package/dist/permissions/AIPermissions.js

@@ -0,0 +1,7 @@
+var AIPermissions;
+(function (AIPermissions) {
+    AIPermissions["Prompt"] = "ai:prompt";
+    AIPermissions["PromptCrud"] = "ai:promptCrud";
+})(AIPermissions || (AIPermissions = {}));
+export { AIPermissions };
+export default AIPermissions;

package/dist/routes/AIRoutes.js

@@ -0,0 +1,10 @@
+import AICrudController from "../controllers/AICrudController.js";
+import AIGenericController from "../controllers/AIGenericController.js";
+async function AiFastifyRoutes(fastify, options) {
+    const crudController = new AICrudController();
+    const genericController = new AIGenericController();
+    fastify.post('/api/ai/prompt/crud', (req, rep) => crudController.prompt(req, rep));
+    fastify.post('/api/ai/prompt/generic', (req, rep) => genericController.prompt(req, rep));
+}
+export default AiFastifyRoutes;
+export { AiFastifyRoutes };

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "3.16.0",
+  "version": "3.17.1",
   "description": "Ai utils",
   "main": "dist/index.js",
   "types": "types/index.d.ts",
@@ -18,8 +18,8 @@
   "author": "Cristian Incarnato & Drax Team",
   "license": "ISC",
   "dependencies": {
-    "@drax/ai-share": "^3.16.0",
-    "@drax/crud-back": "^3.15.0",
+    "@drax/ai-share": "^3.17.0",
+    "@drax/crud-back": "^3.17.0",
     "mongoose": "^8.23.0",
     "mongoose-paginate-v2": "^1.8.3"
   },
@@ -44,5 +44,5 @@
     "typescript": "^5.9.3",
     "vitest": "^3.0.8"
   },
-  "gitHead": "5254c8b1b2b59de46654f135122f38bd29ae5452"
+  "gitHead": "2fc35edc0338b4bc5437346faa6787a24abcf0e5"
 }

package/src/controllers/AICrudController.ts

@@ -0,0 +1,168 @@
+import {z} from "zod";
+import {CommonController} from "@drax/common-back";
+import AiProviderFactory from "../factory/AiProviderFactory.js";
+import AIPermissions from "../permissions/AIPermissions.js";
+
+const CrudAiFieldSchema: z.ZodType<any> = z.lazy(() => z.object({
+    name: z.string(),
+    type: z.string(),
+    label: z.string(),
+    hint: z.string().nullable().default(null),
+    placeholder: z.string().nullable().default(null),
+    readonly: z.boolean().nullable().default(null),
+    default: z.any().nullable().default(null),
+    enum: z.array(z.string()).nullable().default(null),
+    items: z.array(z.object({
+        title: z.string().nullable().default(null),
+        value: z.any().nullable().default(null),
+    })).nullable().default(null),
+    ref: z.string().nullable().default(null),
+    refDisplay: z.string().nullable().default(null),
+    objectFields: z.array(CrudAiFieldSchema).nullable().default(null),
+}))
+
+const PromptRequestSchema = z.object({
+    prompt: z.string().min(1),
+    operation: z.enum(["create", "edit"]).default("create"),
+    entity: z.object({
+        name: z.string(),
+        identifier: z.string().optional().nullable(),
+    }),
+    currentValues: z.record(z.string(), z.any()).default({}),
+    fields: z.array(CrudAiFieldSchema).min(1),
+})
+
+class AICrudController extends CommonController {
+
+    protected buildFieldValueSchema(field: any): z.ZodTypeAny {
+        switch (field.type) {
+            case "number":
+                return z.number().nullable().default(null)
+            case "boolean":
+                return z.boolean().nullable().default(null)
+            case "array.string":
+            case "array.enum":
+            case "array.ref":
+                return z.array(z.string()).nullable().default(null)
+            case "array.number":
+                return z.array(z.number()).nullable().default(null)
+            case "object":
+                if (field.objectFields && Array.isArray(field.objectFields) && field.objectFields.length > 0) {
+                    return z.object(this.buildFieldShape(field.objectFields)).nullable().default(null)
+                }
+                return z.string().nullable().default(null)
+            case "record":
+            case "array.object":
+            case "array.record":
+            case "array.fullFile":
+            case "file":
+            case "fullFile":
+                return z.string().nullable().default(null)
+            case "id":
+            case "string":
+            case "longString":
+            case "date":
+            case "ref":
+            case "enum":
+            case "select":
+            case "password":
+            default:
+                return z.string().nullable().default(null)
+        }
+    }
+
+    protected buildFieldShape(fields: any[]): Record<string, z.ZodTypeAny> {
+        return fields.reduce((acc, field) => {
+            acc[field.name] = this.buildFieldValueSchema(field)
+            return acc
+        }, {} as Record<string, z.ZodTypeAny>)
+    }
+
+    protected buildSystemPrompt(input: z.infer<typeof PromptRequestSchema>) {
+        return [
+            "Sos un asistente de formularios para un sistema CRUD.",
+            "Tu tarea es proponer valores JSON para completar o editar una entidad.",
+            "Debes respetar exactamente los nombres de campo entregados.",
+            "No inventes campos adicionales.",
+            "Si un campo no tiene una propuesta razonable, omitilo.",
+            "Usa tipos compatibles con JSON.",
+            "Para campos enum o select, elegí solamente valores válidos de la lista provista.",
+            "Para campos record, object sin estructura fija, array.object o array.record, devolve un string JSON serializado valido.",
+            "La respuesta debe describir sugerencias concretas, breves y aplicables.",
+            `Operacion actual: ${input.operation}.`,
+            `Entidad actual: ${input.entity.name}.`,
+        ].join("\n")
+    }
+
+    protected buildUserInput(input: z.infer<typeof PromptRequestSchema>) {
+        return JSON.stringify({
+            task: input.prompt,
+            operation: input.operation,
+            entity: input.entity,
+            currentValues: input.currentValues,
+            fields: input.fields,
+            expectedResponse: {
+                message: "string",
+                suggestions: "object with proposed values indexed by field name",
+                warnings: ["optional string warnings"],
+            }
+        })
+    }
+
+    async prompt(req, rep) {
+        try {
+            req.rbac.assertPermission(AIPermissions.PromptCrud)
+
+            const input = PromptRequestSchema.parse(req.body ?? {})
+            const aiProvider = AiProviderFactory.instance()
+
+            const responseSchema = z.object({
+                message: z.string().nullable().default(null),
+                suggestions: z.object(this.buildFieldShape(input.fields)).strict().default({}),
+                warnings: z.array(z.string()).default([]),
+            })
+
+            const response = await aiProvider.prompt({
+                systemPrompt: this.buildSystemPrompt(input),
+                userInput: this.buildUserInput(input),
+                zodSchema: responseSchema,
+                operationTitle: `crud-${input.operation}-assistant`,
+                operationGroup: "crud-form-assistant",
+                ip: req.ip,
+                userAgent: req.headers["user-agent"],
+                tenant: req.rbac?.tenantId ?? null,
+                user: req.rbac?.userId ?? null,
+            })
+
+            const parsedOutput = responseSchema.parse(
+                typeof response.output === "string"
+                    ? JSON.parse(response.output)
+                    : response.output
+            )
+
+            return rep.send({
+                message: parsedOutput.message,
+                suggestions: parsedOutput.suggestions,
+                warnings: parsedOutput.warnings,
+                meta: {
+                    tokens: response.tokens,
+                    inputTokens: response.inputTokens,
+                    outputTokens: response.outputTokens,
+                    time: response.time,
+                }
+            })
+        } catch (e: any) {
+            console.error("AIController.prompt error", e)
+
+            const statusCode = e?.name === "ZodError" ? 400 : 500
+
+            return rep.status(statusCode).send({
+                message: e?.message || "AI prompt error",
+            })
+        }
+    }
+
+}
+
+export default AICrudController;
+export {AICrudController};