Update posts_categorization.py

posts_categorization.py

@@ -40,7 +40,10 @@ def summarize_post(text):
         response = client.chat.completions.create(
             model=PRIMARY_SUMMARIZER_MODEL,
             response_model=SummaryOutput,
-            messages=messages,
+            messages=messages=[
+                {"role": "system", "content": "You are a precise summarizer. Only return a JSON object with a 'summary' string."},
+                {"role": "user", "content": prompt}
+                ],
             temperature=0.3
         )
         return response.summary
@@ -51,7 +54,10 @@ def summarize_post(text):
             response = client.chat.completions.create(
                 model=FALLBACK_SUMMARIZER_MODEL,
                 response_model=SummaryOutput,
-                messages=messages,
+                messages=messages=[
+                    {"role": "system", "content": "You are a precise summarizer. Only return a JSON object with a 'summary' string."},
+                    {"role": "user", "content": prompt}
+                    ],
                 temperature=0.3 # Keep temperature consistent or adjust as needed for fallback
             )
             print(f"Summarization successful with fallback model: {FALLBACK_SUMMARIZER_MODEL}")