✨Feat: 加入协程机制

main.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"sync"
 
 	"github.com/joho/godotenv"
 	"github.com/tmc/langchaingo/llms"
@@ -44,10 +45,24 @@ func init() {
 }
 
 func main() {
+	var wg sync.WaitGroup
 
-	agent.CallWithAgent(llm, ctx, "文件", ".\\prompt\\card.txt")
+	files := []string{
-	agent.CallWithAgent(llm, ctx, "文件", ".\\prompt\\guodegang.txt")
+		".\\prompt\\card.txt",
-	agent.CallWithAgent(llm, ctx, "文件", ".\\prompt\\mermaid.txt")
+		".\\prompt\\guodegang.txt",
-	agent.CallWithAgent(llm, ctx, "文件", ".\\prompt\\quiz.txt")
+		".\\prompt\\mermaid.txt",
-	agent.CallWithAgent(llm, ctx, "文件", ".\\prompt\\scenario.txt")
+		".\\prompt\\quiz.txt",
+		".\\prompt\\scenario.txt",
+	}
+
+	wg.Add(len(files))
+
+	for _, file := range files {
+		go func(f string) {
+			defer wg.Done()
+			agent.CallWithAgent(llm, ctx, "文件", f)
+		}(file)
+	}
+
+	wg.Wait()
 }