@infuro/cms-core 1.0.7 → 1.0.8

package/dist/api.cjs

@@ -1271,6 +1271,129 @@ function createSettingsApiHandlers(config) {
     }
   };
 }
+var KB_CHUNK_LIMIT = 10;
+var KB_CONTEXT_MAX_CHARS = 4e3;
+function getQueryTerms(message) {
+  return message.replace(/[^\w\s]/g, " ").split(/\s+/).filter((w) => w.length > 2).slice(0, 6);
+}
+function createChatHandlers(config) {
+  const { dataSource, entityMap, json, getCms } = config;
+  const contactRepo = () => dataSource.getRepository(entityMap.contacts);
+  const convRepo = () => dataSource.getRepository(entityMap.chat_conversations);
+  const msgRepo = () => dataSource.getRepository(entityMap.chat_messages);
+  const chunkRepo = () => dataSource.getRepository(entityMap.knowledge_base_chunks);
+  return {
+    async identify(req) {
+      try {
+        const body = await req.json();
+        const name = body?.name?.trim();
+        const email = body?.email?.trim();
+        if (!name || !email) return json({ error: "name and email required" }, { status: 400 });
+        const repo = contactRepo();
+        let contact = await repo.findOne({ where: { email, deleted: false } });
+        if (!contact) {
+          const created = repo.create({ name, email, phone: body.phone?.trim() || null });
+          contact = await repo.save(created);
+        }
+        const convRepoInst = convRepo();
+        const conv = await convRepoInst.save(convRepoInst.create({ contactId: contact.id }));
+        return json({
+          contactId: contact.id,
+          conversationId: conv.id
+        });
+      } catch (err) {
+        const message = err instanceof Error ? err.message : "Failed to identify";
+        return json({ error: "Failed to identify", detail: message }, { status: 500 });
+      }
+    },
+    async getMessages(req, conversationId) {
+      try {
+        const conv = await convRepo().findOne({
+          where: { id: parseInt(conversationId, 10) },
+          relations: ["messages"]
+        });
+        if (!conv) return json({ error: "Conversation not found" }, { status: 404 });
+        const messages = (conv.messages ?? []).sort((a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime()).map((m) => ({ role: m.role, content: m.content }));
+        return json({ messages });
+      } catch {
+        return json({ error: "Failed to fetch messages" }, { status: 500 });
+      }
+    },
+    async postMessage(req) {
+      try {
+        const body = await req.json();
+        const conversationId = body?.conversationId;
+        const message = body?.message?.trim();
+        if (!conversationId || !message) return json({ error: "conversationId and message required" }, { status: 400 });
+        const conv = await convRepo().findOne({
+          where: { id: conversationId },
+          relations: ["messages"]
+        });
+        if (!conv) return json({ error: "Conversation not found" }, { status: 404 });
+        const msgRepoInst = msgRepo();
+        await msgRepoInst.save(msgRepoInst.create({ conversationId, role: "user", content: message }));
+        const cms = await getCms();
+        const llm = cms.getPlugin("llm");
+        if (!llm?.chat) return json({ error: "LLM not configured" }, { status: 503 });
+        let contextParts = [];
+        const queryEmbedding = llm.embed ? await llm.embed(message) : null;
+        if (queryEmbedding && queryEmbedding.length > 0) {
+          const vectorStr = "[" + queryEmbedding.join(",") + "]";
+          try {
+            const rows = await dataSource.query(
+              `SELECT id, content FROM knowledge_base_chunks WHERE embedding IS NOT NULL ORDER BY embedding <=> $1::vector LIMIT $2`,
+              [vectorStr, KB_CHUNK_LIMIT]
+            );
+            let totalLen = 0;
+            for (const r of rows) {
+              const text = (r.content ?? "").trim();
+              if (!text || totalLen + text.length > KB_CONTEXT_MAX_CHARS) continue;
+              contextParts.push(text);
+              totalLen += text.length;
+            }
+          } catch {
+          }
+        }
+        if (contextParts.length === 0) {
+          const terms = getQueryTerms(message);
+          if (terms.length > 0) {
+            const conditions = terms.map((t) => ({ content: (0, import_typeorm2.ILike)(`%${t}%`) }));
+            const chunks = await chunkRepo().find({
+              where: conditions,
+              take: KB_CHUNK_LIMIT,
+              order: { id: "ASC" }
+            });
+            const seen = /* @__PURE__ */ new Set();
+            let totalLen = 0;
+            for (const c of chunks) {
+              const text = c.content.trim();
+              if (seen.has(text) || totalLen + text.length > KB_CONTEXT_MAX_CHARS) continue;
+              seen.add(text);
+              contextParts.push(text);
+              totalLen += text.length;
+            }
+          }
+        }
+        const history = (conv.messages ?? []).sort((a, b) => new Date(a.createdAt ?? 0).getTime() - new Date(b.createdAt ?? 0).getTime()).map((m) => ({ role: m.role, content: m.content }));
+        const systemContent = contextParts.length > 0 ? `Use the following context about the company and its products to answer. If the answer is not in the context, say so.
+
+Context:
+${contextParts.join("\n\n")}` : "You are a helpful assistant for the company. If you do not have specific information, say so.";
+        const messages = [
+          { role: "system", content: systemContent },
+          ...history,
+          { role: "user", content: message }
+        ];
+        const { content } = await llm.chat(messages);
+        await msgRepoInst.save(msgRepoInst.create({ conversationId, role: "assistant", content }));
+        return json({ content });
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : "";
+        return json({ error: msg || "Failed to send message" }, { status: 500 });
+      }
+    }
+  };
+}
 
 // src/api/cms-api-handler.ts
 var DEFAULT_EXCLUDE = /* @__PURE__ */ new Set(["users", "password_reset_tokens", "user_groups", "permissions", "comments", "form_fields", "configs"]);
@@ -1293,7 +1416,8 @@ function createCmsApiHandler(config) {
     usersApi,
     userAvatar,
     userProfile,
-    settings: settingsConfig
+    settings: settingsConfig,
+    chat: chatConfig
   } = config;
   const analytics = analyticsConfig ?? (getCms ? {
     json: config.json,
@@ -1338,6 +1462,7 @@ function createCmsApiHandler(config) {
   const avatarPost = userAvatar ? createUserAvatarHandler(userAvatar) : null;
   const profilePut = userProfile ? createUserProfileHandler(userProfile) : null;
   const settingsHandlers = settingsConfig ? createSettingsApiHandlers(settingsConfig) : null;
+  const chatHandlers = chatConfig ? createChatHandlers(chatConfig) : null;
   function resolveResource(segment) {
     const model = pathToModel(segment);
     return crudResources.includes(model) ? model : segment;
@@ -1397,6 +1522,11 @@ function createCmsApiHandler(config) {
         if (method === "GET") return settingsHandlers.GET(req, path[1]);
         if (method === "PUT") return settingsHandlers.PUT(req, path[1]);
       }
+      if (path[0] === "chat" && chatHandlers) {
+        if (path.length === 2 && path[1] === "identify" && method === "POST") return chatHandlers.identify(req);
+        if (path.length === 4 && path[1] === "conversations" && path[3] === "messages" && method === "GET") return chatHandlers.getMessages(req, path[2]);
+        if (path.length === 2 && path[1] === "messages" && method === "POST") return chatHandlers.postMessage(req);
+      }
       if (path.length === 0) return config.json({ error: "Not found" }, { status: 404 });
       const resource = resolveResource(path[0]);
       if (!crudResources.includes(resource)) return config.json({ error: "Invalid resource" }, { status: 400 });