Skip to content

Commit 75adb51

Browse files
committed
add tests
1 parent 3d20d25 commit 75adb51

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_thread_aware_grounding.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ async def test_thread_aware_pronoun_resolution(self):
7979

8080
print(f"\nExtracted memories: {len(extracted_memories)}")
8181
for i, mem in enumerate(extracted_memories):
82-
print(f"{i+1}. [{mem.memory_type}] {mem.text}")
82+
print(f"{i + 1}. [{mem.memory_type}] {mem.text}")
8383

8484
print(f"\nCombined memory text: {all_memory_text}")
8585

@@ -200,7 +200,7 @@ async def test_multi_entity_conversation(self):
200200

201201
print(f"\nMulti-entity extracted memories: {len(extracted_memories)}")
202202
for i, mem in enumerate(extracted_memories):
203-
print(f"{i+1}. [{mem.memory_type}] {mem.text}")
203+
print(f"{i + 1}. [{mem.memory_type}] {mem.text}")
204204

205205
# Should mention both John and Sarah by name
206206
assert "john" in all_memory_text.lower(), "Should mention John by name"

0 commit comments

Comments
 (0)