diff --git a/neo/load.go b/neo/load.go index 5742e12c..7224a7e8 100644 --- a/neo/load.go +++ b/neo/load.go @@ -23,6 +23,7 @@ func Load(cfg config.Config) error { Prompts: []aigc.Prompt{}, Option: map[string]interface{}{}, Allows: []string{}, + Command: Command{Parser: ""}, ConversationSetting: conversation.Setting{ Table: "yao_neo_conversation", Connector: "default", @@ -58,7 +59,11 @@ func Load(cfg config.Config) error { } // Command Setting - store, err := driver.NewMemory("gpt-3_5-turbo", nil) + parser := setting.Command.Parser + if parser == "" || parser == "default" { + parser = setting.Connector + } + store, err := driver.NewMemory(setting.Command.Parser, nil) if err != nil { return err } diff --git a/neo/types.go b/neo/types.go index f020ee93..ad6aa02c 100644 --- a/neo/types.go +++ b/neo/types.go @@ -19,6 +19,7 @@ type DSL struct { Prepare string `json:"prepare,omitempty"` Prompts []aigc.Prompt `json:"prompts,omitempty"` Allows []string `json:"allows,omitempty"` + Command Command `json:"command,omitempty"` AI aigc.AI `json:"-" yaml:"-"` Conversation conversation.Conversation `json:"-" yaml:"-"` } @@ -29,3 +30,8 @@ type Answer interface { Status(code int) Header(key, value string) } + +// Command setting +type Command struct { + Parser string `json:"parser,omitempty"` +}