18 lines
356 B
Go
18 lines
356 B
Go
package service
|
|
|
|
import "github.com/sid-sun/openwebui-bot/pkg/bot/store"
|
|
|
|
func getRole(from string) string {
|
|
if from == store.BotUsername {
|
|
return "assistant"
|
|
}
|
|
return "user"
|
|
}
|
|
|
|
func getSystemPrompt(chatID int64) string {
|
|
if store.SystemPromptStore[chatID] == "" {
|
|
return "You are a friendly assistant"
|
|
}
|
|
return store.SystemPromptStore[chatID]
|
|
}
|