Coverage for src / local_deep_research / embeddings / __init__.py: 100%

2 statements  

« prev     ^ index     » next       coverage.py v7.12.0, created at 2026-01-11 00:51 +0000

1""" 

2Embedding providers for Local Deep Research. 

3 

4This module provides a unified interface for different embedding providers, 

5similar to the LLM provider system. Supports: 

6- Sentence Transformers (HuggingFace) 

7- Ollama 

8- OpenAI 

9- Future: Cohere, Google Vertex AI, Azure OpenAI, etc. 

10 

11Example: 

12 from local_deep_research.embeddings import get_embeddings 

13 

14 embeddings = get_embeddings( 

15 provider="openai", 

16 model="text-embedding-3-small", 

17 settings_snapshot=settings 

18 ) 

19""" 

20 

21from .embeddings_config import ( 

22 get_embeddings, 

23 get_available_embedding_providers, 

24 is_openai_embeddings_available, 

25 is_ollama_embeddings_available, 

26 is_sentence_transformers_available, 

27) 

28 

29__all__ = [ 

30 "get_embeddings", 

31 "get_available_embedding_providers", 

32 "is_openai_embeddings_available", 

33 "is_ollama_embeddings_available", 

34 "is_sentence_transformers_available", 

35]