from sqlalchemy import Column, String, ForeignKey, Text, Integer, JSON
from sqlalchemy.orm import relationship
from app.db.base import BaseModel


class AIConversation(BaseModel):
    __tablename__ = "ai_conversations"
    
    user_id = Column(String, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
    conversation_id = Column(String, nullable=False, unique=True, index=True)
    message = Column(Text, nullable=False)
    response = Column(Text, nullable=False)
    tokens_used = Column(Integer, nullable=True)
    model = Column(String(100), nullable=True)
    context = Column(JSON, nullable=True)  # Сохранение контекста для продолжения диалога
    
    # Relationships
    user = relationship("User", back_populates="conversations")