Update query.py

query.py

@@ -124,6 +124,8 @@ class VectaraQuery():
                 if key == 'data':
                     line = json.loads(value)
                     if line['type'] == 'generation_chunk':
-                        chunks.append(line['generation_chunk'])
+                        chunk = line['generation_chunk']
+                        chunks.append(chunk)
+                        yield chunk
 
         return ''.join(chunks)