重构了一下 rag的方法, 使用 goframe的框架, 还有redis连接部分

ragflow/dataset.go

@@ -2,7 +2,8 @@ package ragflow
 
 import (
 	"context"
-	"fmt"
+
+	"github.com/gogf/gf/v2/errors/gerror"
 )
 
 // 数据集管理
@@ -90,7 +91,7 @@ func (c *Client) CreateDataset(ctx context.Context, req *CreateDatasetReq) (*Dat
 		return nil, err
 	}
 	if res.Code != 0 {
-		return nil, fmt.Errorf("create dataset failed: %s", res.Msg)
+		return nil, gerror.Newf("create dataset failed: %s", res.Msg)
 	}
 	return res.Data, nil
 }
@@ -134,33 +135,33 @@ func (c *Client) ListDatasets(ctx context.Context, req *ListDatasetsReq) (*ListD
 		return nil, err
 	}
 	if res.Code != 0 {
-		return nil, fmt.Errorf("list datasets failed: code=%d", res.Code)
+		return nil, gerror.Newf("list datasets failed: code=%d", res.Code)
 	}
 	return &res, nil
 }
 
 // DeleteDataset 删除数据集
-func (c *Client) DeleteDataset(ctx context.Context, ids []string) error {
+func (c *Client) DeleteDataset(ctx context.Context, ids []string) (err error) {
 	req := DeleteDatasetsReq{Ids: ids}
 	var res CommonResponse
-	if err := c.request(ctx, "DELETE", "/api/v1/datasets", req, &res); err != nil {
-		return err
+	if err = c.request(ctx, "DELETE", "/api/v1/datasets", req, &res); err != nil {
+		return
 	}
 	if !res.IsSuccess() {
-		return fmt.Errorf("delete dataset failed: %s", res.Message)
+		return gerror.Newf("delete dataset failed: %s", res.Message)
 	}
-	return nil
+	return
 }
 
 // UpdateDataset 更新数据集
-func (c *Client) UpdateDataset(ctx context.Context, id string, req *UpdateDatasetReq) error {
+func (c *Client) UpdateDataset(ctx context.Context, id string, req *UpdateDatasetReq) (err 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
+	path := "/api/v1/datasets/" + id
+	if err = c.request(ctx, "PUT", path, req, &res); err != nil {
+		return
 	}
 	if !res.IsSuccess() {
-		return fmt.Errorf("update dataset failed: %s", res.Message)
+		return gerror.Newf("update dataset failed: %s", res.Message)
 	}
-	return nil
+	return
 }