forked from HSE_team/BetterCallPraskovia
26 lines
695 B
Python
26 lines
695 B
Python
"""
|
|
Доменная сущность Embedding
|
|
"""
|
|
from datetime import datetime
|
|
from uuid import UUID, uuid4
|
|
from typing import Any
|
|
|
|
|
|
class Embedding:
|
|
"""Эмбеддинг документа"""
|
|
|
|
def __init__(
|
|
self,
|
|
document_id: UUID,
|
|
embedding: list[float] | None = None,
|
|
model_version: str = "",
|
|
embedding_id: UUID | None = None,
|
|
created_at: datetime | None = None
|
|
):
|
|
self.embedding_id = embedding_id or uuid4()
|
|
self.document_id = document_id
|
|
self.embedding = embedding or []
|
|
self.model_version = model_version
|
|
self.created_at = created_at or datetime.utcnow()
|
|
|