Merge pull request #98 from enhay/bugfix/openai_draw_config

whyiyhw · web-flow · commit 397063597041 · 2023-11-21T12:01:41.000+08:00
fix draw host config
diff --git a/chat/common/openai/draw.go b/chat/common/openai/draw.go
@@ -24,9 +24,9 @@ type Draw struct {
 	Proxy  string
 }
 
-func NewOpenaiDraw(key, proxy string) *Draw {
+func NewOpenaiDraw(host, key, proxy string) *Draw {
 	return &Draw{
-		Host:   "https://api.openai.com",
+		Host:   host,
 		APIKey: key,
 		Proxy:  proxy,
 	}
diff --git a/chat/service/chat/api/internal/logic/chatlogic.go b/chat/service/chat/api/internal/logic/chatlogic.go
@@ -820,6 +820,7 @@ func (p CommendDraw) exec(l *ChatLogic, req *types.ChatReq) bool {
 					)
 				} else if l.svcCtx.Config.Draw.Company == draw.OPENAI {
 					d = openai.NewOpenaiDraw(
+						l.svcCtx.Config.Draw.OpenAi.Host,
 						l.svcCtx.Config.Draw.OpenAi.Key,
 						l.svcCtx.Config.Draw.OpenAi.Proxy,
 					)

Original file line number	Diff line number	Diff line change
`@@ -24,9 +24,9 @@ type Draw struct {`
`24`	`24`	`Proxy string`
`25`	`25`	`}`
`26`	`26`
`27`		`-func NewOpenaiDraw(key, proxy string) *Draw {`
	`27`	`+func NewOpenaiDraw(host, key, proxy string) *Draw {`
`28`	`28`	`return &Draw{`
`29`		`- Host: "https://api.openai.com",`
	`29`	`+ Host: host,`
`30`	`30`	`APIKey: key,`
`31`	`31`	`Proxy: proxy,`
`32`	`32`	`}`
Original file line number	Diff line number	Diff line change
`@@ -820,6 +820,7 @@ func (p CommendDraw) exec(l ChatLogic, req types.ChatReq) bool {`
`820`	`820`	`)`
`821`	`821`	`} else if l.svcCtx.Config.Draw.Company == draw.OPENAI {`
`822`	`822`	`d = openai.NewOpenaiDraw(`
	`823`	`+ l.svcCtx.Config.Draw.OpenAi.Host,`
`823`	`824`	`l.svcCtx.Config.Draw.OpenAi.Key,`
`824`	`825`	`l.svcCtx.Config.Draw.OpenAi.Proxy,`
`825`	`826`	`)`