admin/PrivHunterAI
1
0
Fork 0
mirror of https://github.com/Ed1s0nZ/PrivHunterAI.git synced 2025-05-05 10:17:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update aiapis.go

Browse Source
This commit is contained in:
公明 2025-04-25 10:36:37 +08:00 committed by GitHub
parent c3f93ee66c
commit 573ba870db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
AIAPIS/aiapis.go
View File

@ -88,11 +88,12 @@ func AIScan(model, aiurl, apikey, reqA, respA, respB, statusB string) (string, e
"statusB": statusB,
}
inputJSON, err := json.Marshal(input)
inputBytes, err := json.MarshalIndent(input, "", " ")
if err != nil {
fmt.Printf("Error marshaling input: %v\n", err)
return "", err
}
fmt.Println(string(inputBytes))
request := ChatCompletionRequest{
Model: model, // 根据实际模型名称修改
Messages: []Message{
@ -102,11 +103,11 @@ func AIScan(model, aiurl, apikey, reqA, respA, respB, statusB string) (string, e
},
{
Role: "user",
Content: string(inputJSON),
Content: string(inputBytes),
},
},
Temperature: 0.5,
MaxTokens: 1000,
Temperature: 0.2,
MaxTokens: 2500,
}
response, err := CreateChatCompletion(request, aiurl, apikey)