19 lines
411 B
Go
19 lines
411 B
Go
|
package agent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"eino-test/config"
|
||
|
|
|
||
|
|
"github.com/cloudwego/eino-ext/components/model/openai"
|
||
|
|
"github.com/cloudwego/eino/components/model"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewChatModel(ctx context.Context, cfg *config.Config) (model.ToolCallingChatModel, error) {
|
||
|
|
return openai.NewChatModel(ctx, &openai.ChatModelConfig{
|
||
|
|
BaseURL: cfg.OpenAIBaseURL,
|
||
|
|
APIKey: cfg.OpenAIAPIKey,
|
||
|
|
Model: cfg.OpenAIModel,
|
||
|
|
})
|
||
|
|
}
|