Rename generators/summaru.py to generators/summary.py

generators/summaru.py

@@ -1,17 +0,0 @@
-from rag.retriever import retrieve, format_citations
-from rag.prompts import SUMMARY_SYS, SUMMARY_USER
-from irpr.deps import generate_chat
-
-def make_summary(query: str):
-    hits = retrieve(query, top_k=10)
-    links = format_citations(hits)
-    link_lines = "\n".join([f"[{i}] {u}" for i, u in links])
-    ctx = "\n".join([f"[chunk:{h['chunk_id']}] {h['text']}" for h in hits])
-
-    messages = [
-        {"role": "system", "content": SUMMARY_SYS},
-        {"role": "user", "content": SUMMARY_USER.format(links=link_lines, contexts=ctx)}
-    ]
-
-    out = generate_chat(messages, max_new_tokens=800, temperature=0.2)
-    return out, links

generators/summary.py

@@ -0,0 +1,69 @@
+# generators/summary.py
+from __future__ import annotations
+from typing import List, Tuple
+from irpr.deps import search, generate_chat
+
+SYS = "あなたは日本語のIRアナリストです。引用部分から経営サマリを作ります。冗長な敬語は避け、要点を構造化してください。マークダウンで出力。"
+
+TPL = """以下は関連文書の抜粋です。これを根拠に、経営サマリを作成してください。
+
+# 出力フォーマット
+業績ハイライト:
+- 箇条書きで3-5項目
+
+見通し:
+- 2-4項目
+
+セグメント:
+- 主要セグメントごとに1-2行
+
+財務:
+- 売上/利益、CF、BSの要点
+
+株主還元:
+- 配当/自社株買い等
+
+ESG:
+- 重要事項があれば簡潔に
+
+リスク:
+- 2-3項目
+
+# 根拠抜粋
+{context}
+"""
+
+def make_summary(query: str, k: int = 10) -> Tuple[str, List[str]]:
+    hits = search(query, top_k=k)
+    links, ctx_lines = [], []
+    for i, h in enumerate(hits, 1):
+        src = h.get("source_url") or ""
+        if src and src not in links: links.append(src)
+        ctx_lines.append(f"[{i}] {h.get('title') or ''} {src}\n{h['text'][:1200]}")
+    context = "\n\n".join(ctx_lines) if ctx_lines else "（根拠なし）"
+
+    # LLM試行（失敗しても握りつぶす）
+    try:
+        text = generate_chat(
+            [{"role":"system","content":SYS},
+             {"role":"user","content":TPL.format(context=context)}],
+            max_new_tokens=700
+        )
+        if text.strip():
+            return text.strip(), links
+    except Exception:
+        pass
+
+    # フォールバック（抽出）
+    if not hits:
+        return "根拠が見つかりませんでした。", links
+    bullets = []
+    for h in hits[:5]:
+        for line in (h["text"][:500]).splitlines():
+            line = line.strip()
+            if 6 <= len(line) <= 120:
+                bullets.append(f"- {line}")
+                if len(bullets) >= 5: break
+        if len(bullets) >= 5: break
+    text = "業績ハイライト:\n" + "\n".join(bullets) + "\n\n見通し:\n- （要約生成なし／抽出）"
+    return text, links