✨Feat: 添加GetAll()添加并行处理

backend/handlers/agent.go

@@ -3,6 +3,9 @@ package handle
 import (
 	"agent/agent"
 	"context"
+	"fmt"
+
+	"sync"
 
 	"github.com/tmc/langchaingo/llms"
 )
@@ -44,3 +47,42 @@ func GetScenario(llm llms.Model, ctx context.Context, topic string) (response st
 	response = agent.CallWithAgent(llm, ctx, topic, file)
 	return
 }
+
+func GetAll(llm llms.Model, ctx context.Context, topic string) (response map[string]string) {
+	response = make(map[string]string)
+
+	var wg sync.WaitGroup
+
+	wg.Add(len(files))
+
+	go func() {
+		defer wg.Done()
+		response["card"] = GetCard(llm, ctx, topic)
+	}()
+
+	go func() {
+		defer wg.Done()
+		response["guodegang"] = GetGuodegang(llm, ctx, topic)
+	}()
+
+	go func() {
+		defer wg.Done()
+		response["mermaid"] = GetMermaid(llm, ctx, topic)
+	}()
+
+	go func() {
+		defer wg.Done()
+		response["quiz"] = GetQuiz(llm, ctx, topic)
+	}()
+
+	go func() {
+		defer wg.Done()
+		response["scenario"] = GetScenario(llm, ctx, topic)
+	}()
+
+	wg.Wait()
+
+	fmt.Println("======== All responses received ========")
+
+	return response
+}