ScuroNeko/YaeMikoBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

some fixes

Browse Source
This commit is contained in:
ScuroNeko
2026-01-19 21:56:41 +03:00
parent ad6e52890b
commit 65acbab94e
9 changed files with 88 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
utils/ai/openai.go
View File

@@ -89,12 +89,12 @@ func NewOpenAIAPI(baseURL, token, model string) *OpenAIAPI {
type CreateCompletionReq struct {
Model string `json:"model"`
Messages []Message `json:"messages"`
Verbosity string `json:"verbosity"`
Temperature float32 `json:"temperature"`
PresencePenalty int `json:"presence_penalty"`
FrequencyPenalty int `json:"frequency_penalty"`
TopP int `json:"top_p"`
MaxCompletionTokens int `json:"max_completition_tokens"`
Verbosity string `json:"verbosity,omitempty"`
Temperature float32 `json:"temperature,omitempty"`
PresencePenalty int `json:"presence_penalty,omitempty"`
FrequencyPenalty int `json:"frequency_penalty,omitempty"`
TopP int `json:"top_p,omitempty"`
MaxCompletionTokens int `json:"max_completition_tokens,omitempty"`
}
func (o *OpenAIAPI) CreateCompletion(request CreateCompletionReq) (*OpenAIResponse, error) {
@@ -164,6 +164,13 @@ func (o *OpenAIAPI) CompressChat(history []Message) (*OpenAIResponse, error) {
if err != nil {
return nil, err
}
if resp.StatusCode == 504 || resp.StatusCode == 400 {
time.Sleep(5 * time.Second)
resp, err = http.DefaultClient.Do(req)
if err != nil {
return nil, err
}
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {