Even stricter

Author: johnd0e

openai/init.moon

@@ -215,12 +215,14 @@ class OpenAI
   create_stream_filter: (chunk_callback) =>
     assert types.function(chunk_callback), "Must provide chunk_callback function when streaming response"
 
-    accumulation_buffer = ""
+    accumulation_buffer,streamed = ""
 
     (...) ->
       chunk = ...
 
-      if type(chunk) == "string"
+      if chunk == nil
+        assert not streamed or accumulation_buffer\match"^%s*$", "buffer not empty"
+      elseif type(chunk) == "string"
         accumulation_buffer ..= chunk
 
         while true
@@ -233,8 +235,9 @@ class OpenAI
             field, value, rest_evt = event\match "^(.-):%s+([^\r\n]+)(.-)$"
             switch field
               when "data"
+                streamed = true
                 unless value=="[DONE]"
-                  chunk_callback (json.decode value)
+                  chunk_callback (cjson.decode value)
               when "event","id","retry","" --comment
                 nil -- noop
               when nil