66 lines
1.2 KiB
Go
66 lines
1.2 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/tmc/langchaingo/llms"
|
|
"github.com/tmc/langchaingo/prompts"
|
|
)
|
|
|
|
func Call(llm llms.Model, ctx context.Context) (response string) {
|
|
|
|
response, err := llms.GenerateFromSinglePrompt(
|
|
ctx,
|
|
llm,
|
|
"Hello! My name is wonder!",
|
|
)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("AI:", response)
|
|
|
|
return response
|
|
|
|
}
|
|
func CallWithMessage(llm llms.Model, ctx context.Context, prompt string) (response string) {
|
|
|
|
response, err := llms.GenerateFromSinglePrompt(
|
|
ctx,
|
|
llm,
|
|
"用户正在学习Go语言，你是用户的助手。\n"+prompt,
|
|
)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("AI:", response)
|
|
|
|
return response
|
|
|
|
}
|
|
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)
|
|
return
|
|
}
|
|
|
|
text := string(content)
|
|
|
|
template := prompts.NewPromptTemplate(text, []string{".Knowledge"})
|
|
prompt, err := template.Format(map[string]any{
|
|
"Knowledge": topic,
|
|
})
|
|
|
|
fmt.Println("Me: ", prompt)
|
|
if err != nil {
|
|
fmt.Printf("Prompt err: %v", err)
|
|
}
|
|
response = CallWithMessage(llm, ctx, prompt)
|
|
return response
|
|
}
|