better searching

stream.go

@@ -10,7 +10,7 @@ import (
 
 type Chunk struct {
 	Type string `json:"type"`
-	Text string `json:"text"`
+	Text any    `json:"text"`
 }
 
 type Stream struct {
@@ -64,6 +64,13 @@ func TextChunk(text string) Chunk {
 	}
 }
 
+func ToolChunk(tool *ToolCall) Chunk {
+	return Chunk{
+		Type: "tool",
+		Text: tool,
+	}
+}
+
 func IDChunk(id string) Chunk {
 	return Chunk{
 		Type: "id",