53 lines
2.0 KiB
Go
53 lines
2.0 KiB
Go
package proto
|
||
|
||
type KnowledgeBaseReq struct {
|
||
ID uint `json:"id" form:"id"` // 知识库ID
|
||
UUID string `json:"uuid" form:"uuid"` // 知识库UUID
|
||
Name string `json:"name" form:"name"` // 知识库名称
|
||
Description string `json:"description" form:"description"` // 知识库描述
|
||
FileIDS string `json:"file_ids" form:"file_ids"` // 文件ID列表，逗号分隔
|
||
}
|
||
|
||
type KnowledgeBaseServerResponse struct {
|
||
QuerySelect KnowledgeBaseServerResponseSelect `json:"query_select"` //查询结果
|
||
KnowledgeBase KBase `json:"knowledge_base"` //知识库信息
|
||
IMContext IMParamContext `json:"im_context"` //消息上下文
|
||
}
|
||
|
||
// KBase 定义与 TypeScript 接口对应的 Go 结构体
|
||
type KBase struct {
|
||
ID uint `json:"ID"`
|
||
Name string `json:"Name"`
|
||
Description string `json:"Description"`
|
||
UUID string `json:"UUID"`
|
||
SessionID uint `json:"SessionID"`
|
||
CreatedAt string `json:"CreatedAt"`
|
||
UpdatedAt string `json:"UpdatedAt"`
|
||
DeletedAt string `json:"DeletedAt"`
|
||
AuthID int `json:"AuthID"`
|
||
FileNameList string `json:"FileNameList"`
|
||
FileIDs string `json:"FileIDs"`
|
||
}
|
||
|
||
// 知识库向量查询返回结果Response 定义 JSON 数据对应的结构体
|
||
type KnowledgeBaseServerResponseSelect struct {
|
||
IDs [][]string `json:"ids"`
|
||
Embeddings interface{} `json:"embeddings"`
|
||
Documents [][]string `json:"documents"`
|
||
URIs interface{} `json:"uris"`
|
||
Included []string `json:"included"`
|
||
Data interface{} `json:"data"`
|
||
Metadatas [][]Metadata `json:"metadatas"`
|
||
Distances [][]float64 `json:"distances"`
|
||
}
|
||
|
||
// Metadata 定义元数据结构体
|
||
type Metadata struct {
|
||
Source string `json:"source"`
|
||
}
|
||
|
||
type KnowledgeBaseServerRequest struct {
|
||
IMContext IMParamContext `json:"im_context"` //消息上下文
|
||
KnowledgeBase KBase `json:"knowledge_base"` //知识库信息
|
||
}
|