Create modules/result_synthesizer.py

modules/result_synthesizer.py

@@ -0,0 +1,34 @@
+# modules/result_synthesizer.py
+import os
+from openai import AsyncOpenAI
+
+client = AsyncOpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+class ResultSynthesizer:
+    def __init__(self):
+        pass
+
+    async def synthesize(self, results, output_lang):
+        formatted_results = self._format_results(results)
+
+        response = await client.chat.completions.create(
+            model="gpt-4o",
+            messages=[
+                {"role": "system", "content": f"You are a multilingual summarizer. Your job is to synthesize information into well-organized output in {output_lang}."},
+                {"role": "user", "content": formatted_results}
+            ]
+        )
+        return response.choices[0].message.content
+
+    def _format_results(self, results):
+        texts = []
+        for r in results:
+            q = r.get("query")
+            if "results" in r:
+                for res in r["results"]:
+                    text = res.get("summary", "")
+                    url = res.get("url", "")
+                    texts.append(f"[Query: {q}]\n{text}\n(Source: {url})")
+            elif "error" in r:
+                texts.append(f"[Query: {q}]\nError: {r['error']}")
+        return "\n\n".join(texts)