package mainimport ("context""fmt""github.com/tmc/langchaingo/llms""github.com/tmc/langchaingo/llms/openai""github.com/tmc/langgraphgo/graph")func main() {model, err := openai.New(openai.WithBaseURL("http://127.0.0.1:11434/v1/"),openai.WithToken("ollama"),openai.WithModel("deepseek-r1:1.5b"),)if err != nil {panic(err)}g := graph.NewMessageGraph()g.AddNode("oracle", func(ctx context.Context, state []llms.MessageContent) ([]llms.MessageContent, error) {r, err := model.GenerateContent(ctx, state, llms.WithTemperature(0.0))if err != nil {return nil, err}return append(state,llms.TextParts(llms.ChatMessageTypeAI, r.Choices[0].Content),), nil})g.AddNode(graph.END, func(ctx context.Context, state []llms.MessageContent) ([]llms.MessageContent, error) {return state, nil})g.AddEdge("oracle", graph.END)g.SetEntryPoint("oracle")runnable, err := g.Compile()if err != nil {panic(err)}ctx := context.Background()// Let's run it!res, err := runnable.Invoke(ctx, []llms.MessageContent{llms.TextParts(llms.ChatMessageTypeHuman, "What is 1 + 1?"),})if err != nil {panic(err)}fmt.Println(res)}
[{human [{What is 1 + 1?}]} {ai [{1 + 1 equals 2.}]}]


文章转载自golang算法架构leetcode技术php,如果涉嫌侵权,请发送邮件至:contact@modb.pro进行举报,并提供相关证据,一经查实,墨天轮将立刻删除相关内容。




