/
githubmirror
/
Open-Assistant
Обзор
Документация
Войти
/
githubmirror
/
Open-Assistant
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
backend/oasst_backend/models/message_embedding.py
21 строка
860 B
Andreas Köpf
last_uddated for user_stats, swtich to timestamptz
20 янв 2023, 00:07
20 янв 2023, 00:07
038d1cc
Код
Авторство
О чём код?
from datetime import datetime from typing import List, Optional from uuid import UUID import sqlalchemy as sa import sqlalchemy.dialects.postgresql as pg from sqlmodel import ARRAY, Field, Float, SQLModel class MessageEmbedding(SQLModel, table=True): __tablename__ = "message_embedding" __table_args__ = (sa.PrimaryKeyConstraint("message_id", "model"),) message_id: UUID = Field(sa_column=sa.Column(pg.UUID(as_uuid=True), sa.ForeignKey("message.id"), nullable=False)) model: str = Field(max_length=256, nullable=False) embedding: List[float] = Field(sa_column=sa.Column(ARRAY(Float)), nullable=True) # In the case that the Message Embedding is created afterwards created_date: Optional[datetime] = Field( sa_column=sa.Column(sa.DateTime(timezone=True), nullable=False, server_default=sa.func.current_timestamp()) )