89 lines
3.6 KiB
Python
89 lines
3.6 KiB
Python
from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, Text
|
|
from sqlalchemy.orm import relationship
|
|
from datetime import datetime, timezone
|
|
from .database import Base
|
|
from sqlalchemy import Float, DateTime
|
|
from datetime import datetime, timezone
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
email = Column(String, unique=True, index=True, nullable=False)
|
|
username = Column(String, unique=True, index=True, nullable=False)
|
|
hashed_password = Column(String, nullable=False)
|
|
totp_secret = Column(String, nullable=False)
|
|
is_admin = Column(Boolean, default=False)
|
|
|
|
# Relation : Un utilisateur peut avoir plusieurs projets
|
|
projects = relationship("Project", back_populates="owner", cascade="all, delete-orphan")
|
|
|
|
|
|
class Project(Base):
|
|
__tablename__ = "projects"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
title = Column(String, nullable=False)
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
is_pinned = Column(Boolean, default=False)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
|
|
|
|
# Relations
|
|
owner = relationship("User", back_populates="projects")
|
|
messages = relationship("Message", back_populates="project", cascade="all, delete-orphan")
|
|
|
|
|
|
class Message(Base):
|
|
__tablename__ = "messages"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
role = Column(String, nullable=False) # Ex: 'user', 'assistant', 'worker_1', 'system'
|
|
content = Column(Text, nullable=False)
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
project_id = Column(Integer, ForeignKey("projects.id"), nullable=False)
|
|
|
|
# Relation
|
|
project = relationship("Project", back_populates="messages")
|
|
|
|
class SystemSettings(Base):
|
|
__tablename__ = "system_settings"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
# Serveur Mail (Obligatoire)
|
|
smtp_host = Column(String, nullable=False)
|
|
smtp_port = Column(Integer, nullable=False)
|
|
smtp_user = Column(String, nullable=False)
|
|
smtp_password = Column(String, nullable=False)
|
|
# Clés API (Optionnelles)
|
|
openrouter_api_key = Column(String, nullable=True)
|
|
openrouter_management_key = Column(String, nullable=True)
|
|
groq_api_key = Column(String, nullable=True)
|
|
gemini_api_key = Column(String, nullable=True)
|
|
deepseek_api_key = Column(String, nullable=True)
|
|
mistral_api_key = Column(String, nullable=True)
|
|
cloudflare_account_id = Column(String, nullable=True)
|
|
cloudflare_api_token = Column(String, nullable=True)
|
|
huggingface_api_key = Column(String, nullable=True)
|
|
|
|
class AIModel(Base):
|
|
"""Stocke la liste des modèles extraits depuis les fournisseurs."""
|
|
__tablename__ = "ai_models"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
provider = Column(String, index=True) # ex: openrouter, groq
|
|
model_id = Column(String, unique=True, index=True) # ex: google/gemini-1.5-pro
|
|
name = Column(String)
|
|
context_length = Column(Integer)
|
|
pricing_prompt = Column(Float) # Coût pour 1M tokens
|
|
pricing_completion = Column(Float)
|
|
last_updated = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
|
|
class FinancialLog(Base):
|
|
"""Suivi financier par fournisseur."""
|
|
__tablename__ = "financial_logs"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
provider = Column(String, index=True)
|
|
balance = Column(Float) # Crédit restant
|
|
total_usage = Column(Float) # Consommation totale
|
|
checked_at = Column(DateTime, default=lambda: datetime.now(timezone.utc)) |