package main

import (
	"context"
	"enio_meowrain/chat"
	"enio_meowrain/config"
	"log"
)

func main() {
	// 加载配置
	cfg, err := config.LoadConfig("config.yaml")
	if err != nil {
		log.Fatalf("无法加载配置文件: %v", err)
	}

	ctx := context.Background()
	model, err := chat.CreateChatModel(ctx, cfg)
	if err != nil {
		log.Fatalf("创建聊天模型失败: %v", err)
	}

	question := "我写的代码老是出bug，唉，搞得我很焦虑"
	role := "心理咨询师"
	style := "温柔可爱"

	// 使用 ChatStream
	if err := chat.ChatStream(model, ctx, role, style, question); err != nil {
		log.Printf("聊天过程出错: %v", err)
	}
}