69 lines
3.4 KiB
Python
69 lines
3.4 KiB
Python
import unittest
|
||
from unittest.mock import AsyncMock
|
||
|
||
from app.services.ai_steward import AIStewardService
|
||
|
||
|
||
class AIStewardTitleExtractTest(unittest.IsolatedAsyncioTestCase):
|
||
async def test_extract_title_from_plain_project_create_instruction(self) -> None:
|
||
title = AIStewardService._extract_project_title_from_instruction("给我新增个项目，openclaw自动化部署。")
|
||
self.assertEqual(title, "openclaw自动化部署")
|
||
|
||
async def test_extract_title_from_requirement_name_instruction(self) -> None:
|
||
title = AIStewardService._extract_project_title_from_instruction("给我新增个项目，需求名字为openclaw自动化部署。")
|
||
self.assertEqual(title, "openclaw自动化部署")
|
||
|
||
async def test_extract_title_from_project_name_instruction(self) -> None:
|
||
title = AIStewardService._extract_project_title_from_instruction("给我新增个项目，项目名为openclaw自动化部署。")
|
||
self.assertEqual(title, "openclaw自动化部署")
|
||
|
||
async def test_extract_title_from_correction_sentence(self) -> None:
|
||
title = AIStewardService._extract_project_title_from_instruction("不是创建信号，是创建项目名为openclaw自动化部署。")
|
||
self.assertEqual(title, "openclaw自动化部署")
|
||
|
||
async def test_normalize_project_draft_should_not_use_instruction_sentence_as_title(self) -> None:
|
||
service = AIStewardService(db=None) # type: ignore[arg-type]
|
||
normalized = await service._normalize_project_creation_draft(
|
||
draft={
|
||
"analysis": {},
|
||
"project": {
|
||
"title": "不是创建信号，是创建项目名为openclaw自动化部署。",
|
||
"description": "不是创建信号，是创建项目名为openclaw自动化部署。",
|
||
},
|
||
"tasks": [],
|
||
},
|
||
input_text="不是创建信号，是创建项目名为openclaw自动化部署。",
|
||
creator_id=1,
|
||
tenant_id=1,
|
||
)
|
||
self.assertEqual(normalized["project"]["title"], "openclaw自动化部署")
|
||
self.assertEqual(normalized["project"]["description"], "项目：openclaw自动化部署")
|
||
|
||
async def test_project_creation_text_retries_with_glm47_fallback_function(self) -> None:
|
||
service = AIStewardService(db=None) # type: ignore[arg-type]
|
||
called_functions: list[str] = []
|
||
|
||
async def _mock_generate(_: str, **kwargs):
|
||
fn = str(kwargs.get("function_name") or "")
|
||
called_functions.append(fn)
|
||
if len(called_functions) <= 2:
|
||
return "我暂时无法生成可靠回答（模型服务异常）。请稍后重试。"
|
||
return '{"project_name":"openclaw自动化部署"}'
|
||
|
||
service.llm.generate = AsyncMock(side_effect=_mock_generate) # type: ignore[method-assign]
|
||
text = await service._generate_project_creation_text(
|
||
prompt="生成分析结果",
|
||
tenant_id=1,
|
||
function_name="project_from_text_analysis",
|
||
timeout_seconds=5.0,
|
||
)
|
||
self.assertEqual(text, '{"project_name":"openclaw自动化部署"}')
|
||
self.assertEqual(
|
||
called_functions,
|
||
[
|
||
"project_from_text_analysis",
|
||
"project_from_text_analysis",
|
||
"project_draft_analysis_glm47_fallback",
|
||
],
|
||
)
|