File size: 4,924 Bytes
b4f2238 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 | """Business logic services."""
from __future__ import annotations
import logging
from typing import TYPE_CHECKING, Any
if TYPE_CHECKING:
from apps.documents.services.generation.outputs import ComplaintOutput, DefenseOutput
from apps.core.models.enums import LegalStatus
from apps.documents.services.infrastructure.wiring import get_case_service
from apps.documents.services.placeholders import EnhancedContextBuilder
from apps.litigation_ai.placeholders.spec import LitigationPlaceholderKeys
logger = logging.getLogger("apps.documents.generation")
class LitigationContextBuilder:
def __init__(self, enhanced_context_builder: EnhancedContextBuilder | None = None) -> None:
self._enhanced_context_builder = enhanced_context_builder
@property
def enhanced_context_builder(self) -> EnhancedContextBuilder:
if self._enhanced_context_builder is None:
self._enhanced_context_builder = EnhancedContextBuilder()
return self._enhanced_context_builder
def convert_to_paragraphs(self, text: str) -> str:
if not text:
return ""
return text.replace("\r\n", "\a").replace("\n", "\a")
def _get_party_name_for_prompt(self, *, case_id: int, legal_status: str) -> Any:
case_service = get_case_service()
party_names = case_service.get_case_parties_by_legal_status_internal(case_id=case_id, legal_status=legal_status)
if party_names:
return party_names[0]
return "未指定"
def extract_complaint_prompt_data(self, case_dto: Any) -> dict[str, Any]:
case_id = case_dto.id
plaintiff = self._get_party_name_for_prompt(case_id=case_id, legal_status=LegalStatus.PLAINTIFF)
defendant = self._get_party_name_for_prompt(case_id=case_id, legal_status=LegalStatus.DEFENDANT)
return {
"cause_of_action": case_dto.cause_of_action or "民事纠纷",
"plaintiff": plaintiff,
"defendant": defendant,
"litigation_request": "请求依法判决",
"facts_and_reasons": "事实与理由待补充",
}
def extract_defense_prompt_data(self, case_dto: Any) -> dict[str, Any]:
case_id = case_dto.id
plaintiff = self._get_party_name_for_prompt(case_id=case_id, legal_status=LegalStatus.PLAINTIFF)
defendant = self._get_party_name_for_prompt(case_id=case_id, legal_status=LegalStatus.DEFENDANT)
return {
"cause_of_action": case_dto.cause_of_action or "民事纠纷",
"plaintiff": plaintiff,
"defendant": defendant,
"defense_opinion": "不同意原告的诉讼请求",
"defense_reasons": "答辩理由待补充",
}
def build_complaint_context(self, *, case_dto: Any, llm_result: ComplaintOutput) -> dict[str, Any]:
context_data = {"case_id": case_dto.id, "case_dto": case_dto}
required = [
LitigationPlaceholderKeys.PLAINTIFF,
LitigationPlaceholderKeys.DEFENDANT,
LitigationPlaceholderKeys.CAUSE_OF_ACTION,
LitigationPlaceholderKeys.DATE,
LitigationPlaceholderKeys.COMPLAINT_PARTY,
LitigationPlaceholderKeys.COURT,
LitigationPlaceholderKeys.COMPLAINT_SIGNATURE,
]
context = self.enhanced_context_builder.build_context(context_data, required_placeholders=required) # type: ignore[no-any-return]
context.update(
{
LitigationPlaceholderKeys.VARIABLE_LITIGATION_REQUEST: self.convert_to_paragraphs(
llm_result.litigation_request
),
LitigationPlaceholderKeys.VARIABLE_FACTS_AND_REASONS: self.convert_to_paragraphs(
llm_result.facts_and_reasons
),
}
)
return context
def build_defense_context(self, *, case_dto: Any, llm_result: DefenseOutput) -> dict[str, Any]:
context_data = {"case_id": case_dto.id, "case_dto": case_dto}
required = [
LitigationPlaceholderKeys.PLAINTIFF,
LitigationPlaceholderKeys.DEFENDANT,
LitigationPlaceholderKeys.RESPONDENT,
LitigationPlaceholderKeys.CAUSE_OF_ACTION,
LitigationPlaceholderKeys.DATE,
LitigationPlaceholderKeys.DEFENSE_PARTY,
LitigationPlaceholderKeys.COURT,
LitigationPlaceholderKeys.DEFENSE_SIGNATURE,
]
context = self.enhanced_context_builder.build_context(context_data, required_placeholders=required) # type: ignore[no-any-return]
context.update(
{
LitigationPlaceholderKeys.VARIABLE_DEFENSE_OPINION: llm_result.defense_opinion,
LitigationPlaceholderKeys.VARIABLE_DEFENSE_REASONS: self.convert_to_paragraphs(
llm_result.defense_reasons
),
}
)
return context
|