opencode/yao
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: Integrate AIGC & OpenAI with Moapi service

Browse source
This commit is contained in:
Max 2024-07-30 16:59:13 +08:00
parent db397f8cda
commit 953d592669
3 changed files with 18 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aigc/aigc.go
View file

@ -99,7 +99,7 @@ func (ai *DSL) newAI() (AI, error) {
if ai.Connector == "" || strings.HasPrefix(ai.Connector, "moapi") { if ai.Connector == "" || strings.HasPrefix(ai.Connector, "moapi") {
model := "gpt-3.5-turbo" model := "gpt-3.5-turbo"
if ai.Connector != "" { if strings.HasPrefix(ai.Connector, "moapi:") {
model = strings.TrimPrefix(ai.Connector, "moapi:") model = strings.TrimPrefix(ai.Connector, "moapi:")
} }

10
neo/neo.go
View file

@ -343,12 +343,16 @@ func (neo *DSL) write(msg *message.JSON, w io.Writer, ctx command.Context, messa
args := []interface{}{ctx, messages, msg, string(content)} args := []interface{}{ctx, messages, msg, string(content)}
p, err := process.Of(neo.Write, args...) p, err := process.Of(neo.Write, args...)
if err != nil { if err != nil {
return err log.Error("Neo custom write process error: %s", err.Error())
msg.Write(w)
return nil
} }
res, err := p.WithSID(ctx.Sid).Exec() res, err := p.WithSID(ctx.Sid).Exec()
if err != nil { if err != nil {
return err log.Error("Neo custom write error: %s", err.Error())
msg.Write(w)
return nil
} }
if res == nil { if res == nil {
@ -549,7 +553,7 @@ func (neo *DSL) newAI() error {
if neo.Connector == "" || strings.HasPrefix(neo.Connector, "moapi") { if neo.Connector == "" || strings.HasPrefix(neo.Connector, "moapi") {
model := "gpt-3.5-turbo" model := "gpt-3.5-turbo"
if neo.Connector != "" { if strings.HasPrefix(neo.Connector, "moapi:") {
model = strings.TrimPrefix(neo.Connector, "moapi:") model = strings.TrimPrefix(neo.Connector, "moapi:")
} }

10
openai/openai.go
View file

@ -35,6 +35,16 @@ type OpenAI struct {
// New create a new OpenAI instance by connector id // New create a new OpenAI instance by connector id
func New(id string) (*OpenAI, error) { func New(id string) (*OpenAI, error) {
// Moapi integration
if id == "" || strings.HasPrefix(id, "moapi") {
model := "gpt-3.5-turbo"
if strings.HasPrefix(id, "moapi:") {
model = strings.TrimPrefix(id, "moapi:")
}
return NewMoapi(model)
}
c, err := connector.Select(id) c, err := connector.Select(id)
if err != nil { if err != nil {
return nil, err return nil, err