address comments for edge cases

copilot-extensions · Nov 8, 2024 · 1ce7211 · 1ce7211
1 parent c08602f
commit 1ce7211
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/stream/parse.go b/pkg/stream/parse.go
@@ -40,6 +40,14 @@ func ParseFile(filename string) error {
 			continue // skip lines without "data: "
 		}
 
+		// Handle special cases
+		if line == "[DONE]" {
+			break // stop processing if we encounter [DONE]
+		}
+		if line == "" {
+			continue // skip empty data lines
+		}
+
 		// Parse the JSON line into our `Data` struct
 		var data Data
 		err := json.Unmarshal([]byte(line), &data)