fix: stream yeild problem

nobu007 · Aug 18, 2024 · e8c30fb · e8c30fb
1 parent 9195094
commit e8c30fb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/codeinterpreterapi/session.py b/src/codeinterpreterapi/session.py
@@ -403,11 +403,10 @@ def generate_response_stream(
                     output = chunk["output"]
                 else:
                     output = str(chunk)
-                yield output
                 full_output += output
 
             print("generate_response_stream brain.stream full_output=", full_output)
-            self._aoutput_handler(full_output)
+            yield self._output_handler(full_output)
         except Exception as e:
             if self.verbose:
                 traceback.print_exc()