diff --git a/kimi.go b/kimi.go deleted file mode 100644 index 3ae1c78..0000000 --- a/kimi.go +++ /dev/null @@ -1,97 +0,0 @@ -package main - -import ( - "bytes" - "encoding/json" - "errors" - "fmt" - "net/http" -) - -// 定义请求参数结构体 -type ChatCompletionRequestKimi struct { - Model string `json:"model"` - Messages []MessageKimi `json:"messages"` - Temperature float32 `json:"temperature"` -} - -// 定义消息结构体 -type MessageKimi struct { - Role string `json:"role"` - Content string `json:"content"` -} - -// 定义响应结果结构体 -type ChatCompletionResponseKimi struct { - Choices []ChoiceKimi `json:"choices"` -} - -type ChoiceKimi struct { - Message MessageKimi `json:"message"` -} - -func kimi(respA, respB string) (string, error) { - // 设置 API Key 和请求 URL - apiURL := "https://api.moonshot.cn/v1/chat/completions" - - // 创建请求参数 - request := ChatCompletionRequestKimi{ - Model: "moonshot-v1-8k", - Messages: []MessageKimi{ - { - Role: "system", - Content: prompt, - }, - { - Role: "user", - Content: "A:" + respA + "B:" + respB, - }, - }, - Temperature: 0.3, - } - - // 将请求参数编码为 JSON - jsonData, err := json.Marshal(request) - if err != nil { - fmt.Println("JSON 编码失败:", err) - return "", err - } - - // 创建 HTTP 请求 - req, err := http.NewRequest("POST", apiURL, bytes.NewBuffer(jsonData)) - if err != nil { - fmt.Println("创建请求失败:", err) - return "", err - } - req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", apiKeyKimi)) - req.Header.Set("Content-Type", "application/json") - - // 发送请求 - client := &http.Client{} - resp, err := client.Do(req) - if err != nil { - fmt.Println("发送请求失败:", err) - return "", err - } - defer resp.Body.Close() - - // 解析响应结果 - var response ChatCompletionResponseKimi - err = json.NewDecoder(resp.Body).Decode(&response) - if err != nil { - fmt.Println("解析响应失败:", err) - return "", err - } - - // 输出回复内容 - if len(response.Choices) > 0 { - result := response.Choices[0].Message.Content - return result, nil - } else { - // 处理 Choices 为空的情况,例如返回一个默认值或错误 - return "", errors.New("error: choices is empty") - } - // result := response.Choices[0].Message.Content - // return result, nil - -}