File size: 2,189 Bytes
63a5dee
 
 
 
 
6b0f762
63a5dee
6b0f762
63a5dee
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6b0f762
63a5dee
 
 
 
6b0f762
 
 
63a5dee
 
 
6b0f762
63a5dee
 
 
 
6b0f762
63a5dee
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6b0f762
63a5dee
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# 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 = 12) -> 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)
        # 1抜粋1200字まで
        ctx_lines.append(f"[{i}] {h.get('title') or ''} {src}\n{h['text'][:1200]}")
    context = "\n\n".join(ctx_lines) if ctx_lines else "(根拠なし)"

    # OpenAIで生成(失敗時は抽出フォールバック)
    try:
        text = generate_chat(
            [{"role":"system","content":SYS},
             {"role":"user","content":TPL.format(context=context)}],
            max_new_tokens=900
        )
        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