Ensure GPT labels formatted correctly

IDinsight · May 2, 2024 · 3800d73 · 3800d73
1 parent 3d15d53
commit 3800d73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/scripts/send_slack_report/initial_query_classification.py b/backend/scripts/send_slack_report/initial_query_classification.py
@@ -125,4 +125,4 @@ def label_question(user_question, client):
     )
     output = chat_completion.choices[0].message.content
     label = re.findall(r"<Response>\s*([\s\S]*?)\s*</Response>", output)[0]
-    return label
+    return label.upper()