Spaces:
Runtime error
Runtime error
File size: 2,264 Bytes
a066e5a 55c0d78 a066e5a 55c0d78 a066e5a 55c0d78 a066e5a 55c0d78 a066e5a 55c0d78 a066e5a 55c0d78 | 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 | from datetime import datetime
from typing import Literal, Optional
import uuid
from pydantic import BaseModel, ConfigDict, Field
class KBEntryCreate(BaseModel):
title: str
content: str = Field(min_length=1)
source_url: Optional[str] = None
class KBEntryResponse(BaseModel):
# set it because we need to pass SQLAlchemy ORM object (KnowledgeBase)
# into this model for validation
model_config = ConfigDict(from_attributes=True)
id: uuid.UUID
created_at: datetime
updated_at: datetime
tenant_id: uuid.UUID
title: str
content: str
source_url: Optional[str] = None
origin: Literal["manual", "synced"]
is_indexed: bool = False
class KBEntryUpdate(BaseModel):
title: Optional[str] = None
content: Optional[str] = Field(default=None, min_length=1)
source_url: Optional[str] = None
class BulkUploadError(BaseModel):
row: int
reason: str
class BulkUploadResponse(BaseModel):
created: int
errors: list[BulkUploadError]
class WPSyncRequest(BaseModel):
site_url: str
post_type: str = "post"
force: bool = False
class WPSyncResponse(BaseModel):
synced: int
errors: list[str] = []
class WPSyncPreviewEntry(BaseModel):
title: str
content: str
source_url: str
warnings: list[str] = []
class WPSyncPreviewResponse(BaseModel):
site_url: str
entries: list[WPSyncPreviewEntry]
errors: list[str] = []
class WPSyncConfirmResponse(BaseModel):
job_id: uuid.UUID
class WPSyncStatusResponse(BaseModel):
job_id: uuid.UUID
status: str
source_url: str
total: int
processed: int
failed: int
error_message: Optional[str] = None
created: int = 0
updated: int = 0
skipped: int = 0
class WPSingleSyncRequest(BaseModel):
site_url: str
source_url: str
post_type: str = "post"
force: bool = False
class WPSingleSyncResponse(BaseModel):
action: Literal["created", "updated", "skipped"]
entry: KBEntryResponse
class KBReindexResponse(BaseModel):
task_id: str
status: str = "queued"
class WPSyncJobControlResponse(BaseModel):
job_id: uuid.UUID
status: str
class WPSyncResumeResponse(BaseModel):
original_job_id: uuid.UUID
new_job_id: uuid.UUID
|