ragflow的dto

ragflow/chat.go

@@ -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
+}