adminandClaude Opus 4.8
e7fab724fc
把标题生成从「逐行看图、各生成1条、每行一次请求」改为一次请求、纯提示词 （不传图）、生成多条、按序回填（数量由用户写进提示词，docs/11 §17.1/§17.7）。 - ai_text_service：新增 extract_titles_from_response（多行→多条、逐行去 序号/引号、丢空）+ AiTextClient.generate_texts（一次POST返回多条，与 generate_text 共用 _post）；generate_text/extract_text 改为取首条 - ai_title：generate_titles（一次请求、image_path=None 纯文本）；移除 generate_title/_reference_image/render_title_prompt（不再逐行看图/替换占位符） - config_service：DEFAULT_TITLE_PROMPT 改批量风格（生成多条、每行一条） - 面板 _TitleWorker：一次 generate_titles → 按序 write_title_result 回填； N>行数多的丢+日志、N<行数后面行留空+日志；请求异常→日志+成功0；去掉行间节流 - _start_title 不再传 request_interval 测试：extract_titles 多行→多条/去序号引号、generate_texts 多条且无图、 generate_titles 一次请求/异常。全套 py37 通过（test_config_service 的 packaging 模板失败属并行 §19.13，与本改动无关）。离屏冒烟：3/3 精确、 5/3 丢弃、2/3 留空 三种分发均按序回填 Excel A + 日志正确。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
"""Tests for AI title batch generation core (docs/11 §17.1 / §17.7)."""
|
|
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
|
|
|
|
|
|
class TestAiTitleCore(unittest.TestCase):
|
|
def test_generate_titles_one_request_returns_list(self):
|
|
from core.ai_title import generate_titles
|
|
|
|
client = _RecordingTextClient(["纯棉短袖T恤", "印花连衣裙", "碎花雪纺衫"])
|
|
titles = generate_titles("生成 3 条标题", model_config={}, api_client=client)
|
|
|
|
self.assertEqual(titles, ["纯棉短袖T恤", "印花连衣裙", "碎花雪纺衫"])
|
|
self.assertEqual(client.calls, 1) # 只请求一次
|
|
self.assertEqual(client.image_paths, [None]) # 纯文本，不传图
|
|
|
|
def test_generate_titles_propagates_client_error(self):
|
|
from core.ai_title import generate_titles
|
|
|
|
with self.assertRaises(RuntimeError):
|
|
generate_titles("x", model_config={}, api_client=_FailingTextClient())
|
|
|
|
|
|
class _RecordingTextClient:
|
|
def __init__(self, titles):
|
|
self._titles = titles
|
|
self.calls = 0
|
|
self.image_paths = []
|
|
|
|
def generate_texts(self, prompt, image_path=None, resolution="1K"):
|
|
self.calls += 1
|
|
self.image_paths.append(image_path)
|
|
return list(self._titles)
|
|
|
|
|
|
class _FailingTextClient:
|
|
def generate_texts(self, prompt, image_path=None, resolution="1K"):
|
|
raise RuntimeError("boom")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|