mirror of
https://github.com/coaidev/coai.git
synced 2025-05-19 13:00:14 +09:00
34 lines
736 B
Go
34 lines
736 B
Go
package skylark
|
|
|
|
import (
|
|
factory "chat/adapter/common"
|
|
"chat/globals"
|
|
|
|
"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
|
|
)
|
|
|
|
type ChatInstance struct {
|
|
Instance *arkruntime.Client
|
|
isFirstReasoning bool
|
|
isReasonOver bool
|
|
}
|
|
|
|
func NewChatInstance(endpoint, apiKey string) *ChatInstance {
|
|
//https://ark.cn-beijing.volces.com/api/v3
|
|
instance := arkruntime.NewClientWithApiKey(apiKey, arkruntime.WithBaseUrl(endpoint))
|
|
return &ChatInstance{
|
|
Instance: instance,
|
|
isFirstReasoning: true,
|
|
isReasonOver: false,
|
|
}
|
|
}
|
|
|
|
func NewChatInstanceFromConfig(conf globals.ChannelConfig) factory.Factory {
|
|
params := conf.SplitRandomSecret(1)
|
|
|
|
return NewChatInstance(
|
|
conf.GetEndpoint(),
|
|
params[0],
|
|
)
|
|
}
|