amorfo77/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2026-09-27 20:00:31 +00:00
Code Issues Projects Releases Wiki Activity

Chore: Lazy imports of the heavy AI modules (#12275)

Browse Source
This commit is contained in:
Trenton H
2026-03-07 12:53:22 -08:00
committed by GitHub
parent 91ddda9256
commit f5c0c21922
9 changed files with 78 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/paperless_ai/tests/test_embedding.py
+2 -2
View File
@@ -67,7 +67,7 @@ def test_get_embedding_model_openai(mock_ai_config):
mock_ai_config.return_value.llm_api_key = "test_api_key"
mock_ai_config.return_value.llm_endpoint = "http://test-url"
with patch("paperless_ai.embedding.OpenAIEmbedding") as MockOpenAIEmbedding:
with patch("llama_index.embeddings.openai.OpenAIEmbedding") as MockOpenAIEmbedding:
model = get_embedding_model()
MockOpenAIEmbedding.assert_called_once_with(
model="text-embedding-3-small",
@@ -84,7 +84,7 @@ def test_get_embedding_model_huggingface(mock_ai_config):
)
with patch(
"paperless_ai.embedding.HuggingFaceEmbedding",
"llama_index.embeddings.huggingface.HuggingFaceEmbedding",
) as MockHuggingFaceEmbedding:
model = get_embedding_model()
MockHuggingFaceEmbedding.assert_called_once_with(