fix(tests): fix remaining 6 failures — mock setup for v3 pipeline

- _get_all_from_vector_store: top_k -> limit (2 missed occurrences)
- test_empty_llm_response_fact_extraction (sync+async): add
  db.get_last_messages mock returning [], set custom_instructions=None,
  fix log assertion to use record.message not record.msg
- test_thinking_tags (sync+async): add embed_batch mock, db mock
  with get_last_messages=[], set custom_instructions=None

Root cause: v3 pipeline calls self.db.get_last_messages() and
self.embedding_model.embed_batch() which weren't mocked, causing
the pipeline to silently fail before reaching the LLM call.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
chaithanyak42
2026-04-14 02:32:29 +05:30
co-authored by Claude Opus 4.6
parent ce38ab7840
commit 60065f3792
2 changed files with 30 additions and 20 deletions
+10 -2
View File
@@ -144,6 +144,7 @@ def create_mocked_memory():
mock_embedder = MagicMock()
mock_embedder.embed.return_value = [0.1, 0.2, 0.3]
mock_embedder.embed_batch.return_value = [[0.1, 0.2, 0.3]]
mock_embedder_factory.return_value = mock_embedder
mock_vector_store = MagicMock()
@@ -151,9 +152,12 @@ def create_mocked_memory():
mock_vector_store.add.return_value = None
mock_vector_factory.return_value = mock_vector_store
mock_sqlite.return_value = MagicMock()
mock_db = MagicMock()
mock_db.get_last_messages.return_value = []
mock_sqlite.return_value = mock_db
memory = Memory()
memory.custom_instructions = None
memory.api_version = "v1.0"
return memory, mock_llm, mock_vector_store
@@ -170,6 +174,7 @@ def create_mocked_async_memory():
mock_embedder = MagicMock()
mock_embedder.embed.return_value = [0.1, 0.2, 0.3]
mock_embedder.embed_batch.return_value = [[0.1, 0.2, 0.3]]
mock_embedder_factory.return_value = mock_embedder
mock_vector_store = MagicMock()
@@ -177,9 +182,12 @@ def create_mocked_async_memory():
mock_vector_store.add.return_value = None
mock_vector_factory.return_value = mock_vector_store
mock_sqlite.return_value = MagicMock()
mock_db = MagicMock()
mock_db.get_last_messages.return_value = []
mock_sqlite.return_value = mock_db
memory = AsyncMemory()
memory.custom_instructions = None
memory.api_version = "v1.0"
return memory, mock_llm, mock_vector_store