Improve multi-entity contextual grounding in memory extraction

Enhanced DISCRETE_EXTRACTION_PROMPT with explicit multi-entity handling instructions and improved test robustness to focus on core grounding functionality.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: abrookins

tests/test_thread_aware_grounding.py

@@ -90,13 +90,13 @@ async def test_thread_aware_pronoun_resolution(self):
         ), "Memories should contain the grounded name 'John'"
 
         # Ideally, there should be minimal or no ungrounded pronouns
-        ungrounded_pronouns = [
-            "he ",
-            "his ",
-            "him ",
-        ]  # Note: spaces to avoid false positives
+        # Use word boundary matching to avoid false positives like "the" containing "he"
+        import re
+
+        ungrounded_pronouns = [r"\bhe\b", r"\bhis\b", r"\bhim\b"]
         ungrounded_count = sum(
-            all_memory_text.lower().count(pronoun) for pronoun in ungrounded_pronouns
+            len(re.findall(pattern, all_memory_text, re.IGNORECASE))
+            for pattern in ungrounded_pronouns
         )
 
         print(f"Ungrounded pronouns found: {ungrounded_count}")
@@ -227,8 +227,14 @@ async def test_multi_entity_conversation(self):
             # Still consider it a pass if we have some entity grounding
 
         # Check for reduced pronoun usage - this is the key improvement
-        pronouns = ["he ", "she ", "his ", "her ", "him "]
-        pronoun_count = sum(all_memory_text.lower().count(p) for p in pronouns)
+        # Use word boundary matching to avoid false positives like "the" containing "he"
+        import re
+
+        pronouns = [r"\bhe\b", r"\bshe\b", r"\bhis\b", r"\bher\b", r"\bhim\b"]
+        pronoun_count = sum(
+            len(re.findall(pattern, all_memory_text, re.IGNORECASE))
+            for pattern in pronouns
+        )
         print(f"Remaining pronouns: {pronoun_count}")
 
         # The main success criterion: significantly reduced pronoun usage