diff --git a/kimi.go b/kimi.go index 1d1196d..6768bc5 100644 --- a/kimi.go +++ b/kimi.go @@ -9,25 +9,25 @@ import ( ) // 定义请求参数结构体 -type ChatCompletionRequest struct { - Model string `json:"model"` - Messages []Message `json:"messages"` - Temperature float32 `json:"temperature"` +type ChatCompletionRequestKimi struct { + Model string `json:"model"` + Messages []MessageKimi `json:"messages"` + Temperature float32 `json:"temperature"` } // 定义消息结构体 -type Message struct { +type MessageKimi struct { Role string `json:"role"` Content string `json:"content"` } // 定义响应结果结构体 -type ChatCompletionResponse struct { - Choices []Choice `json:"choices"` +type ChatCompletionResponseKimi struct { + Choices []ChoiceKimi `json:"choices"` } -type Choice struct { - Message Message `json:"message"` +type ChoiceKimi struct { + Message MessageKimi `json:"message"` } func kimi(respA, respB string) (string, error) { @@ -35,9 +35,9 @@ func kimi(respA, respB string) (string, error) { apiURL := "https://api.moonshot.cn/v1/chat/completions" // 创建请求参数 - request := ChatCompletionRequest{ + request := ChatCompletionRequestKimi{ Model: "moonshot-v1-8k", - Messages: []Message{ + Messages: []MessageKimi{ { Role: "system", Content: "{\"role\": \"你是一个AI,负责通过比较两个HTTP响应数据包来检测潜在的越权行为,并自行做出判断。\",\"inputs\": {\"responseA\": \"账号A请求某接口的响应。\",\"responseB\": \"使用账号B的Cookie重放请求的响应。\"},\"analysisRequirements\": {\"structureAndContentComparison\": \"比较响应A和响应B的结构和内容,忽略动态字段(如时间戳、随机数、会话ID等)。\",\"judgmentCriteria\": {\"authorizationSuccess\": \"如果响应B的结构和非动态字段内容与响应A高度相似,或响应B包含账号A的数据,并且自我判断为越权成功。\",\"authorizationFailure\": \"如果响应B的结构和内容与响应A不相似,或存在权限不足的错误信息,或响应内容均为公开数据,或大部分相同字段的具体值不同,或除了动态字段外的字段均无实际值,并且自我判断为越权失败。\",\"unknown\": \"其他情况,或无法确定是否存在越权,并且自我判断为无法确定。\"}},\"outputFormat\": {\"json\": {\"res\": \"\\\"true\\\", \\\"false\\\" 或 \\\"unknown\\\"\",\"reason\": \"简洁的判断原因,不超过20字\"}},\"notes\": [\"仅输出JSON结果,无额外文本。\",\"确保JSON格式正确,便于后续处理。\",\"保持客观,仅根据响应内容进行分析。\"],\"process\": [\"接收并理解响应A和响应B。\",\"分析响应A和响应B,忽略动态字段。\",\"基于响应的结构、内容和相关性进行自我判断,包括但不限于:\",\"- 识别响应中可能的敏感数据或权限信息。\",\"- 评估响应与预期结果之间的一致性。\",\"- 确定是否存在明显的越权迹象。\",\"输出指定格式的JSON结果,包括判断和判断原因。\"]}", @@ -63,7 +63,7 @@ func kimi(respA, respB string) (string, error) { fmt.Println("创建请求失败:", err) return "", err } - req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", apiKey)) + req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", apiKeyKimi)) req.Header.Set("Content-Type", "application/json") // 发送请求 @@ -76,7 +76,7 @@ func kimi(respA, respB string) (string, error) { defer resp.Body.Close() // 解析响应结果 - var response ChatCompletionResponse + var response ChatCompletionResponseKimi err = json.NewDecoder(resp.Body).Decode(&response) if err != nil { fmt.Println("解析响应失败:", err)