Coverage for src / qdrant_loader_mcp_server / mcp / schemas / __init__.py: 100%

1from typing import Any 

2 

3from .analyze_relationships import get_analyze_relationships_tool_schema 

4from .attachment import get_attachment_search_tool_schema 

5from .cluster_documents import get_cluster_documents_tool_schema 

6from .detect_conflicts import get_detect_conflicts_tool_schema 

7from .expand_chunk_context import get_expand_chunk_context_tool_schema 

8from .expand_cluster import get_expand_cluster_tool_schema 

9from .expand_document import get_expand_document_tool_schema 

10from .find_complementary import get_find_complementary_tool_schema 

11from .find_similar import get_find_similar_tool_schema 

12from .hierarchy import get_hierarchy_search_tool_schema 

13from .search import get_search_tool_schema 

14 

15 

16def get_all_tool_schemas() -> list[dict[str, Any]]: 

17 # Backward-compatibility alias: expose detect_conflicts under legacy name 

18 # 'detect_document_conflicts' in the tools list while the underlying schema 

19 # function uses the canonical 'detect_conflicts' name. 

20 conflicts_schema = get_detect_conflicts_tool_schema() 

21 conflicts_schema_legacy = {**conflicts_schema, "name": "detect_document_conflicts"} 

22 

23 return [ 

24 get_search_tool_schema(), 

25 get_hierarchy_search_tool_schema(), 

26 get_attachment_search_tool_schema(), 

27 get_analyze_relationships_tool_schema(), 

28 get_find_similar_tool_schema(), 

29 conflicts_schema_legacy, 

30 get_find_complementary_tool_schema(), 

31 get_cluster_documents_tool_schema(), 

32 get_expand_document_tool_schema(), 

33 get_expand_cluster_tool_schema(), 

34 get_expand_chunk_context_tool_schema(), 

35 ] 

36 

37 

38class MCPSchemas: 

39 """Backward-compatible wrapper exposing static methods for tests/importers.""" 

40 

41 get_search_tool_schema = staticmethod(get_search_tool_schema) 

42 get_hierarchy_search_tool_schema = staticmethod(get_hierarchy_search_tool_schema) 

43 get_attachment_search_tool_schema = staticmethod(get_attachment_search_tool_schema) 

44 get_analyze_relationships_tool_schema = staticmethod( 

45 get_analyze_relationships_tool_schema 

46 ) 

47 get_find_similar_tool_schema = staticmethod(get_find_similar_tool_schema) 

48 get_detect_conflicts_tool_schema = staticmethod(get_detect_conflicts_tool_schema) 

49 get_find_complementary_tool_schema = staticmethod( 

50 get_find_complementary_tool_schema 

51 ) 

52 get_cluster_documents_tool_schema = staticmethod(get_cluster_documents_tool_schema) 

53 get_expand_document_tool_schema = staticmethod(get_expand_document_tool_schema) 

54 get_expand_cluster_tool_schema = staticmethod(get_expand_cluster_tool_schema) 

55 get_expand_chunk_context_tool_schema = staticmethod( 

56 get_expand_chunk_context_tool_schema 

57 ) 

58 

59 @classmethod 

60 def get_all_tool_schemas(cls) -> list[dict[str, Any]]: 

61 return get_all_tool_schemas() 

62 

63 

64__all__ = [ 

65 "get_search_tool_schema", 

66 "get_hierarchy_search_tool_schema", 

67 "get_attachment_search_tool_schema", 

68 "get_analyze_relationships_tool_schema", 

69 "get_find_similar_tool_schema", 

70 "get_detect_conflicts_tool_schema", 

71 "get_find_complementary_tool_schema", 

72 "get_cluster_documents_tool_schema", 

73 "get_expand_document_tool_schema", 

74 "get_expand_cluster_tool_schema", 

75 "get_expand_chunk_context_tool_schema", 

76 "get_all_tool_schemas", 

77 "MCPSchemas", 

78]