feat: api add list files and delete files

Author: shenhaofang

client.go

@@ -109,6 +109,44 @@ func (c *AIClient) UpdateFile(method string, param OpenAIFileCreateParam) (*File
 	return &res.FileInfo, err
 }
 
+func (c *AIClient) ListFiles(param OpenAIListFilesParam) (*ListFilesResp, error) {
+	query := param.ToQuery()
+	if query != "" {
+		query = "?" + query
+	}
+	url := c.BaseURL + "/files" + query
+
+	// 发送请求
+	httpReq, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_client]make request to send msg error")
+	}
+	httpReq.Header.Set("Authorization", "Bearer "+c.Key)
+	// req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15")
+
+	res := new(ListFilesResp)
+	httpResp, err := c.client.Do(httpReq)
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_client]send request to send msg error")
+	}
+	defer httpResp.Body.Close()
+	bodyBytes, err := io.ReadAll(httpResp.Body)
+	// fmt.Println(string(bodyBytes))
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_resp]read resp body failed")
+	}
+	err = json.Unmarshal(bodyBytes, res)
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_resp]unmarshal resp body failed")
+	}
+
+	if res.Error != nil {
+		return res, errors.New(res.Error.Message)
+	}
+
+	return res, err
+}
+
 func (c *AIClient) RetrieveFile(method string, fileID string) (*FileInfo, error) {
 	if fileID == "" {
 		return nil, errors.New("[ai_client]file id is empty")
@@ -148,6 +186,44 @@ func (c *AIClient) RetrieveFile(method string, fileID string) (*FileInfo, error)
 	return &res.FileInfo, err
 }
 
+func (c *AIClient) DeleteFile(fileID string) (*DeleteFileResp, error) {
+	if fileID == "" {
+		return nil, errors.New("[ai_client]file id is empty")
+	}
+	url := c.BaseURL + "/files/" + fileID
+
+	// 发送请求
+	httpReq, err := http.NewRequest("DELETE", url, nil)
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_client]make request to send msg error")
+	}
+	httpReq.Header.Set("Authorization", "Bearer "+c.Key)
+	// req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15")
+
+	res := new(DeleteFileResp)
+
+	httpResp, err := c.client.Do(httpReq)
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_client]send request to send msg error")
+	}
+	defer httpResp.Body.Close()
+
+	bodyBytes, err := io.ReadAll(httpResp.Body)
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_resp]read resp body failed")
+	}
+	err = json.Unmarshal(bodyBytes, res)
+	if err != nil {
+		return nil, errors.Wrap(err, "[ai_resp]unmarshal resp body failed")
+	}
+
+	if res.Error != nil {
+		return res, errors.New(res.Error.Message)
+	}
+
+	return res, err
+}
+
 func (c *AIClient) MakeChatReqBytes(param OpenAIChatParam) (reqByts []byte, err error) {
 	if param.Model == "" {
 		param.Model = "gpt-3.5-turbo-0301" //gpt-3.5-turbo or gpt-3.5-turbo-0301

example/qwen-vl/main.go

@@ -176,6 +176,16 @@ func main() {
 	}
 	fmt.Println(fileInfo2)
 
+	fileListResp, err := aiClient.ListFiles(openai.OpenAIListFilesParam{
+		Limit: 10,
+		Order: "desc",
+	})
+	if err != nil {
+		log.Fatalf("Error fileList request: %v", err)
+		return
+	}
+	fmt.Println(fileListResp)
+
 	// 请求ai大模型
 	chatParam = openai.OpenAIChatParam{
 		Model: "qwen-long",
@@ -217,4 +227,11 @@ func main() {
 	}
 	// Print the response content
 	fmt.Println(resChat.Choices[0].Message.Content)
+
+	delFileResp, err := aiClient.DeleteFile(fileInfo.ID)
+	if err != nil {
+		log.Fatalf("Error delete file request: %v", err)
+		return
+	}
+	fmt.Println(delFileResp)
 }

request.go

@@ -6,8 +6,10 @@ import (
 	"io"
 	"mime/multipart"
 	"net/http"
+	"net/url"
 	"os"
 	"path"
+	"strconv"
 
 	jsoniter "github.com/json-iterator/go"
 	"github.com/pkg/errors"
@@ -103,6 +105,30 @@ func (p *OpenAIFileCreateParam) buildMultipartWriter(body io.Writer) error {
 	return err
 }
 
+type OpenAIListFilesParam struct {
+	Purpose string
+	Limit   int
+	Order   string
+	After   string
+}
+
+func (o OpenAIListFilesParam) ToQuery() string {
+	vals := make(url.Values, 4)
+	if o.Purpose != "" {
+		vals.Set("purpose", o.Purpose)
+	}
+	if o.Limit != 0 {
+		vals.Set("limit", strconv.Itoa(o.Limit))
+	}
+	if o.Order != "" {
+		vals.Set("order", o.Order)
+	}
+	if o.After != "" {
+		vals.Set("after", o.After)
+	}
+	return vals.Encode()
+}
+
 // https://platform.openai.com/docs/api-reference/files/object
 type FileInfo struct {
 	ID            string `json:"id"`             // The file identifier, which can be referenced in the API endpoints.

response.go

@@ -53,6 +53,21 @@ type FileResp struct {
 	Error *AIError `json:"error,omitempty"`
 }
 
+type ListFilesResp struct {
+	Data     []FileInfo `json:"data"`
+	Object   string     `json:"object"`
+	NextPage int        `json:"next_page"`
+	HasMore  bool       `json:"has_more"`
+	Error    *AIError   `json:"error,omitempty"`
+}
+
+type DeleteFileResp struct {
+	ID      string `json:"id"`
+	Object  string `json:"object"`
+	Deleted bool   `json:"deleted"`
+	Error   *AIError
+}
+
 type FinishReason string
 
 // https://platform.openai.com/docs/api-reference/chat/object