From 4bfe57af40d98331721124837950c4329ad12baf Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Sep 2025 15:07:03 +0800
Subject: [PATCH] feat(ai): 增加公司 ID 筛选功能

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java |    4 ++++
 1 files changed, 4 insertions(+), 0 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
index 79ae1f2..0f361cb 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
@@ -55,7 +55,9 @@
         entity.setCompanyId(dto.getCompanyId());
         entity.setName(dto.getName());
         entity.setDescription(dto.getDescription());
+        entity.setPromptHead(dto.getPromptHead());
         entity.setPromptTemplate(dto.getPromptTemplate());
+        entity.setJsonTemplate(dto.getJsonTemplate());
         entity.setIconImg(dto.getIconImg());
         entity.setModelId(dto.getModelId());
         entity.setLinkId(dto.getLinkId());
@@ -74,7 +76,9 @@
                             .set(AiProductRole::getCompanyId, dto.getCompanyId())
                             .set(AiProductRole::getName, dto.getName())
                             .set(AiProductRole::getDescription, dto.getDescription())
+                            .set(AiProductRole::getPromptHead, dto.getPromptHead())
                             .set(AiProductRole::getPromptTemplate, dto.getPromptTemplate())
+                            .set(AiProductRole::getJsonTemplate, dto.getJsonTemplate())
                             .set(AiProductRole::getIconImg, dto.getIconImg())
                             .set(AiProductRole::getModelId, dto.getModelId())
                             .set(AiProductRole::getLinkId, dto.getLinkId())

--
Gitblit v1.9.1