mirror of
https://github.com/langgenius/dify.git
synced 2026-08-29 19:54:46 +08:00
70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Annotated
|
|
|
|
from pydantic import Field, WithJsonSchema, field_validator
|
|
|
|
from fields.base import ResponseModel
|
|
from libs.helper import to_timestamp
|
|
|
|
UUIDString = Annotated[str, WithJsonSchema({"format": "uuid", "type": "string"})]
|
|
Int64 = Annotated[int, WithJsonSchema({"format": "int64", "type": "integer"})]
|
|
|
|
|
|
class Annotation(ResponseModel):
|
|
id: UUIDString
|
|
question: str | None = None
|
|
answer: str | None = Field(default=None, validation_alias="content")
|
|
hit_count: int | None = None
|
|
created_at: Int64 | None = None
|
|
|
|
@field_validator("created_at", mode="before")
|
|
@classmethod
|
|
def _normalize_created_at(cls, value: datetime | int | None) -> int | None:
|
|
return to_timestamp(value)
|
|
|
|
|
|
class AnnotationList(ResponseModel):
|
|
data: list[Annotation]
|
|
has_more: bool
|
|
limit: int
|
|
total: int
|
|
page: int
|
|
|
|
|
|
class AnnotationJobStatusResponse(ResponseModel):
|
|
job_id: UUIDString
|
|
job_status: str
|
|
|
|
|
|
class AnnotationJobStatusDetailResponse(AnnotationJobStatusResponse):
|
|
error_msg: str = ""
|
|
|
|
|
|
class AnnotationExportList(ResponseModel):
|
|
data: list[Annotation]
|
|
|
|
|
|
class AnnotationHitHistory(ResponseModel):
|
|
id: str
|
|
source: str | None = None
|
|
score: float | None = None
|
|
question: str | None = None
|
|
created_at: int | None = None
|
|
match: str | None = Field(default=None, validation_alias="annotation_question")
|
|
response: str | None = Field(default=None, validation_alias="annotation_content")
|
|
|
|
@field_validator("created_at", mode="before")
|
|
@classmethod
|
|
def _normalize_created_at(cls, value: datetime | int | None) -> int | None:
|
|
return to_timestamp(value)
|
|
|
|
|
|
class AnnotationHitHistoryList(ResponseModel):
|
|
data: list[AnnotationHitHistory]
|
|
has_more: bool
|
|
limit: int
|
|
total: int
|
|
page: int
|