From 6e7d055aab549627a6dadcbef9ed9e21161d66cc Mon Sep 17 00:00:00 2001
From: Travis Cline <travis.cline@gmail.com>
Date: Wed, 20 Mar 2024 14:43:55 -0700
Subject: [PATCH] readme: Include imports

---
 README.md | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 783b138..4a011e5 100644
--- a/README.md
+++ b/README.md
@@ -9,7 +9,19 @@
 This is a simple example of how to use the library to create a simple chatbot that uses OpenAI to generate responses.
 
 ```go
-func ExampleMessageGraph() {
+import (
+	"context"
+	"errors"
+	"fmt"
+	"testing"
+
+	"github.com/tmc/langchaingo/llms"
+	"github.com/tmc/langchaingo/llms/openai"
+	"github.com/tmc/langchaingo/schema"
+	"github.com/tmc/langgraphgo/graph"
+)
+
+func main() {
 	model, err := openai.New()
 	if err != nil {
 		panic(err)