LangChainMilvusService向量数据库操作

llm-back/cmc-agent/pom.xml

@@ -0,0 +1,108 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <artifactId>ruoyi</artifactId>
+        <groupId>com.ruoyi</groupId>
+        <version>3.8.9</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+    <packaging>jar</packaging>
+    <artifactId>cmc-agent</artifactId>
+
+    <description>
+        cmc智能体
+    </description>
+
+    <dependencies>
+
+        <dependency>
+            <groupId>com.ruoyi</groupId>
+            <artifactId>ruoyi-framework</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.ruoyi</groupId>
+            <artifactId>ruoyi-system</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.ruoyi</groupId>
+            <artifactId>ruoyi-common</artifactId>
+        </dependency>
+
+        <!-- spring-boot-devtools -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-devtools</artifactId>
+            <optional>true</optional> <!-- 表示依赖不会传递 -->
+        </dependency>
+
+        <!-- 向量数据库-->
+        <dependency>
+            <groupId>io.milvus</groupId>
+            <artifactId>milvus-sdk-java</artifactId>
+            <version>2.3.3</version>
+        </dependency>
+
+        <!-- LangChain4j -->
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j</artifactId>
+            <version>0.35.0</version>
+        </dependency>
+
+        <!-- LangChain4j Milvus 集成 -->
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-milvus</artifactId>
+            <version>0.35.0</version> <!-- 版本需与核心一致 -->
+        </dependency>
+
+        <!-- LangChain4j embedding 集成 -->
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-embeddings-bge-small-zh-v15</artifactId>
+            <version>0.35.0</version>
+        </dependency>
+
+        <!-- 解析PDF -->
+        <dependency>
+            <groupId>org.apache.pdfbox</groupId>
+            <artifactId>pdfbox</artifactId>
+            <version>3.0.3</version>
+        </dependency>
+
+
+    </dependencies>
+
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.springframework.boot</groupId>
+                <artifactId>spring-boot-maven-plugin</artifactId>
+                <version>2.5.15</version>
+                <configuration>
+                    <fork>true</fork> <!-- 如果没有该配置，devtools不会生效 -->
+                </configuration>
+                <executions>
+                    <execution>
+                        <goals>
+                            <goal>repackage</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
+            <plugin>   
+                <groupId>org.apache.maven.plugins</groupId>   
+                <artifactId>maven-war-plugin</artifactId>   
+                <version>3.1.0</version>   
+                <configuration>
+                    <failOnMissingWebXml>false</failOnMissingWebXml>
+                    <warName>${project.artifactId}</warName>
+                </configuration>   
+           </plugin>   
+        </plugins>
+        <finalName>${project.artifactId}</finalName>
+    </build>
+
+</project>

llm-back/cmc-agent/src/main/java/com/cmc/agent/controller/KnowLedgeController.java

@@ -0,0 +1,63 @@
+package com.cmc.agent.controller;
+
+import com.ruoyi.common.config.RuoYiConfig;
+import com.ruoyi.common.core.controller.BaseController;
+import com.ruoyi.common.core.domain.AjaxResult;
+import com.cmc.agent.service.LangChainMilvusService;
+import dev.langchain4j.model.embedding.onnx.bgesmallzhv15.BgeSmallZhV15EmbeddingModel;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.File;
+import java.io.IOException;
+
+/**
+ * cmc知识库Controller
+ * 
+ * @author cmc
+ * @date 2025-04-08
+ */
+@RestController
+@RequestMapping("/llm/knowledge")
+public class KnowLedgeController extends BaseController
+{
+    private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
+
+    /**
+     * 新建知识库
+     */
+    @PostMapping("/create")
+    public AjaxResult createKnowLedgeBase(String collectionName)
+    {
+        LangChainMilvusService langChainMilvusService = new LangChainMilvusService(
+                "localhost",
+                19530,
+                collectionName,
+                embeddingModel);
+        langChainMilvusService.createCollection(512);
+        return success();
+    }
+
+    /**
+     * 插入知识库文件
+     */
+    @PostMapping("/insert")
+    public AjaxResult insertKnowledge(MultipartFile file, String collectionName) throws IOException {
+        LangChainMilvusService langChainMilvusService = new LangChainMilvusService(
+                "localhost",
+                19530,
+                collectionName,
+                embeddingModel);
+
+        File profilePath = new File( RuoYiConfig.getProfile() + "/upload/knowledge");
+        if (!profilePath.exists())
+            profilePath.mkdirs();
+        File transferFile = new File( profilePath + "/" + file.getOriginalFilename());
+        if (!transferFile.exists())
+            file.transferTo(transferFile);
+        langChainMilvusService.insertLangchainEmbeddingDocument(transferFile);
+        return success();
+    }
+
+}

llm-back/cmc-agent/src/main/java/com/cmc/agent/controller/RagController.java

@@ -0,0 +1,45 @@
+package com.cmc.agent.controller;
+
+import com.ruoyi.common.core.controller.BaseController;
+import com.ruoyi.common.core.domain.AjaxResult;
+import com.cmc.agent.service.LangChainMilvusService;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import dev.langchain4j.model.embedding.onnx.bgesmallzhv15.BgeSmallZhV15EmbeddingModel;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.io.IOException;
+import java.util.List;
+
+/**
+ * cmc知识库Controller
+ * 
+ * @author cmc
+ * @date 2025-04-08
+ */
+@RestController
+@RequestMapping("/llm/rag")
+public class RagController extends BaseController
+{
+    private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
+
+    /**
+     * 增强检索生成回答
+     */
+    @PostMapping("/answer")
+    public AjaxResult answerWithRAG(String question, String collectionName) throws IOException {
+        LangChainMilvusService langChainMilvusService = new LangChainMilvusService(
+                "localhost",
+                19530,
+                collectionName,
+                embeddingModel);
+
+        // 2. Milvus检索
+        List<String> contexts = langChainMilvusService.retrieveFromMilvus(question, 3);
+
+        // 3. 调用本地LLM或HTTP服务
+        return success(langChainMilvusService.generateAnswer(question, contexts));
+    }
+
+}

llm-back/cmc-agent/src/main/java/com/cmc/agent/service/LangChainMilvusService.java

@@ -0,0 +1,194 @@
+package com.cmc.agent.service;
+
+import com.google.gson.Gson;
+import dev.langchain4j.data.document.Document;
+import dev.langchain4j.data.document.parser.apache.pdfbox.ApachePdfBoxDocumentParser;
+import dev.langchain4j.data.document.splitter.DocumentByParagraphSplitter;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+
+import io.milvus.client.MilvusServiceClient;
+import io.milvus.grpc.DataType;
+import io.milvus.grpc.MutationResult;
+import io.milvus.grpc.SearchResults;
+import io.milvus.param.ConnectParam;
+import io.milvus.param.IndexType;
+import io.milvus.param.MetricType;
+import io.milvus.param.R;
+import io.milvus.param.collection.CreateCollectionParam;
+import io.milvus.param.collection.FieldType;
+import io.milvus.param.dml.InsertParam;
+import io.milvus.param.dml.SearchParam;
+import io.milvus.param.index.CreateIndexParam;
+import io.milvus.response.SearchResultsWrapper;
+import okhttp3.*;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.*;
+import java.util.stream.Collectors;
+
+public class LangChainMilvusService {
+    private static final String LLM_SERVICE_URL = "http://localhost:8000/generate";
+    private final MilvusServiceClient milvusClient;
+    private final String collectionName;
+    private  EmbeddingModel embeddingModel;
+
+    /**
+     * 连接milvus知识库，加入langchain自带emdding
+     */
+    public LangChainMilvusService(String host, int port, String collectionName, EmbeddingModel embeddingModel) {
+        this.milvusClient = new MilvusServiceClient(
+                ConnectParam.newBuilder()
+                        .withHost(host)
+                        .withPort(port)
+                        .build()
+        );
+        this.collectionName = collectionName;
+        this.embeddingModel = embeddingModel;
+    }
+
+    /**
+     * 新建知识库Collection（含Schema、Field、Index）
+     */
+    public void createCollection(int dimension) {
+        FieldType idField = FieldType.newBuilder()
+                .withName("id")
+                .withDataType(DataType.Int64)
+                .withPrimaryKey(true)
+                .withAutoID(true)
+                .build();
+
+        FieldType fileNameField = FieldType.newBuilder()
+                .withName("file_name")
+                .withDataType(DataType.VarChar)
+                .withMaxLength(256)
+                .build();
+
+        FieldType contentField = FieldType.newBuilder()
+                .withName("content")
+                .withDataType(DataType.VarChar)
+                .withMaxLength(65535)
+                .build();
+
+        FieldType vectorField = FieldType.newBuilder()
+                .withName("embedding")
+                .withDataType(DataType.Float16Vector)
+                .withDimension(dimension)
+                .build();
+
+        CreateCollectionParam createCollectionParam = CreateCollectionParam.newBuilder()
+                .withCollectionName(collectionName)
+                .addFieldType(idField)
+                .addFieldType(fileNameField)
+                .addFieldType(contentField)
+                .addFieldType(vectorField)
+                .build();
+
+        milvusClient.createCollection(createCollectionParam);
+
+        // 创建索引
+        CreateIndexParam createIndexParam = CreateIndexParam.newBuilder()
+                .withCollectionName(collectionName)
+                .withFieldName("embedding")
+                .withIndexType(IndexType.IVF_FLAT)
+                .withMetricType(MetricType.COSINE)
+                .withExtraParam("{\"nlist\": 64}")
+                .build();
+
+        milvusClient.createIndex(createIndexParam);
+    }
+
+    public void insertLangchainEmbeddingDocument(File file) throws IOException {
+        // 加载文档
+        InputStream fileInputStream = new FileInputStream(file);
+        Document document = new ApachePdfBoxDocumentParser().parse(fileInputStream);
+        DocumentByParagraphSplitter splitter = new DocumentByParagraphSplitter(150,10);;
+        List<TextSegment> segments = splitter.split(document);
+
+        // 提取文本和生成嵌入
+        List<String> texts = new ArrayList<>();
+        List<List<Float>> embeddings = new ArrayList<>();
+
+        for (TextSegment segment : segments) {
+            String text = segment.text();
+            if (text.trim().isEmpty())
+                continue;
+            texts.add(text);
+            embeddings.add(embeddingModel.embed(text).content().vectorAsList());
+        }
+
+        // 准备插入数据
+        List<InsertParam.Field> fields = new ArrayList<>();
+        fields.add(new InsertParam.Field("file_name", Arrays.asList(file.getName())));
+        fields.add(new InsertParam.Field("content", Arrays.asList(texts)));
+        fields.add(new InsertParam.Field("embedding", Arrays.asList(embeddings)));
+
+        InsertParam insertParam = InsertParam.newBuilder()
+                .withCollectionName(collectionName)
+                .withFields(fields)
+                .build();
+
+        // 执行插入
+        R<MutationResult> insertResult = milvusClient.insert(insertParam);
+
+        if (insertResult.getStatus() != R.Status.Success.getCode()) {
+            throw new RuntimeException("Failed to insert document: " + insertResult.getMessage());
+        }
+    }
+
+    // 从Milvus检索相关文档
+    public List<String> retrieveFromMilvus(String query, int topK) throws IOException {
+        List<Float> queryVector = embeddingModel.embed(query).content().vectorAsList();
+
+        SearchParam searchParam = SearchParam.newBuilder()
+                .withCollectionName(collectionName)
+                .withVectors(queryVector)
+                .withTopK(topK)
+                .withOutFields(Arrays.asList("content"))
+                .build();
+
+        R<SearchResults> response = milvusClient.search(searchParam);
+        SearchResultsWrapper wrapper = new SearchResultsWrapper(response.getData().getResults());
+
+        return wrapper.getRowRecords(0).stream()
+                .map(record -> (String) record.get("content"))
+                .collect(Collectors.toList());
+    }
+
+    // 调用LLM生成回答
+    public String generateAnswer(String question, List<String> contexts) throws IOException {
+        String prompt = buildPrompt(question, contexts);
+        Gson gson = new Gson();
+        Map<String, String> hashMap = new HashMap<>();
+        hashMap.put("prompt", prompt);
+        hashMap.put("max_tokens", "512");
+        RequestBody body = RequestBody.create(
+                MediaType.parse("application/json"),
+                new Gson().toJson(hashMap));
+
+        Request request = new Request.Builder()
+                .url(LLM_SERVICE_URL)
+                .post(body)
+                .build();
+
+        try (Response response = new OkHttpClient().newCall(request).execute()) {
+            return gson.fromJson(response.body().string(), Map.class).get("text").toString();
+        }
+    }
+
+    private String buildPrompt(String question, List<String> contexts) {
+        StringBuilder sb = new StringBuilder();
+        sb.append("根据以下上下文回答问题：\n\n");
+        for (int i = 0; i < contexts.size(); i++) {
+            sb.append("上下文").append(i+1).append(": ").append(contexts.get(i)).append("\n\n");
+        }
+        sb.append("问题: ").append(question).append("\n回答: ");
+        return sb.toString();
+    }
+    public void close() {
+        milvusClient.close();
+    }
+}

llm-back/pom.xml

@@ -218,10 +218,18 @@
                 <version>${ruoyi.version}</version>
             </dependency>
 
+            <!-- 智能体-->
+            <dependency>
+                <groupId>com.cmc</groupId>
+                <artifactId>cmc-agent</artifactId>
+                <version>${ruoyi.version}</version>
+            </dependency>
+
         </dependencies>
     </dependencyManagement>
 
     <modules>
+        <module>cmc-agent</module>
         <module>ruoyi-admin</module>
         <module>ruoyi-framework</module>
         <module>ruoyi-system</module>

llm-back/ruoyi-admin/src/main/java/com/ruoyi/RuoYiApplication.java

@@ -16,15 +16,6 @@ public class RuoYiApplication
     {
         // System.setProperty("spring.devtools.restart.enabled", "false");
         SpringApplication.run(RuoYiApplication.class, args);
-        System.out.println("(♥◠‿◠)ﾉﾞ  若依启动成功   ლ(´ڡ`ლ)ﾞ  \n" +
-                " .-------.       ____     __        \n" +
-                " |  _ _   \\      \\   \\   /  /    \n" +
-                " | ( ' )  |       \\  _. /  '       \n" +
-                " |(_ o _) /        _( )_ .'         \n" +
-                " | (_,_).' __  ___(_ o _)'          \n" +
-                " |  |\\ \\  |  ||   |(_,_)'         \n" +
-                " |  | \\ `'   /|   `-'  /           \n" +
-                " |  |  \\    /  \\      /           \n" +
-                " ''-'   `'-'    `-..-'              ");
+        System.out.println("办公助手启动成功 \n");
     }
 }

llm-back/ruoyi-admin/src/main/java/com/ruoyi/web/controller/llm/CmcChatController.java

@@ -2,7 +2,8 @@ package com.ruoyi.web.controller.llm;
 
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
-import org.springframework.security.access.prepost.PreAuthorize;
+
+import com.ruoyi.common.utils.SnowFlake;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -37,7 +38,6 @@ public class CmcChatController extends BaseController
     /**
      * 查询cmc聊天记录列表
      */
-    @PreAuthorize("@ss.hasPermi('llm:chat:list')")
     @GetMapping("/list")
     public TableDataInfo list(CmcChat cmcChat)
     {
@@ -49,7 +49,6 @@ public class CmcChatController extends BaseController
     /**
      * 导出cmc聊天记录列表
      */
-    @PreAuthorize("@ss.hasPermi('llm:chat:export')")
     @Log(title = "cmc聊天记录", businessType = BusinessType.EXPORT)
     @PostMapping("/export")
     public void export(HttpServletResponse response, CmcChat cmcChat)
@@ -62,7 +61,6 @@ public class CmcChatController extends BaseController
     /**
      * 获取cmc聊天记录详细信息
      */
-    @PreAuthorize("@ss.hasPermi('llm:chat:query')")
     @GetMapping(value = "/{chatId}")
     public AjaxResult getInfo(@PathVariable("chatId") String chatId)
     {
@@ -72,18 +70,17 @@ public class CmcChatController extends BaseController
     /**
      * 新增cmc聊天记录
      */
-    @PreAuthorize("@ss.hasPermi('llm:chat:add')")
     @Log(title = "cmc聊天记录", businessType = BusinessType.INSERT)
     @PostMapping
     public AjaxResult add(@RequestBody CmcChat cmcChat)
     {
+        cmcChat.setChatId(new SnowFlake().generateId());
         return toAjax(cmcChatService.insertCmcChat(cmcChat));
     }
 
     /**
      * 修改cmc聊天记录
      */
-    @PreAuthorize("@ss.hasPermi('llm:chat:edit')")
     @Log(title = "cmc聊天记录", businessType = BusinessType.UPDATE)
     @PutMapping
     public AjaxResult edit(@RequestBody CmcChat cmcChat)
@@ -94,7 +91,6 @@ public class CmcChatController extends BaseController
     /**
      * 删除cmc聊天记录
      */
-    @PreAuthorize("@ss.hasPermi('llm:chat:remove')")
     @Log(title = "cmc聊天记录", businessType = BusinessType.DELETE)
 	@DeleteMapping("/{chatIds}")
     public AjaxResult remove(@PathVariable String[] chatIds)

llm-back/ruoyi-admin/src/main/java/com/ruoyi/web/controller/llm/CmcDocumentController.java

@@ -2,7 +2,8 @@ package com.ruoyi.web.controller.llm;
 
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
-import org.springframework.security.access.prepost.PreAuthorize;
+
+import com.ruoyi.common.utils.SnowFlake;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -37,7 +38,6 @@ public class CmcDocumentController extends BaseController
     /**
      * 查询cmc聊天附件列表
      */
-    @PreAuthorize("@ss.hasPermi('llm:document:list')")
     @GetMapping("/list")
     public TableDataInfo list(CmcDocument cmcDocument)
     {
@@ -49,7 +49,6 @@ public class CmcDocumentController extends BaseController
     /**
      * 导出cmc聊天附件列表
      */
-    @PreAuthorize("@ss.hasPermi('llm:document:export')")
     @Log(title = "cmc聊天附件", businessType = BusinessType.EXPORT)
     @PostMapping("/export")
     public void export(HttpServletResponse response, CmcDocument cmcDocument)
@@ -62,7 +61,6 @@ public class CmcDocumentController extends BaseController
     /**
      * 获取cmc聊天附件详细信息
      */
-    @PreAuthorize("@ss.hasPermi('llm:document:query')")
     @GetMapping(value = "/{documentId}")
     public AjaxResult getInfo(@PathVariable("documentId") String documentId)
     {
@@ -72,18 +70,17 @@ public class CmcDocumentController extends BaseController
     /**
      * 新增cmc聊天附件
      */
-    @PreAuthorize("@ss.hasPermi('llm:document:add')")
     @Log(title = "cmc聊天附件", businessType = BusinessType.INSERT)
     @PostMapping
     public AjaxResult add(@RequestBody CmcDocument cmcDocument)
     {
+        cmcDocument.setDocumentId(new SnowFlake().generateId());
         return toAjax(cmcDocumentService.insertCmcDocument(cmcDocument));
     }
 
     /**
      * 修改cmc聊天附件
      */
-    @PreAuthorize("@ss.hasPermi('llm:document:edit')")
     @Log(title = "cmc聊天附件", businessType = BusinessType.UPDATE)
     @PutMapping
     public AjaxResult edit(@RequestBody CmcDocument cmcDocument)
@@ -94,7 +91,6 @@ public class CmcDocumentController extends BaseController
     /**
      * 删除cmc聊天附件
      */
-    @PreAuthorize("@ss.hasPermi('llm:document:remove')")
     @Log(title = "cmc聊天附件", businessType = BusinessType.DELETE)
 	@DeleteMapping("/{documentIds}")
     public AjaxResult remove(@PathVariable String[] documentIds)

llm-back/ruoyi-admin/src/main/java/com/ruoyi/web/controller/llm/CmcTopicController.java

@@ -3,6 +3,7 @@ package com.ruoyi.web.controller.llm;
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
 
+import com.ruoyi.common.utils.SnowFlake;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -73,6 +74,7 @@ public class CmcTopicController extends BaseController
     @PostMapping
     public AjaxResult add(@RequestBody CmcTopic cmcTopic)
     {
+        cmcTopic.setTopicId(new SnowFlake().generateId());
         return toAjax(cmcTopicService.insertCmcTopic(cmcTopic));
     }
 

llm-back/ruoyi-admin/src/main/resources/application.yml

@@ -1,13 +1,13 @@
 # 项目相关配置
-ruoyi:
+cmc:
   # 名称
-  name: RuoYi
+  name: llm
   # 版本
   version: 3.8.9
   # 版权年份
   copyrightYear: 2024
   # 文件路径 示例（ Windows配置D:/ruoyi/uploadPath，Linux配置 /home/ruoyi/uploadPath）
-  profile: D:/ruoyi/uploadPath
+  profile: /home/cmc/projects/cmc-llm
   # 获取ip地址开关
   addressEnabled: false
   # 验证码类型 math 数字计算 char 字符验证
@@ -57,9 +57,9 @@ spring:
   servlet:
     multipart:
       # 单个文件大小
-      max-file-size: 10MB
+      max-file-size: 1024MB
       # 设置总上传的文件大小
-      max-request-size: 20MB
+      max-request-size: 1024MB
   # 服务模块
   devtools:
     restart:
@@ -95,7 +95,7 @@ token:
   # 令牌密钥
   secret: abcdefghijklmnopqrstuvwxyz
   # 令牌有效期（默认30分钟）
-  expireTime: 30
+  expireTime: 480
 
 # MyBatis配置
 mybatis:

llm-back/ruoyi-admin/src/main/resources/banner.txt

@@ -1,24 +1,2 @@
 Application Version: ${ruoyi.version}
-Spring Boot Version: ${spring-boot.version}
-////////////////////////////////////////////////////////////////////
-//                          _ooOoo_                               //
-//                         o8888888o                              //
-//                         88" . "88                              //
-//                         (| ^_^ |)                              //
-//                         O\  =  /O                              //
-//                      ____/`---'\____                           //
-//                    .'  \\|     |//  `.                         //
-//                   /  \\|||  :  |||//  \                        //
-//                  /  _||||| -:- |||||-  \                       //
-//                  |   | \\\  -  /// |   |                       //
-//                  | \_|  ''\---/''  |   |                       //
-//                  \  .-\__  `-`  ___/-. /                       //
-//                ___`. .'  /--.--\  `. . ___                     //
-//              ."" '<  `.___\_<|>_/___.'  >'"".                  //
-//            | | :  `- \`.;`\ _ /`;.`/ - ` : | |                 //
-//            \  \ `-.   \_ __\ /__ _/   .-` /  /                 //
-//      ========`-.____`-.___\_____/___.-`____.-'========         //
-//                           `=---='                              //
-//      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^        //
-//             佛祖保佑       永不宕机      永无BUG               //
-////////////////////////////////////////////////////////////////////
+Spring Boot Version: ${spring-boot.version}

llm-back/ruoyi-admin/src/main/resources/logback.xml

@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <configuration>
     <!-- 日志存放路径 -->
-	<property name="log.path" value="/home/ruoyi/logs" />
+	<property name="log.path" value="/home/cmc/projects/cmc-llm/logs" />
     <!-- 日志输出格式 -->
 	<property name="log.pattern" value="%d{HH:mm:ss.SSS} [%thread] %-5level %logger{20} - [%method,%line] - %msg%n" />
 

llm-back/ruoyi-common/src/main/java/com/ruoyi/common/config/RuoYiConfig.java

@@ -9,7 +9,7 @@ import org.springframework.stereotype.Component;
  * @author ruoyi
  */
 @Component
-@ConfigurationProperties(prefix = "ruoyi")
+@ConfigurationProperties(prefix = "cmc")
 public class RuoYiConfig
 {
     /** 项目名称 */

llm-back/ruoyi-common/src/main/java/com/ruoyi/common/utils/SnowFlake.java

@@ -0,0 +1,159 @@
+package com.ruoyi.common.utils;
+
+import org.springframework.stereotype.Component;
+
+import java.lang.management.ManagementFactory;
+import java.net.InetAddress;
+import java.net.NetworkInterface;
+import java.util.concurrent.ThreadLocalRandom;
+
+/**
+ * 雪花id生成类
+ * @author zyl
+ */
+@Component
+public class SnowFlake {
+    /* 时间起始标记点，作为基准，一般取系统的最近时间（一旦确定不能变动） */
+    private final long twepoch = 1715236540000L;
+    private final long workerIdBits = 5L;/* 机器标识位数 */
+    private final long datacenterIdBits = 5L;
+    private final long maxWorkerId = -1L ^ (-1L << workerIdBits);
+    private final long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
+    private final long sequenceBits = 12L;/* 毫秒内自增位 */
+    private final long workerIdShift = sequenceBits;
+    private final long datacenterIdShift = sequenceBits + workerIdBits;
+    /* 时间戳左移动位 */
+    private final long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
+    private final long sequenceMask = -1L ^ (-1L << sequenceBits);
+
+    private long workerId;
+
+    /* 数据标识id部分 */
+    private long datacenterId;
+    private long sequence = 0L;/* 0，并发控制 */
+    private long lastTimestamp = -1L;/* 上次生产id时间戳 */
+
+    public SnowFlake() {
+        this.datacenterId = getDatacenterId(maxDatacenterId);
+        this.workerId = getMaxWorkerId(datacenterId, maxWorkerId);
+    }
+
+    /**
+     * @param workerId     工作机器ID
+     * @param datacenterId 序列号
+     */
+    public SnowFlake(long workerId, long datacenterId) {
+        if (workerId > maxWorkerId || workerId < 0) {
+            System.out.println(
+                    String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
+        }
+        if (datacenterId > maxDatacenterId || datacenterId < 0) {
+            System.out.println(
+                    String.format("datacenter Id can't be greater than %d or less than 0", maxDatacenterId));
+        }
+        this.workerId = workerId;
+        this.datacenterId = datacenterId;
+    }
+
+    /**
+     * 获取 maxWorkerId
+     */
+    protected static long getMaxWorkerId(long datacenterId, long maxWorkerId) {
+        StringBuilder mpid = new StringBuilder();
+        mpid.append(datacenterId);
+        String name = ManagementFactory.getRuntimeMXBean().getName();
+        if (name != null && "".equals(name)) {
+            /*
+             * GET jvmPid
+             */
+            mpid.append(name.split("@")[0]);
+        }
+        /*
+         * MAC + PID 的 hashcode 获取16个低位
+         */
+        return (mpid.toString().hashCode() & 0xffff) % (maxWorkerId + 1);
+    }
+
+    /**
+     * 数据标识id部分
+     */
+    protected static long getDatacenterId(long maxDatacenterId) {
+        long id = 0L;
+        try {
+            InetAddress ip = InetAddress.getLocalHost();
+            NetworkInterface network = NetworkInterface.getByInetAddress(ip);
+            if (network == null) {
+                id = 1L;
+            } else {
+                byte[] mac = network.getHardwareAddress();
+                if (null != mac) {
+                    id = ((0x000000FF & (long) mac[mac.length - 1]) | (0x0000FF00 & (((long) mac[mac.length - 2]) << 8))) >> 6;
+                    id = id % (maxDatacenterId + 1);
+                }
+            }
+        } catch (Exception e) {
+            System.out.println(" getDatacenterId: " + e.getMessage());
+        }
+        return id;
+    }
+
+    /**
+     * 获取下一个ID
+     *
+     * @return
+     */
+    public synchronized long nextId() {
+        long timestamp = timeGen();
+        if (timestamp < lastTimestamp) {//闰秒
+            long offset = lastTimestamp - timestamp;
+            if (offset <= 5) {
+                try {
+                    wait(offset << 1);
+                    timestamp = timeGen();
+                    if (timestamp < lastTimestamp) {
+                        throw new RuntimeException(String.format("Clock moved backwards.  Refusing to generate id for %d milliseconds", offset));
+                    }
+                } catch (Exception e) {
+                    throw new RuntimeException(e);
+                }
+            } else {
+                throw new RuntimeException(String.format("Clock moved backwards.  Refusing to generate id for %d milliseconds", offset));
+            }
+        }
+
+        if (lastTimestamp == timestamp) {
+            // 相同毫秒内，序列号自增
+            sequence = (sequence + 1) & sequenceMask;
+            if (sequence == 0) {
+                // 同一毫秒的序列数已经达到最大
+                timestamp = tilNextMillis(lastTimestamp);
+            }
+        } else {
+            // 不同毫秒内，序列号置为 1 - 3 随机数
+            sequence = ThreadLocalRandom.current().nextLong(1, 3);
+        }
+
+        lastTimestamp = timestamp;
+
+        return ((timestamp - twepoch) << timestampLeftShift)    // 时间戳部分
+                | (datacenterId << datacenterIdShift)           // 数据中心部分
+                | (workerId << workerIdShift)                   // 机器标识部分
+                | sequence;                                     // 序列号部分
+    }
+
+    protected long tilNextMillis(long lastTimestamp) {
+        long timestamp = timeGen();
+        while (timestamp <= lastTimestamp) {
+            timestamp = timeGen();
+        }
+        return timestamp;
+    }
+
+    protected long timeGen() {
+        return System.currentTimeMillis();
+    }
+
+    public String generateId() {
+        return String.valueOf(this.nextId());
+    }
+}