From 8e2a61d804ecdb99709b5f230129fc69fbdd3580 Mon Sep 17 00:00:00 2001
From: HeZhaohui <hezhaohui0807@163.com>
Date: Sat, 28 Feb 2026 14:28:38 +0800
Subject: [PATCH] =?UTF-8?q?Feat:=20=E5=AE=9E=E7=8E=B0=E7=AE=80=E5=8D=95?=
 =?UTF-8?q?=E5=AF=B9=E8=AF=9D=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/single/main.go | 43 +++++++++++++++++++++++++++------------
 cmd/test/main.go   | 50 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 80 insertions(+), 13 deletions(-)
 create mode 100644 cmd/test/main.go

diff --git a/cmd/single/main.go b/cmd/single/main.go
index c3ad5d6..a7fce44 100644
--- a/cmd/single/main.go
+++ b/cmd/single/main.go
@@ -1,9 +1,11 @@
 package main
 
 import (
+	"bufio"
 	"context"
 	"fmt"
 	"log"
+	"os"
 
 	"github.com/tmc/langchaingo/llms"
 	"github.com/tmc/langchaingo/llms/openai"
@@ -30,21 +32,36 @@ func main() {
 	}
 
 	ctx := context.Background()
+	reader := bufio.NewReader(os.Stdin)
 
-	prompt := "哟什么风把您给吹来了"
+	fmt.Println("========= 请输入您的问题 (输入 'exit' 退出) =========")
+	for {
+		fmt.Print("😇 You: ")
+		prompt, err := reader.ReadString('\n')
+		if err != nil {
+			log.Fatalf("读取输入失败: %v", err)
+		}
+
+		if prompt == "exit\n" {
+			fmt.Println("========= 谢谢使用，再见! =========")
+			break
+		}
+
+		// Chat
+		message := []llms.MessageContent{
+			llms.TextParts(llms.ChatMessageTypeHuman, prompt),
+		}
+
+		response, err := llm.GenerateContent(ctx, message,
+			llms.WithModel(model),
+		)
+
+		if err != nil {
+			log.Fatalf("调用失败: %v\n模型: %s", err, model)
+		}
+
+		fmt.Println("🤖 AI: ", response.Choices[0].Content)
 
-	message := []llms.MessageContent{
-		llms.TextParts(llms.ChatMessageTypeHuman, prompt),
 	}
 
-	response, err := llm.GenerateContent(ctx, message,
-		llms.WithModel(model),
-	)
-
-	if err != nil {
-		log.Fatalf("调用失败: %v\n模型: %s", err, model)
-	}
-
-	fmt.Println("模型响应:", response.Choices[0].Content)
-
 }
diff --git a/cmd/test/main.go b/cmd/test/main.go
new file mode 100644
index 0000000..c3ad5d6
--- /dev/null
+++ b/cmd/test/main.go
@@ -0,0 +1,50 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+
+	"github.com/tmc/langchaingo/llms"
+	"github.com/tmc/langchaingo/llms/openai"
+)
+
+func main() {
+	apiKey := "sk-angtzujyplaqsrxvlxkmjahpyxwnhhaonnfamokggxclmdlv"
+	baseUrl := "https://api.siliconflow.cn/v1"
+
+	//baseUrl := "https://api.siliconflow.cn/v1/chat/completions"
+	model := "Pro/deepseek-ai/DeepSeek-V3.2"
+
+	if apiKey == "" {
+		log.Fatal("apiKey is empty")
+	}
+
+	llm, err := openai.New(
+		openai.WithToken(apiKey),
+		openai.WithBaseURL(baseUrl),
+	)
+
+	if err != nil {
+		log.Fatalf("初始化失败: %v\nAPI Key: %s\nBase URL: %s", err, apiKey, baseUrl)
+	}
+
+	ctx := context.Background()
+
+	prompt := "哟什么风把您给吹来了"
+
+	message := []llms.MessageContent{
+		llms.TextParts(llms.ChatMessageTypeHuman, prompt),
+	}
+
+	response, err := llm.GenerateContent(ctx, message,
+		llms.WithModel(model),
+	)
+
+	if err != nil {
+		log.Fatalf("调用失败: %v\n模型: %s", err, model)
+	}
+
+	fmt.Println("模型响应:", response.Choices[0].Content)
+
+}