revise message parsing

Author: binary-husky

ajet/context_tracker/multiagent_tracking.py

@@ -82,7 +82,7 @@ def extract_text_content_from_content_dict(self, msg):
         #        },
         #    ],
         # }
-        # or tool_result format:
+        # or tool_result format?? not observed yet:
         # msg = {
         #    "role": "tool",
         #    "content": [
@@ -97,31 +97,27 @@ def extract_text_content_from_content_dict(self, msg):
 
         str_content = ""
         for item in msg["content"]:
-            assert isinstance(item, dict), f"Unsupported non-dict item in message content: {item}. Full message: {msg}"
-
+            # item = {
+            #   "type": "text",
+            #   "text": "some text"
+            # },
             item_type = item.get("type", "")
+            assert not item_type == "tool_use", f"never observed such protocal yet"
+            assert not item_type == "tool_result", f"never observed such protocal yet"
 
-            # Handle text content block
-            if "text" in item:
-                if isinstance(item["text"], str):
-                    str_content += item["text"]
-            # Handle tool_result content block (AgentScope format)
-            elif item_type == "tool_result" and "output" in item:
-                output = item["output"]
-                if isinstance(output, str):
-                    str_content += output
-                else:
-                    str_content += str(output)
-            # Handle tool_use content block (for completeness)
-            elif item_type == "tool_use":
-                # tool_use blocks are handled via tool_calls field, skip content extraction
-                continue
-            else:
+            assert isinstance(item, dict), f"Unsupported non-dict item in message content: {item}. Full message: {msg}"
+
+            if ("text" not in item):
                 logger.warning(
-                    f"Non-text content in message content detected: {item}. Ignoring this item."
+                    f"Non-text content in message content detected: {item}. Ignoring."
                 )
-                # Continue processing other items instead of skipping the entire message
-                continue
+                should_skip_message = True
+                return str_content, should_skip_message
+
+            if isinstance(item["text"], str):
+                str_content += str(item["text"])
+            else:
+                str_content = ""
 
         should_skip_message = False
         return str_content, should_skip_message

ajet/launcher.py

@@ -99,7 +99,7 @@ def parse_args():
         default=False,
         help="Kill system processes (ray + vllm + python) that may block the current experiment",
     )
-    parser.add_argument("--prefix", type=str, default="", required=False, help="Prefix for service names")
+    parser.add_argument("--prefix", type=str, default="", required=False, help="Prefix for deepfinance service names")
     return parser.parse_args()