common/ragflow/document.go

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
}