Using deepseek example for streaming reasoning content

tmc · Feb 5, 2025 · 959619f · 959619f
1 parent 78079b7
commit 959619f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/examples/deepseek-completion-example/deepseek-completion-example.go b/examples/deepseek-completion-example/deepseek-completion-example.go
@@ -32,8 +32,13 @@ func main() {
 		content,
 		llms.WithMaxTokens(2000),
 		llms.WithTemperature(0.7),
-		llms.WithStreamingFunc(func(ctx context.Context, chunk []byte) error {
-			fmt.Print(string(chunk))
+		llms.WithStreamingReasoningFunc(func(ctx context.Context, reasoningChunk []byte, chunk []byte) error {
+			if len(reasoningChunk) > 0 {
+				fmt.Printf("Streaming Reasoning: %s\n", string(reasoningChunk))
+			}
+			if len(chunk) > 0 {
+				fmt.Printf("Streaming Content: %s\n", string(chunk))
+			}
 			return nil
 		}),
 	)