mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-06 20:15:17 +02:00
feat(tests): Update tests for summary-free indexing
This commit is contained in:
parent
dc6a17930b
commit
ddfe60c2f0
26 changed files with 123 additions and 294 deletions
|
|
@ -206,23 +206,23 @@ def _patch_llm_bindings() -> None:
|
|||
fake_create_chat_litellm_from_agent_config,
|
||||
fake_create_chat_litellm_from_config,
|
||||
)
|
||||
from tests.e2e.fakes.llm import fake_get_user_long_context_llm
|
||||
from tests.e2e.fakes.llm import fake_get_agent_llm
|
||||
|
||||
targets = [
|
||||
"app.services.llm_service.get_user_long_context_llm",
|
||||
"app.tasks.connector_indexers.confluence_indexer.get_user_long_context_llm",
|
||||
"app.tasks.connector_indexers.google_drive_indexer.get_user_long_context_llm",
|
||||
"app.tasks.connector_indexers.google_gmail_indexer.get_user_long_context_llm",
|
||||
"app.tasks.connector_indexers.notion_indexer.get_user_long_context_llm",
|
||||
"app.tasks.connector_indexers.onedrive_indexer.get_user_long_context_llm",
|
||||
"app.tasks.connector_indexers.dropbox_indexer.get_user_long_context_llm",
|
||||
"app.tasks.connector_indexers.local_folder_indexer.get_user_long_context_llm",
|
||||
"app.tasks.document_processors._save.get_user_long_context_llm",
|
||||
"app.tasks.document_processors.markdown_processor.get_user_long_context_llm",
|
||||
"app.services.llm_service.get_agent_llm",
|
||||
"app.tasks.connector_indexers.confluence_indexer.get_agent_llm",
|
||||
"app.tasks.connector_indexers.google_drive_indexer.get_agent_llm",
|
||||
"app.tasks.connector_indexers.google_gmail_indexer.get_agent_llm",
|
||||
"app.tasks.connector_indexers.notion_indexer.get_agent_llm",
|
||||
"app.tasks.connector_indexers.onedrive_indexer.get_agent_llm",
|
||||
"app.tasks.connector_indexers.dropbox_indexer.get_agent_llm",
|
||||
"app.tasks.connector_indexers.local_folder_indexer.get_agent_llm",
|
||||
"app.tasks.document_processors._save.get_agent_llm",
|
||||
"app.tasks.document_processors.markdown_processor.get_agent_llm",
|
||||
]
|
||||
for target in targets:
|
||||
try:
|
||||
p = patch(target, fake_get_user_long_context_llm)
|
||||
p = patch(target, fake_get_agent_llm)
|
||||
p.start()
|
||||
_active_patches.append(p)
|
||||
logger.info("[fake-llm] patched %s", target)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue