sino-bc-cli 1.0.0-beta.0

package/dist/skills/biz-line-strategy-generate/scripts/test_ai_strategy.py

@@ -0,0 +1,235 @@
+"""AI 文案交接的离线回归：只核对请求归属与回包，不执行投放分析。"""
+import json
+import sys
+import tempfile
+import threading
+import unittest
+import zipfile
+from pathlib import Path
+from xml.etree import ElementTree as ET
+
+sys.dont_write_bytecode = True
+sys.path.insert(0, str(Path(__file__).resolve().parent))
+
+from ai_strategy import AiStrategyExhausted, review_ai_strategy
+from build_report import read_ai_sheet, validate_detail_workbook, validate_xlsx, write_detail_workbook, write_xlsx
+
+AI_TEXT = ('一页速览\n- 本期观察。\n'
+           '【预算策略】\n【策略总结】\n- **有效**：消耗 **$12.00**。\n'
+           '【广告产品策略】\n【调整方向】\n- 保持观察。\n'
+           '【受众策略】\n【关键数据】\n- 样本。\n'
+           '【创意策略】\n【Learning】\n- 待观察。')
+
+
+class AiStrategyReview(unittest.TestCase):
+    """复用已完成的明细-only 工作簿，三次请求各有独立身份。"""
+
+    def fixture(self, root):
+        """请求中的提示词与工作簿路径均指向本次目录的真实文件。"""
+        workbook = root / 'ai-detail.xlsx'
+        prompt = root / 'prompt.md'
+        workbook.write_bytes(b'workbook fixture')
+        prompt.write_text('请根据 Excel 明细表生成复盘。', encoding='utf-8')
+        return workbook, prompt
+
+    def test_bound_reply_reads_current_prompt(self):
+        """Agent 只把当前请求的身份写回，脚本不制造策略文本。"""
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp)
+            workbook, prompt = self.fixture(root)
+            seen = []
+
+            def progress(event):
+                seen.append(event)
+                if event.get('event') != 'ai_strategy_required':
+                    return
+                request = json.loads(Path(event['request_path']).read_text(encoding='utf-8'))
+                self.assertEqual(request['prompt'], prompt.read_text(encoding='utf-8'))
+                self.assertEqual(request['workbookPath'], str(workbook))
+                self.assertNotIn('payloadPath', request)
+                self.assertEqual(request['bizLineId'], 321)
+                reply = {key: request[key] for key in ('bizLineId', 'attempt', 'requestId')}
+                reply['text'] = AI_TEXT
+                Path(event['response_path']).write_text(json.dumps(reply, ensure_ascii=False), encoding='utf-8')
+
+            value = review_ai_strategy(root, workbook, prompt, threading.Event(), progress,
+                                       biz_line_id=321, attempt=2, wait_seconds=1)
+            self.assertIn('消耗', value)
+            self.assertEqual([item['event'] for item in seen],
+                             ['ai_strategy_required', 'ai_strategy_accepted'])
+
+    def test_wrong_identity_then_valid_reply_retries_only_ai(self):
+        """第一次错线被拒，第二次使用新请求 ID；草稿不重建。"""
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp)
+            workbook, prompt = self.fixture(root)
+            requests = []
+
+            def progress(event):
+                if event.get('event') != 'ai_strategy_required':
+                    return
+                request = json.loads(Path(event['request_path']).read_text(encoding='utf-8'))
+                requests.append(request)
+                reply = {key: request[key] for key in ('bizLineId', 'attempt', 'requestId')}
+                reply['text'] = AI_TEXT
+                if len(requests) == 1:
+                    reply['bizLineId'] = 999
+                Path(event['response_path']).write_text(json.dumps(reply), encoding='utf-8')
+
+            value = review_ai_strategy(root, workbook, prompt, threading.Event(), progress,
+                                       biz_line_id=321, attempt=1, wait_seconds=1)
+            self.assertEqual(value, AI_TEXT)
+            self.assertEqual(len(requests), 2)
+            self.assertNotEqual(requests[0]['requestId'], requests[1]['requestId'])
+            self.assertEqual(workbook.read_bytes(), b'workbook fixture')
+
+    def test_empty_and_timeout_exhaust_three_attempts(self):
+        """无效文案与未响应只消耗 AI 尝试，第三次后抛终态错误。"""
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp)
+            workbook, prompt = self.fixture(root)
+            issued = []
+
+            def progress(event):
+                if event.get('event') != 'ai_strategy_required':
+                    return
+                issued.append(event['requestId'])
+                if len(issued) == 1:
+                    request = json.loads(Path(event['request_path']).read_text(encoding='utf-8'))
+                    reply = {key: request[key] for key in ('bizLineId', 'attempt', 'requestId')}
+                    reply['text'] = '   '
+                    Path(event['response_path']).write_text(json.dumps(reply), encoding='utf-8')
+
+            with self.assertRaises(AiStrategyExhausted):
+                review_ai_strategy(root, workbook, prompt, threading.Event(), progress,
+                                   biz_line_id=321, attempt=1, wait_seconds=0.01)
+            self.assertEqual(len(issued), 3)
+
+    def test_missing_section_retries_only_ai(self):
+        """缺少任一策略列时不能接受不完整工作簿。"""
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp)
+            workbook, prompt = self.fixture(root)
+            rejected = []
+
+            def progress(event):
+                if event['event'] == 'ai_strategy_rejected':
+                    rejected.append(event['reason'])
+                if event['event'] == 'ai_strategy_required':
+                    request = json.loads(Path(event['request_path']).read_text(encoding='utf-8'))
+                    reply = {key: request[key] for key in ('bizLineId', 'attempt', 'requestId')}
+                    reply['text'] = AI_TEXT.replace('【创意策略】', '创意策略')
+                    Path(event['response_path']).write_text(json.dumps(reply, ensure_ascii=False), encoding='utf-8')
+
+            with self.assertRaises(AiStrategyExhausted):
+                review_ai_strategy(root, workbook, prompt, threading.Event(), progress,
+                                   biz_line_id=321, attempt=1, wait_seconds=1)
+            self.assertEqual(len(rejected), 3)
+
+
+class AiStrategyWorkbook(unittest.TestCase):
+    """第三张表把固定策略标签与 AI 文案放在同一行，前两张表保持不变。"""
+
+    def test_ai_sheet_preserves_draft_and_native_bold(self):
+        """三类标签复制到各策略左列，素材无标签，正文保留换行和加粗。"""
+        rows = [{'bizLine': '样本', 'budgetTags': '预算标签',
+                 'productTags': '产品标签', 'audienceTags': '受众标签'}]
+        details = [{'kind': '预算', 'channel': 'Meta', 'id': 'fb', 'name': 'Meta'}]
+        ai_text = AI_TEXT
+        with tempfile.TemporaryDirectory() as tmp:
+            draft = Path(tmp) / 'draft.xlsx'
+            final = Path(tmp) / 'final.xlsx'
+            write_xlsx(rows, str(draft), details)
+            validate_xlsx(draft, rows, details)
+            write_xlsx(rows, str(final), details, ai_strategy=ai_text)
+            self.assertEqual(validate_xlsx(final, rows, details, ai_strategy=ai_text)['ai_columns'], 8)
+            with zipfile.ZipFile(draft) as old, zipfile.ZipFile(final) as new:
+                self.assertEqual(old.read('xl/worksheets/sheet1.xml'), new.read('xl/worksheets/sheet1.xml'))
+                self.assertEqual(old.read('xl/worksheets/sheet2.xml'), new.read('xl/worksheets/sheet2.xml'))
+                workbook = ET.fromstring(new.read('xl/workbook.xml'))
+                ns = {'s': 'http://schemas.openxmlformats.org/spreadsheetml/2006/main'}
+                self.assertEqual([item.attrib['name'] for item in workbook.findall('s:sheets/s:sheet', ns)],
+                                 ['策略', '明细', 'AI策略'])
+                strings = list(ET.fromstring(new.read('xl/sharedStrings.xml')))
+                ai = ET.fromstring(new.read('xl/worksheets/sheet3.xml'))
+                sheet_rows = ai.findall('s:sheetData/s:row', ns)
+                self.assertEqual(len(sheet_rows), 2)
+                cells = sheet_rows[1].findall('s:c', ns)
+                self.assertEqual(len(cells), 8)
+                self.assertTrue(all(cell.get('t') == 's' for row in sheet_rows for cell in row))
+                self.assertNotIn(b'inlineStr', new.read('xl/worksheets/sheet3.xml'))
+                self.assertEqual([''.join(strings[int(cell.find('s:v', ns).text)].itertext()) for cell in cells],
+                                 ['一页速览\n- 本期观察。', '预算标签',
+                                  '【策略总结】\n- 有效：消耗 $12.00。', '产品标签',
+                                  '【调整方向】\n- 保持观察。', '受众标签',
+                                  '【关键数据】\n- 样本。', '【Learning】\n- 待观察。'])
+                self.assertEqual([''.join(strings[int(cell.find('s:v', ns).text)].itertext()) for cell in sheet_rows[0].findall('s:c', ns)],
+                                 ['一页速览', '预算策略标签', '预算策略', '广告产品策略标签',
+                                  '广告产品策略', '受众策略标签', '受众策略', '创意策略'])
+                runs = strings[int(cells[2].find('s:v', ns).text)].findall('s:r', ns)
+                self.assertEqual([''.join(run.itertext()) for run in runs if run.find('s:rPr/s:b', ns) is not None],
+                                 ['有效', '$12.00'])
+
+    def test_readback_rejects_extra_rows_columns_and_wrong_sheet(self):
+        """回读拒绝错表、多行业和错列，不能用位置猜入库内容。"""
+        mutations = (
+            ('xl/workbook.xml', lambda text: text.replace('name="AI策略"', 'name="其他"')),
+            ('xl/worksheets/sheet3.xml', lambda text: text.replace('</sheetData>', '<row r="3"/></sheetData>')),
+            ('xl/worksheets/sheet3.xml', lambda text: text.replace('r="C2"', 'r="D2"')),
+            ('xl/sharedStrings.xml', lambda text: text.replace('预算策略标签', '错误标签列')),
+            ('xl/worksheets/sheet3.xml', lambda text: text.replace('t="s"', 't="inlineStr"')),
+            ('xl/_rels/workbook.xml.rels', lambda text: text.replace('Target="worksheets/sheet3.xml"',
+                                                                      'Target="worksheets/sheet1.xml"')),
+        )
+        for name, mutate in mutations:
+            with self.subTest(name=name), tempfile.TemporaryDirectory() as tmp:
+                source = Path(tmp) / 'source.xlsx'
+                damaged = Path(tmp) / 'damaged.xlsx'
+                write_xlsx([{'bizLine': '样本'}], str(source), [], ai_strategy=AI_TEXT)
+                with zipfile.ZipFile(source) as archive, zipfile.ZipFile(damaged, 'x') as target:
+                    for member in archive.namelist():
+                        value = archive.read(member)
+                        if member == name:
+                            value = mutate(value.decode('utf-8')).encode('utf-8')
+                        target.writestr(member, value)
+                with self.assertRaises(ValueError):
+                    read_ai_sheet(damaged)
+
+    def test_ai_line_over_excel_limit_rejected(self):
+        """不能为保住长文本而静默截断 AI 句子。"""
+        with tempfile.TemporaryDirectory() as tmp:
+            with self.assertRaises(ValueError):
+                write_xlsx([{'bizLine': '样本'}], str(Path(tmp) / 'out.xlsx'), [],
+                           ai_strategy=AI_TEXT.replace('样本。', '字' * 32768))
+
+    def test_missing_fixed_tag_stays_empty(self):
+        """第一张表标签为空时，AI sheet 不自行补标签或改变策略正文。"""
+        rows = [{'bizLine': '样本', 'budgetTags': '其他', 'productTags': '', 'audienceTags': ''}]
+        with tempfile.TemporaryDirectory() as tmp:
+            path = Path(tmp) / 'final.xlsx'
+            write_xlsx(rows, str(path), [], ai_strategy=AI_TEXT)
+            self.assertEqual(validate_xlsx(path, rows, [], ai_strategy=AI_TEXT)['ai_columns'], 8)
+            with zipfile.ZipFile(path) as workbook:
+                ns = {'s': 'http://schemas.openxmlformats.org/spreadsheetml/2006/main'}
+                sheet = ET.fromstring(workbook.read('xl/worksheets/sheet3.xml'))
+                cells = sheet.findall('s:sheetData/s:row', ns)[1].findall('s:c', ns)
+                strings = list(ET.fromstring(workbook.read('xl/sharedStrings.xml')))
+                self.assertEqual([''.join(strings[int(cells[index].find('s:v', ns).text)].itertext()) for index in (1, 3, 5)],
+                                 ['其他', '', ''])
+
+    def test_ai_input_contains_only_detail_sheet(self):
+        """交给 AI 的工作簿不含固定策略正文或其他 sheet。"""
+        details = [{'kind': '预算', 'channel': 'Meta', 'id': 'fb', 'name': '测试维度'}]
+        with tempfile.TemporaryDirectory() as tmp:
+            path = Path(tmp) / 'ai-detail.xlsx'
+            write_detail_workbook(details, str(path))
+            self.assertEqual(validate_detail_workbook(path, details)['details'], 1)
+            with zipfile.ZipFile(path) as workbook:
+                xml = ET.fromstring(workbook.read('xl/workbook.xml'))
+                ns = {'s': 'http://schemas.openxmlformats.org/spreadsheetml/2006/main'}
+                self.assertEqual([item.attrib['name'] for item in xml.findall('s:sheets/s:sheet', ns)], ['明细'])
+                self.assertNotIn('策略内容', workbook.read('xl/sharedStrings.xml').decode('utf-8'))
+
+
+if __name__ == '__main__':
+    unittest.main()

package/dist/skills/biz-line-strategy-generate/scripts/test_cli_json.py

@@ -0,0 +1,97 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""不发网：验证信封解包、词包截断、双维度判定。"""
+from __future__ import annotations
+
+import sys
+from pathlib import Path
+
+SCRIPTS = Path(__file__).resolve().parent
+if str(SCRIPTS) not in sys.path:
+    sys.path.insert(0, str(SCRIPTS))
+
+from cli_json import dual_verdict, parse_json_text, short_label, unwrap_page, zh
+from summarize import advice_of, dim_layer_text, fmt_share, pct_pair, setting_line
+
+
+def test_unwrap_metrics_uses_data_data():
+    env = {"ok": True, "data": {"data": [{"spend_usd": 1}], "total": 1808, "result": None}}
+    rows, total = unwrap_page(env)
+    assert len(rows) == 1 and total == 1808
+
+
+def test_unwrap_list_uses_result():
+    env = {"ok": True, "data": {"result": [{"id": 1258, "name": "x"}], "total": 1}}
+    rows, total = unwrap_page(env)
+    assert rows[0]["id"] == 1258 and total == 1
+
+
+def test_unwrap_project_list_is_array():
+    env = {"ok": True, "data": [{"projectName": "A"}]}
+    rows, total = unwrap_page(env)
+    assert total == 1 and rows[0]["projectName"] == "A"
+
+
+def test_parse_array_not_first_object():
+    raw = '[{"spend_usd": 10}, {"spend_usd": 20}]'
+    parsed = parse_json_text(raw)
+    assert isinstance(parsed, list) and len(parsed) == 2
+
+
+def test_short_label_truncates_keyword_pack():
+    pack = "3d,a3,battle camp," + ",".join(["genshin"] * 80)
+    out = short_label(pack)
+    assert "genshin" in out or "3d" in out
+    assert len(out) < 80
+    assert pack not in out
+
+
+def test_dual_verdict_table():
+    assert dual_verdict(True, "变好") == "有效"
+    assert dual_verdict(False, "变好") == "有效"
+    assert dual_verdict(False, "稳定") == "无效"
+    assert dual_verdict(True, "无法对比") == "待观察"
+
+
+def test_share_uses_arrow():
+    assert pct_pair(50, 55) == "50.0% → 55.0%"
+    out = fmt_share({"应用推广": 100}, {"应用推广": 100})
+    assert ">>" not in out
+    assert "100.0% → 100.0%" in out
+
+
+def test_setting_and_learning_are_multiline():
+    text = setting_line("【国家】", {"美国"}, {"越南"})
+    assert "\n" in text and text.startswith("【国家】")
+    items = [
+        {"label": f"广告{i}", "spend": 80, "imps": 10000, "cpm": 8.0} for i in range(12)
+    ]
+    body, _hit, _hs, _ts = dim_layer_text("广告", items, [])
+    assert body.startswith("广告")
+    assert body.count("· 新增") == 8
+    assert "另有 4 个新增未展开" in body
+    assert "{" not in body
+    assert advice_of("有效") == "建议：继续执行"
+
+
+def test_zh_web_conversions():
+    assert zh("WEB_CONVERSIONS") == "网站转化"
+    assert zh("CUSTOM&LOOKALIKE") == "自定义/相似受众"
+
+
+if __name__ == "__main__":
+    tests = [
+        test_unwrap_metrics_uses_data_data,
+        test_unwrap_list_uses_result,
+        test_unwrap_project_list_is_array,
+        test_parse_array_not_first_object,
+        test_short_label_truncates_keyword_pack,
+        test_dual_verdict_table,
+        test_share_uses_arrow,
+        test_setting_and_learning_are_multiline,
+        test_zh_web_conversions,
+    ]
+    for fn in tests:
+        fn()
+        print("ok", fn.__name__)
+    print("DONE")