Coverage for /home/admin/Documents/AI/applications/lexigram-dev/lexigram/experimental/ai/lexigram-ai-workers/src/lexigram/ai/workers/batch_embedding/__init__.py: 100%

1""" 

2Batch embedding worker for efficient embedding generation. 

3 

4Handles batch processing of embeddings with: 

5- Queue-based job management 

6- Batch optimization (minimize API calls) 

7- Progress tracking and resumability 

8- Automatic retry on failures 

9- Integration with embedding cache 

10""" 

11 

12from __future__ import annotations 

13 

14from lexigram.ai.workers.batch_embedding.cache import EmbeddingCache 

15from lexigram.ai.workers.batch_embedding.progress import ProgressTracker 

16from lexigram.ai.workers.batch_embedding.protocols import EmbeddingProvider 

17from lexigram.ai.workers.batch_embedding.types import ( 

18 BatchEmbeddingJob, 

19 BatchEmbeddingProgress, 

20 BatchEmbeddingResult, 

21 EmbeddingStatus, 

22) 

23from lexigram.ai.workers.batch_embedding.worker import BatchEmbeddingWorker 

24 

25__all__ = [ 

26 "BatchEmbeddingJob", 

27 "BatchEmbeddingProgress", 

28 "BatchEmbeddingResult", 

29 "BatchEmbeddingWorker", 

30 "EmbeddingCache", 

31 "EmbeddingProvider", 

32 "EmbeddingStatus", 

33 "ProgressTracker", 

34]