Update src/txagent/txagent.py

src/txagent/txagent.py

@@ -778,7 +778,7 @@ Generate **one summarized sentence** about "function calls' responses" with nece
         """
         print("\033[1;32;40mstart\033[0m")
         print("len(message)", len(message))
-        if len(message) <= 10:
+        if not message or len(message.strip()) <= 10:
             yield "Hi, I am TxAgent, an assistant for answering biomedical questions. Please provide a valid message with a string longer than 10 characters."
             return "Please provide a valid message."
         outputs = []
@@ -820,13 +820,17 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     history.extend(current_gradio_history)
                     if special_tool_call == 'Finish':
                         next_round = False
-                        conversation.extend(function_call_messages)
-                        return function_call_messages[0]['content']
+                        if function_call_messages and len(function_call_messages) > 0:
+                            conversation.extend(function_call_messages)
+                            return function_call_messages[0]['content']
+                        else:
+                            return "No final response available."
                     elif special_tool_call == 'RequireClarification' or special_tool_call == 'DirectResponse':
-                        history.append(
-                            ChatMessage(role="assistant", content=history[-1].content))
-                        yield history
-                        return history[-1].content
+                        if history:
+                            history.append(
+                                ChatMessage(role="assistant", content=history[-1].content))
+                            yield history
+                            return history[-1].content
                     if (self.enable_summary or token_overflow) and not call_agent:
                         if token_overflow:
                             print("token_overflow, using summary")
@@ -834,10 +838,8 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                     last_status = self.function_result_summary(
                         conversation, status=last_status,
                         enable_summary=enable_summary)
-                    if function_call_messages is not None:
+                    if function_call_messages:
                         conversation.extend(function_call_messages)
-                        # Hide intermediate output
-                        pass
                     else:
                         next_round = False
                         conversation.extend(
@@ -904,7 +906,7 @@ Generate **one summarized sentence** about "function calls' responses" with nece
                 for each in history:
                     if each.metadata is not None:
                         each.metadata['status'] = 'done'
-                if '[FinalAnswer]' in last_thought or '"name": "Finish",' in last_outputs_str:
+                if '[FinalAnswer]' in last_thought or 'Finish' in last_outputs_str:
                     if '[FinalAnswer]' in last_thought:
                         _, final_answer = last_thought.split('[FinalAnswer]', 1)
                     else: