relax check for matching names when removing inline tags (#82)

Author: vbarda

langgraph_supervisor/agent_name.py

@@ -59,7 +59,7 @@ def remove_inline_agent_name(message: BaseMessage) -> BaseMessage:
         >>> remove_inline_agent_name(AIMessage(content=[{"type": "text", "text": "<name>assistant</name><content>Hello</content>"}], name="assistant"))
         AIMessage(content=[{"type": "text", "text": "Hello"}], name="assistant")
     """
-    if not isinstance(message, AIMessage) or not message.name:
+    if not isinstance(message, AIMessage):
         return message
 
     is_content_blocks_content = _is_content_blocks_content(message.content)
@@ -78,9 +78,6 @@ def remove_inline_agent_name(message: BaseMessage) -> BaseMessage:
     if not name_match or not content_match:
         return message
 
-    if name_match.group(1) != message.name:
-        return message
-
     parsed_content = content_match.group(1)
     parsed_message = message.model_copy()
     if is_content_blocks_content:

tests/test_agent_name.py

@@ -67,13 +67,6 @@ def test_remove_inline_agent_name():
     result = remove_inline_agent_name(ai_message)
     assert result == ai_message
 
-    # Test that messages with mismatched name are returned unchanged.
-    ai_message = AIMessage(
-        content="<name>different_name</name><content>Hello world</content>", name="assistant"
-    )
-    result = remove_inline_agent_name(ai_message)
-    assert result == ai_message
-
     # Test that content is correctly extracted from tags.
     ai_message = AIMessage(
         content="<name>assistant</name><content>Hello world</content>", name="assistant"