mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-10 18:05:48 +02:00
24 lines
847 B
Go
24 lines
847 B
Go
package settings
|
|
|
|
import "github.com/sashabaranov/go-openai"
|
|
|
|
type OpenAI struct {
|
|
BaseUrl string `json:"base_url" binding:"omitempty,url"`
|
|
Token string `json:"token" binding:"omitempty,safety_text"`
|
|
Proxy string `json:"proxy" binding:"omitempty,url"`
|
|
Model string `json:"model" binding:"omitempty,safety_text"`
|
|
APIType string `json:"api_type" binding:"omitempty,oneof=OPEN_AI AZURE"`
|
|
EnableCodeCompletion bool `json:"enable_code_completion" binding:"omitempty"`
|
|
CodeCompletionModel string `json:"code_completion_model" binding:"omitempty,safety_text"`
|
|
}
|
|
|
|
var OpenAISettings = &OpenAI{
|
|
APIType: string(openai.APITypeOpenAI),
|
|
}
|
|
|
|
func (o *OpenAI) GetCodeCompletionModel() string {
|
|
if o.CodeCompletionModel == "" {
|
|
return o.Model
|
|
}
|
|
return o.CodeCompletionModel
|
|
}
|