89 lines
1.9 KiB
Python
89 lines
1.9 KiB
Python
from datetime import datetime
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class CustomerBase(BaseModel):
|
|
name: str = Field(..., description="Customer name")
|
|
contact_info: Optional[str] = Field(None, description="Contact information")
|
|
|
|
|
|
class CustomerCreate(CustomerBase):
|
|
pass
|
|
|
|
|
|
class CustomerUpdate(BaseModel):
|
|
name: Optional[str] = None
|
|
contact_info: Optional[str] = None
|
|
|
|
|
|
class CustomerRead(CustomerBase):
|
|
id: int
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ProjectRead(BaseModel):
|
|
id: int
|
|
customer_id: int
|
|
raw_requirement: str
|
|
ai_solution_md: Optional[str] = None
|
|
status: str
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ProjectUpdate(BaseModel):
|
|
ai_solution_md: Optional[str] = None
|
|
status: Optional[str] = None
|
|
|
|
|
|
class RequirementAnalyzeRequest(BaseModel):
|
|
customer_id: int = Field(..., description="Related customer id")
|
|
raw_text: str = Field(..., description="Raw requirement text from chat or WeChat")
|
|
|
|
|
|
class RequirementAnalyzeResponse(BaseModel):
|
|
project_id: int
|
|
ai_solution_md: str
|
|
ai_solution_json: Dict[str, Any]
|
|
|
|
|
|
class QuoteGenerateResponse(BaseModel):
|
|
quote_id: int
|
|
project_id: int
|
|
total_amount: float
|
|
excel_path: str
|
|
pdf_path: str
|
|
|
|
|
|
class ContractGenerateRequest(BaseModel):
|
|
delivery_date: str = Field(..., description="Delivery date, e.g. 2026-03-31")
|
|
extra_placeholders: Dict[str, str] = Field(
|
|
default_factory=dict,
|
|
description="Additional placeholder mappings for the contract template",
|
|
)
|
|
|
|
|
|
class ContractGenerateResponse(BaseModel):
|
|
project_id: int
|
|
contract_path: str
|
|
|
|
|
|
class FinanceSyncResult(BaseModel):
|
|
id: int
|
|
month: str
|
|
type: str
|
|
file_name: str
|
|
file_path: str
|
|
|
|
|
|
class FinanceSyncResponse(BaseModel):
|
|
items: List[FinanceSyncResult]
|
|
|