up

2026-08-13 18:54:34 +08:00
parent 3288df482e
commit 7b03f54bb5
6 changed files with 235 additions and 2 deletions
@@ -29,6 +29,9 @@
 .env.*
 !.env.example

+# Runtime data
+/data/
+
 # Go workspace files
 go.work
 go.work.sum
@@ -1,3 +1,15 @@
 module myaibot

 go 1.26.4
+
+require (
+	github.com/openai/openai-go v1.12.0
+	gopkg.in/yaml.v3 v3.0.1
+)
+
+require (
+	github.com/tidwall/gjson v1.14.4 // indirect
+	github.com/tidwall/match v1.1.1 // indirect
+	github.com/tidwall/pretty v1.2.1 // indirect
+	github.com/tidwall/sjson v1.2.5 // indirect
+)
@@ -0,0 +1,16 @@
+github.com/openai/openai-go v1.12.0 h1:NBQCnXzqOTv5wsgNC36PrFEiskGfO5wccfCWDo9S1U0=
+github.com/openai/openai-go v1.12.0/go.mod h1:g461MYGXEXBVdV5SaR/5tNzNbSfwTBBefwc+LlDCK0Y=
+github.com/tidwall/gjson v1.14.2/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
+github.com/tidwall/gjson v1.14.4 h1:uo0p8EbA09J7RQaflQ1aBRffTR7xedD2bcIVSYxLnkM=
+github.com/tidwall/gjson v1.14.4/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
+github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
+github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
+github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
+github.com/tidwall/pretty v1.2.1 h1:qjsOFOWWQl+N3RsoF5/ssm1pHmJJwhjlSbZ51I6wMl4=
+github.com/tidwall/pretty v1.2.1/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
+github.com/tidwall/sjson v1.2.5 h1:kLy8mja+1c9jlljvWTlSazM7cKDRfJuR/bOJhcY5NcY=
+github.com/tidwall/sjson v1.2.5/go.mod h1:Fvgq9kS/6ociJEDnK0Fk1cpYF4FIW6ZF7LAe+6jwd28=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
@@ -0,0 +1,59 @@
+package bot
+
+import (
+	"context"
+	"errors"
+	"fmt"
+
+	"github.com/openai/openai-go"
+	"github.com/openai/openai-go/option"
+	"myaibot/internal/config"
+)
+
+const maxHistory = 20
+
+type Bot struct {
+	client  *openai.Client
+	cfg     *config.Config
+	history []openai.ChatCompletionMessageParamUnion
+}
+
+func New(cfg *config.Config) *Bot {
+	client := openai.NewClient(
+		option.WithAPIKey(cfg.APIKey),
+		option.WithBaseURL(cfg.BaseURL),
+	)
+	return &Bot{
+		client: &client,
+		cfg:    cfg,
+	}
+}
+
+func (b *Bot) Chat(ctx context.Context, userMsg string) (string, error) {
+	if b.cfg.APIKey == "" {
+		return "", errors.New("未配置 api_key，请编辑 data/config.yaml")
+	}
+	history := make([]openai.ChatCompletionMessageParamUnion, 0, len(b.history)+2)
+	history = append(history, openai.SystemMessage(b.cfg.SystemPrompt))
+	history = append(history, b.history...)
+	history = append(history, openai.UserMessage(userMsg))
+
+	stream := b.client.Chat.Completions.NewStreaming(ctx, openai.ChatCompletionNewParams{
+		Model:    b.cfg.Model,
+		Messages: history,
+	})
+	answer := ""
+	for stream.Next() {
+		for _, delta := range stream.Current().Choices {
+			answer += delta.Delta.Content
+		}
+	}
+	if err := stream.Err(); err != nil {
+		return "", fmt.Errorf("调用 AI 接口失败: %w", err)
+	}
+	b.history = append(b.history, openai.UserMessage(userMsg), openai.AssistantMessage(answer))
+	if len(b.history) > maxHistory {
+		b.history = b.history[len(b.history)-maxHistory:]
+	}
+	return answer, nil
+}
@@ -0,0 +1,100 @@
+package config
+
+import (
+	"os"
+	"path/filepath"
+
+	"gopkg.in/yaml.v3"
+)
+
+const (
+	configDir  = "data"
+	configFile = "config.yaml"
+)
+
+type Config struct {
+	BotName      string `yaml:"bot_name"`
+	Port         int    `yaml:"port"`
+	LogLevel     string `yaml:"log_level"`
+	APIKey       string `yaml:"api_key"`
+	BaseURL      string `yaml:"base_url"`
+	Model        string `yaml:"model"`
+	SystemPrompt string `yaml:"system_prompt"`
+}
+
+var cfg *Config
+
+func Load() (*Config, error) {
+	if cfg != nil {
+		return cfg, nil
+	}
+	if err := os.MkdirAll(configDir, 0o755); err != nil {
+		return nil, err
+	}
+	path := filepath.Join(configDir, configFile)
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		if err := writeDefault(path); err != nil {
+			return nil, err
+		}
+	}
+	data, err := os.ReadFile(path)
+	if err != nil {
+		return nil, err
+	}
+	cfg = &Config{}
+	if err := yaml.Unmarshal(data, cfg); err != nil {
+		return nil, err
+	}
+	applyDefaults(cfg)
+	return cfg, nil
+}
+
+func applyDefaults(c *Config) {
+	def := &Config{
+		BotName:      "ai-bot",
+		Port:         8080,
+		LogLevel:     "info",
+		BaseURL:      "https://api.openai.com/v1",
+		Model:        "gpt-4o-mini",
+		SystemPrompt: "你是一个乐于助人的 AI 助手。",
+	}
+	if c.BotName == "" {
+		c.BotName = def.BotName
+	}
+	if c.Port == 0 {
+		c.Port = def.Port
+	}
+	if c.LogLevel == "" {
+		c.LogLevel = def.LogLevel
+	}
+	if c.BaseURL == "" {
+		c.BaseURL = def.BaseURL
+	}
+	if c.Model == "" {
+		c.Model = def.Model
+	}
+	if c.SystemPrompt == "" {
+		c.SystemPrompt = def.SystemPrompt
+	}
+}
+
+func GetConfig() *Config {
+	return cfg
+}
+
+func writeDefault(path string) error {
+	cfg = &Config{
+		BotName:      "ai-bot",
+		Port:         8080,
+		LogLevel:     "info",
+		APIKey:       "",
+		BaseURL:      "https://api.openai.com/v1",
+		Model:        "gpt-4o-mini",
+		SystemPrompt: "你是一个乐于助人的 AI 助手。",
+	}
+	data, err := yaml.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+	return os.WriteFile(path, data, 0o644)
+}
@@ -1,7 +1,50 @@
 package main

-import "fmt"
+import (
+	"bufio"
+	"context"
+	"fmt"
+	"log"
+	"os"
+	"strings"
+
+	"myaibot/internal/bot"
+	"myaibot/internal/config"
+)

 func main() {
-	fmt.Println("Hello, World!")
+	cfg, err := config.Load()
+	if err != nil {
+		log.Fatalf("加载配置失败: %v", err)
+	}
+	if cfg.APIKey == "" {
+		log.Println("提示: 未配置 api_key，请编辑 data/config.yaml")
+	}
+	fmt.Printf("🤖 %s 已启动 (模型: %s)。输入问题开始对话，输入 /exit 退出。\n", cfg.BotName, cfg.Model)
+
+	b := bot.New(cfg)
+	scanner := bufio.NewScanner(os.Stdin)
+	for {
+		fmt.Print("你: ")
+		if !scanner.Scan() {
+			break
+		}
+		input := strings.TrimSpace(scanner.Text())
+		if input == "" {
+			continue
+		}
+		if input == "/exit" || input == "/quit" {
+			fmt.Println("再见！")
+			break
+		}
+		answer, err := b.Chat(context.Background(), input)
+		if err != nil {
+			fmt.Printf("⚠️  %v\n", err)
+			continue
+		}
+		fmt.Printf("%s: %s\n", cfg.BotName, answer)
+	}
+	if err := scanner.Err(); err != nil {
+		log.Fatal(err)
+	}
 }