知识检索响应结构更新

Author: yuhongxiao

core/src/main/java/ai/z/openapi/service/model/ModelData.java

@@ -1,5 +1,6 @@
 package ai.z.openapi.service.model;
 
+import ai.z.openapi.service.tools.KnowledgeV2Result;
 import com.fasterxml.jackson.annotation.JsonProperty;
 import ai.z.openapi.service.web_search.WebSearchResp;
 import lombok.AllArgsConstructor;
@@ -54,4 +55,6 @@ public final class ModelData {
 
 	private String delta;
 
+	private KnowledgeV2Result knowledgeV2;
+
 }

core/src/main/java/ai/z/openapi/service/model/ToolCalls.java

@@ -1,11 +1,14 @@
 package ai.z.openapi.service.model;
 
+import ai.z.openapi.service.tools.DocReference;
 import com.fasterxml.jackson.annotation.JsonProperty;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
+import java.util.List;
+
 /**
  * Represents tool calls made by the model during conversation. This class contains
  * information about function calls including the function details, unique identifier, and
@@ -35,4 +38,7 @@ public class ToolCalls {
 	@JsonProperty("mcp")
 	private MCPToolCall mcp;
 
+	@JsonProperty("doc_reference")
+	private List<DocReference> docReferenceList;
+
 }

core/src/main/java/ai/z/openapi/service/tools/DocReference.java

@@ -0,0 +1,24 @@
+package ai.z.openapi.service.tools;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import java.io.Serializable;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+@JsonIgnoreProperties(ignoreUnknown = true)
+public class DocReference implements Serializable {
+
+    /** 文档索引 */
+    private Integer index;
+
+    /** 文档类型 */
+    private String doc_type;
+
+    /** 文档名称 */
+    private String doc_name;
+
+}

core/src/main/java/ai/z/openapi/service/tools/KnowledgeV2Result.java

@@ -0,0 +1,163 @@
+package ai.z.openapi.service.tools;
+
+import java.io.Serializable;
+import java.util.List;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+/** 知识库检索V2返回结果 */
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+public class KnowledgeV2Result implements Serializable {
+
+    /** 数据来源列表 */
+    private List<Content> contents;
+
+    /** query重写结果 */
+    private RewrittenQuery rewritten_query;
+
+    private String type;
+
+    /** 数据来源 */
+    @lombok.Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public static class Content {
+
+        /** 切片ID（uuid） */
+        private String id;
+
+        /** 模态类型：text、image_url、video_url */
+        private String type;
+
+        /** 文本内容 */
+        private String text;
+
+        /** 文本中的媒体文件列表 */
+        private List<Media> medias;
+
+        /** 图像URL对象 */
+        private ImageUrl image_url;
+
+        /** 视频URL对象 */
+        private VideoUrl video_url;
+
+        /** 召回位次（rankIndex） */
+        private Integer index;
+
+        /** 召回分数（rankScore） */
+        private Double score;
+
+        /** 重排位次（rerankIndex） */
+        private Integer rerank_index;
+
+        /** 重排分数（rerankScore） */
+        private Double rerank_score;
+
+        /** 元数据 */
+        private Metadata metadata;
+        
+    }
+
+    /** 媒体文件 */
+    @lombok.Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public static class Media {
+
+        /** 图片ID */
+        private String id;
+
+        /** 图片URL */
+        private String url;
+
+        /** 图片描述 */
+        private String description;
+
+    }
+
+    /** 图像URL对象 */
+    @lombok.Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public static class ImageUrl {
+
+        /** URL */
+        private String url;
+
+    }
+
+    /** 视频URL对象 */
+    @lombok.Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public static class VideoUrl {
+
+        /** URL */
+        private String url;
+
+    }
+
+    /** 元数据 */
+    @lombok.Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public static class Metadata {
+
+        /** 切片ID */
+        private String _id;
+
+        /** 知识库ID */
+        private String know_id;
+
+        /** 文档ID */
+        private String doc_id;
+
+        /** 文档类型（dtype） */
+        private String doc_type;
+
+        /** 文档名称（filename） */
+        private String doc_name;
+
+        /** 文档URL */
+        private String doc_url;
+
+        /** 切片下标 */
+        private Integer index;
+
+        /** 文档页码 */
+        private Integer page_index;
+
+        /** 视频切片下标 */
+        private Integer clip_index;
+
+        /** 首帧时间戳 */
+        private Long start_time;
+
+        /** 尾帧时间戳 */
+        private Long end_time;
+
+        /** 视频切片时长 */
+        private Long duration;
+
+        /** 关键帧列表 */
+        private List<Object> frames;
+
+    }
+
+    /** query重写结果 */
+    @lombok.Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public static class RewrittenQuery {
+
+        /** 生效query */
+        private String primary_query;
+
+        /** 备选query列表 */
+        private List<String> multi_queries;
+
+    }
+}