chore/ 删除无用文件
This commit is contained in:
@@ -16,7 +16,10 @@ def _mock_get_sync_db(db):
|
||||
|
||||
class ProcessMemoirSegmentsImageEnqueueTest(unittest.TestCase):
|
||||
@patch("app.tasks.memoir_tasks._chapter_has_cover_to_generate", return_value=True)
|
||||
@patch("app.tasks.memoir_tasks._chapter_has_any_section_images_to_generate", return_value=True)
|
||||
@patch(
|
||||
"app.tasks.memoir_tasks._chapter_has_any_section_images_to_generate",
|
||||
return_value=True,
|
||||
)
|
||||
@patch("app.tasks.memoir_tasks._update_task_status_sync")
|
||||
@patch("app.tasks.memoir_tasks._release_chapter_lock")
|
||||
@patch("app.tasks.memoir_tasks._acquire_chapter_lock", return_value=True)
|
||||
@@ -55,8 +58,12 @@ class ProcessMemoirSegmentsImageEnqueueTest(unittest.TestCase):
|
||||
max_attempts=20,
|
||||
liblib_template_uuid="tpl-uuid",
|
||||
)
|
||||
get_state_mock.return_value = SimpleNamespace(current_stage="childhood", slots={})
|
||||
update_slot_mock.return_value = SimpleNamespace(current_stage="childhood", slots={})
|
||||
get_state_mock.return_value = SimpleNamespace(
|
||||
current_stage="childhood", slots={}
|
||||
)
|
||||
update_slot_mock.return_value = SimpleNamespace(
|
||||
current_stage="childhood", slots={}
|
||||
)
|
||||
llm = Mock()
|
||||
bound_llm = Mock()
|
||||
bound_llm.invoke.side_effect = [
|
||||
@@ -70,7 +77,9 @@ class ProcessMemoirSegmentsImageEnqueueTest(unittest.TestCase):
|
||||
}
|
||||
```"""
|
||||
),
|
||||
SimpleNamespace(content='{"paragraphs":[{"content":"新的章节正文","image_description":"南方小镇的青石板路"}]}'),
|
||||
SimpleNamespace(
|
||||
content='{"paragraphs":[{"content":"新的章节正文","image_description":"南方小镇的青石板路"}]}'
|
||||
),
|
||||
]
|
||||
llm.bind.return_value = bound_llm
|
||||
llm.invoke.side_effect = [
|
||||
@@ -170,7 +179,9 @@ class ProcessMemoirSegmentsImageEnqueueTest(unittest.TestCase):
|
||||
max_attempts=20,
|
||||
liblib_template_uuid="tpl-uuid",
|
||||
)
|
||||
get_state_mock.return_value = SimpleNamespace(current_stage="childhood", slots={})
|
||||
get_state_mock.return_value = SimpleNamespace(
|
||||
current_stage="childhood", slots={}
|
||||
)
|
||||
|
||||
segment = SimpleNamespace(
|
||||
id="segment-1",
|
||||
|
||||
Reference in New Issue
Block a user