feat(ai-outfit): 左栏新增「标题生成」(看图→提示词→AI 文字标题→写回A列) (§19.18)
AI 穿搭页左栏新增独立的「生成标题」流程：用户写标题提示词，AI 看该行 衣服图（目录行取首图）生成电商标题，逐行立即写回 Excel A 列并刷新明细表； 完成后重载 Excel，紧接「开始生成」跑图即用新标题。移除原「最终生成要求预览」 腾出版面（docs/11 §17）。 - ai_text_service.py：AiTextClient 复用图像服务 HTTP 管道做文本输出； chat/gemini 带图视觉，images/images_edits 明确报错；extract_text 取首条标题 - ai_title.py + TitleResult：单行编排，never raises - excel_service.write_title_result：只写 A 列、不动 D/E/F - config_service：title_model 默认 + load/save_title_prompt + 默认标题话术 - 面板：标题生成组（提示词+标题模型下拉+保存+生成标题）置于话术组上方； _TitleWorker 顺序逐行+立即回填+刷新；与「开始生成」互斥；删预览相关组件 - 测试：文本解析/payload、generate_title（单文件/目录首图/失败）、 write_title_result、面板标题组存在且无预览；全套 py37 通过 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Claude Opus 4.8
@@ -0,0 +1,136 @@
|
||||
"""Tests for the AI text (title) service (docs/11 §17)."""
|
||||
import sys
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
|
||||
|
||||
|
||||
CHAT_CFG = {"url": "https://relay.example.com/v1/chat/completions",
|
||||
"model": "gpt-4o", "api_key": "sk-x", "api_type": "chat"}
|
||||
GEMINI_CFG = {"url": "https://relay.example.com/v1beta/models/{model}:generateContent",
|
||||
"model": "gemini-2.5-flash", "api_key": "sk-x", "api_type": "gemini"}
|
||||
IMAGES_EDITS_CFG = {"url": "https://relay.example.com/v1/images/edits",
|
||||
"model": "img", "api_key": "sk-x", "api_type": "images_edits"}
|
||||
|
||||
|
||||
class TestExtractText(unittest.TestCase):
|
||||
def test_chat_string_content(self):
|
||||
from services.ai_text_service import extract_text_from_response
|
||||
|
||||
data = {"choices": [{"message": {"content": "纯棉宽松短袖T恤"}}]}
|
||||
self.assertEqual(extract_text_from_response(data), "纯棉宽松短袖T恤")
|
||||
|
||||
def test_chat_list_content(self):
|
||||
from services.ai_text_service import extract_text_from_response
|
||||
|
||||
data = {"choices": [{"message": {"content": [
|
||||
{"type": "text", "text": "印花连衣裙"}]}}]}
|
||||
self.assertEqual(extract_text_from_response(data), "印花连衣裙")
|
||||
|
||||
def test_gemini_parts(self):
|
||||
from services.ai_text_service import extract_text_from_response
|
||||
|
||||
data = {"candidates": [{"content": {"parts": [{"text": "碎花雪纺衫"}]}}]}
|
||||
self.assertEqual(extract_text_from_response(data), "碎花雪纺衫")
|
||||
|
||||
def test_multiline_takes_first_and_strips_numbering_quotes(self):
|
||||
from services.ai_text_service import extract_text_from_response
|
||||
|
||||
data = {"choices": [{"message": {"content":
|
||||
"1. 「韩版宽松卫衣」\n2. 复古工装外套\n3. 简约百搭T恤"}}]}
|
||||
self.assertEqual(extract_text_from_response(data), "韩版宽松卫衣")
|
||||
|
||||
def test_missing_text_returns_empty(self):
|
||||
from services.ai_text_service import extract_text_from_response
|
||||
|
||||
self.assertEqual(extract_text_from_response({"choices": []}), "")
|
||||
self.assertEqual(extract_text_from_response({}), "")
|
||||
|
||||
|
||||
class TestBuildTextPayload(unittest.TestCase):
|
||||
def test_chat_with_image_includes_image_url(self):
|
||||
from services.ai_text_service import build_text_payload
|
||||
|
||||
payload = build_text_payload(CHAT_CFG, "写标题", "data:image/png;base64,AAA")
|
||||
content = payload["messages"][0]["content"]
|
||||
types = [part["type"] for part in content]
|
||||
self.assertEqual(types, ["text", "image_url"])
|
||||
self.assertEqual(content[1]["image_url"]["url"], "data:image/png;base64,AAA")
|
||||
|
||||
def test_chat_without_image_text_only(self):
|
||||
from services.ai_text_service import build_text_payload
|
||||
|
||||
payload = build_text_payload(CHAT_CFG, "写标题")
|
||||
content = payload["messages"][0]["content"]
|
||||
self.assertEqual([p["type"] for p in content], ["text"])
|
||||
|
||||
def test_gemini_text_only_modality(self):
|
||||
from services.ai_text_service import build_text_payload
|
||||
|
||||
payload = build_text_payload(GEMINI_CFG, "写标题",
|
||||
"data:image/jpeg;base64,BBB")
|
||||
self.assertEqual(
|
||||
payload["generationConfig"]["responseModalities"], ["TEXT"])
|
||||
parts = payload["contents"][0]["parts"]
|
||||
self.assertIn("inlineData", parts[1])
|
||||
|
||||
def test_images_edits_rejected(self):
|
||||
from services.ai_text_service import AiTextServiceError, build_text_payload
|
||||
|
||||
with self.assertRaises(AiTextServiceError):
|
||||
build_text_payload(IMAGES_EDITS_CFG, "写标题")
|
||||
|
||||
|
||||
class _FakeResponse:
|
||||
def __init__(self, payload):
|
||||
self._payload = payload
|
||||
|
||||
def raise_for_status(self):
|
||||
pass
|
||||
|
||||
def json(self):
|
||||
return self._payload
|
||||
|
||||
|
||||
class _FakeSession:
|
||||
def __init__(self, payload):
|
||||
self._payload = payload
|
||||
self.posted = None
|
||||
|
||||
def post(self, url, headers=None, json=None, timeout=None):
|
||||
self.posted = {"url": url, "headers": headers, "json": json}
|
||||
return _FakeResponse(self._payload)
|
||||
|
||||
|
||||
class TestGenerateText(unittest.TestCase):
|
||||
def test_generate_text_posts_and_returns_title(self):
|
||||
from services.ai_text_service import AiTextClient
|
||||
|
||||
session = _FakeSession({"choices": [{"message": {"content": "夏季冰丝防晒衣"}}]})
|
||||
client = AiTextClient(CHAT_CFG, session=session)
|
||||
|
||||
# No image_path -> text-only payload; still returns the parsed title.
|
||||
title = client.generate_text("给这件衣服起标题")
|
||||
|
||||
self.assertEqual(title, "夏季冰丝防晒衣")
|
||||
self.assertIn("chat/completions", session.posted["url"])
|
||||
self.assertEqual(session.posted["headers"]["Authorization"], "Bearer sk-x")
|
||||
|
||||
def test_generate_text_images_edits_raises(self):
|
||||
from services.ai_text_service import AiTextClient, AiTextServiceError
|
||||
|
||||
client = AiTextClient(IMAGES_EDITS_CFG, session=_FakeSession({}))
|
||||
with self.assertRaises(AiTextServiceError):
|
||||
client.generate_text("标题")
|
||||
|
||||
def test_generate_text_empty_response_raises(self):
|
||||
from services.ai_text_service import AiTextClient, AiTextServiceError
|
||||
|
||||
client = AiTextClient(CHAT_CFG, session=_FakeSession({"choices": []}))
|
||||
with self.assertRaises(AiTextServiceError):
|
||||
client.generate_text("标题")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user