Coverage for apps/inners/models/dtos/contracts/requests/passage_searches/input_setting_body.py: 100%

1from typing import List, Optional 

2from uuid import UUID 

3 

4from apps.inners.models.dtos.contracts.requests.base_request import BaseRequest 

5 

6 

7class LlmSetting(BaseRequest): 

8 model_name: str 

9 max_token: int 

10 

11 

12class PreprocessorSetting(BaseRequest): 

13 is_force_refresh_categorized_element: bool 

14 is_force_refresh_categorized_document: bool 

15 file_partition_strategy: str 

16 chunk_size: int 

17 overlap_size: int 

18 is_include_table: bool 

19 is_include_image: bool 

20 

21 

22class EmbedderSetting(BaseRequest): 

23 is_force_refresh_embedding: bool 

24 is_force_refresh_document: bool 

25 model_name: str 

26 query_instruction: Optional[str] 

27 

28 

29class RetrieverSetting(BaseRequest): 

30 is_force_refresh_relevant_document: bool 

31 top_k: int 

32 

33 

34class RerankerSetting(BaseRequest): 

35 is_force_refresh_re_ranked_document: bool 

36 model_name: str 

37 top_k: int 

38 

39 

40class InputSettingBody(BaseRequest): 

41 document_ids: List[UUID] 

42 llm_setting: LlmSetting 

43 preprocessor_setting: PreprocessorSetting 

44 embedder_setting: EmbedderSetting 

45 retriever_setting: RetrieverSetting 

46 reranker_setting: RerankerSetting 

47 question: str