知识库创建

schoolNewsServ/common/common-dto/src/main/java/org/xyzh/common/dto/ai/TbAiKnowledge.java

@@ -18,6 +18,11 @@ public class TbAiKnowledge extends BaseDTO {
      */
     private String title;
     
+    /**
+     * @description 知识库头像
+     */
+    private String avatar;
+    
     /**
      * @description 知识库描述
      */
@@ -73,6 +78,31 @@ public class TbAiKnowledge extends BaseDTO {
      */
     private String embeddingModel;
     
+    /**
+     * @description 向量模型提供商
+     */
+    private String embeddingModelProvider;
+    
+    /**
+     * @description Rerank模型名称
+     */
+    private String rerankModel;
+    
+    /**
+     * @description Rerank模型提供商
+     */
+    private String rerankModelProvider;
+    
+    /**
+     * @description 检索Top K（返回前K个结果）
+     */
+    private Integer retrievalTopK;
+    
+    /**
+     * @description 检索分数阈值（0.00-1.00）
+     */
+    private Double retrievalScoreThreshold;
+    
     /**
      * @description 向量ID（用于向量检索）
      */
@@ -116,6 +146,15 @@ public class TbAiKnowledge extends BaseDTO {
         this.title = title;
     }
 
+    public String getAvatar() {
+        return avatar;
+    }
+
+    public void setAvatar(String avatar) {
+        this.avatar = avatar;
+    }
+
+    
     public String getDescription() {
         return description;
     }
@@ -260,6 +299,45 @@ public class TbAiKnowledge extends BaseDTO {
         this.updater = updater;
     }
 
+    public String getEmbeddingModelProvider() {
+        return embeddingModelProvider;
+    }
+    public void setEmbeddingModelProvider(String embeddingModelProvider) {
+        this.embeddingModelProvider = embeddingModelProvider;
+    }
+
+    public String getRerankModel() {
+        return rerankModel;
+    }
+
+    public void setRerankModel(String rerankModel) {
+        this.rerankModel = rerankModel;
+    }
+
+    public String getRerankModelProvider() {
+        return rerankModelProvider;
+    }
+
+    public void setRerankModelProvider(String rerankModelProvider) {
+        this.rerankModelProvider = rerankModelProvider;
+    }
+
+    public Integer getRetrievalTopK() {
+        return retrievalTopK;
+    }
+
+    public void setRetrievalTopK(Integer retrievalTopK) {
+        this.retrievalTopK = retrievalTopK;
+    }
+
+    public Double getRetrievalScoreThreshold() {
+        return retrievalScoreThreshold;
+    }
+
+    public void setRetrievalScoreThreshold(Double retrievalScoreThreshold) {
+        this.retrievalScoreThreshold = retrievalScoreThreshold;
+    }
+
     @Override
     public String toString() {
         return "TbAiKnowledge{" +