Spaces:
Running
Running
File size: 1,994 Bytes
24f95f0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | from sqlalchemy import Column, String, Float, DateTime, JSON, ForeignKey, Boolean
from sqlalchemy.dialects.postgresql import UUID
from app.db.session import Base
from datetime import datetime
import uuid
# Note: pgvector requires 'pgvector' extension in Postgres
# and 'pgvector' python package.
try:
from pgvector.sqlalchemy import Vector
except ImportError:
# Fallback for environments where pgvector isn't installed yet
class Vector:
def __init__(self, *args, **kwargs): pass
class ScamEvent(Base):
__tablename__ = "scam_events"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
text = Column(String, nullable=False)
source = Column(String)
risk_score = Column(Float)
decision = Column(String)
created_at = Column(DateTime, default=datetime.utcnow)
event_metadata = Column("metadata", JSON) # Store intent and extra details
embedding = Column(Vector(384)) # matches sentence-transformers (e.g. all-MiniLM-L6-v2)
class Entity(Base):
__tablename__ = "entities"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
type = Column(String) # "phone", "upi", "domain", "brand"
value = Column(String, unique=True)
first_seen = Column(DateTime, default=datetime.utcnow)
last_seen = Column(DateTime, default=datetime.utcnow)
reputation_score = Column(Float, default=0.0)
class EventEntity(Base):
__tablename__ = "event_entities"
event_id = Column(UUID(as_uuid=True), ForeignKey("scam_events.id"), primary_key=True)
entity_id = Column(UUID(as_uuid=True), ForeignKey("entities.id"), primary_key=True)
class Feedback(Base):
__tablename__ = "feedback"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
event_id = Column(UUID(as_uuid=True), ForeignKey("scam_events.id"))
is_scam = Column(Boolean)
correct_category = Column(String)
notes = Column(String)
created_at = Column(DateTime, default=datetime.utcnow)
|