jack 4 days ago
parent
commit
0959e66181
2 changed files with 6 additions and 2 deletions
  1. 1 1
      app/core/runner/pub_handler.py
  2. 5 1
      app/core/runner/thread_runner.py

+ 1 - 1
app/core/runner/pub_handler.py

@@ -344,7 +344,7 @@ class StreamEventHandler:
         pub MessageDelta
         """
         delta = {
-            "content": [{"index": index, "type": "text", "text": {"value": content}}],
+            "content": [{"index": index, "type": "text", "text": {"value": ""}}],
             "role": "assistant",
             "tools_content": [
                 {

+ 5 - 1
app/core/runner/thread_runner.py

@@ -236,6 +236,8 @@ class ThreadRunner:
         if self.stream == False and hasattr(response_stream, "choices"):
             response_stream = [response_stream]
         response_msg = llm_callback_handler.handle_llm_response(response_stream)
+        print("response_msg================================================================response_msg")
+        print(response_msg)
         message_creation_run_step = llm_callback_handler.step
         print("444444444444444444444444455555555577777777777777777777777")
         logging.info("chat_response_message: %s", response_msg)
@@ -293,8 +295,10 @@ class ThreadRunner:
                         },
                         completed=not external_tool_call_dict,
                     )
+                    print("llm_callback_handler.message================================================================llm_callback_handler.message")
+                    print(llm_callback_handler.message)
                     self.event_handler.pub_message_delta_tool(
-                        message_id=new_run_step.id,
+                        message_id=llm_callback_handler.message.id,
                         index=0,
                         content=json.dumps(tool_calls_with_outputs)
                     )