mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-07-07 12:31:15 +03:00
4ee12e6237
Some checks are pending
CI / AMD64 Build (push) Waiting to run
CI / ARM Build (push) Waiting to run
CI / Jetson Jetpack 6 (push) Waiting to run
CI / AMD64 Extra Build (push) Blocked by required conditions
CI / ARM Extra Build (push) Blocked by required conditions
CI / Synaptics Build (push) Blocked by required conditions
CI / Assemble and push default build (push) Blocked by required conditions
* Pin ruff * Add python upgrade fixes This enables python upgrade checks in ruff to look for deprecated types and patterns. This namely fixes: - usage of deprecated `Typing` which is now built in - some specific exceptions which are caught and have new aliases Some specific UP checks were also ignored as they are stylistic / unimportant and likely to cause bugs * Remove async blocking calls Use asyncio.to_thread on two remaining blocking calls to fix hanging event thread loop. Enable this specific rule to block it in the future. * Use proper logging mechanism * Correctly format logs * Raise with context When raising an exception include the from context to improve debugging * Cleanup
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from pydantic import BaseModel, Field, RootModel
|
|
|
|
|
|
class FacesResponse(RootModel[dict[str, list[str]]]):
|
|
"""Response model for the get_faces endpoint.
|
|
|
|
Returns a mapping of face names to lists of image filenames.
|
|
Each face name corresponds to a directory in the faces folder,
|
|
and the list contains the names of image files for that face.
|
|
|
|
Example:
|
|
{
|
|
"john_doe": ["face1.webp", "face2.jpg"],
|
|
"jane_smith": ["face3.png"]
|
|
}
|
|
"""
|
|
|
|
root: dict[str, list[str]] = Field(
|
|
default_factory=dict,
|
|
description="Dictionary mapping face names to lists of image filenames",
|
|
)
|
|
|
|
|
|
class FaceRecognitionResponse(BaseModel):
|
|
"""Response model for face recognition endpoint.
|
|
|
|
Returns the result of attempting to recognize a face from an uploaded image.
|
|
"""
|
|
|
|
success: bool = Field(description="Whether the face recognition was successful")
|
|
score: float | None = Field(
|
|
default=None, description="Confidence score of the recognition (0-1)"
|
|
)
|
|
face_name: str | None = Field(
|
|
default=None, description="The recognized face name if successful"
|
|
)
|