Prospectio MCP API

from typing import List, Optional from sqlalchemy import ARRAY, String, Text, JSON from sqlalchemy.orm import Mapped, mapped_column from sqlalchemy.dialects.postgresql import UUID import uuid from infrastructure.dto.database.base import Base class Company(Base): """ Represents a company with optional fields to match frontend requirements. SQLAlchemy model for database persistence. """ __tablename__ = "companies" id: Mapped[str] = mapped_column( UUID(as_uuid=False), primary_key=True, default=lambda: str(uuid.uuid4()), doc="Unique identifier for the company", ) name: Mapped[Optional[str]] = mapped_column(String(255), doc="Name of the company") industry: Mapped[Optional[str]] = mapped_column( String(255), doc="Industry sector of the company" ) compatibility: Mapped[Optional[str]] = mapped_column( String(50), doc="Compatibility rate with the prospect (e.g., '95% compatibility')", ) source: Mapped[Optional[str]] = mapped_column( String(255), doc="Source of the information (e.g., 'LinkedIn Sales Navigator')" ) location: Mapped[Optional[str]] = mapped_column( String(255), doc="Location of the company" ) size: Mapped[Optional[str]] = mapped_column( String(100), doc="Company size (e.g., '50-200 employees')" ) revenue: Mapped[Optional[str]] = mapped_column( String(100), doc="Company revenue (e.g., '5-10M€')" ) website: Mapped[Optional[str]] = mapped_column(String(500), doc="Company website") description: Mapped[Optional[str]] = mapped_column( Text, doc="Description of the company ('About')" ) opportunities: Mapped[Optional[List[str]]] = mapped_column( ARRAY(String), doc="List of opportunities or keywords associated with the company", ) def __repr__(self) -> str: """ String representation of the Company object. Returns: str: A string representation showing id, name, and industry. """ return ( f"Company(id={self.id!r}, name={self.name!r}, industry={self.industry!r})" ) # Type alias for a list of companies CompanyList = List[Company]