101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"learning-assistant/internal/ai"
|
|
"learning-assistant/internal/tui"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func main() {
|
|
// 加载配置
|
|
config, err := loadConfig("../config/config.yaml")
|
|
if err != nil {
|
|
fmt.Printf("加载配置文件失败: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 获取API密钥
|
|
// apiKey := getAPIKey()
|
|
// if apiKey == "" {
|
|
// fmt.Println("错误: 未设置API密钥")
|
|
// fmt.Println("请设置环境变量 SILICON_API_KEY 或在配置文件中指定")
|
|
// os.Exit(1)
|
|
// }
|
|
|
|
// 创建AI客户端
|
|
client := ai.NewClient(config.API)
|
|
|
|
// 创建提示词管理器
|
|
promptManager := ai.NewPromptManager()
|
|
|
|
// 读取用户输入
|
|
fmt.Println("🎓 学习助手")
|
|
fmt.Println("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━")
|
|
fmt.Println()
|
|
fmt.Print("请输入您想学习的知识点: ")
|
|
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
if !scanner.Scan() {
|
|
return
|
|
}
|
|
|
|
knowledge := scanner.Text()
|
|
if knowledge == "" {
|
|
fmt.Println("输入不能为空")
|
|
return
|
|
}
|
|
|
|
// 创建模型
|
|
model := tui.NewModel(client, promptManager)
|
|
model.SetInput(knowledge)
|
|
|
|
// 设置信号处理
|
|
sigChan := make(chan os.Signal, 1)
|
|
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
// 创建程序
|
|
p := tea.NewProgram(&model, tea.WithAltScreen())
|
|
|
|
// 启动程序
|
|
if _, err := p.Run(); err != nil {
|
|
fmt.Printf("启动失败: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
// Config 配置
|
|
type Config struct {
|
|
API ai.Config
|
|
}
|
|
|
|
func loadConfig(path string) (*Config, error) {
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var config Config
|
|
if err := yaml.Unmarshal(data, &config); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &config, nil
|
|
}
|
|
|
|
// getAPIKey 获取API密钥
|
|
func getAPIKey() string {
|
|
// 优先从环境变量获取
|
|
if key := os.Getenv("SILICON_API_KEY"); key != "" {
|
|
return key
|
|
}
|
|
|
|
return ""
|
|
}
|