Coverage for apps/outers/containers/use_cases/graph_container.py: 100%

1from dependency_injector import providers 

2from dependency_injector.containers import DeclarativeContainer 

3 

4from apps.inners.use_cases.graphs.long_form_qa_graph import LongFormQaGraph 

5from apps.inners.use_cases.graphs.passage_search_graph import PassageSearchGraph 

6 

7 

8class GraphContainer(DeclarativeContainer): 

9 settings = providers.DependenciesContainer() 

10 datastores = providers.DependenciesContainer() 

11 document_processors = providers.DependenciesContainer() 

12 

13 passage_search = providers.Singleton( 

14 PassageSearchGraph, 

15 one_llm_setting=settings.one_llm, 

16 two_llm_setting=settings.two_llm, 

17 two_datastore=datastores.two, 

18 four_datastore=datastores.four, 

19 partition_document_processor=document_processors.partition, 

20 category_document_processor=document_processors.category, 

21 ) 

22 

23 long_form_qa = providers.Singleton( 

24 LongFormQaGraph, 

25 one_llm_setting=settings.one_llm, 

26 two_llm_setting=settings.two_llm, 

27 two_datastore=datastores.two, 

28 four_datastore=datastores.four, 

29 partition_document_processor=document_processors.partition, 

30 category_document_processor=document_processors.category, 

31 )