Add debug prints

weni-ai · Nov 22, 2024 · 6db03fd · 6db03fd
1 parent 9fd16be
commit 6db03fd
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/router/classifiers/chatgpt_function.py b/router/classifiers/chatgpt_function.py
@@ -109,12 +109,20 @@ def predict(
             classification = self.CLASSIFICATION_OTHER
             return classification
 
+        print("[+ ChatGPT message function classification data +]")
+        print(f"Tools: {flows_list}")
+        print(f"Model: {self.chatgpt_model}")
+        print(f"Message: {msg}")
+
         response = self.client.chat_completions_create(
             model=self.chatgpt_model,
             messages=msg,
             tools=flows_list,
             tool_choice="auto"
         )
+        print("[+ ChatGPT message function classification response +]")
+        print(f"Response: {response}")
+        print("[++++++++++++++++++++++++++++++++++++++++++++++++]")
 
         tool_calls = response.choices[0].message.tool_calls