refactor(ai-outfit): 标题生成改为「一次请求·纯提示词·多条按序填」 (§19.22)
把标题生成从「逐行看图、各生成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>
This commit is contained in:
@@ -47,6 +47,21 @@ class TestExtractText(unittest.TestCase):
|
||||
self.assertEqual(extract_text_from_response({"choices": []}), "")
|
||||
self.assertEqual(extract_text_from_response({}), "")
|
||||
|
||||
def test_extract_titles_multiline_to_list_cleaned(self):
|
||||
"""§17.1: 多行 → 多条,逐行去序号/引号、丢空行。"""
|
||||
from services.ai_text_service import extract_titles_from_response
|
||||
|
||||
data = {"choices": [{"message": {"content":
|
||||
"1. 「韩版宽松卫衣」\n\n2) 复古工装外套\n- 简约百搭T恤\n "}}]}
|
||||
self.assertEqual(
|
||||
extract_titles_from_response(data),
|
||||
["韩版宽松卫衣", "复古工装外套", "简约百搭T恤"])
|
||||
|
||||
def test_extract_titles_empty_when_no_text(self):
|
||||
from services.ai_text_service import extract_titles_from_response
|
||||
|
||||
self.assertEqual(extract_titles_from_response({"choices": []}), [])
|
||||
|
||||
|
||||
class TestBuildTextPayload(unittest.TestCase):
|
||||
def test_chat_with_image_includes_image_url(self):
|
||||
@@ -131,6 +146,27 @@ class TestGenerateText(unittest.TestCase):
|
||||
with self.assertRaises(AiTextServiceError):
|
||||
client.generate_text("标题")
|
||||
|
||||
def test_generate_texts_returns_all_titles_text_only(self):
|
||||
"""§17.1: 一次请求 → 多条;纯文本(payload 不含 image_url)。"""
|
||||
from services.ai_text_service import AiTextClient
|
||||
|
||||
session = _FakeSession({"choices": [{"message": {"content":
|
||||
"标题一\n标题二\n标题三"}}]})
|
||||
client = AiTextClient(CHAT_CFG, session=session)
|
||||
|
||||
titles = client.generate_texts("生成 3 条标题")
|
||||
|
||||
self.assertEqual(titles, ["标题一", "标题二", "标题三"])
|
||||
content = session.posted["json"]["messages"][0]["content"]
|
||||
self.assertEqual([p["type"] for p in content], ["text"]) # 无图
|
||||
|
||||
def test_generate_texts_empty_raises(self):
|
||||
from services.ai_text_service import AiTextClient, AiTextServiceError
|
||||
|
||||
client = AiTextClient(CHAT_CFG, session=_FakeSession({"choices": []}))
|
||||
with self.assertRaises(AiTextServiceError):
|
||||
client.generate_texts("标题")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user