fix(reranker): support nested llm config in LLMReranker for non-OpenAI providers (#4405)
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user