jack 4 달 전
부모
커밋
60c6bd3b0e
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      app/core/runner/thread_runner.py

+ 2 - 0
app/core/runner/thread_runner.py

@@ -194,6 +194,8 @@ class ThreadRunner:
             on_message_create_func=create_message_callback,
             on_message_create_func=create_message_callback,
             event_handler=self.event_handler,
             event_handler=self.event_handler,
         )
         )
+        if self.stream == False and hasattr(response_stream, "choices"):
+            response_stream = [response_stream]
         response_msg = llm_callback_handler.handle_llm_response(response_stream)
         response_msg = llm_callback_handler.handle_llm_response(response_stream)
         message_creation_run_step = llm_callback_handler.step
         message_creation_run_step = llm_callback_handler.step
         print("444444444444444444444444455555555577777777777777777777777")
         print("444444444444444444444444455555555577777777777777777777777")