mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-11 02:15:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat(wip): code completion with llm

Browse source
This commit is contained in:
Jacky 2025-04-15 21:54:51 +08:00
parent 63fb823344
commit a57748a432
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
22 changed files with 623 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

19
settings/openai.go
View file

@ -3,13 +3,22 @@ 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"`
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
}