Skip to content

Commit a3bc126

Browse files
committed
test: fix test
1 parent 010ea92 commit a3bc126

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/mem_reader/test_simple_structure.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ def test_process_doc_data(self):
7575
info = {"user_id": "user1", "session_id": "session1"}
7676

7777
# Mock LLM response
78-
mock_response = '{"summary": "A sample document about testing.", "tags": ["document"]}'
78+
mock_response = (
79+
'{"value": "A sample document about testing.", "tags": ["document"], "key": "title"}'
80+
)
7981
self.reader.llm.generate.return_value = mock_response
8082
self.reader.chunker.chunk.return_value = [
8183
Chunk(text="Parsed document text", token_count=3, sentences=["Parsed document text"])

0 commit comments

Comments
 (0)