| """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) |
| 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) |
| context.update( |
| { |
| LitigationPlaceholderKeys.VARIABLE_DEFENSE_OPINION: llm_result.defense_opinion, |
| LitigationPlaceholderKeys.VARIABLE_DEFENSE_REASONS: self.convert_to_paragraphs( |
| llm_result.defense_reasons |
| ), |
| } |
| ) |
| return context |
|
|
