llvy.ltd/casdoor
1
0
Fork 0
mirror of https://github.com/casdoor/casdoor.git synced 2025-07-04 05:10:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ai package

Browse Source
This commit is contained in:
Yang Luo
2023-05-01 17:19:45 +08:00
parent 48c5bd942c
commit eea2e1d271
7 changed files with 149 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
object/chat.go
View File

@ -121,6 +121,13 @@ func UpdateChat(id string, chat *Chat) bool {
}
func AddChat(chat *Chat) bool {
if chat.Type == "AI" && chat.User2 == "" {
provider := getDefaultAiProvider()
if provider != nil {
chat.User2 = provider.Name
}
}
affected, err := adapter.Engine.Insert(chat)
if err != nil {
panic(err)

4
object/provider.go
View File

@ -177,8 +177,8 @@ func GetProvider(id string) *Provider {
return getProvider(owner, name)
}
func GetDefaultCaptchaProvider() *Provider {
provider := Provider{Owner: "admin", Category: "Captcha"}
func getDefaultAiProvider() *Provider {
provider := Provider{Owner: "admin", Category: "AI"}
existed, err := adapter.Engine.Get(&provider)
if err != nil {
panic(err)