openai_fix (#161)

Co-authored-by: xxYyh <[email protected]>

Author: YYH211

src/legacy/multi_agent.py

@@ -278,6 +278,9 @@ async def supervisor_tools(state: ReportState, config: RunnableConfig)  -> Comma
             question_obj = cast(Question, observation)
             result.append({"role": "assistant", "content": question_obj.question})
             return Command(goto=END, update={"messages": result})
+        elif tool_call["name"] == "FinishReport":
+            result.append({"role": "user", "content": "Report is Finish"})
+            return Command(goto=END, update={"messages": result})
         elif tool_call["name"] == "Sections":
             sections_list = cast(Sections, observation).sections
         elif tool_call["name"] == "Introduction":
@@ -340,7 +343,7 @@ async def supervisor_should_continue(state: ReportState) -> str:
     messages = state["messages"]
     last_message = messages[-1]
     # End because the supervisor asked a question or is finished
-    if not last_message.tool_calls or (len(last_message.tool_calls) == 1 and last_message.tool_calls[0]["name"] == "FinishReport"):
+    if not last_message.tool_calls:
         # Exit the graph
         return END