47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
|
package handle
|
||
|
|
|
||
|
|
import (
|
||
|
|
"agent/agent"
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/tmc/langchaingo/llms"
|
||
|
|
)
|
||
|
|
|
||
|
|
var files = []string{
|
||
|
|
".\\prompt\\card.txt",
|
||
|
|
".\\prompt\\guodegang.txt",
|
||
|
|
".\\prompt\\mermaid.txt",
|
||
|
|
".\\prompt\\quiz.txt",
|
||
|
|
".\\prompt\\scenario.txt",
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetCard(llm llms.Model, ctx context.Context, topic string) (response string) {
|
||
|
|
file := files[0]
|
||
|
|
response = agent.CallWithAgent(llm, ctx, topic, file)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetGuodegang(llm llms.Model, ctx context.Context, topic string) (response string) {
|
||
|
|
file := files[1]
|
||
|
|
response = agent.CallWithAgent(llm, ctx, topic, file)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetMermaid(llm llms.Model, ctx context.Context, topic string) (response string) {
|
||
|
|
file := files[2]
|
||
|
|
response = agent.CallWithAgent(llm, ctx, topic, file)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetQuiz(llm llms.Model, ctx context.Context, topic string) (response string) {
|
||
|
|
file := files[3]
|
||
|
|
response = agent.CallWithAgent(llm, ctx, topic, file)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetScenario(llm llms.Model, ctx context.Context, topic string) (response string) {
|
||
|
|
file := files[4]
|
||
|
|
response = agent.CallWithAgent(llm, ctx, topic, file)
|
||
|
|
return
|
||
|
|
}
|