ragflow的dto
This commit is contained in:
Cold
172
ragflow/chat.go
Normal file
172
ragflow/chat.go
Normal file
@@ -0,0 +1,172 @@
|
|||||||
|
package ragflow
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Chat 结构体
|
||||||
|
type Chat struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Avatar string `json:"avatar"`
|
||||||
|
DatasetIds []string `json:"dataset_ids"`
|
||||||
|
Llm Llm `json:"llm"`
|
||||||
|
Prompt Prompt `json:"prompt"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
DoRefer string `json:"do_refer"`
|
||||||
|
Language string `json:"language"`
|
||||||
|
PromptType string `json:"prompt_type"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
TenantId string `json:"tenant_id"`
|
||||||
|
TopK int `json:"top_k"`
|
||||||
|
CreateDate string `json:"create_date"`
|
||||||
|
CreateTime int64 `json:"create_time"`
|
||||||
|
UpdateDate string `json:"update_date"`
|
||||||
|
UpdateTime int64 `json:"update_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Llm struct {
|
||||||
|
ModelName string `json:"model_name,omitempty"`
|
||||||
|
Temperature float64 `json:"temperature,omitempty"`
|
||||||
|
TopP float64 `json:"top_p,omitempty"`
|
||||||
|
PresencePenalty float64 `json:"presence_penalty,omitempty"`
|
||||||
|
FrequencyPenalty float64 `json:"frequency_penalty,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Prompt struct {
|
||||||
|
SimilarityThreshold float64 `json:"similarity_threshold,omitempty"`
|
||||||
|
KeywordsSimilarityWeight float64 `json:"keywords_similarity_weight,omitempty"`
|
||||||
|
Opener string `json:"opener,omitempty"`
|
||||||
|
Prompt string `json:"prompt,omitempty"`
|
||||||
|
RerankModel string `json:"rerank_model,omitempty"`
|
||||||
|
TopN int `json:"top_n,omitempty"`
|
||||||
|
Variables []Variable `json:"variables,omitempty"`
|
||||||
|
EmptyResponse string `json:"empty_response,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Variable struct {
|
||||||
|
Key string `json:"key"`
|
||||||
|
Optional bool `json:"optional"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateChatReq 创建聊天助手请求
|
||||||
|
type CreateChatReq struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Avatar string `json:"avatar,omitempty"`
|
||||||
|
DatasetIds []string `json:"dataset_ids,omitempty"`
|
||||||
|
Llm *Llm `json:"llm,omitempty"`
|
||||||
|
Prompt *Prompt `json:"prompt,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateChatReq 更新聊天助手请求
|
||||||
|
type UpdateChatReq struct {
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
Avatar string `json:"avatar,omitempty"`
|
||||||
|
DatasetIds []string `json:"dataset_ids,omitempty"`
|
||||||
|
Llm *Llm `json:"llm,omitempty"`
|
||||||
|
Prompt *Prompt `json:"prompt,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListChatsReq 列出聊天助手请求
|
||||||
|
type ListChatsReq struct {
|
||||||
|
Page int `json:"page,omitempty"`
|
||||||
|
PageSize int `json:"page_size,omitempty"`
|
||||||
|
OrderBy string `json:"orderby,omitempty"`
|
||||||
|
Desc bool `json:"desc,omitempty"`
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
Id string `json:"id,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListChatsRes 列出聊天助手响应
|
||||||
|
type ListChatsRes struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data []*Chat `json:"data"`
|
||||||
|
Total int `json:"total"` // API 文档中未明确 total 字段，但通常列表接口会有
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteChatsReq 删除聊天助手请求
|
||||||
|
type DeleteChatsReq struct {
|
||||||
|
Ids []string `json:"ids"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateChat 创建聊天助手
|
||||||
|
func (c *Client) CreateChat(ctx context.Context, req *CreateChatReq) (*Chat, error) {
|
||||||
|
var res struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data *Chat `json:"data"`
|
||||||
|
Msg string `json:"message"`
|
||||||
|
}
|
||||||
|
if err := c.request(ctx, "POST", "/api/v1/chats", req, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("create chat failed: %s", res.Msg)
|
||||||
|
}
|
||||||
|
return res.Data, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListChats 列出聊天助手
|
||||||
|
func (c *Client) ListChats(ctx context.Context, req *ListChatsReq) (*ListChatsRes, error) {
|
||||||
|
path := "/api/v1/chats?"
|
||||||
|
params := map[string]interface{}{}
|
||||||
|
if req.Page > 0 {
|
||||||
|
params["page"] = req.Page
|
||||||
|
}
|
||||||
|
if req.PageSize > 0 {
|
||||||
|
params["page_size"] = req.PageSize
|
||||||
|
}
|
||||||
|
if req.OrderBy != "" {
|
||||||
|
params["orderby"] = req.OrderBy
|
||||||
|
}
|
||||||
|
if req.Desc {
|
||||||
|
params["desc"] = "true"
|
||||||
|
} else {
|
||||||
|
params["desc"] = "false"
|
||||||
|
}
|
||||||
|
if req.Name != "" {
|
||||||
|
params["name"] = req.Name
|
||||||
|
}
|
||||||
|
if req.Id != "" {
|
||||||
|
params["id"] = req.Id
|
||||||
|
}
|
||||||
|
|
||||||
|
for k, v := range params {
|
||||||
|
path += fmt.Sprintf("%s=%v&", k, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
var res ListChatsRes
|
||||||
|
if err := c.request(ctx, "GET", path, nil, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("list chats failed: code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return &res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteChats 删除聊天助手
|
||||||
|
func (c *Client) DeleteChats(ctx context.Context, ids []string) error {
|
||||||
|
req := DeleteChatsReq{Ids: ids}
|
||||||
|
var res CommonResponse
|
||||||
|
if err := c.request(ctx, "DELETE", "/api/v1/chats", req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("delete chats failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateChat 更新聊天助手
|
||||||
|
func (c *Client) UpdateChat(ctx context.Context, id string, req *UpdateChatReq) error {
|
||||||
|
var res CommonResponse
|
||||||
|
path := fmt.Sprintf("/api/v1/chats/%s", id)
|
||||||
|
if err := c.request(ctx, "PUT", path, req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("update chat failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
174
ragflow/chunk.go
Normal file
174
ragflow/chunk.go
Normal file
@@ -0,0 +1,174 @@
|
|||||||
|
package ragflow
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Chunk 结构体
|
||||||
|
type Chunk struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
Content string `json:"content"`
|
||||||
|
DocumentId string `json:"document_id"`
|
||||||
|
DatasetId string `json:"dataset_id"`
|
||||||
|
CreateTime string `json:"create_time"`
|
||||||
|
CreateTimestamp float64 `json:"create_timestamp"`
|
||||||
|
ImportantKeywords []string `json:"important_keywords"`
|
||||||
|
Questions []string `json:"questions"`
|
||||||
|
Available bool `json:"available"`
|
||||||
|
ImageId string `json:"image_id"`
|
||||||
|
Positions []string `json:"positions"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddChunkReq 添加知识块请求
|
||||||
|
type AddChunkReq struct {
|
||||||
|
Content string `json:"content"`
|
||||||
|
ImportantKeywords []string `json:"important_keywords,omitempty"`
|
||||||
|
Questions []string `json:"questions,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListChunksReq 列出知识块请求
|
||||||
|
type ListChunksReq struct {
|
||||||
|
Keywords string `json:"keywords,omitempty"`
|
||||||
|
Page int `json:"page,omitempty"`
|
||||||
|
PageSize int `json:"page_size,omitempty"`
|
||||||
|
Id string `json:"id,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListChunksRes 列出知识块响应
|
||||||
|
type ListChunksRes struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data struct {
|
||||||
|
Chunks []*Chunk `json:"chunks"`
|
||||||
|
Doc interface{} `json:"doc"` // 文档信息，暂时用 interface{}
|
||||||
|
Total int `json:"total"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteChunksReq 删除知识块请求
|
||||||
|
type DeleteChunksReq struct {
|
||||||
|
ChunkIds []string `json:"chunk_ids,omitempty"` // 如果为空，删除所有
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateChunkReq 更新知识块请求
|
||||||
|
type UpdateChunkReq struct {
|
||||||
|
Content string `json:"content,omitempty"`
|
||||||
|
ImportantKeywords []string `json:"important_keywords,omitempty"`
|
||||||
|
Available *bool `json:"available,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RetrieveChunksReq 检索知识块请求
|
||||||
|
type RetrieveChunksReq struct {
|
||||||
|
Question string `json:"question"`
|
||||||
|
DatasetIds []string `json:"dataset_ids,omitempty"`
|
||||||
|
DocumentIds []string `json:"document_ids,omitempty"`
|
||||||
|
Page int `json:"page,omitempty"`
|
||||||
|
PageSize int `json:"page_size,omitempty"`
|
||||||
|
SimilarityThreshold float64 `json:"similarity_threshold,omitempty"`
|
||||||
|
VectorSimilarityWeight float64 `json:"vector_similarity_weight,omitempty"`
|
||||||
|
TopK int `json:"top_k,omitempty"`
|
||||||
|
RerankId string `json:"rerank_id,omitempty"`
|
||||||
|
Keyword bool `json:"keyword,omitempty"`
|
||||||
|
Highlight bool `json:"highlight,omitempty"`
|
||||||
|
CrossLanguages []string `json:"cross_languages,omitempty"`
|
||||||
|
MetadataCondition map[string]interface{} `json:"metadata_condition,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RetrieveChunksRes 检索知识块响应 (结构比较复杂，暂时简化，根据实际返回调整)
|
||||||
|
// 官方文档未给出详细响应结构，假设返回 chunks 列表
|
||||||
|
type RetrieveChunksRes struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data struct {
|
||||||
|
Chunks []interface{} `json:"chunks"` // 检索结果可能包含额外信息
|
||||||
|
Total int `json:"total"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddChunk 添加知识块
|
||||||
|
func (c *Client) AddChunk(ctx context.Context, datasetId, documentId string, req *AddChunkReq) (*Chunk, error) {
|
||||||
|
path := fmt.Sprintf("/api/v1/datasets/%s/documents/%s/chunks", datasetId, documentId)
|
||||||
|
var res struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data struct {
|
||||||
|
Chunk *Chunk `json:"chunk"`
|
||||||
|
} `json:"data"`
|
||||||
|
Msg string `json:"message"`
|
||||||
|
}
|
||||||
|
if err := c.request(ctx, "POST", path, req, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("add chunk failed: %s", res.Msg)
|
||||||
|
}
|
||||||
|
return res.Data.Chunk, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListChunks 列出知识块
|
||||||
|
func (c *Client) ListChunks(ctx context.Context, datasetId, documentId string, req *ListChunksReq) (*ListChunksRes, error) {
|
||||||
|
path := fmt.Sprintf("/api/v1/datasets/%s/documents/%s/chunks?", datasetId, documentId)
|
||||||
|
params := map[string]interface{}{}
|
||||||
|
if req.Keywords != "" {
|
||||||
|
params["keywords"] = req.Keywords
|
||||||
|
}
|
||||||
|
if req.Page > 0 {
|
||||||
|
params["page"] = req.Page
|
||||||
|
}
|
||||||
|
if req.PageSize > 0 {
|
||||||
|
params["page_size"] = req.PageSize
|
||||||
|
}
|
||||||
|
if req.Id != "" {
|
||||||
|
params["id"] = req.Id
|
||||||
|
}
|
||||||
|
|
||||||
|
for k, v := range params {
|
||||||
|
path += fmt.Sprintf("%s=%v&", k, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
var res ListChunksRes
|
||||||
|
if err := c.request(ctx, "GET", path, nil, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("list chunks failed: code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return &res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteChunks 删除知识块
|
||||||
|
func (c *Client) DeleteChunks(ctx context.Context, datasetId, documentId string, chunkIds []string) error {
|
||||||
|
req := DeleteChunksReq{ChunkIds: chunkIds}
|
||||||
|
var res CommonResponse
|
||||||
|
path := fmt.Sprintf("/api/v1/datasets/%s/documents/%s/chunks", datasetId, documentId)
|
||||||
|
if err := c.request(ctx, "DELETE", path, req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("delete chunks failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateChunk 更新知识块
|
||||||
|
func (c *Client) UpdateChunk(ctx context.Context, datasetId, documentId, chunkId string, req *UpdateChunkReq) error {
|
||||||
|
var res CommonResponse
|
||||||
|
path := fmt.Sprintf("/api/v1/datasets/%s/documents/%s/chunks/%s", datasetId, documentId, chunkId)
|
||||||
|
if err := c.request(ctx, "PUT", path, req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("update chunk failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RetrieveChunks 检索知识块
|
||||||
|
func (c *Client) RetrieveChunks(ctx context.Context, req *RetrieveChunksReq) (*RetrieveChunksRes, error) {
|
||||||
|
var res RetrieveChunksRes
|
||||||
|
if err := c.request(ctx, "POST", "/api/v1/retrieval", req, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("retrieve chunks failed: code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return &res, nil
|
||||||
|
}
|
||||||
161
ragflow/dataset.go
Normal file
161
ragflow/dataset.go
Normal file
@@ -0,0 +1,161 @@
|
|||||||
|
package ragflow
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Dataset 结构体
|
||||||
|
type Dataset struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Avatar string `json:"avatar"`
|
||||||
|
TenantId string `json:"tenant_id"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
Language string `json:"language"`
|
||||||
|
EmbeddingModel string `json:"embedding_model"`
|
||||||
|
Permission string `json:"permission"`
|
||||||
|
DocumentCount int `json:"document_count"`
|
||||||
|
ChunkCount int `json:"chunk_count"`
|
||||||
|
ParseStatus string `json:"parse_status"`
|
||||||
|
CreatedBy string `json:"created_by"`
|
||||||
|
CreateTime int64 `json:"create_time"`
|
||||||
|
UpdateDate string `json:"update_date"`
|
||||||
|
UpdateTime int64 `json:"update_time"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
ChunkMethod string `json:"chunk_method"`
|
||||||
|
ParserConfig map[string]interface{} `json:"parser_config"`
|
||||||
|
VectorSimilarityWeight float64 `json:"vector_similarity_weight"`
|
||||||
|
SimilarityThreshold float64 `json:"similarity_threshold"`
|
||||||
|
TokenNum int `json:"token_num"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateDatasetReq 创建数据集请求
|
||||||
|
type CreateDatasetReq struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Avatar string `json:"avatar,omitempty"`
|
||||||
|
Description string `json:"description,omitempty"`
|
||||||
|
EmbeddingModel string `json:"embedding_model,omitempty"`
|
||||||
|
Permission string `json:"permission,omitempty"`
|
||||||
|
ChunkMethod string `json:"chunk_method,omitempty"`
|
||||||
|
ParserConfig map[string]interface{} `json:"parser_config,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateDatasetReq 更新数据集请求
|
||||||
|
type UpdateDatasetReq struct {
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
Avatar string `json:"avatar,omitempty"`
|
||||||
|
Description string `json:"description,omitempty"`
|
||||||
|
EmbeddingModel string `json:"embedding_model,omitempty"`
|
||||||
|
Permission string `json:"permission,omitempty"`
|
||||||
|
ChunkMethod string `json:"chunk_method,omitempty"`
|
||||||
|
PageRank int `json:"pagerank,omitempty"`
|
||||||
|
ParserConfig map[string]interface{} `json:"parser_config,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListDatasetsReq 列出数据集请求
|
||||||
|
type ListDatasetsReq struct {
|
||||||
|
Page int `json:"page,omitempty"`
|
||||||
|
PageSize int `json:"page_size,omitempty"`
|
||||||
|
OrderBy string `json:"orderby,omitempty"`
|
||||||
|
Desc bool `json:"desc,omitempty"`
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
Id string `json:"id,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListDatasetsRes 列出数据集响应
|
||||||
|
type ListDatasetsRes struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data []*Dataset `json:"data"`
|
||||||
|
Total int `json:"total"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteDatasetsReq 删除数据集请求
|
||||||
|
type DeleteDatasetsReq struct {
|
||||||
|
Ids []string `json:"ids"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateDataset 创建数据集
|
||||||
|
func (c *Client) CreateDataset(ctx context.Context, req *CreateDatasetReq) (*Dataset, error) {
|
||||||
|
var res struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data *Dataset `json:"data"`
|
||||||
|
Msg string `json:"message"`
|
||||||
|
}
|
||||||
|
if err := c.request(ctx, "POST", "/api/v1/datasets", req, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("create dataset failed: %s", res.Msg)
|
||||||
|
}
|
||||||
|
return res.Data, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListDatasets 列出数据集
|
||||||
|
func (c *Client) ListDatasets(ctx context.Context, req *ListDatasetsReq) (*ListDatasetsRes, error) {
|
||||||
|
// 构建查询参数
|
||||||
|
path := "/api/v1/datasets?"
|
||||||
|
params := map[string]interface{}{}
|
||||||
|
if req.Page > 0 {
|
||||||
|
params["page"] = req.Page
|
||||||
|
}
|
||||||
|
if req.PageSize > 0 {
|
||||||
|
params["page_size"] = req.PageSize
|
||||||
|
}
|
||||||
|
if req.OrderBy != "" {
|
||||||
|
params["orderby"] = req.OrderBy
|
||||||
|
}
|
||||||
|
// desc 默认为 true，如果显式设置为 false 才传递，或者根据 API 行为调整
|
||||||
|
// 这里简单处理，如果设置了就传
|
||||||
|
if req.Desc {
|
||||||
|
params["desc"] = "true"
|
||||||
|
} else {
|
||||||
|
params["desc"] = "false"
|
||||||
|
}
|
||||||
|
if req.Name != "" {
|
||||||
|
params["name"] = req.Name
|
||||||
|
}
|
||||||
|
if req.Id != "" {
|
||||||
|
params["id"] = req.Id
|
||||||
|
}
|
||||||
|
|
||||||
|
// 拼接 query string
|
||||||
|
for k, v := range params {
|
||||||
|
path += fmt.Sprintf("%s=%v&", k, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
var res ListDatasetsRes
|
||||||
|
if err := c.request(ctx, "GET", path, nil, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("list datasets failed: code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return &res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteDataset 删除数据集
|
||||||
|
func (c *Client) DeleteDataset(ctx context.Context, ids []string) error {
|
||||||
|
req := DeleteDatasetsReq{Ids: ids}
|
||||||
|
var res CommonResponse
|
||||||
|
if err := c.request(ctx, "DELETE", "/api/v1/datasets", req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("delete dataset failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateDataset 更新数据集
|
||||||
|
func (c *Client) UpdateDataset(ctx context.Context, id string, req *UpdateDatasetReq) error {
|
||||||
|
var res CommonResponse
|
||||||
|
path := fmt.Sprintf("/api/v1/datasets/%s", id)
|
||||||
|
if err := c.request(ctx, "PUT", path, req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("update dataset failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
132
ragflow/document.go
Normal file
132
ragflow/document.go
Normal file
@@ -0,0 +1,132 @@
|
|||||||
|
package ragflow
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Document 结构体
|
||||||
|
type Document struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
DatasetId string `json:"dataset_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Size int64 `json:"size"`
|
||||||
|
Location string `json:"location"`
|
||||||
|
CreatedBy string `json:"created_by"`
|
||||||
|
CreateTime int64 `json:"create_time"`
|
||||||
|
Thumbnail string `json:"thumbnail"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
RunStatus string `json:"run_status"` // 对应 API 返回的 "run" 字段，可能需要确认
|
||||||
|
Status string `json:"status"`
|
||||||
|
ChunkMethod string `json:"chunk_method"`
|
||||||
|
ParserConfig map[string]interface{} `json:"parser_config"`
|
||||||
|
TokenNum int `json:"token_num"`
|
||||||
|
ChunkCount int `json:"chunk_count"`
|
||||||
|
ProcessBegin int64 `json:"process_begin"`
|
||||||
|
ProcessDu int64 `json:"process_du"`
|
||||||
|
Progress float64 `json:"progress"`
|
||||||
|
ProgressMsg string `json:"progress_msg"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// UploadDocumentReq 上传文档请求
|
||||||
|
// 注意：上传文件通常需要 multipart/form-data，这里仅定义结构，实际逻辑在方法中处理
|
||||||
|
type UploadDocumentReq struct {
|
||||||
|
FilePaths []string // 本地文件路径列表
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListDocumentsReq 列出文档请求
|
||||||
|
type ListDocumentsReq struct {
|
||||||
|
Page int `json:"page,omitempty"`
|
||||||
|
PageSize int `json:"page_size,omitempty"`
|
||||||
|
OrderBy string `json:"orderby,omitempty"`
|
||||||
|
Desc bool `json:"desc,omitempty"`
|
||||||
|
Keywords string `json:"keywords,omitempty"`
|
||||||
|
Id string `json:"id,omitempty"`
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
CreateTimeFrom int64 `json:"create_time_from,omitempty"`
|
||||||
|
CreateTimeTo int64 `json:"create_time_to,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListDocumentsRes 列出文档响应
|
||||||
|
type ListDocumentsRes struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data []*Document `json:"data"`
|
||||||
|
Total int `json:"total"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteDocumentsReq 删除文档请求
|
||||||
|
type DeleteDocumentsReq struct {
|
||||||
|
Ids []string `json:"ids"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListDocuments 列出文档
|
||||||
|
func (c *Client) ListDocuments(ctx context.Context, datasetId string, req *ListDocumentsReq) (*ListDocumentsRes, error) {
|
||||||
|
path := fmt.Sprintf("/api/v1/datasets/%s/documents?", datasetId)
|
||||||
|
params := map[string]interface{}{}
|
||||||
|
if req.Page > 0 {
|
||||||
|
params["page"] = req.Page
|
||||||
|
}
|
||||||
|
if req.PageSize > 0 {
|
||||||
|
params["page_size"] = req.PageSize
|
||||||
|
}
|
||||||
|
if req.OrderBy != "" {
|
||||||
|
params["orderby"] = req.OrderBy
|
||||||
|
}
|
||||||
|
if req.Desc {
|
||||||
|
params["desc"] = "true"
|
||||||
|
} else {
|
||||||
|
params["desc"] = "false"
|
||||||
|
}
|
||||||
|
if req.Keywords != "" {
|
||||||
|
params["keywords"] = req.Keywords
|
||||||
|
}
|
||||||
|
if req.Id != "" {
|
||||||
|
params["id"] = req.Id
|
||||||
|
}
|
||||||
|
if req.Name != "" {
|
||||||
|
params["name"] = req.Name
|
||||||
|
}
|
||||||
|
if req.CreateTimeFrom > 0 {
|
||||||
|
params["create_time_from"] = req.CreateTimeFrom
|
||||||
|
}
|
||||||
|
if req.CreateTimeTo > 0 {
|
||||||
|
params["create_time_to"] = req.CreateTimeTo
|
||||||
|
}
|
||||||
|
|
||||||
|
for k, v := range params {
|
||||||
|
path += fmt.Sprintf("%s=%v&", k, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
var res ListDocumentsRes
|
||||||
|
if err := c.request(ctx, "GET", path, nil, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("list documents failed: code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return &res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UploadDocument 上传文档
|
||||||
|
// 注意：此方法需要特殊处理 multipart/form-data，目前的 request 方法可能不支持
|
||||||
|
// 我们需要扩展 request 方法或在此处单独实现
|
||||||
|
func (c *Client) UploadDocument(ctx context.Context, datasetId string, filePaths []string) error {
|
||||||
|
// TODO: 实现文件上传逻辑，需要使用 gclient 的 UploadFile 功能
|
||||||
|
// 由于 request 方法封装了 JSON 处理，这里可能需要绕过 request 方法直接使用 c.Client
|
||||||
|
// 暂时留空或仅做简单提示，待完善 Client 封装以支持文件上传
|
||||||
|
return fmt.Errorf("upload document not implemented yet")
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteDocument 删除文档
|
||||||
|
func (c *Client) DeleteDocument(ctx context.Context, datasetId string, ids []string) error {
|
||||||
|
req := DeleteDocumentsReq{Ids: ids}
|
||||||
|
var res CommonResponse
|
||||||
|
path := fmt.Sprintf("/api/v1/datasets/%s/documents", datasetId)
|
||||||
|
if err := c.request(ctx, "DELETE", path, req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("delete document failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -1,5 +0,0 @@
|
|||||||
package ragflow
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
|
|
||||||
}
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
package ragflow
|
|
||||||
77
ragflow/service/client.go
Normal file
77
ragflow/service/client.go
Normal file
@@ -0,0 +1,77 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gitee.com/red-future---jilin-g/common/ragflow/dto"
|
||||||
|
"github.com/gogf/gf/v2/encoding/gjson"
|
||||||
|
"github.com/gogf/gf/v2/frame/g"
|
||||||
|
"github.com/gogf/gf/v2/net/gclient"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Client struct {
|
||||||
|
BaseURL string
|
||||||
|
ApiKey string
|
||||||
|
Client *gclient.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewClient 创建一个新的 RAGFlow 客户端
|
||||||
|
func NewClient(baseUrl, apiKey string) *Client {
|
||||||
|
return &Client{
|
||||||
|
BaseURL: baseUrl,
|
||||||
|
ApiKey: apiKey,
|
||||||
|
Client: g.Client().SetTimeout(30 * time.Second),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// request 发送 HTTP 请求
|
||||||
|
func (c *Client) request(ctx context.Context, method, path string, data interface{}, result interface{}) error {
|
||||||
|
url := fmt.Sprintf("%s%s", c.BaseURL, path)
|
||||||
|
|
||||||
|
req := c.Client.Header(map[string]string{
|
||||||
|
"Authorization": fmt.Sprintf("Bearer %s", c.ApiKey),
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
})
|
||||||
|
|
||||||
|
var res *gclient.Response
|
||||||
|
var err error
|
||||||
|
|
||||||
|
switch method {
|
||||||
|
case "GET":
|
||||||
|
res, err = req.Get(ctx, url, data)
|
||||||
|
case "POST":
|
||||||
|
res, err = req.Post(ctx, url, data)
|
||||||
|
case "PUT":
|
||||||
|
res, err = req.Put(ctx, url, data)
|
||||||
|
case "DELETE":
|
||||||
|
res, err = req.Delete(ctx, url, data)
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("unsupported method: %s", method)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer res.Close()
|
||||||
|
|
||||||
|
// 读取响应体
|
||||||
|
body := res.ReadAllString()
|
||||||
|
|
||||||
|
// 解析响应
|
||||||
|
if result != nil {
|
||||||
|
if err := gjson.DecodeTo(body, result); err != nil {
|
||||||
|
return fmt.Errorf("failed to decode response: %v, body: %s", err, body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查业务错误码
|
||||||
|
if commonRes, ok := result.(*dto.CommonResponse); ok {
|
||||||
|
if !commonRes.IsSuccess() {
|
||||||
|
return fmt.Errorf("api error: code=%d, message=%s", commonRes.Code, commonRes.Message)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
164
ragflow/session.go
Normal file
164
ragflow/session.go
Normal file
@@ -0,0 +1,164 @@
|
|||||||
|
package ragflow
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Session 结构体
|
||||||
|
type Session struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
ChatId string `json:"chat_id"` // 响应中是 "chat" 或 "chat_id"，根据文档示例调整
|
||||||
|
Messages []Message `json:"messages"`
|
||||||
|
CreateDate string `json:"create_date"`
|
||||||
|
CreateTime int64 `json:"create_time"`
|
||||||
|
UpdateDate string `json:"update_date"`
|
||||||
|
UpdateTime int64 `json:"update_time"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Message struct {
|
||||||
|
Content string `json:"content"`
|
||||||
|
Role string `json:"role"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateSessionReq 创建会话请求
|
||||||
|
type CreateSessionReq struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
UserId string `json:"user_id,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListSessionsReq 列出会话请求
|
||||||
|
type ListSessionsReq struct {
|
||||||
|
Page int `json:"page,omitempty"`
|
||||||
|
PageSize int `json:"page_size,omitempty"`
|
||||||
|
OrderBy string `json:"orderby,omitempty"`
|
||||||
|
Desc bool `json:"desc,omitempty"`
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
Id string `json:"id,omitempty"`
|
||||||
|
UserId string `json:"user_id,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListSessionsRes 列出会话响应
|
||||||
|
type ListSessionsRes struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data []*Session `json:"data"`
|
||||||
|
Total int `json:"total"` // API 文档未明确
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteSessionsReq 删除会话请求
|
||||||
|
type DeleteSessionsReq struct {
|
||||||
|
Ids []string `json:"ids"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ChatCompletionReq 对话请求
|
||||||
|
type ChatCompletionReq struct {
|
||||||
|
Question string `json:"question"`
|
||||||
|
Stream bool `json:"stream"`
|
||||||
|
SessionId string `json:"session_id,omitempty"`
|
||||||
|
UserId string `json:"user_id,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ChatCompletionRes 对话响应 (非流式)
|
||||||
|
type ChatCompletionRes struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data struct {
|
||||||
|
Answer string `json:"answer"`
|
||||||
|
Reference interface{} `json:"reference"`
|
||||||
|
AudioBinary interface{} `json:"audio_binary"`
|
||||||
|
Id interface{} `json:"id"`
|
||||||
|
SessionId string `json:"session_id"`
|
||||||
|
} `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateSession 创建会话
|
||||||
|
func (c *Client) CreateSession(ctx context.Context, chatId string, req *CreateSessionReq) (*Session, error) {
|
||||||
|
path := fmt.Sprintf("/api/v1/chats/%s/sessions", chatId)
|
||||||
|
var res struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Data *Session `json:"data"`
|
||||||
|
Msg string `json:"message"`
|
||||||
|
}
|
||||||
|
if err := c.request(ctx, "POST", path, req, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("create session failed: %s", res.Msg)
|
||||||
|
}
|
||||||
|
return res.Data, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListSessions 列出会话
|
||||||
|
func (c *Client) ListSessions(ctx context.Context, chatId string, req *ListSessionsReq) (*ListSessionsRes, error) {
|
||||||
|
path := fmt.Sprintf("/api/v1/chats/%s/sessions?", chatId)
|
||||||
|
params := map[string]interface{}{}
|
||||||
|
if req.Page > 0 {
|
||||||
|
params["page"] = req.Page
|
||||||
|
}
|
||||||
|
if req.PageSize > 0 {
|
||||||
|
params["page_size"] = req.PageSize
|
||||||
|
}
|
||||||
|
if req.OrderBy != "" {
|
||||||
|
params["orderby"] = req.OrderBy
|
||||||
|
}
|
||||||
|
if req.Desc {
|
||||||
|
params["desc"] = "true"
|
||||||
|
} else {
|
||||||
|
params["desc"] = "false"
|
||||||
|
}
|
||||||
|
if req.Name != "" {
|
||||||
|
params["name"] = req.Name
|
||||||
|
}
|
||||||
|
if req.Id != "" {
|
||||||
|
params["id"] = req.Id
|
||||||
|
}
|
||||||
|
if req.UserId != "" {
|
||||||
|
params["user_id"] = req.UserId
|
||||||
|
}
|
||||||
|
|
||||||
|
for k, v := range params {
|
||||||
|
path += fmt.Sprintf("%s=%v&", k, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
var res ListSessionsRes
|
||||||
|
if err := c.request(ctx, "GET", path, nil, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("list sessions failed: code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return &res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteSessions 删除会话
|
||||||
|
func (c *Client) DeleteSessions(ctx context.Context, chatId string, ids []string) error {
|
||||||
|
req := DeleteSessionsReq{Ids: ids}
|
||||||
|
var res CommonResponse
|
||||||
|
path := fmt.Sprintf("/api/v1/chats/%s/sessions", chatId)
|
||||||
|
if err := c.request(ctx, "DELETE", path, req, &res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !res.IsSuccess() {
|
||||||
|
return fmt.Errorf("delete sessions failed: %s", res.Message)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ChatCompletion 对话 (目前仅支持非流式)
|
||||||
|
func (c *Client) ChatCompletion(ctx context.Context, chatId string, req *ChatCompletionReq) (*ChatCompletionRes, error) {
|
||||||
|
path := fmt.Sprintf("/api/v1/chats/%s/completions", chatId)
|
||||||
|
var res ChatCompletionRes
|
||||||
|
|
||||||
|
// 如果需要流式支持，需要使用 gclient 的流式处理能力，这里暂只实现非流式
|
||||||
|
if req.Stream {
|
||||||
|
return nil, fmt.Errorf("stream mode not supported yet")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.request(ctx, "POST", path, req, &res); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.Code != 0 {
|
||||||
|
return nil, fmt.Errorf("chat completion failed: code=%d", res.Code)
|
||||||
|
}
|
||||||
|
return &res, nil
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user