""" 使用 ``crawler_copy/jd_pc_search`` 中的采集脚本执行流水线；竞品 Markdown 由 ``pipeline.competitor_report.jd_report`` 生成。 依赖环境变量 ``LOW_GI_PROJECT_ROOT``（由 Django settings 从 ``market_assistant/.env`` 注入）。 """ from __future__ import annotations import json import os import sys from pathlib import Path from typing import Any from django.conf import settings from ..csv.schema import MERGED_FIELD_TO_CSV_HEADER from ..models import PipelineJob def merge_llm_supplement_with_rules_report(llm_md: str, rules_md: str) -> str: """ **以规则引擎全文为正文**（含第五章完整竞品矩阵、各章内嵌统计图与表格）。 大模型稿作为 **8.5 小节**嵌入在 **第八章末、第九章策略** 之前，与第八章第二、三节等具体分析同卷连贯， **不再**插在篇首「## 一、」之前。 注：API「重新生成报告」已不再调用本函数，避免整篇 LLM 与矩阵/图表**数据含义**冲突；保留供脚本或将来显式开关复用。 """ body = (rules_md or "").strip() sup = (llm_md or "").strip() if not sup: return body if not body: return sup marker = "\n---\n\n## 九、策略与机会提示（假设清单，待验证）" insert = ( "\n\n---\n\n" "### 8.5 大模型深度补充（与第二章至第八章第二节正文中的定量内容互补）\n\n" "> **说明**：本段位于**第八章末**；**竞品矩阵、价盘表、统计图与第八章第二、三节等各节正文以相应章节为准**，" "此处为跨小节语义整合，便于衔接第九章。\n\n" f"{sup.strip()}\n" "\n---\n\n## 九、策略与机会提示（假设清单，待验证）" ) if marker in body: return body.replace(marker, insert, 1) # 旧版报告标题或语言差异时的回退 for alt in ( "\n## 九、策略与机会提示（假设清单，待验证）", "\n## 九、策略与机会提示", ): if alt in body and marker not in body: return body.replace( alt, "\n\n---\n\n### 8.5 大模型深度补充（与第二章至第八章第二节正文中的定量内容互补）\n\n" "> **说明**：位于第八章末；**矩阵与图表以正文为准**。\n\n" f"{sup.strip()}\n" + alt, 1, ) app = "\n## 附录 A：数据留存说明" if app in body: tail = ( "\n\n---\n\n### 8.5 大模型深度补充（与第二章至第八章第二节正文中的定量内容互补）\n\n" f"{sup.strip()}\n" ) return body.replace(app, tail + app, 1) return body + "\n\n---\n\n### 8.5 大模型深度补充\n\n" + sup.strip() + "\n" def merge_llm_report_with_rules_charts(llm_md: str, rules_md: str) -> str: """兼容旧名：等价于 ``merge_llm_supplement_with_rules_report``。""" return merge_llm_supplement_with_rules_report(llm_md, rules_md) def _flat_comment_texts(comment_rows: list[dict[str, str]]) -> list[str]: """全部非空评价正文（与报告统计同源）。""" out: list[str] = [] for row in comment_rows: t = (row.get("tagCommentContent") or "").strip() if t: out.append(t) return out def _safe_dir_segment_for_job(s: str, max_len: int = 48) -> str: """与 ``jd_keyword_pipeline._safe_dir_segment`` 一致，避免多线程下改模块全局。""" bad = '<>:"/\\|?*\n\r\t' t = "".join("_" if c in bad else c for c in (s or "").strip())[:max_len] t = t.strip(" .") or "run" return t def resolve_pipeline_run_directory_for_job(job: PipelineJob) -> Path: """ 在拉起子进程前固定本次 ``run_dir``（与 ``jd_keyword_pipeline._resolve_pipeline_run_dir`` **同一规则**）。 调用方负责 ``mkdir``。 """ root = (settings.LOW_GI_PROJECT_ROOT or "").strip() if not root: raise RuntimeError("LOW_GI_PROJECT_ROOT 未配置") project_data = Path(root).resolve() / "data" / "JD" prd = (job.pipeline_run_dir or "").strip() kw = (job.keyword or "").strip() if prd: p = Path(prd).expanduser() if not p.is_absolute(): p = project_data / p return p.resolve() import time stamp = time.strftime("%Y%m%d_%H%M%S") seg = _safe_dir_segment_for_job(kw) return (project_data / "pipeline_runs" / f"{stamp}_{seg}").resolve() def try_write_competitor_report_if_merged_exists( run_dir: Path, keyword: str, *, report_config: dict[str, Any] | None = None, ) -> None: """若已有合并表则补写竞品 Markdown（用于子进程被 terminate 后的部分产物）。""" _, kpl = _jd_crawler_modules() base = Path(run_dir).resolve() merged = base / kpl.FILE_MERGED_CSV if not merged.is_file(): return try: write_competitor_analysis_for_run_dir( base, keyword, report_config=report_config ) except Exception: pass def _jd_crawler_modules(): from pipeline.competitor_report import jd_report as jcr # noqa: WPS433 root = Path(settings.CRAWLER_JD_ROOT) if not root.is_dir(): raise FileNotFoundError(f"爬虫副本目录不存在: {root}") root_s = str(root.resolve()) if root_s not in sys.path: sys.path.insert(0, root_s) import jd_keyword_pipeline as kpl # noqa: WPS433 return jcr, kpl def use_chunked_group_summaries_llm(report_config: dict[str, Any] | None) -> bool: """ 是否按矩阵细类**拆分**第五/六/八章等 group 归纳的 LLM 请求（默认开启）。 关闭方式：``report_config`` 中 ``llm_group_summaries_chunk_by_matrix``: false， 或环境变量 ``MA_LLM_GROUP_SUMMARIES_BULK=1``（恢复单次打包调用）。 """ rc = report_config if isinstance(report_config, dict) else {} if os.environ.get("MA_LLM_GROUP_SUMMARIES_BULK", "").strip().lower() in ( "1", "true", "yes", ): return False return bool(rc.get("llm_group_summaries_chunk_by_matrix", True)) def get_default_report_config() -> dict[str, Any]: """与 ``pipeline.competitor_report.jd_report`` 模块常量一致的默认报告调参（供前端回填）。""" jcr, _ = _jd_crawler_modules() return { "llm_comment_sentiment": True, "llm_matrix_group_summaries": True, "llm_comment_group_summaries": True, "llm_price_group_summaries": True, "llm_promo_group_summaries": True, # 全任务第九章大模型长文已弃用：可执行策略由「策略制定」按矩阵细类生成（见 jd_report 第九章固定说明）。 "llm_strategy_opportunities": False, "llm_group_summaries_chunk_by_matrix": True, "chapter8_text_mining_probe": True, "chapter8_text_mining_probe_live_llm": True, "chapter8_text_mining_probe_llm_chunked": True, "chapter8_text_mining_probe_wordcloud": True, "external_market_table_rows": [ {"indicator": a, "value_and_scope": b, "source": c, "year": d} for a, b, c, d in jcr.EXTERNAL_MARKET_TABLE_ROWS ], } def get_default_strategy_config() -> dict[str, Any]: """策略生成页独立默认（与 ``report_config`` 无关），供前端回填与 PATCH 合并。""" return { # 默认走大模型润色；用户可在单次生成时勾选「仅规则稿」取消 "use_llm_default": True, } def write_competitor_analysis_for_run_dir( run_dir: Path, keyword: str, *, report_config: dict[str, Any] | None = None, ) -> Path: """ 在已有流水线目录上读取 CSV / meta，写入 ``competitor_analysis.md``（不重新爬取）。 """ jcr, kpl = _jd_crawler_modules() kw = (keyword or "").strip() if not kw: raise ValueError("keyword 不能为空") run_dir = Path(run_dir).resolve() merged_path = run_dir / kpl.FILE_MERGED_CSV if not merged_path.is_file(): raise FileNotFoundError(f"缺少合并表，无法生成报告: {merged_path.name}") _, merged_rows = jcr._read_csv_rows(merged_path) _, search_export_rows = jcr._read_csv_rows(run_dir / kpl.FILE_PC_SEARCH_CSV) _, comment_rows = jcr._read_csv_rows(run_dir / kpl.FILE_COMMENTS_FLAT_CSV) meta_path = run_dir / kpl.FILE_RUN_META_JSON meta: dict[str, Any] | None = None if meta_path.is_file(): try: meta = json.loads(meta_path.read_text(encoding="utf-8")) except json.JSONDecodeError: meta = None eff_rc: dict[str, Any] = ( dict(report_config) if isinstance(report_config, dict) else {} ) # 任务仅保存部分调参时，补齐默认项（含各 llm_* 开关）；显式写入的键不被覆盖 for _k, _v in get_default_report_config().items(): if _k not in eff_rc: eff_rc[_k] = _v all_tx = _flat_comment_texts(comment_rows) suggest_path = run_dir / "keyword_suggest_llm.json" suggest_record: dict[str, Any] = { "schema_version": 3, "total_comment_texts": len(all_tx), } skip_kw = os.environ.get("MA_SKIP_LLM_KEYWORD_SUGGEST", "").strip().lower() in ( "1", "true", "yes", ) if not skip_kw: try: from ..llm.keyword_suggest import suggest_focus_keywords_from_all_comments brief_pre = jcr.build_competitor_brief( run_dir=run_dir, keyword=kw, merged_rows=merged_rows, search_export_rows=search_export_rows, comment_rows=comment_rows, meta=meta, report_config=eff_rc, ) brief_slice = { "keyword": brief_pre.get("keyword"), "comment_focus_keywords": [], "usage_scenarios": [], "category_mix_top": (brief_pre.get("category_mix_top") or [])[:6], "scope": brief_pre.get("scope"), } sug = suggest_focus_keywords_from_all_comments( keyword=kw, brief_slice=brief_slice, all_comment_texts=all_tx, ) suggest_record.update(sug) except Exception as e: suggest_record["error"] = str(e) suggest_record["suggested_focus_keywords"] = [] else: suggest_record["skipped"] = True suggest_record["suggested_focus_keywords"] = [] suggest_record["suggested_scenario_groups"] = [] suggest_record["scenario_note"] = "预设场景词组已废弃，不再写入 report_config。" suggest_path.write_text( json.dumps(suggest_record, ensure_ascii=False, indent=2), encoding="utf-8", ) (run_dir / "effective_report_config.json").write_text( json.dumps(eff_rc, ensure_ascii=False, indent=2), encoding="utf-8", ) brief_final = jcr.build_competitor_brief( run_dir=run_dir, keyword=kw, merged_rows=merged_rows, search_export_rows=search_export_rows, comment_rows=comment_rows, meta=meta, report_config=eff_rc, ) from ..reporting.charts import generate_report_charts generate_report_charts(run_dir, brief_final, report_config=eff_rc) llm_sentiment_md = "" sentiment_llm_record: dict[str, Any] = { "schema_version": 2, "attempted": False, "groups": [], } skip_sent = os.environ.get( "MA_SKIP_LLM_COMMENT_SENTIMENT", "" ).strip().lower() in ("1", "true", "yes") env_on = os.environ.get("MA_ENABLE_LLM_COMMENT_SENTIMENT", "").strip().lower() in ( "1", "true", "yes", ) want_sent = bool(eff_rc.get("llm_comment_sentiment")) or env_on if want_sent and not skip_sent: feedback_groups_sg = jcr._consumer_feedback_by_matrix_group( merged_rows=merged_rows, comment_rows=comment_rows, sku_header=MERGED_FIELD_TO_CSV_HEADER["sku_id"], ) if not feedback_groups_sg: sentiment_llm_record["skipped"] = "no_feedback_groups" else: sentiment_llm_record["attempted"] = True from ..llm.generate import generate_comment_sentiment_analysis_llm parts_sent: list[str] = [] grp_logs: list[dict[str, Any]] = [] sku_h = MERGED_FIELD_TO_CSV_HEADER["sku_id"] title_h = MERGED_FIELD_TO_CSV_HEADER["title"] for gname, cr_g, _tu in feedback_groups_sg: sub_merged = [ r for r in merged_rows if jcr._competitor_matrix_group_key(r) == gname ] units, scores = jcr._iter_comment_text_units_and_scores( cr_g, sub_merged ) if len(units) < 2: grp_logs.append( { "group": gname, "skipped": "insufficient_comment_texts", "n_texts": len(units), } ) continue try: attr_units = jcr._comment_lines_with_product_context( cr_g, merged_rows, sku_header=sku_h, title_h=title_h, ) if len(attr_units) != len(units): attr_units = list(units) pl = jcr.build_comment_sentiment_llm_payload( units, scores=scores, attributed_texts=attr_units, max_samples_positive=16, max_samples_negative=30, max_samples_mixed=10, max_chars_per_review=360, semantic_pool_max=40, shuffle_seed=f"{kw}|{gname}", ) pl["keyword"] = kw pl["matrix_group_focus"] = gname md_one = generate_comment_sentiment_analysis_llm(pl) parts_sent.append(f"#### {gname}\n\n{md_one.strip()}\n") grp_logs.append( {"group": gname, "ok": True, "chars": len(md_one)} ) except Exception as e: grp_logs.append({"group": gname, "ok": False, "error": str(e)}) sentiment_llm_record["groups"] = grp_logs llm_sentiment_md = "\n".join(parts_sent).strip() sentiment_llm_record["chars"] = len(llm_sentiment_md) if llm_sentiment_md: sentiment_llm_record["ok"] = True else: sentiment_llm_record["ok"] = False if grp_logs and all("skipped" in x for x in grp_logs): sentiment_llm_record["skipped"] = "insufficient_comment_texts" elif grp_logs and any("error" in x for x in grp_logs): sentiment_llm_record["error"] = "all_groups_failed_or_skipped" else: sentiment_llm_record["error"] = "no_output" elif skip_sent: sentiment_llm_record["skipped"] = "MA_SKIP_LLM_COMMENT_SENTIMENT" elif not want_sent: sentiment_llm_record["skipped"] = "not_enabled" (run_dir / "comment_sentiment_llm.json").write_text( json.dumps(sentiment_llm_record, ensure_ascii=False, indent=2), encoding="utf-8", ) llm_matrix_md = "" llm_price_md = "" llm_promo_md = "" llm_comment_gr_md = "" llm_strategy_opp_md = "" matrix_llm_rec: dict[str, Any] = {"schema_version": 1, "attempted": False} price_llm_rec: dict[str, Any] = {"schema_version": 1, "attempted": False} promo_llm_rec: dict[str, Any] = {"schema_version": 1, "attempted": False} scenario_gr_llm_rec: dict[str, Any] = {"schema_version": 1, "attempted": False} comment_gr_llm_rec: dict[str, Any] = {"schema_version": 1, "attempted": False} strategy_opp_llm_rec: dict[str, Any] = {"schema_version": 1, "attempted": False} sku_h = MERGED_FIELD_TO_CSV_HEADER["sku_id"] title_h = MERGED_FIELD_TO_CSV_HEADER["title"] def _env_on(name: str) -> bool: return os.environ.get(name, "").strip().lower() in ("1", "true", "yes") skip_mx = _env_on("MA_SKIP_LLM_MATRIX_GROUP_SUMMARIES") skip_pr = _env_on("MA_SKIP_LLM_PRICE_GROUP_SUMMARIES") skip_po = _env_on("MA_SKIP_LLM_PROMO_GROUP_SUMMARIES") skip_cg = _env_on("MA_SKIP_LLM_COMMENT_GROUP_SUMMARIES") want_mx = bool(eff_rc.get("llm_matrix_group_summaries")) or _env_on( "MA_ENABLE_LLM_MATRIX_GROUP_SUMMARIES" ) want_pr = bool(eff_rc.get("llm_price_group_summaries")) or _env_on( "MA_ENABLE_LLM_PRICE_GROUP_SUMMARIES" ) want_po = bool(eff_rc.get("llm_promo_group_summaries")) or _env_on( "MA_ENABLE_LLM_PROMO_GROUP_SUMMARIES" ) want_cg = bool(eff_rc.get("llm_comment_group_summaries")) or _env_on( "MA_ENABLE_LLM_COMMENT_GROUP_SUMMARIES" ) skip_st = _env_on("MA_SKIP_LLM_STRATEGY_OPPORTUNITIES") want_st = bool(eff_rc.get("llm_strategy_opportunities")) or _env_on( "MA_ENABLE_LLM_STRATEGY_OPPORTUNITIES" ) use_ch8_probe = bool(eff_rc.get("chapter8_text_mining_probe")) chapter8_probe_embed_md = "" ch8_probe_rec: dict[str, Any] = {"schema_version": 1, "attempted": False} if use_ch8_probe: ch8_probe_rec["attempted"] = True try: from ..demos.chapter8_text_mining_probe import ( build_markdown as build_ch8_probe_full_md, markdown_embed_body_for_competitor_report, ) _rc = eff_rc full_probe = build_ch8_probe_full_md( run_dir, min_texts=int(_rc.get("chapter8_probe_min_texts") or 8), lda_topics_n=int(_rc.get("chapter8_probe_lda_topics") or 4), top_k_words=int(_rc.get("chapter8_probe_top_k_words") or 30), cooc_vocab=int(_rc.get("chapter8_probe_cooc_vocab") or 80), cooc_pairs=int(_rc.get("chapter8_probe_cooc_pairs") or 25), live_llm=bool(_rc.get("chapter8_text_mining_probe_live_llm", True)), llm_chunked=bool( _rc.get("chapter8_text_mining_probe_llm_chunked", True) ), wordcloud_enabled=bool( _rc.get("chapter8_text_mining_probe_wordcloud", True) ), wordcloud_max=int(_rc.get("chapter8_probe_wordcloud_max") or 40), ) (run_dir / "chapter8_text_mining_probe.md").write_text( full_probe, encoding="utf-8" ) chapter8_probe_embed_md = markdown_embed_body_for_competitor_report( full_probe ) ch8_probe_rec["ok"] = True ch8_probe_rec["chars_embed"] = len(chapter8_probe_embed_md) except Exception as e: ch8_probe_rec["ok"] = False ch8_probe_rec["error"] = str(e) if use_ch8_probe and chapter8_probe_embed_md: want_cg = False chunk_gr = use_chunked_group_summaries_llm(eff_rc) if want_mx and not skip_mx and merged_rows: pl_mx = jcr.build_matrix_groups_llm_payload( merged_rows, sku_header=sku_h, title_h=title_h ) if pl_mx: matrix_llm_rec["attempted"] = True try: if chunk_gr: from ..llm.generate import ( generate_matrix_group_summaries_llm_chunked, ) llm_matrix_md = generate_matrix_group_summaries_llm_chunked( pl_mx, keyword=kw ) else: from ..llm.generate import generate_matrix_group_summaries_llm llm_matrix_md = generate_matrix_group_summaries_llm( pl_mx, keyword=kw ) matrix_llm_rec["ok"] = True matrix_llm_rec["chars"] = len(llm_matrix_md) matrix_llm_rec["chunked_by_matrix"] = chunk_gr if chunk_gr: matrix_llm_rec["chunk_count"] = len(pl_mx) except Exception as e: matrix_llm_rec["ok"] = False matrix_llm_rec["error"] = str(e) else: matrix_llm_rec["skipped"] = "empty_matrix_groups_payload" elif skip_mx: matrix_llm_rec["skipped"] = "MA_SKIP_LLM_MATRIX_GROUP_SUMMARIES" elif not want_mx: matrix_llm_rec["skipped"] = "not_enabled" if want_pr and not skip_pr and merged_rows: pl_pr = jcr.build_price_groups_llm_payload( merged_rows, sku_header=sku_h, title_h=title_h ) if pl_pr: price_llm_rec["attempted"] = True try: if chunk_gr: from ..llm.generate import ( generate_price_group_summaries_llm_chunked, ) llm_price_md = generate_price_group_summaries_llm_chunked( pl_pr, keyword=kw ) else: from ..llm.generate import generate_price_group_summaries_llm llm_price_md = generate_price_group_summaries_llm( pl_pr, keyword=kw ) price_llm_rec["ok"] = True price_llm_rec["chars"] = len(llm_price_md) price_llm_rec["chunked_by_matrix"] = chunk_gr if chunk_gr: price_llm_rec["chunk_count"] = len(pl_pr) except Exception as e: price_llm_rec["ok"] = False price_llm_rec["error"] = str(e) else: price_llm_rec["skipped"] = "empty_price_payload" elif skip_pr: price_llm_rec["skipped"] = "MA_SKIP_LLM_PRICE_GROUP_SUMMARIES" elif not want_pr: price_llm_rec["skipped"] = "not_enabled" if want_po and not skip_po and merged_rows: pl_po = jcr.build_promo_groups_llm_payload( merged_rows, sku_header=sku_h, title_h=title_h ) if pl_po: promo_llm_rec["attempted"] = True try: if chunk_gr: from ..llm.generate import ( generate_promo_group_summaries_llm_chunked, ) llm_promo_md = generate_promo_group_summaries_llm_chunked( pl_po, keyword=kw ) else: from ..llm.generate import generate_promo_group_summaries_llm llm_promo_md = generate_promo_group_summaries_llm( pl_po, keyword=kw ) promo_llm_rec["ok"] = True promo_llm_rec["chars"] = len(llm_promo_md) promo_llm_rec["chunked_by_matrix"] = chunk_gr if chunk_gr: promo_llm_rec["chunk_count"] = len(pl_po) except Exception as e: promo_llm_rec["ok"] = False promo_llm_rec["error"] = str(e) else: promo_llm_rec["skipped"] = "empty_promo_payload" elif skip_po: promo_llm_rec["skipped"] = "MA_SKIP_LLM_PROMO_GROUP_SUMMARIES" elif not want_po: promo_llm_rec["skipped"] = "not_enabled" scenario_gr_llm_rec["skipped"] = "preset_scenario_summaries_removed" if want_cg and not skip_cg and merged_rows: fb_cg = jcr._consumer_feedback_by_matrix_group( merged_rows=merged_rows, comment_rows=comment_rows, sku_header=sku_h, ) pl_cg = jcr.build_comment_groups_llm_payload( feedback_groups=fb_cg, merged_rows=merged_rows, sku_header=sku_h, title_h=title_h, ) if pl_cg: comment_gr_llm_rec["attempted"] = True try: if chunk_gr: from ..llm.generate import ( generate_comment_group_summaries_llm_chunked, ) llm_comment_gr_md = generate_comment_group_summaries_llm_chunked( pl_cg, keyword=kw ) else: from ..llm.generate import generate_comment_group_summaries_llm llm_comment_gr_md = generate_comment_group_summaries_llm( pl_cg, keyword=kw ) comment_gr_llm_rec["ok"] = True comment_gr_llm_rec["chars"] = len(llm_comment_gr_md) comment_gr_llm_rec["chunked_by_matrix"] = chunk_gr if chunk_gr: comment_gr_llm_rec["chunk_count"] = len(pl_cg) except Exception as e: comment_gr_llm_rec["ok"] = False comment_gr_llm_rec["error"] = str(e) else: comment_gr_llm_rec["skipped"] = "empty_comment_groups_payload" elif skip_cg: comment_gr_llm_rec["skipped"] = "MA_SKIP_LLM_COMMENT_GROUP_SUMMARIES" elif not want_cg: comment_gr_llm_rec["skipped"] = "not_enabled" if want_st and not skip_st and isinstance(brief_final, dict) and brief_final: strategy_opp_llm_rec["attempted"] = True try: from ..llm.generate import generate_strategy_opportunities_llm _strategy_narratives: dict[str, str] = {} if (llm_matrix_md or "").strip(): _strategy_narratives["sec5_matrix_group_summaries"] = llm_matrix_md if (llm_price_md or "").strip(): _strategy_narratives["sec6_price_group_summaries"] = llm_price_md if (llm_promo_md or "").strip(): _strategy_narratives["sec6_promo_group_summaries"] = llm_promo_md if use_ch8_probe and (chapter8_probe_embed_md or "").strip(): _strategy_narratives["sec8_3_text_mining_probe"] = ( chapter8_probe_embed_md ) elif (llm_comment_gr_md or "").strip(): _strategy_narratives["sec8_3_comment_focus_summaries"] = ( llm_comment_gr_md ) llm_strategy_opp_md = generate_strategy_opportunities_llm( brief_final, keyword=kw, chapter_llm_narratives=_strategy_narratives or None, ) strategy_opp_llm_rec["ok"] = True strategy_opp_llm_rec["chars"] = len(llm_strategy_opp_md) strategy_opp_llm_rec["prior_chapter_narrative_keys"] = sorted( _strategy_narratives.keys() ) if (llm_strategy_opp_md or "").strip(): strategy_opp_llm_rec["markdown"] = llm_strategy_opp_md except Exception as e: strategy_opp_llm_rec["ok"] = False strategy_opp_llm_rec["error"] = str(e) elif skip_st: strategy_opp_llm_rec["skipped"] = "MA_SKIP_LLM_STRATEGY_OPPORTUNITIES" elif not want_st: strategy_opp_llm_rec["skipped"] = "not_enabled" (run_dir / "matrix_groups_llm.json").write_text( json.dumps(matrix_llm_rec, ensure_ascii=False, indent=2), encoding="utf-8", ) (run_dir / "price_groups_llm.json").write_text( json.dumps(price_llm_rec, ensure_ascii=False, indent=2), encoding="utf-8", ) (run_dir / "promo_groups_llm.json").write_text( json.dumps(promo_llm_rec, ensure_ascii=False, indent=2), encoding="utf-8", ) (run_dir / "comment_groups_llm.json").write_text( json.dumps(comment_gr_llm_rec, ensure_ascii=False, indent=2), encoding="utf-8", ) (run_dir / "scenario_groups_llm.json").write_text( json.dumps(scenario_gr_llm_rec, ensure_ascii=False, indent=2), encoding="utf-8", ) (run_dir / "strategy_opportunities_llm.json").write_text( json.dumps(strategy_opp_llm_rec, ensure_ascii=False, indent=2), encoding="utf-8", ) if use_ch8_probe: (run_dir / "chapter8_text_mining_probe.json").write_text( json.dumps(ch8_probe_rec, ensure_ascii=False, indent=2), encoding="utf-8", ) md = jcr.build_competitor_markdown( run_dir=run_dir, keyword=kw, merged_rows=merged_rows, search_export_rows=search_export_rows, comment_rows=comment_rows, meta=meta, report_config=eff_rc, llm_sentiment_section_md=llm_sentiment_md or None, llm_matrix_section_md=llm_matrix_md or None, llm_price_groups_section_md=llm_price_md or None, llm_promo_groups_section_md=llm_promo_md or None, llm_scenario_groups_section_md=None, llm_comment_groups_section_md=llm_comment_gr_md or None, llm_strategy_opportunities_section_md=llm_strategy_opp_md or None, chapter8_text_mining_probe_section_md=chapter8_probe_embed_md or None, ) out_md = run_dir / "competitor_analysis.md" out_md.write_text(md, encoding="utf-8") return run_dir def regenerate_competitor_report( run_dir_str: str, keyword: str, *, report_config: dict[str, Any] | None = None, ) -> Path: """校验 ``run_dir`` 位于 ``LOW_GI_PROJECT_ROOT/data/JD`` 下后，重写竞品 Markdown。""" low_root = (settings.LOW_GI_PROJECT_ROOT or "").strip() if not low_root: raise RuntimeError("LOW_GI_PROJECT_ROOT 未配置") base = Path(run_dir_str).expanduser().resolve() jd_root = (Path(low_root) / "data" / "JD").resolve() try: base.relative_to(jd_root) except ValueError as e: raise ValueError("run_dir 不在京东数据目录下") from e return write_competitor_analysis_for_run_dir( base, keyword, report_config=report_config ) def write_competitor_analysis_markdown(run_dir_str: str, markdown: str) -> Path: """将已生成的 Markdown 正文写入 ``run_dir/competitor_analysis.md``（与规则重生成同路径）。""" low_root = (settings.LOW_GI_PROJECT_ROOT or "").strip() if not low_root: raise RuntimeError("LOW_GI_PROJECT_ROOT 未配置") base = Path(run_dir_str).expanduser().resolve() jd_root = (Path(low_root) / "data" / "JD").resolve() try: base.relative_to(jd_root) except ValueError as e: raise ValueError("run_dir 不在京东数据目录下") from e out = base / "competitor_analysis.md" out.write_text(markdown or "", encoding="utf-8") return out def build_competitor_brief_for_job( run_dir_str: str, keyword: str, *, report_config: dict[str, Any] | None = None, ) -> dict[str, Any]: """ 读取 ``run_dir`` 下合并表 / 搜索导出 / 评价 / meta，返回与 Markdown 报告**同一套计数规则**的 **JSON 结构化摘要**（规则驱动）。 ``run_dir`` 须位于 ``LOW_GI_PROJECT_ROOT/data/JD`` 下。 """ low_root = (settings.LOW_GI_PROJECT_ROOT or "").strip() if not low_root: raise RuntimeError("LOW_GI_PROJECT_ROOT 未配置") base = Path(run_dir_str).expanduser().resolve() jd_root = (Path(low_root) / "data" / "JD").resolve() try: base.relative_to(jd_root) except ValueError as e: raise ValueError("run_dir 不在京东数据目录下") from e jcr, kpl = _jd_crawler_modules() kw = (keyword or "").strip() if not kw: raise ValueError("keyword 不能为空") merged_path = base / kpl.FILE_MERGED_CSV if not merged_path.is_file(): raise FileNotFoundError(f"缺少合并表，无法生成摘要: {merged_path.name}") _, merged_rows = jcr._read_csv_rows(merged_path) _, search_export_rows = jcr._read_csv_rows(base / kpl.FILE_PC_SEARCH_CSV) _, comment_rows = jcr._read_csv_rows(base / kpl.FILE_COMMENTS_FLAT_CSV) meta_path = base / kpl.FILE_RUN_META_JSON meta: dict[str, Any] | None = None if meta_path.is_file(): try: meta = json.loads(meta_path.read_text(encoding="utf-8")) except json.JSONDecodeError: meta = None eff: dict[str, Any] | None = None if isinstance(report_config, dict): eff = dict(report_config) eff_path = base / "effective_report_config.json" if eff_path.is_file(): try: loaded = json.loads(eff_path.read_text(encoding="utf-8")) if isinstance(loaded, dict) and loaded: eff = loaded except json.JSONDecodeError: pass return jcr.build_competitor_brief( run_dir=base, keyword=kw, merged_rows=merged_rows, search_export_rows=search_export_rows, comment_rows=comment_rows, meta=meta, report_config=eff, ) def run_jd_keyword_and_report( keyword: str, *, max_skus: int | None = None, page_start: int | None = None, page_to: int | None = None, pipeline_run_dir: str | None = None, cookie_file_path: str | None = None, pvid: str | None = None, request_delay: str | None = None, list_pages: str | None = None, scenario_filter_enabled: bool | None = None, report_config: dict[str, Any] | None = None, cancel_check: Any | None = None, ) -> Path: _, kpl = _jd_crawler_modules() kw = (keyword or "").strip() if not kw: raise ValueError("keyword 不能为空") backup: dict[str, Any] = {} if cancel_check is not None: backup["PIPELINE_CANCEL_CHECK"] = getattr(kpl, "PIPELINE_CANCEL_CHECK", None) kpl.PIPELINE_CANCEL_CHECK = cancel_check try: if max_skus is not None: backup["MAX_SKUS"] = kpl.MAX_SKUS kpl.MAX_SKUS = max(1, int(max_skus)) if page_start is not None: backup["PAGE_START"] = kpl.PAGE_START kpl.PAGE_START = max(1, int(page_start)) if page_to is not None: backup["PAGE_TO"] = kpl.PAGE_TO kpl.PAGE_TO = max(1, int(page_to)) prd = (pipeline_run_dir or "").strip() if prd: backup["PIPELINE_RUN_DIR"] = kpl.PIPELINE_RUN_DIR kpl.PIPELINE_RUN_DIR = prd cf = (cookie_file_path or "").strip() if cf: backup["PIPELINE_COOKIE_FILE"] = kpl.PIPELINE_COOKIE_FILE kpl.PIPELINE_COOKIE_FILE = cf pv = (pvid or "").strip() if pv: backup["PVID"] = kpl.PVID kpl.PVID = pv rd = (request_delay or "").strip() if rd: backup["REQUEST_DELAY"] = kpl.REQUEST_DELAY kpl.REQUEST_DELAY = rd lp = (list_pages or "").strip() if lp: backup["LIST_PAGES"] = kpl.LIST_PAGES kpl.LIST_PAGES = lp if scenario_filter_enabled is not None: backup["SCENARIO_FILTER_ENABLED"] = kpl.SCENARIO_FILTER_ENABLED kpl.SCENARIO_FILTER_ENABLED = bool(scenario_filter_enabled) run_dir = kpl.main(keyword=kw) except kpl.PipelineCancelled as e: run_dir_path = Path(e.run_dir).resolve() merged = run_dir_path / kpl.FILE_MERGED_CSV if merged.is_file(): try: write_competitor_analysis_for_run_dir( run_dir_path, kw, report_config=report_config ) except Exception: pass raise finally: for name, val in backup.items(): setattr(kpl, name, val) return write_competitor_analysis_for_run_dir( Path(run_dir).resolve(), kw, report_config=report_config )