feat: type whisper transcripts

This commit is contained in:
Felix Spöttel
2023-02-08 16:29:21 +01:00
parent 7c3b86d4a5
commit d9ce63ee39

View File

@@ -49,8 +49,20 @@ class Job(WithDbFields):
config: Optional[JobConfig]
class RawTranscript(BaseModel):
id: int
seek: int
start: float
end: float
text: str
tokens: List[int]
temperature: float
avg_logprob: float
compression_ratio: float
no_speech_prob: float
class Artifact(WithDbFields):
# TODO: narrow type
data: Optional[List[Any]]
data: Optional[List[RawTranscript]]
job_id: UUID
type: ArtifactType