c62156af53
B1: 项目脚手架 + 数据模型 + 租户管理 - Task 1.1: FastAPI 项目脚手架、SQLite + async SQLAlchemy - Task 1.2: 7 个数据模型 (Tenant, TenantConfig, DigitalEmployee, Conversation, Message, KnowledgeBase, Document) - Task 1.3: 租户 CRUD API + LLM 配置（含 API Key AES 加密） B2: 数字员工配置 + LLM Provider 抽象层 - Task 2.1: 数字员工 CRUD API（关联知识库） - Task 2.2: BaseLLMProvider 抽象接口 + OpenAI/Qwen Provider - Task 2.3: Provider 动态实例化 + test-provider 端点 验证: 26 个测试全部通过 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
import json
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class EmployeeCreate(BaseModel):
|
|
tenant_id: str
|
|
name: str = Field(..., max_length=200)
|
|
role: str = Field(..., max_length=100)
|
|
system_prompt: str
|
|
greeting: str | None = None
|
|
avatar_url: str | None = Field(None, max_length=500)
|
|
temperature: float = 0.7
|
|
max_context_messages: int = 20
|
|
knowledge_base_ids: list[str] = []
|
|
|
|
|
|
class EmployeeUpdate(BaseModel):
|
|
name: str | None = Field(None, max_length=200)
|
|
role: str | None = Field(None, max_length=100)
|
|
system_prompt: str | None = None
|
|
greeting: str | None = None
|
|
avatar_url: str | None = Field(None, max_length=500)
|
|
temperature: float | None = None
|
|
max_context_messages: int | None = None
|
|
knowledge_base_ids: list[str] | None = None
|
|
status: str | None = None
|
|
|
|
|
|
class EmployeeResponse(BaseModel):
|
|
id: str
|
|
tenant_id: str
|
|
name: str
|
|
role: str
|
|
system_prompt: str
|
|
greeting: str | None
|
|
avatar_url: str | None
|
|
temperature: float
|
|
max_context_messages: int
|
|
knowledge_base_ids: list[str]
|
|
status: str
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
@classmethod
|
|
def from_model(cls, employee) -> "EmployeeResponse":
|
|
kb_ids = json.loads(employee.knowledge_base_ids) if employee.knowledge_base_ids else []
|
|
return cls(
|
|
id=employee.id,
|
|
tenant_id=employee.tenant_id,
|
|
name=employee.name,
|
|
role=employee.role,
|
|
system_prompt=employee.system_prompt,
|
|
greeting=employee.greeting,
|
|
avatar_url=employee.avatar_url,
|
|
temperature=employee.temperature,
|
|
max_context_messages=employee.max_context_messages,
|
|
knowledge_base_ids=kb_ids,
|
|
status=employee.status.value,
|
|
created_at=employee.created_at,
|
|
updated_at=employee.updated_at,
|
|
)
|