mirror of
https://github.com/coaidev/coai.git
synced 2025-05-20 05:20:15 +09:00
34 lines
574 B
Go
34 lines
574 B
Go
package palm2
|
|
|
|
import (
|
|
"chat/utils"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type ChatInstance struct {
|
|
Endpoint string
|
|
ApiKey string
|
|
}
|
|
|
|
func (c *ChatInstance) GetApiKey() string {
|
|
return c.ApiKey
|
|
}
|
|
|
|
func (c *ChatInstance) GetEndpoint() string {
|
|
return c.Endpoint
|
|
}
|
|
|
|
func NewChatInstance(endpoint string, apiKey string) *ChatInstance {
|
|
return &ChatInstance{
|
|
Endpoint: endpoint,
|
|
ApiKey: apiKey,
|
|
}
|
|
}
|
|
|
|
func NewChatInstanceFromConfig() *ChatInstance {
|
|
return NewChatInstance(
|
|
viper.GetString("palm2.endpoint"),
|
|
utils.GetRandomKey(viper.GetString("palm2.apikey")),
|
|
)
|
|
}
|