wonder/langchain-go
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

✨Feat: 增添结构体解析函数

Browse Source
This commit is contained in:
hhs
2026-03-27 19:30:04 +08:00
parent 4b9670967f
commit 18040d00c4
11 changed files with 179 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/agent/basicAgent.go
+48 -1
View File
@@ -8,6 +8,7 @@ import (
"github.com/tmc/langchaingo/llms"
"github.com/tmc/langchaingo/prompts"
"github.com/tmc/langchaingo/outputparser"
)
func Call(llm llms.Model, ctx context.Context) (response string) {
@@ -42,7 +43,7 @@ func CallWithMessage(llm llms.Model, ctx context.Context, prompt string) (respon
return response
}
func CallWithAgent(llm llms.Model, ctx context.Context, topic string, promptPath string) (response string) {
func CallWithAgent(llm llms.Model, ctx context.Context, topic string, promptPath string,) (response string) {
content, err := os.ReadFile(promptPath)
if err != nil {
fmt.Printf("File err: %v", err)
@@ -63,3 +64,49 @@ func CallWithAgent(llm llms.Model, ctx context.Context, topic string, promptPath
response = CallWithMessage(llm, ctx, prompt)
return response
}
func CallWithAgentDefined[T any](llm llms.Model, ctx context.Context, topic string, promptPath string, responseType T) (response T) {
parse, err := outputparser.NewDefined(responseType)
if err != nil {
fmt.Printf("CreateParse err: %v", err)
return
}
content, err := os.ReadFile(promptPath)
if err != nil {
fmt.Printf("File err: %v", err)
return
}
text := string(content)
textWithFormatInstruction := text + "\n" + parse.GetFormatInstructions()
template := prompts.NewPromptTemplate(textWithFormatInstruction, []string{".Knowledge"})
prompt, err := template.Format(map[string]any{
"Knowledge": topic,
})
fmt.Println("Me: ", prompt)
if err != nil {
fmt.Printf("Prompt err: %v", err)
}
responseString, err := llms.GenerateFromSinglePrompt(
ctx,
llm,
"用户正在学习Go语言，你是用户的助手。\n"+prompt,
)
if err != nil {
log.Fatal(err)
}
response, err = parse.Parse(responseString)
if err != nil {
fmt.Printf("Parse err: %v", err)
return
}
return response
}
backend/agent/response_struct/card_test.go
-18
View File
@@ -1,18 +0,0 @@
package agent
import (
"testing"
"github.com/tmc/langchaingo/outputparser"
)
func TestCard(t *testing.T) {
definedCard, err := outputparser.NewDefined(Card{})
if err != nil {
t.Fatalf("%v", err)
}
output := definedCard.GetFormatInstructions()
t.Log(output)
}
backend/handlers/agent.go
+19 -17
View File
@@ -2,6 +2,7 @@ package handle
import (
"agent/agent"
"agent/response_struct"
"context"
"fmt"
@@ -18,38 +19,38 @@ var files = []string{
".\\prompt\\scenario.txt",
}
func GetCard(llm llms.Model, ctx context.Context, topic string) (response string) {
func GetCard(llm llms.Model, ctx context.Context, topic string) (response response_struct.Card) {
file := files[0]
response = agent.CallWithAgent(llm, ctx, topic, file)
response = agent.CallWithAgentDefined(llm, ctx, topic, file, response_struct.Card{})
return
}
func GetGuodegang(llm llms.Model, ctx context.Context, topic string) (response string) {
func GetGuodegang(llm llms.Model, ctx context.Context, topic string) (response response_struct.Guodegang) {
file := files[1]
response = agent.CallWithAgent(llm, ctx, topic, file)
response = agent.CallWithAgentDefined(llm, ctx, topic, file, response_struct.Guodegang{})
return
}
func GetMermaid(llm llms.Model, ctx context.Context, topic string) (response string) {
func GetMermaid(llm llms.Model, ctx context.Context, topic string) (response response_struct.Mermaid) {
file := files[2]
response = agent.CallWithAgent(llm, ctx, topic, file)
response = agent.CallWithAgentDefined(llm, ctx, topic, file, response_struct.Mermaid{})
return
}
func GetQuiz(llm llms.Model, ctx context.Context, topic string) (response string) {
func GetQuiz(llm llms.Model, ctx context.Context, topic string) (response response_struct.Quiz) {
file := files[3]
response = agent.CallWithAgent(llm, ctx, topic, file)
response = agent.CallWithAgentDefined(llm, ctx, topic, file, response_struct.Quiz{})
return
}
func GetScenario(llm llms.Model, ctx context.Context, topic string) (response string) {
func GetScenario(llm llms.Model, ctx context.Context, topic string) (response response_struct.Scenario) {
file := files[4]
response = agent.CallWithAgent(llm, ctx, topic, file)
response = agent.CallWithAgentDefined(llm, ctx, topic, file, response_struct.Scenario{})
return
}
func GetAll(llm llms.Model, ctx context.Context, topic string) (response map[string]string) {
response = make(map[string]string)
func GetAll(llm llms.Model, ctx context.Context, topic string) (response *response_struct.AllStruct) {
var wg sync.WaitGroup
@@ -57,30 +58,31 @@ func GetAll(llm llms.Model, ctx context.Context, topic string) (response map[str
go func() {
defer wg.Done()
response["card"] = GetCard(llm, ctx, topic)
response.CardStruct = GetCard(llm, ctx, topic)
}()
go func() {
defer wg.Done()
response["guodegang"] = GetGuodegang(llm, ctx, topic)
response.GuodegangStruct = GetGuodegang(llm, ctx, topic)
}()
go func() {
defer wg.Done()
response["mermaid"] = GetMermaid(llm, ctx, topic)
response.MermaidStruct = GetMermaid(llm, ctx, topic)
}()
go func() {
defer wg.Done()
response["quiz"] = GetQuiz(llm, ctx, topic)
response.QuizStruct = GetQuiz(llm, ctx, topic)
}()
go func() {
defer wg.Done()
response["scenario"] = GetScenario(llm, ctx, topic)
response.ScenarioStruct = GetScenario(llm, ctx, topic)
}()
wg.Wait()
fmt.Println("======== All responses received ========")
backend/response_struct/allstruct.go
+9
View File
@@ -0,0 +1,9 @@
package response_struct
type AllStruct struct {
CardStruct Card `json:"card"`
GuodegangStruct Guodegang `json:"guodegang"`
MermaidStruct Mermaid `json:"mermaid"`
QuizStruct Quiz `json:"quiz"`
ScenarioStruct Scenario `json:"scenario"`
}
backend/agent/response_struct/card.go → backend/response_struct/card.go
+1 -1
View File
@@ -1,4 +1,4 @@
package agent
package response_struct
// Card 是顶层结构体，包含所有关于 Go map 的笔记内容
type Card struct {
backend/response_struct/guodegang.go
+6
View File
@@ -0,0 +1,6 @@
package response_struct
type Guodegang struct {
Name string `json:"name"`
Content string `json:"content"`
}
backend/response_struct/mermaid.go
+6
View File
@@ -0,0 +1,6 @@
package response_struct
type Mermaid struct {
Type string `json:"type"`
Content string `json:"content"`
}
backend/response_struct/quiz.go
+12
View File
@@ -0,0 +1,12 @@
package response_struct
type Quiz struct {
Questions []Question `json:"questions"`
}
type Question struct {
Question string `json:"question"`
Options []string `json:"options"`
Answer string `json:"answer"`
Explanation string `json:"explanation"`
}
backend/response_struct/response_test.go
+62
View File
@@ -0,0 +1,62 @@
package response_struct
import (
"testing"
"github.com/tmc/langchaingo/outputparser"
)
func TestCard(t *testing.T) {
definedCard, err := outputparser.NewDefined(Card{})
if err != nil {
t.Fatalf("%v", err)
}
output := definedCard.GetFormatInstructions()
t.Log(output)
}
func TestGuodegang(t *testing.T) {
definedGuodegang, err := outputparser.NewDefined(Guodegang{})
if err != nil {
t.Fatalf("%v", err)
}
output := definedGuodegang.GetFormatInstructions()
t.Log(output)
}
func TestMermaid(t *testing.T) {
definedMermaid, err := outputparser.NewDefined(Mermaid{})
if err != nil {
t.Fatalf("%v", err)
}
output := definedMermaid.GetFormatInstructions()
t.Log(output)
}
func TestQuiz(t *testing.T) {
definedQuiz, err := outputparser.NewDefined(Quiz{})
if err != nil {
t.Fatalf("%v", err)
}
output := definedQuiz.GetFormatInstructions()
t.Log(output)
}
func TestScenario(t *testing.T) {
definedScenario, err := outputparser.NewDefined(Scenario{})
if err != nil {
t.Fatalf("%v", err)
}
output := definedScenario.GetFormatInstructions()
t.Log(output)
}
backend/response_struct/scenario.go
+14
View File
@@ -0,0 +1,14 @@
package response_struct
type ScenarioResponse struct {
Scenarios []Scenario `json:"scenarios"`
}
type Scenario struct {
Name string `json:"name"`
Background string `json:"background"`
Solution string `json:"solution"`
CodeExample string `json:"code_example"`
CoreProblemSolved string `json:"core_problem_solved"`
KeyConsiderations []string `json:"key_considerations"`
}