Update analytics/ai_assistant.py

analytics/ai_assistant.py

@@ -199,6 +199,8 @@ def _infer_data_request(question: str, as_of_month: str | None = None, segment:
         temperature=0.0,
         top_p=0.95
     )
+
+    print(response.choices[0].message.content)
     text = response.choices[0].message.content if hasattr(response, 'choices') else str(response)
     spec = _extract_json(text)
     if not spec:
@@ -341,6 +343,6 @@ def generate_ai_answer(question: str, df: pd.DataFrame, as_of_month: str | None
         temperature=HF_TEMPERATURE,
         top_p=0.95
     )
-
+    print(response.choices[0].message.content)
     output_text = response.choices[0].message.content if hasattr(response, 'choices') else (response[0].get('generated_text') if isinstance(response, list) else str(response))
     return output_text