mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-25 08:46:22 +02:00
aabc24f82c
- Introduced slow callback logging in FastAPI to identify blocking calls. - Added performance logging for agent creation and tool loading processes. - Implemented caching for MCP tools to reduce redundant server calls. - Enhanced sandbox management with in-process caching for improved efficiency. - Refactored several functions for better readability and performance tracking. - Updated tests to ensure proper functionality of new features and optimizations.
38 lines
1,002 B
Python
38 lines
1,002 B
Python
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_summarizer_chain(monkeypatch):
|
|
chain = MagicMock()
|
|
chain.ainvoke = AsyncMock(return_value=MagicMock(content="The summary."))
|
|
|
|
template = MagicMock()
|
|
template.__or__ = MagicMock(return_value=chain)
|
|
|
|
monkeypatch.setattr(
|
|
"app.indexing_pipeline.document_summarizer.SUMMARY_PROMPT_TEMPLATE",
|
|
template,
|
|
)
|
|
return chain
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_chunker_instance(monkeypatch):
|
|
mock = MagicMock()
|
|
mock.chunk.return_value = [MagicMock(text="prose chunk")]
|
|
monkeypatch.setattr(
|
|
"app.indexing_pipeline.document_chunker.config.chunker_instance", mock
|
|
)
|
|
return mock
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_code_chunker_instance(monkeypatch):
|
|
mock = MagicMock()
|
|
mock.chunk.return_value = [MagicMock(text="code chunk")]
|
|
monkeypatch.setattr(
|
|
"app.indexing_pipeline.document_chunker.config.code_chunker_instance", mock
|
|
)
|
|
return mock
|