mirror of
https://github.com/bellingcat/whisperbox-transcribe.git
synced 2026-06-12 21:48:35 +03:00
chore(deps): migrate to pydantic v2
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from datetime import datetime
|
||||
from uuid import UUID
|
||||
|
||||
from pydantic import AnyHttpUrl, BaseModel
|
||||
from pydantic import AnyHttpUrl, BaseModel, ConfigDict
|
||||
|
||||
from app.shared.db.models import (
|
||||
ArtifactData,
|
||||
@@ -18,10 +18,8 @@ from app.shared.db.models import (
|
||||
class WithDbFields(BaseModel):
|
||||
id: UUID
|
||||
created_at: datetime
|
||||
updated_at: datetime | None
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
updated_at: datetime | None = None
|
||||
model_config = ConfigDict(from_attributes=True)
|
||||
|
||||
|
||||
class Job(WithDbFields):
|
||||
@@ -30,8 +28,8 @@ class Job(WithDbFields):
|
||||
status: JobStatus
|
||||
type: JobType
|
||||
url: AnyHttpUrl
|
||||
meta: JobMeta | None
|
||||
config: JobConfig | None
|
||||
meta: JobMeta | None = None
|
||||
config: JobConfig | None = None
|
||||
|
||||
|
||||
class Artifact(WithDbFields):
|
||||
|
||||
@@ -129,10 +129,11 @@ def app_factory(
|
||||
)
|
||||
|
||||
language: str | None = Field(
|
||||
default=None,
|
||||
description=(
|
||||
"Spoken language in the media file. "
|
||||
"While optional, this can improve output when set."
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
@api_router.post(
|
||||
@@ -161,7 +162,7 @@ def app_factory(
|
||||
|
||||
# create a job with status "create" and save it to the database.
|
||||
job = models.Job(
|
||||
url=payload.url,
|
||||
url=str(payload.url),
|
||||
status=dtos.JobStatus.create,
|
||||
type=payload.type,
|
||||
config={"language": payload.language} if payload.language else None,
|
||||
|
||||
Reference in New Issue
Block a user