admin/PrivHunterAI
1
0
Fork 0
mirror of https://github.com/Ed1s0nZ/PrivHunterAI.git synced 2025-06-21 18:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update aiapis.go

Browse Source
This commit is contained in:
公明 2025-04-09 11:16:47 +08:00 committed by GitHub
parent fa25bd9946
commit 1fd14b5b14
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
AIAPIS/aiapis.go
View File

@ -81,6 +81,19 @@ func CreateChatCompletion(request ChatCompletionRequest, aiurl string, aiapikey
}
func AIScan(model, aiurl, apikey, reqA, respA, respB, statusB string) (string, error) {
input := map[string]string{
"reqA": reqA,
"responseA": respA,
"responseB": respB,
"statusB": statusB,
}
inputJSON, err := json.Marshal(input)
if err != nil {
fmt.Printf("Error marshaling input: %v\n", err)
return "", err
}
fmt.Println(string(inputJSON))
request := ChatCompletionRequest{
Model: model, // 根据实际模型名称修改
Messages: []Message{
@ -90,7 +103,7 @@ func AIScan(model, aiurl, apikey, reqA, respA, respB, statusB string) (string, e
},
{
Role: "user",
Content: "reqA:" + reqA + "\n" + "responseA:" + respA + "\n" + "responseB:" + respB + "\n" + "statusB:" + statusB,
Content: string(inputJSON),
},
},
Temperature: 0.7,
@ -104,8 +117,6 @@ func AIScan(model, aiurl, apikey, reqA, respA, respB, statusB string) (string, e
}
if len(response.Choices) > 0 {
// fmt.Println("Response:")
// fmt.Println(response.Choices[0].Message.Content)
return response.Choices[0].Message.Content, nil
} else {
fmt.Println("No response received")