Update generators/qa.py

generators/qa.py

@@ -6,7 +6,7 @@ from irpr.deps import search, generate_chat
 SYS = "あなたは日本語のIR担当です。投資家からの想定質問と模範回答を、根拠に基づいて簡潔に作成します。"
 
 TPL = """次の抜粋を根拠に、投資家向けの想定Q&Aを {n} 問作成してください。
-各問は "Q: ... / A: ..." の2行で、Aは2-4文以内。根拠があれば括弧で短く示してください。
+各問は "Q: ... / A: ..." の2行。Aは2-4文以内。根拠があれば括弧で短く示してください。
 
 # 抜粋
 {context}
@@ -21,21 +21,25 @@ def make_qa(query: str, n: int = 30) -> Tuple[List[dict], List[str]]:
         ctx.append(f"[{i}] {h.get('title') or ''} {src}\n{h['text'][:1000]}")
     context = "\n\n".join(ctx) if ctx else "（根拠なし）"
 
-    # LLM試行
+    # OpenAI で生成
     qa_list: List[dict] = []
     try:
         out = generate_chat(
             [{"role":"system","content":SYS},
              {"role":"user","content":TPL.format(n=n, context=context)}],
-            max_new_tokens=1000
+            max_new_tokens=1500
         )
+        cur = None
         for line in out.splitlines():
             line = line.strip()
             if line.startswith("Q:"):
-                qa_list.append({"q": line[2:].strip(), "a": ""})
-            elif line.startswith("A:") and qa_list:
-                qa_list[-1]["a"] = line[2:].strip()
-        qa_list = [x for x in qa_list if x.get("q") and x.get("a")]
+                if cur and cur.get("q") and cur.get("a"):
+                    qa_list.append(cur)
+                cur = {"q": line[2:].strip(), "a": ""}
+            elif line.startswith("A:") and cur:
+                cur["a"] = line[2:].strip()
+        if cur and cur.get("q") and cur.get("a"):
+            qa_list.append(cur)
         if qa_list:
             return qa_list[:n], links
     except Exception: