Neo initialization to enable ‘moapi’ connector and default model support

This commit is contained in:
Max 2024-11-05 16:06:32 +08:00
parent 16234df9d6
commit 61a292c7a6
3 changed files with 26 additions and 5 deletions

View file

@ -110,8 +110,19 @@ func (cmd *Command) save() error {
// NewAI create a new AI // NewAI create a new AI
func (cmd *Command) newAI() (aigc.AI, error) { func (cmd *Command) newAI() (aigc.AI, error) {
if cmd.Connector == "" { if cmd.Connector == "" || strings.HasPrefix(cmd.Connector, "moapi") {
return nil, fmt.Errorf("%s connector is required", cmd.ID) model := "gpt-3.5-turbo"
if strings.HasPrefix(cmd.Connector, "moapi:") {
model = strings.TrimPrefix(cmd.Connector, "moapi:")
}
ai, err := openai.NewMoapi(model)
if err != nil {
return nil, err
}
cmd.AI = ai
return cmd.AI, nil
} }
conn, err := connector.Select(cmd.Connector) conn, err := connector.Select(cmd.Connector)

View file

@ -198,8 +198,17 @@ func (driver *Memory) GetCommands() ([]Command, error) {
// NewAI create a new AI // NewAI create a new AI
func (driver *Memory) newAI() (aigc.AI, error) { func (driver *Memory) newAI() (aigc.AI, error) {
if driver.model == "" { if driver.model == "" || strings.HasPrefix(driver.model, "moapi") {
return nil, fmt.Errorf("%s connector is required", driver.model) model := "gpt-3.5-turbo"
if strings.HasPrefix(driver.model, "moapi:") {
model = strings.TrimPrefix(driver.model, "moapi:")
}
ai, err := openai.NewMoapi(model)
if err != nil {
return nil, err
}
return ai, nil
} }
conn, err := connector.Select(driver.model) conn, err := connector.Select(driver.model)

View file

@ -63,7 +63,8 @@ func Load(cfg config.Config) error {
if parser == "" || parser == "default" { if parser == "" || parser == "default" {
parser = setting.Connector parser = setting.Connector
} }
store, err := driver.NewMemory(setting.Command.Parser, nil)
store, err := driver.NewMemory(parser, nil)
if err != nil { if err != nil {
return err return err
} }