mirror of
https://github.com/furyhawk/ai_agent.git
synced 2026-07-21 02:05:42 +00:00
furyhawk
8351e73d39
- Implemented `conversation-store` for managing conversations and messages. - Created `file-preview-store` to handle file preview state. - Added `sidebar-store` for sidebar visibility management. - Developed `theme-store` for theme persistence and management. - Introduced `kb-selection-store` for managing active knowledge base selections with persistence. chore: define API and chat types - Added types for API responses, authentication, chat messages, conversations, and projects. - Defined interfaces for various entities including users, sessions, and message ratings. build: configure TypeScript and testing setup - Set up `tsconfig.json` for TypeScript configuration. - Created `vitest.config.ts` for testing configuration with Vitest. - Added `vitest.setup.ts` for global test setup including mocks for Next.js router and media queries. - Configured Vercel deployment settings in `vercel.json`.
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
"""Message rating model for user feedback on AI responses.
|
|
|
|
This module is only imported when JWT auth is enabled (see
|
|
`app/db/models/__init__.py` and `alembic/env.py`).
|
|
"""
|
|
|
|
import uuid
|
|
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import CheckConstraint, ForeignKey, Integer, Text, UniqueConstraint
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base, TimestampMixin
|
|
|
|
if TYPE_CHECKING:
|
|
from app.db.models.conversation import Message
|
|
from app.db.models.user import User
|
|
|
|
|
|
class MessageRating(Base, TimestampMixin):
|
|
"""User rating for AI assistant messages."""
|
|
|
|
__tablename__ = "message_ratings"
|
|
__table_args__ = (
|
|
UniqueConstraint("message_id", "user_id", name="uq_message_user_rating"),
|
|
CheckConstraint("rating IN (1, -1)", name="ck_rating_value"),
|
|
)
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
message_id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("messages.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("users.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
rating: Mapped[int] = mapped_column(Integer, nullable=False) # 1 or -1
|
|
comment: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
message: Mapped["Message"] = relationship(
|
|
"Message",
|
|
foreign_keys="MessageRating.message_id",
|
|
)
|
|
user: Mapped["User"] = relationship(
|
|
"User",
|
|
foreign_keys="MessageRating.user_id",
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<MessageRating(id={self.id}, rating={self.rating})>"
|