🎉 初始化项目
This commit is contained in:
59
server/model/app/ai_preset.go
Normal file
59
server/model/app/ai_preset.go
Normal file
@@ -0,0 +1,59 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"git.echol.cn/loser/ai_proxy/server/global"
|
||||
)
|
||||
|
||||
// AiPreset AI预设模板
|
||||
type AiPreset struct {
|
||||
global.GVA_MODEL
|
||||
UserID uint `json:"userId" gorm:"comment:用户ID;index"`
|
||||
Name string `json:"name" gorm:"comment:预设名称;size:200;not null"`
|
||||
Description string `json:"description" gorm:"comment:预设描述;type:text"`
|
||||
Prompts Prompts `json:"prompts" gorm:"comment:提示词数组;type:jsonb;not null"`
|
||||
RegexScripts RegexScripts `json:"regexScripts" gorm:"comment:正则脚本;type:jsonb"`
|
||||
Temperature float64 `json:"temperature" gorm:"comment:温度;default:1.0"`
|
||||
TopP float64 `json:"topP" gorm:"comment:TopP;default:0.9"`
|
||||
MaxTokens int `json:"maxTokens" gorm:"comment:最大Token数;default:4096"`
|
||||
FrequencyPenalty float64 `json:"frequencyPenalty" gorm:"comment:频率惩罚;default:0"`
|
||||
PresencePenalty float64 `json:"presencePenalty" gorm:"comment:存在惩罚;default:0"`
|
||||
StreamEnabled bool `json:"streamEnabled" gorm:"comment:是否启用流式;default:true"`
|
||||
IsDefault bool `json:"isDefault" gorm:"comment:是否默认;default:false"`
|
||||
IsPublic bool `json:"isPublic" gorm:"comment:是否公开;default:false"`
|
||||
}
|
||||
|
||||
func (AiPreset) TableName() string {
|
||||
return "ai_presets"
|
||||
}
|
||||
|
||||
// Prompt 提示词结构
|
||||
type Prompt struct {
|
||||
Name string `json:"name"`
|
||||
SystemPrompt bool `json:"system_prompt"`
|
||||
Role string `json:"role"`
|
||||
Content string `json:"content"`
|
||||
Identifier string `json:"identifier"`
|
||||
InjectionPosition int `json:"injection_position"`
|
||||
InjectionDepth int `json:"injection_depth"`
|
||||
InjectionOrder int `json:"injection_order"`
|
||||
InjectionTrigger []string `json:"injection_trigger"`
|
||||
Marker bool `json:"marker"`
|
||||
ForbidOverrides bool `json:"forbid_overrides"`
|
||||
}
|
||||
|
||||
// RegexScript 正则脚本
|
||||
type RegexScript struct {
|
||||
ID string `json:"id"`
|
||||
ScriptName string `json:"scriptName"`
|
||||
FindRegex string `json:"findRegex"`
|
||||
ReplaceString string `json:"replaceString"`
|
||||
TrimStrings []string `json:"trimStrings"`
|
||||
Placement []int `json:"placement"`
|
||||
Disabled bool `json:"disabled"`
|
||||
MarkdownOnly bool `json:"markdownOnly"`
|
||||
PromptOnly bool `json:"promptOnly"`
|
||||
RunOnEdit bool `json:"runOnEdit"`
|
||||
SubstituteRegex int `json:"substituteRegex"`
|
||||
MinDepth *int `json:"minDepth"`
|
||||
MaxDepth *int `json:"maxDepth"`
|
||||
}
|
||||
Reference in New Issue
Block a user