package chat

import (
	"context"
	"enio_meowrain/config"

	"github.com/cloudwego/eino-ext/components/model/openai"
)

func CreateChatModel(ctx context.Context, cfg *config.Config) (*openai.ChatModel, error) {
	chatModel, err := openai.NewChatModel(ctx, &openai.ChatModelConfig{
		BaseURL: cfg.OpenAI.BaseURL,
		Model:   cfg.OpenAI.Model,
		APIKey:  cfg.OpenAI.APIKey,
		Timeout: cfg.OpenAI.Timeout,
	})
	if err != nil {
		return nil, err
	}
	return chatModel, nil

}