Update app/openai_client.py

app/openai_client.py

@@ -14,9 +14,7 @@ HEADERS = {
 
 
 async def openai_chat(messages, temperature: float = 0.7, max_tokens: int = 300) -> str:
-    """
-    非同期：通常のチャット生成（UIの広告生成で使用）
-    """
+    """自由回答（プレーンテキスト）"""
     if not OPENAI_API_KEY:
         raise RuntimeError("OPENAI_API_KEY is not set in environment variables.")
 
@@ -34,10 +32,29 @@ async def openai_chat(messages, temperature: float = 0.7, max_tokens: int = 300)
         return data["choices"][0]["message"]["content"].strip()
 
 
+async def openai_chat_json(messages, temperature: float = 0.2, max_tokens: int = 1200) -> dict:
+    """JSONモード（出力は厳密なJSON文字列）"""
+    if not OPENAI_API_KEY:
+        raise RuntimeError("OPENAI_API_KEY is not set in environment variables.")
+
+    url = f"{OPENAI_BASE}/chat/completions"
+    payload = {
+        "model": OPENAI_MODEL,
+        "messages": messages,
+        "temperature": float(temperature),
+        "max_tokens": int(max_tokens),
+        "response_format": {"type": "json_object"},
+    }
+    async with httpx.AsyncClient(timeout=60) as client:
+        r = await client.post(url, headers=HEADERS, json=payload)
+        r.raise_for_status()
+        data = r.json()
+        content = data["choices"][0]["message"]["content"].strip()
+        return json.loads(content)
+
+
 def openai_judge(system: str, user: str) -> dict:
-    """
-    同期：審査用（JSONフォーマットで判定/修正案を返す）
-    """
+    """審査用（JSONモード）。同期呼び出し"""
     if not OPENAI_API_KEY:
         raise RuntimeError("OPENAI_API_KEY is not set in environment variables.")