Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions core/src/main/java/ai/z/openapi/service/model/ModelData.java
Original file line number Diff line number Diff line change
@@ -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;
Expand Down Expand Up @@ -54,4 +55,6 @@ public final class ModelData {

private String delta;

private KnowledgeV2Result knowledgeV2;

}
6 changes: 6 additions & 0 deletions core/src/main/java/ai/z/openapi/service/model/ToolCalls.java
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -35,4 +38,7 @@ public class ToolCalls {
@JsonProperty("mcp")
private MCPToolCall mcp;

@JsonProperty("doc_reference")
private List<DocReference> docReferenceList;

}
25 changes: 25 additions & 0 deletions core/src/main/java/ai/z/openapi/service/tools/DocReference.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
package ai.z.openapi.service.tools;

import com.fasterxml.jackson.annotation.JsonIgnoreProperties;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serializable;

@Data
@NoArgsConstructor
@AllArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public class DocReference implements Serializable {

/** 文档索引 */
private Integer index;

/** 文档类型 */
private String doc_type;

/** 文档名称 */
private String doc_name;
}
158 changes: 158 additions & 0 deletions core/src/main/java/ai/z/openapi/service/tools/KnowledgeV2Result.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,158 @@
package ai.z.openapi.service.tools;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serializable;
import java.util.List;

/** 知识库检索V2返回结果 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class KnowledgeV2Result implements Serializable {

/** 数据来源列表 */
private List<Content> contents;

/** query重写结果 */
private RewrittenQuery rewritten_query;

private String type;

/** 数据来源 */
@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 MetadataDTO metadata;
}

/** 媒体文件 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public static class Media {

/** 图片ID */
private String id;

/** 图片URL */
private String url;

/** 图片描述 */
private String description;
}

/** 图像URL对象 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public static class ImageUrl {

/** URL */
private String url;
}

/** 视频URL对象 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public static class VideoUrl {

/** URL */
private String url;
}

/** 元数据 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public static class MetadataDTO {

/** 切片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重写结果 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public static class RewrittenQuery {

/** 生效query */
private String primary_query;

/** 备选query列表 */
private List<String> multi_queries;
}
}
2 changes: 1 addition & 1 deletion pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
</scm>

<properties>
<revision>0.3.3</revision>
<revision>0.3.3.3</revision>
<java.version>8</java.version>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
Expand Down
Loading