diff --git a/content/en/docs/eino/core_modules/components/chat_model_guide.md b/content/en/docs/eino/core_modules/components/chat_model_guide.md
index ba11d9c485..d5b4d5b403 100644
--- a/content/en/docs/eino/core_modules/components/chat_model_guide.md
+++ b/content/en/docs/eino/core_modules/components/chat_model_guide.md
@@ -265,14 +265,16 @@ response, err := cm.Generate(ctx, messages,
 
 ```go
 import (
-    "context"
-    "fmt"
-
-    "github.com/cloudwego/eino/callbacks"
-    "github.com/cloudwego/eino/components/model"
-    "github.com/cloudwego/eino/compose"
-    "github.com/cloudwego/eino/schema"
-    callbacksHelper "github.com/cloudwego/eino/utils/callbacks"
+	"context"
+	"errors"
+	"fmt"
+	"io"
+
+	"github.com/cloudwego/eino/callbacks"
+	"github.com/cloudwego/eino/components/model"
+	"github.com/cloudwego/eino/compose"
+	"github.com/cloudwego/eino/schema"
+	callbacksHelper "github.com/cloudwego/eino/utils/callbacks"
 )
 
 // Create callback handler
@@ -296,7 +298,7 @@ handler := &callbacksHelper.ModelCallbackHandler{
             }
             if err != nil {
                 fmt.Printf("Stream read error: %v\n", err)
-                return
+                return ctx
             }
             if chunk == nil || chunk.Message == nil {
                 continue
diff --git a/content/zh/docs/eino/core_modules/components/chat_model_guide.md b/content/zh/docs/eino/core_modules/components/chat_model_guide.md
index 6f73bc5ae4..33aaa81c3f 100644
--- a/content/zh/docs/eino/core_modules/components/chat_model_guide.md
+++ b/content/zh/docs/eino/core_modules/components/chat_model_guide.md
@@ -265,14 +265,16 @@ response, err := cm.Generate(ctx, messages,
 
 ```go
 import (
-    "context"
-    "fmt"
-
-    "github.com/cloudwego/eino/callbacks"
-    "github.com/cloudwego/eino/components/model"
-    "github.com/cloudwego/eino/compose"
-    "github.com/cloudwego/eino/schema"
-    callbacksHelper "github.com/cloudwego/eino/utils/callbacks"
+	"context"
+	"errors"
+	"fmt"
+	"io"
+
+	"github.com/cloudwego/eino/callbacks"
+	"github.com/cloudwego/eino/components/model"
+	"github.com/cloudwego/eino/compose"
+	"github.com/cloudwego/eino/schema"
+	callbacksHelper "github.com/cloudwego/eino/utils/callbacks"
 )
 
 // 创建 callback handler
@@ -296,7 +298,7 @@ handler := &callbacksHelper.ModelCallbackHandler{
             }
             if err != nil {
                 fmt.Printf("流读取错误: %v\n", err)
-                return
+                return ctx
             }
             if chunk == nil || chunk.Message == nil {
                 continue