Coverage for intelligence_toolkit/query_text_data/answer_schema.py: 100%

1theme_integration_format = { 

2 "type": "json_schema", 

3 "json_schema": { 

4 "name": "final_report", 

5 "strict": True, 

6 "schema": { 

7 "type": "object", 

8 "properties": { 

9 "report_title": { 

10 "type": "string" 

11 }, 

12 "report_overview": { 

13 "type": "string" 

14 }, 

15 "report_implications": { 

16 "type": "string" 

17 }, 

18 "answer": { 

19 "type": "string" 

20 }, 

21 }, 

22 "required": ["report_title", "report_overview", "report_implications", "answer"], 

23 "additionalProperties": False, 

24 } 

25 } 

26} 

27 

28 

29theme_summarization_format = { 

30 "type": "json_schema", 

31 "json_schema": { 

32 "name": "theme_summary", 

33 "strict": True, 

34 "schema": { 

35 "type": "object", 

36 "properties": { 

37 "theme_title": { 

38 "type": "string" 

39 }, 

40 "theme_points": { 

41 "type": "array", 

42 "items": { 

43 "type": "object", 

44 "properties": { 

45 "point_title": { 

46 "type": "string" 

47 }, 

48 "point_evidence": { 

49 "type": "string" 

50 }, 

51 "point_commentary": { 

52 "type": "string" 

53 } 

54 }, 

55 "required": ["point_title", "point_evidence", "point_commentary"], 

56 "additionalProperties": False, 

57 } 

58 } 

59 

60 }, 

61 "required": ["theme_title", "theme_points"], 

62 "additionalProperties": False, 

63 } 

64 } 

65} 

66 

67thematic_update_format = { 

68 "type": "json_schema", 

69 "json_schema": { 

70 "name": "thematic_analysis", 

71 "strict": True, 

72 "schema": { 

73 "type": "object", 

74 "properties": { 

75 "updates": { 

76 "type": "array", 

77 "items": { 

78 "type": "object", 

79 "properties": { 

80 "point_id": { 

81 "type": "number" 

82 }, 

83 "point_title": { 

84 "type": "string" 

85 }, 

86 "source_ids": { 

87 "type": "array", 

88 "items": { 

89 "type": "number" 

90 } 

91 } 

92 }, 

93 "required": ["point_id", "point_title", "source_ids"], 

94 "additionalProperties": False, 

95 } 

96 }, 

97 "themes": { 

98 "type": "array", 

99 "items": { 

100 "type": "object", 

101 "properties": { 

102 "theme_title": { 

103 "type": "string" 

104 }, 

105 "point_ids": { 

106 "type": "array", 

107 "items": { 

108 "type": "number" 

109 } 

110 } 

111 }, 

112 "required": ["theme_title", "point_ids"], 

113 "additionalProperties": False, 

114 }, 

115 } 

116 }, 

117 "required": ["updates", "themes"], 

118 "additionalProperties": False, 

119 } 

120 } 

121}