Coverage for presidio_analyzer / llm_utils / __init__.py: 100%

1"""Utilities for LLM-based recognizers.""" 

2 

3from .config_loader import ( 

4 get_conf_path, 

5 get_model_config, 

6 load_yaml_file, 

7 resolve_config_path, 

8 validate_config_fields, 

9) 

10from .entity_mapper import ( 

11 GENERIC_PII_ENTITY, 

12 consolidate_generic_entities, 

13 ensure_generic_entity_support, 

14 filter_results_by_entities, 

15 filter_results_by_labels, 

16 filter_results_by_score, 

17 skip_unmapped_entities, 

18 validate_result_positions, 

19) 

20from .examples_loader import convert_to_langextract_format, load_yaml_examples 

21from .langextract_helper import ( 

22 calculate_extraction_confidence, 

23 check_langextract_available, 

24 convert_langextract_to_presidio_results, 

25 create_reverse_entity_mapping, 

26 extract_lm_config, 

27 get_supported_entities, 

28 lx, 

29 lx_factory, 

30) 

31from .prompt_loader import load_file_from_conf, load_prompt_file, render_jinja_template 

32 

33__all__ = [ 

34 "get_conf_path", 

35 "get_model_config", 

36 "load_yaml_file", 

37 "resolve_config_path", 

38 "validate_config_fields", 

39 "GENERIC_PII_ENTITY", 

40 "consolidate_generic_entities", 

41 "ensure_generic_entity_support", 

42 "filter_results_by_entities", 

43 "filter_results_by_labels", 

44 "filter_results_by_score", 

45 "skip_unmapped_entities", 

46 "validate_result_positions", 

47 "convert_to_langextract_format", 

48 "load_yaml_examples", 

49 "calculate_extraction_confidence", 

50 "check_langextract_available", 

51 "convert_langextract_to_presidio_results", 

52 "create_reverse_entity_mapping", 

53 "extract_lm_config", 

54 "get_supported_entities", 

55 "lx", 

56 "lx_factory", 

57 "load_file_from_conf", 

58 "load_prompt_file", 

59 "render_jinja_template", 

60] 

61