generated from kgod/ai-review-template
77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
"""Import parsing speed-ups: slim schema without model evidence (A) + sha256 parse cache (B)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import io
|
|
from typing import Any
|
|
|
|
from docx import Document
|
|
|
|
from app.import_parser import ImportParseOutput, OpenAIResumeImportParser
|
|
from app.resume_import_service import ResumeImportService
|
|
|
|
|
|
class FakeCompletion:
|
|
def __init__(self) -> None:
|
|
self.calls: list[dict[str, Any]] = []
|
|
|
|
def complete(self, **kwargs: Any) -> Any:
|
|
self.calls.append(kwargs)
|
|
return kwargs["schema"].model_validate(
|
|
{
|
|
"basics": {"name": "张三"},
|
|
"sections": [
|
|
{
|
|
"kind": "education",
|
|
"heading": "教育经历",
|
|
"items": [{"fields": {"school": "示例大学", "major": "软件工程"}}],
|
|
}
|
|
],
|
|
"skill_groups": [],
|
|
}
|
|
)
|
|
|
|
|
|
def _docx(text: str) -> bytes:
|
|
document = Document()
|
|
document.add_paragraph(text)
|
|
buffer = io.BytesIO()
|
|
document.save(buffer)
|
|
return buffer.getvalue()
|
|
|
|
|
|
def _service(tmp_path, completion: FakeCompletion) -> ResumeImportService:
|
|
return ResumeImportService(
|
|
storage_root=tmp_path / "imports",
|
|
parser=OpenAIResumeImportParser(completion=completion),
|
|
)
|
|
|
|
|
|
def test_service_uses_slim_schema_without_model_evidence(tmp_path) -> None:
|
|
"""模型不再逐条输出 evidence 引用（本地匹配已覆盖），输出 token 与时间同降。"""
|
|
from app.import_parser_fast import SlimImportParseOutput
|
|
|
|
completion = FakeCompletion()
|
|
prepared = _service(tmp_path, completion).prepare(
|
|
file_name="r.docx", declared_mime=None, content=_docx("张三 示例大学 软件工程")
|
|
)
|
|
|
|
assert completion.calls[0]["schema"] is SlimImportParseOutput
|
|
assert "evidence" not in completion.calls[0]["system_prompt"].casefold()
|
|
assert prepared["document"]["sections"][0]["items"][0]["school"] == "示例大学"
|
|
assert all(item["evidence"] for item in prepared["field_reviews"]) # 本地匹配仍然提供证据
|
|
|
|
|
|
def test_repeated_upload_of_same_file_skips_llm_parse(tmp_path) -> None:
|
|
"""同一文件跨会话重复上传命中 sha256 缓存，不再调 LLM 解析。"""
|
|
completion = FakeCompletion()
|
|
service = _service(tmp_path, completion)
|
|
content = _docx("张三 示例大学 软件工程")
|
|
|
|
first = service.prepare(file_name="a.docx", declared_mime=None, content=content)
|
|
second = service.prepare(file_name="b.docx", declared_mime=None, content=content)
|
|
|
|
assert len(completion.calls) == 1
|
|
assert second["document"] == first["document"]
|
|
assert second["sha256"] == first["sha256"]
|