brainsqueeze commited on
Commit
a72a624
·
verified ·
1 Parent(s): 1cf41ac

Fixing postprocessing of editorial AI response

Browse files
Files changed (1) hide show
  1. services.py +8 -6
services.py CHANGED
@@ -104,12 +104,14 @@ def cost_estimator(
104
 
105
  def identify_vague_statements(text: str) -> Tuple[List[str], List[str], List[str]]:
106
  data = _get_json("/editorialai/vaguestatement", input_section=text)
107
- return (
108
- data.get("vague_statements") or [],
109
- data.get("alternative_texts") or [],
110
- data.get("reasons") or [],
111
- # data.get("alternative_texts") or [],
112
- )
 
 
113
 
114
 
115
  def help_mission_statement(recipient_name: str, recipient_mission_info: str):
 
104
 
105
  def identify_vague_statements(text: str) -> Tuple[List[str], List[str], List[str]]:
106
  data = _get_json("/editorialai/vaguestatement", input_section=text)
107
+
108
+ statements, alternatives, reasons = [], [], []
109
+ for record in data:
110
+ statements.append(record["vague_statement"])
111
+ alternatives.append(record["alternative_text"])
112
+ reasons.append(record["reason"])
113
+
114
+ return statements, alternatives, reasons
115
 
116
 
117
  def help_mission_statement(recipient_name: str, recipient_mission_info: str):