5632bf88f8
- 新增config.yaml和config.go用于管理配置 - 重构chatmodel.go使用配置初始化模型 - 修改GenerateChatMessage和ChatStream函数签名添加context参数 - 更新main.go加载配置并初始化聊天模型 - 优化错误处理和日志输出
23 lines
468 B
Go
23 lines
468 B
Go
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
|
|
|
|
}
|