chore: adding test for llm reranker

This commit is contained in:
kartik-mem0
2026-03-19 00:31:30 +05:30
parent 5154174342
commit d14470d6bb
4 changed files with 353 additions and 0 deletions
+11
View File
@@ -0,0 +1,11 @@
from unittest.mock import MagicMock, patch
import pytest
@pytest.fixture
def mock_llm():
with patch("mem0.reranker.llm_reranker.LlmFactory") as mock_factory:
mock_llm_instance = MagicMock()
mock_factory.create.return_value = mock_llm_instance
yield mock_factory, mock_llm_instance