Update modules/workflow.py

modules/workflow.py

@@ -2,18 +2,77 @@
 import os
 from typing import Dict, Any, List
 
-# --- retriever は相対インポートを優先、失敗時は絶対インポートにフォールバック ---
+# =========================
+# retriever のロード（相対 → 絶対 → フォールバック実装）
+# =========================
+def _fallback_retrieve_contexts(query: str, top_k: int = 5) -> List[str]:
+    """
+    フォールバックの簡易 RAG:
+    - modules.utils.data_dir()/chunks.jsonl を読み込み
+    - キーワード一致スコアで上位を返す（埋め込みなし）
+    """
+    import json
+    from pathlib import Path
+    try:
+        # 遅延 import（循環回避 & 起動高速化）
+        from modules.utils import data_dir, ensure_dirs
+        ensure_dirs()
+        p = data_dir() / "chunks.jsonl"
+        if not p.exists():
+            return []
+        rows = []
+        with open(p, "r", encoding="utf-8") as f:
+            for line in f:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    obj = json.loads(line)
+                    if isinstance(obj, dict) and obj.get("text"):
+                        rows.append(obj)
+                except Exception:
+                    continue
+        if not rows:
+            return []
+
+        # ごく簡易なキーワード一致スコア
+        q = (query or "").lower()
+        def score(t: str) -> int:
+            t = (t or "").lower()
+            s = 0
+            for w in q.split():
+                if w and w in t:
+                    s += 1
+            return s
+
+        scored = sorted(rows, key=lambda r: score(str(r.get("text",""))), reverse=True)
+        top = scored[:max(1, int(top_k))]
+        out = []
+        for r in top:
+            txt = str(r.get("text","")).strip()
+            src = r.get("source")
+            out.append(f"{txt}\n[source] {src}" if src else txt)
+        return out
+    except Exception:
+        return []
+
+# まず相対 import を試す
 try:
-    from .rag_retriever import retrieve_contexts
+    from .rag_retriever import retrieve_contexts  # type: ignore
 except Exception:
-    from modules.rag_retriever import retrieve_contexts  # type: ignore
+    # 絶対 import
+    try:
+        from modules.rag_retriever import retrieve_contexts  # type: ignore
+    except Exception:
+        # 最後の手段: フォールバックを使う
+        retrieve_contexts = _fallback_retrieve_contexts  # type: ignore
+
 
-# exporters / utils も相対で
+# exporters / utils は相対を優先
 try:
     from .exporters import export_docx, export_pptx
     from .utils import ensure_dirs, export_dir, make_tracking_token
 except Exception:
-    # 万一相対が失敗する環境でも動くように
     from modules.exporters import export_docx, export_pptx  # type: ignore
     from modules.utils import ensure_dirs, export_dir, make_tracking_token  # type: ignore
 
@@ -77,7 +136,7 @@ def run_full_workflow(
 ) -> Dict[str, Any]:
     """
     app.py から呼ばれるメイン処理。
-    - RAG で上位コンテキスト取得
+    - RAG で上位コンテキスト取得（retriever が無い場合はフォールバック）
     - 簡易スコアリング
     - 提案ドラフト生成
     - DOCX/PPTX にエクスポート