Changeset 0.22.7 (#149)
This commit is contained in:
@@ -47,8 +47,8 @@ type editRequest struct {
|
||||
|
||||
type regenerateRequest struct {
|
||||
Model string `json:"model,omitempty"`
|
||||
PresetID string `json:"preset_id,omitempty"`
|
||||
APIConfigID string `json:"provider_config_id,omitempty"`
|
||||
PersonaID string `json:"persona_id,omitempty"`
|
||||
ProviderConfigID string `json:"provider_config_id,omitempty"`
|
||||
MaxTokens int `json:"max_tokens,omitempty"`
|
||||
Temperature *float64 `json:"temperature,omitempty"`
|
||||
DisabledTools []string `json:"disabled_tools,omitempty"`
|
||||
@@ -417,34 +417,34 @@ func (h *MessageHandler) Regenerate(c *gin.Context) {
|
||||
|
||||
comp := NewCompletionHandler(h.vault, h.stores, h.hub, h.objStore, nil)
|
||||
|
||||
var presetSystemPrompt string
|
||||
var personaSystemPrompt string
|
||||
var personaID string
|
||||
model := req.Model
|
||||
apiConfigID := req.APIConfigID
|
||||
providerConfigID := req.ProviderConfigID
|
||||
temperature := req.Temperature
|
||||
maxTokens := req.MaxTokens
|
||||
|
||||
if req.PresetID != "" {
|
||||
preset := ResolvePreset(h.stores, req.PresetID, userID)
|
||||
if preset == nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "preset not found"})
|
||||
if req.PersonaID != "" {
|
||||
persona := ResolvePersona(h.stores, req.PersonaID, userID)
|
||||
if persona == nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "persona not found"})
|
||||
return
|
||||
}
|
||||
personaID = preset.ID
|
||||
personaID = persona.ID
|
||||
if model == "" {
|
||||
model = preset.BaseModelID
|
||||
model = persona.BaseModelID
|
||||
}
|
||||
if apiConfigID == "" && preset.ProviderConfigID != nil {
|
||||
apiConfigID = *preset.ProviderConfigID
|
||||
if providerConfigID == "" && persona.ProviderConfigID != nil {
|
||||
providerConfigID = *persona.ProviderConfigID
|
||||
}
|
||||
if temperature == nil && preset.Temperature != nil {
|
||||
temperature = preset.Temperature
|
||||
if temperature == nil && persona.Temperature != nil {
|
||||
temperature = persona.Temperature
|
||||
}
|
||||
if maxTokens == 0 && preset.MaxTokens != nil {
|
||||
maxTokens = *preset.MaxTokens
|
||||
if maxTokens == 0 && persona.MaxTokens != nil {
|
||||
maxTokens = *persona.MaxTokens
|
||||
}
|
||||
if preset.SystemPrompt != "" {
|
||||
presetSystemPrompt = preset.SystemPrompt
|
||||
if persona.SystemPrompt != "" {
|
||||
personaSystemPrompt = persona.SystemPrompt
|
||||
}
|
||||
}
|
||||
|
||||
@@ -459,7 +459,7 @@ func (h *MessageHandler) Regenerate(c *gin.Context) {
|
||||
|
||||
providerCfg, providerID, model, configID, providerScope, err := comp.resolveConfig(userID, channelID, completionRequest{
|
||||
Model: model,
|
||||
APIConfigID: apiConfigID,
|
||||
ProviderConfigID: providerConfigID,
|
||||
})
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
@@ -475,8 +475,8 @@ func (h *MessageHandler) Regenerate(c *gin.Context) {
|
||||
// Build LLM message array
|
||||
llmMessages := make([]providers.Message, 0, len(contextPath)+1)
|
||||
|
||||
if presetSystemPrompt != "" {
|
||||
llmMessages = append(llmMessages, providers.Message{Role: "system", Content: presetSystemPrompt})
|
||||
if personaSystemPrompt != "" {
|
||||
llmMessages = append(llmMessages, providers.Message{Role: "system", Content: personaSystemPrompt})
|
||||
} else {
|
||||
var systemPrompt *string
|
||||
_ = database.DB.QueryRow(database.Q(`SELECT system_prompt FROM channels WHERE id = $1`), channelID).Scan(&systemPrompt)
|
||||
|
||||
Reference in New Issue
Block a user