kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: 移除_parser_方法中的source_type参数，并更新相关测试用例

Browse Source
This commit is contained in:
Wenjie Zhang 2026-03-20 12:42:46 +08:00
parent 8804c7446c
commit 78f2ae2a03
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/package/yuxi/services/conversation_service.py
View File

@ -83,7 +83,7 @@ async def _convert_upload_to_markdown(upload: UploadFile) -> ConversionResult:
try: try:
file_size = await _write_upload_to_disk(upload, temp_path) file_size = await _write_upload_to_disk(upload, temp_path)
markdown = await Parser.aparse(str(temp_path), source_type="file") markdown = await Parser.aparse(str(temp_path))
markdown, truncated = _truncate_markdown(markdown) markdown, truncated = _truncate_markdown(markdown)
return ConversionResult( return ConversionResult(
file_id=uuid.uuid4().hex, file_id=uuid.uuid4().hex,
@ -94,7 +94,7 @@ async def _convert_upload_to_markdown(upload: UploadFile) -> ConversionResult:
truncated=truncated, truncated=truncated,
) )
except Exception as exc: # noqa: BLE001 except Exception as exc: # noqa: BLE001
logger.error("Attachment conversion failed: %s", exc) logger.error(f"Attachment conversion failed: {exc}")
raise raise

6
backend/test/test_conversation_service_attachment_state.py
View File

@ -123,8 +123,7 @@ async def test_convert_upload_to_markdown_returns_conversion_result(
tmp_path, tmp_path,
monkeypatch: pytest.MonkeyPatch, monkeypatch: pytest.MonkeyPatch,
): ):
async def _fake_aparse(source: str, params=None, source_type: str = "file") -> str: async def _fake_aparse(source: str, params=None) -> str:
assert source_type == "file"
return "converted markdown" return "converted markdown"
monkeypatch.setattr(svc, "_ensure_workdir", lambda: tmp_path) monkeypatch.setattr(svc, "_ensure_workdir", lambda: tmp_path)
@ -147,8 +146,7 @@ async def test_convert_upload_to_markdown_truncates_content(
tmp_path, tmp_path,
monkeypatch: pytest.MonkeyPatch, monkeypatch: pytest.MonkeyPatch,
): ):
async def _fake_aparse(source: str, params=None, source_type: str = "file") -> str: async def _fake_aparse(source: str, params=None) -> str:
assert source_type == "file"
return "x" * (svc.MAX_ATTACHMENT_MARKDOWN_CHARS + 200) return "x" * (svc.MAX_ATTACHMENT_MARKDOWN_CHARS + 200)
monkeypatch.setattr(svc, "_ensure_workdir", lambda: tmp_path) monkeypatch.setattr(svc, "_ensure_workdir", lambda: tmp_path)