1f4e9e5a16
Add the SelectModel command to the DSL, allowing the selection of a specific model. This command takes a model name as input and sets the AI to use that model. It returns a success message with a status code of 200 if the selection is successful.
41 lines
1.5 KiB
Go
41 lines
1.5 KiB
Go
package neo
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/yaoapp/yao/aigc"
|
|
"github.com/yaoapp/yao/neo/conversation"
|
|
)
|
|
|
|
// DSL AI assistant
|
|
type DSL struct {
|
|
ID string `json:"-" yaml:"-"`
|
|
Name string `json:"name,omitempty"`
|
|
Use string `json:"use,omitempty"`
|
|
Guard string `json:"guard,omitempty"`
|
|
Connector string `json:"connector"`
|
|
ConversationSetting conversation.Setting `json:"conversation" yaml:"conversation"`
|
|
Option map[string]interface{} `json:"option"`
|
|
Prepare string `json:"prepare,omitempty"`
|
|
Write string `json:"write,omitempty"`
|
|
Prompts []aigc.Prompt `json:"prompts,omitempty"`
|
|
Allows []string `json:"allows,omitempty"`
|
|
Command Command `json:"command,omitempty"`
|
|
Models []string `json:"models,omitempty"`
|
|
AI aigc.AI `json:"-" yaml:"-"`
|
|
Conversation conversation.Conversation `json:"-" yaml:"-"`
|
|
GuardHandlers []gin.HandlerFunc `json:"-" yaml:"-"`
|
|
}
|
|
|
|
// Answer the answer interface
|
|
type Answer interface {
|
|
Stream(func(w io.Writer) bool) bool
|
|
Status(code int)
|
|
Header(key, value string)
|
|
}
|
|
|
|
// Command setting
|
|
type Command struct {
|
|
Parser string `json:"parser,omitempty"`
|
|
}
|