feat: define ORM models for JD, Candidate, MatchResult with Alembic migration

backend/src/models/__init__.py

@@ -0,0 +1,5 @@
+from .jd import JobDescription
+from .candidate import Candidate
+from .match_result import MatchResult
+
+__all__ = ["JobDescription", "Candidate", "MatchResult"]

backend/src/models/candidate.py

@@ -0,0 +1,53 @@
+import uuid
+from datetime import datetime
+from sqlalchemy import String, Text, DateTime, JSON, Float, Integer, Boolean, func
+from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.dialects.postgresql import UUID
+from ..database import Base
+
+
+class Candidate(Base):
+    __tablename__ = "candidates"
+
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    external_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
+
+    name: Mapped[str | None] = mapped_column(String(255), nullable=True)
+    email: Mapped[str | None] = mapped_column(String(255), nullable=True)
+
+    looking_for: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    currently_employed: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
+    notice_period: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    open_to_working_at: Mapped[str | None] = mapped_column(String(255), nullable=True)
+    role_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    engineer_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
+
+    years_of_experience: Mapped[float | None] = mapped_column(Float, nullable=True)
+    programming_languages: Mapped[list] = mapped_column(JSON, default=list)
+    backend_frameworks: Mapped[list] = mapped_column(JSON, default=list)
+    frontend_technologies: Mapped[list] = mapped_column(JSON, default=list)
+    gen_ai_experience: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
+    recent_experience_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    education_status: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    degree: Mapped[str | None] = mapped_column(String(255), nullable=True)
+
+    parsed_summary: Mapped[str | None] = mapped_column(Text, nullable=True)
+    parsed_skills: Mapped[str | None] = mapped_column(Text, nullable=True)
+    parsed_work_experience: Mapped[list] = mapped_column(JSON, default=list)
+
+    most_recent_company: Mapped[str | None] = mapped_column(String(255), nullable=True)
+    most_recent_company_description: Mapped[str | None] = mapped_column(Text, nullable=True)
+    most_recent_company_is_funded: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
+    most_recent_company_is_product_company: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
+    most_recent_company_total_funding: Mapped[float | None] = mapped_column(Float, nullable=True)
+    most_recent_company_funding_status: Mapped[str | None] = mapped_column(String(100), nullable=True)
+
+    time_in_current_company: Mapped[float | None] = mapped_column(Float, nullable=True)
+    is_actively_or_passively_looking: Mapped[str | None] = mapped_column(String(100), nullable=True)
+
+    growth_velocity: Mapped[float] = mapped_column(Float, default=0.5)
+    embedding_hash: Mapped[str | None] = mapped_column(String(64), nullable=True)
+    qdrant_id: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
+
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
+    updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())

backend/src/models/jd.py

@@ -0,0 +1,28 @@
+import uuid
+from datetime import datetime
+from sqlalchemy import String, Text, DateTime, JSON, Float, Integer, func
+from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.dialects.postgresql import UUID, ARRAY
+from ..database import Base
+
+
+class JobDescription(Base):
+    __tablename__ = "job_descriptions"
+
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    title: Mapped[str] = mapped_column(String(255))
+    raw_text: Mapped[str] = mapped_column(Text)
+    parsed_requirements: Mapped[dict] = mapped_column(JSON, default=dict)
+    required_skills: Mapped[list] = mapped_column(JSON, default=list)
+    min_yoe: Mapped[float | None] = mapped_column(Float, nullable=True)
+    max_yoe: Mapped[float | None] = mapped_column(Float, nullable=True)
+    role_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    engineer_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    location: Mapped[str | None] = mapped_column(String(255), nullable=True)
+    remote_allowed: Mapped[bool | None] = mapped_column(nullable=True)
+    jd_quality: Mapped[dict] = mapped_column(JSON, default=dict)
+    embedding_text: Mapped[str | None] = mapped_column(Text, nullable=True)
+    qdrant_id: Mapped[str | None] = mapped_column(String(64), nullable=True)
+    status: Mapped[str] = mapped_column(String(32), default="pending")
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
+    updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())

backend/src/models/match_result.py

@@ -0,0 +1,27 @@
+import uuid
+from datetime import datetime
+from sqlalchemy import String, DateTime, JSON, Float, ForeignKey, func
+from sqlalchemy.orm import Mapped, mapped_column
+from sqlalchemy.dialects.postgresql import UUID
+from ..database import Base
+
+
+class MatchResult(Base):
+    __tablename__ = "match_results"
+
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    jd_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("job_descriptions.id", ondelete="CASCADE"), index=True)
+    candidate_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("candidates.id", ondelete="CASCADE"), index=True)
+
+    rank: Mapped[int | None] = mapped_column(nullable=True)
+    stage1_score: Mapped[float] = mapped_column(Float, default=0.0)
+    stage2_score: Mapped[float | None] = mapped_column(Float, nullable=True)
+    final_score: Mapped[float] = mapped_column(Float, default=0.0)
+
+    component_scores: Mapped[dict] = mapped_column(JSON, default=dict)
+    gaps: Mapped[list] = mapped_column(JSON, default=list)
+    explanation: Mapped[str | None] = mapped_column(String, nullable=True)
+    explanation_generated_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+
+    weights_used: Mapped[dict] = mapped_column(JSON, default=dict)
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())