mirror of
https://github.com/Ed1s0nZ/PrivHunterAI.git
synced 2025-09-17 20:41:37 +00:00
Delete kimi.go
This commit is contained in:
parent
985e6f751c
commit
bf6693be57
97
kimi.go
97
kimi.go
@ -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
|
|
||||||
|
|
||||||
}
|
|
Loading…
x
Reference in New Issue
Block a user