From b19684057babd0a0b88b6f453da0f78ff00c759a Mon Sep 17 00:00:00 2001 From: Mohammad Amin Date: Wed, 7 Aug 2024 16:23:23 +0330 Subject: [PATCH] In order to reduce costs feat: limiting the number of chat completions to 1 in order to reduce costs --- dags/violation_detection_helpers/utils/classify.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dags/violation_detection_helpers/utils/classify.py b/dags/violation_detection_helpers/utils/classify.py index a6040c65..232d8244 100644 --- a/dags/violation_detection_helpers/utils/classify.py +++ b/dags/violation_detection_helpers/utils/classify.py @@ -42,6 +42,7 @@ def classify(self, text: str) -> str: + "\n\n[VIOLATION DETECTIONS]\n", }, ], + n=1, ) # the label to add to general data structure