compair-core 0.3.1__py3-none-any.whl → 0.3.2__py3-none-any.whl

compair_core-0.3.2.dist-info/RECORD

@@ -0,0 +1,36 @@
+compair/__init__.py,sha256=OUXUSHK8JTlWMgCpzodE1B6Xc7glkCcRMHZQi02sRCM,2594
+compair/celery_app.py,sha256=OM_Saza9yC9Q0kz_WXctfswrKkG7ruT52Zl5E4guiT0,640
+compair/default_groups.py,sha256=dbacrFkSjqEQZ_uoFU5gYhgIoP_3lmvz6LJNHCJvxlw,498
+compair/embeddings.py,sha256=ELpMYWBw3nCjxTx9vLrDiHYptzlzj_7JW6jZAf2Iql4,2341
+compair/feedback.py,sha256=jgDxYKo5PzW2p-uLf5ETmlL-mDAfqzJazO1NZDK-Z-g,2755
+compair/logger.py,sha256=mB9gV3FfC0qE_G9NBErpEAJhyGxDxEVKqWKu3n8hOkc,802
+compair/main.py,sha256=uc3q5TY3TZdC9UsqnbHX741rplNoveK0BwYOoCi0cCA,8073
+compair/models.py,sha256=J24eVQSPdiz13VdrKe8QsZh-nte8OZySf4LQwp4wPuU,13891
+compair/schema.py,sha256=TxQpDQ96J_tIj-Y_1C_x2eUYn9n_LOG6XiNLCX1-GYY,2902
+compair/tasks.py,sha256=_wvwjeHMyH7dD9zbMyGDfwMp9KqRxR1901LX5rSsLbo,3590
+compair/utils.py,sha256=5Cw3FrOsCc5jfO4R5Ghg_TXvh3DwHICl2PdwLZDm3XY,1581
+compair_core-0.3.2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+compair_email/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+compair_email/email.py,sha256=bedTrcfmi4K2kB0B2bSbryhq5k_4C-QNtb8wZjvzKAs,238
+compair_email/email_core.py,sha256=da7JxTo5ude55mB7UNLlpNp8xenYwoPaqyTunxjU7to,316
+compair_email/templates.py,sha256=JVlLdJEcpu14mVKRAYRIPIw2JGy70kG70mfjXgby-To,206
+compair_email/templates_core.py,sha256=1jPbXo36TjzNDCgYwL_4FyT5SsAZe5r6ufJe6W94eeE,424
+server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+server/app.py,sha256=XjdNPBX7978BWjWYS_66ekfrDuDsTWAEfndemaMFQF4,3413
+server/deps.py,sha256=0X-Z5JQGeXwbMooWIOC2kXVmsiJIvgUtqkK2PmDjKpI,1557
+server/settings.py,sha256=eqHrrv62h-l2VTDtiKPVfB5BV_ysA4W2qfX7Nxn7SNo,1476
+server/local_model/__init__.py,sha256=YlzDgorgAjGou9d_W29Xp3TVu08e4t9x8csFxn8cgSE,50
+server/local_model/app.py,sha256=2bOLjgAyKnFcng2lmDVMB0G6WgnnVGjHyj2X8TeJjnU,1626
+server/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+server/providers/console_mailer.py,sha256=7ady954yOPlT8t3spkDvdMdO3BTwosJUj1cpVNOwj8U,308
+server/providers/contracts.py,sha256=pYA_2AaPHw089O_UP2JtWRHbIiVkoNGhLuesuNATowU,1858
+server/providers/local_storage.py,sha256=UNCDHejeWlDlPVDRdwsN13toNRi_qHiaX-kbmzxf0OA,1054
+server/providers/noop_analytics.py,sha256=OKw23SObxBlQzFdB0xEBg5qD1Fcq_bcx6OOb4S3Mbd0,194
+server/providers/noop_billing.py,sha256=V18Cpl1D1reM3xhgw-lShGliVpYO8IsiAPWOAIR34jM,1358
+server/providers/noop_ocr.py,sha256=fMaJrivDef38-ECgIuTXUBCIm_avgvZf3nQ3UTdFPNI,341
+server/routers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+server/routers/capabilities.py,sha256=FZBLgmlIw6tWZiuihpzRmZ9AvbrY1jolVs7zqnhNPRU,1150
+compair_core-0.3.2.dist-info/METADATA,sha256=AQeN3eKdTcJGKJZKF4UAaKL7Mje41ONRutMhXW6d58s,4533
+compair_core-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+compair_core-0.3.2.dist-info/top_level.txt,sha256=-UjF8_GOz0G56FNiZDZnMSkEpnzBX__7qNrKpKLOqNA,29
+compair_core-0.3.2.dist-info/RECORD,,

compair_core-0.3.2.dist-info/top_level.txt

@@ -0,0 +1,3 @@
+compair
+compair_email
+server

compair_email/email.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+try:  # Cloud build ships the premium mailer
+    from compair_cloud.compair_email.email import *  # type: ignore
+except (ImportError, ModuleNotFoundError):
+    from .email_core import *  # type: ignore

compair_email/email_core.py

@@ -0,0 +1,15 @@
+import os
+
+from redmail import EmailSender
+
+EMAIL_HOST = f"{os.environ.get('EMAIL_HOST')}"
+EMAIL_USER = f"{os.environ.get('EMAIL_USER')}"
+EMAIL_PW = f"{os.environ.get('EMAIL_PW')}"
+
+emailer = EmailSender(
+    host=EMAIL_HOST,  
+    port=587, 
+    username=EMAIL_USER,
+    password=EMAIL_PW,
+    use_starttls=True 
+)

compair_email/templates.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+try:
+    from compair_cloud.compair_email.templates import *  # type: ignore
+except (ImportError, ModuleNotFoundError):
+    from .templates_core import *  # type: ignore

compair_email/templates_core.py

@@ -0,0 +1,13 @@
+"""Minimal email templates for the core edition."""
+
+ACCOUNT_VERIFY_TEMPLATE = """
+<p>Hi {{user_name}},</p>
+<p>Please verify your Compair account by clicking the link below:</p>
+<p><a href="{{verify_link}}">Verify my account</a></p>
+<p>Thanks!</p>
+""".strip()
+
+PASSWORD_RESET_TEMPLATE = """
+<p>We received a request to reset your password.</p>
+<p>Your password reset code is: <strong>{{reset_code}}</strong></p>
+""".strip()

server/app.py

@@ -0,0 +1,90 @@
+"""FastAPI app factory supporting Core and Cloud editions."""
+from __future__ import annotations
+
+from fastapi import FastAPI
+
+from .deps import (
+    get_analytics,
+    get_billing,
+    get_mailer,
+    get_ocr,
+    get_settings_dependency,
+    get_storage,
+)
+from .providers.local_storage import LocalStorage
+from .routers.capabilities import router as capabilities_router
+from .settings import Settings, get_settings
+
+
+def _normalize_edition(value: str) -> str:
+    return (value or "core").lower()
+
+
+def create_app(settings: Settings | None = None) -> FastAPI:
+    """Instantiate the FastAPI application with edition-specific wiring."""
+
+    resolved_settings = settings or get_settings()
+    edition = _normalize_edition(resolved_settings.edition)
+
+    app = FastAPI(title="Compair API", version=resolved_settings.version)
+
+    from api import router as legacy_router
+
+    app.include_router(legacy_router)
+    app.include_router(capabilities_router)
+
+    # Share the resolved settings with request handlers
+    app.dependency_overrides[get_settings_dependency] = lambda: resolved_settings
+
+    if edition == "cloud":
+        try:
+            from compair_cloud.analytics.ga4 import GA4Analytics
+            from compair_cloud.billing.stripe_provider import StripeBilling
+            from compair_cloud.mailer.transactional import TransactionalMailer
+            from compair_cloud.ocr.claude_ocr import ClaudeOCR
+            from compair_cloud.storage.r2_storage import R2Storage
+        except ImportError as exc:  # pragma: no cover - only triggered in misconfigured builds
+            raise RuntimeError(
+                "Cloud edition requires the private 'compair_cloud' package to be installed."
+            ) from exc
+
+        storage_provider = R2Storage(
+            bucket=resolved_settings.r2_bucket,
+            cdn_base=resolved_settings.r2_cdn_base,
+            access_key=resolved_settings.r2_access_key,
+            secret_key=resolved_settings.r2_secret_key,
+            endpoint_url=resolved_settings.r2_endpoint_url,
+        )
+        billing_provider = StripeBilling(
+            stripe_key=resolved_settings.stripe_key,
+            endpoint_secret=resolved_settings.stripe_endpoint_secret,
+        )
+        ocr_provider = ClaudeOCR()
+        mailer_provider = TransactionalMailer()
+
+        analytics_provider = None
+        if resolved_settings.ga4_measurement_id and resolved_settings.ga4_api_secret:
+            analytics_provider = GA4Analytics(
+                measurement_id=resolved_settings.ga4_measurement_id,
+                api_secret=resolved_settings.ga4_api_secret,
+            )
+
+        app.dependency_overrides[get_storage] = lambda sp=storage_provider: sp
+        app.dependency_overrides[get_billing] = lambda bp=billing_provider: bp
+        app.dependency_overrides[get_ocr] = lambda op=ocr_provider: op
+        if analytics_provider is not None:
+            app.dependency_overrides[get_analytics] = lambda ap=analytics_provider: ap
+        app.dependency_overrides[get_mailer] = lambda mp=mailer_provider: mp
+
+    else:
+        storage_provider = LocalStorage(
+            base_dir=resolved_settings.local_upload_dir,
+            base_url=resolved_settings.local_upload_base_url,
+        )
+        app.dependency_overrides[get_storage] = lambda sp=storage_provider: sp
+
+    return app
+
+
+# Uvicorn compatibility: allow ``uvicorn server.app:app``
+app = create_app()

server/deps.py

@@ -0,0 +1,67 @@
+"""Dependency entry points for features that differ by edition."""
+from __future__ import annotations
+
+from functools import lru_cache
+
+from fastapi import Depends
+
+from .providers.console_mailer import ConsoleMailer
+from .providers.contracts import Analytics, BillingProvider, Mailer, OCRProvider, StorageProvider
+from .providers.local_storage import LocalStorage
+from .providers.noop_analytics import NoopAnalytics
+from .providers.noop_billing import NoopBilling
+from .providers.noop_ocr import NoopOCR
+from .settings import Settings, get_settings
+
+
+@lru_cache
+def _local_storage_factory(base_dir: str, base_url: str) -> LocalStorage:
+    return LocalStorage(base_dir=base_dir, base_url=base_url)
+
+
+@lru_cache
+def _noop_billing() -> NoopBilling:
+    return NoopBilling()
+
+
+@lru_cache
+def _noop_ocr() -> NoopOCR:
+    return NoopOCR()
+
+
+@lru_cache
+def _console_mailer() -> ConsoleMailer:
+    return ConsoleMailer()
+
+
+@lru_cache
+def _noop_analytics() -> NoopAnalytics:
+    return NoopAnalytics()
+
+
+def get_settings_dependency() -> Settings:
+    """Expose settings as a FastAPI dependency."""
+    return get_settings()
+
+
+
+def get_storage(
+    settings: Settings = Depends(get_settings_dependency),
+) -> StorageProvider:
+    return _local_storage_factory(settings.local_upload_dir, settings.local_upload_base_url)
+
+
+def get_billing() -> BillingProvider:
+    return _noop_billing()
+
+
+def get_ocr() -> OCRProvider:
+    return _noop_ocr()
+
+
+def get_mailer() -> Mailer:
+    return _console_mailer()
+
+
+def get_analytics() -> Analytics:
+    return _noop_analytics()

server/local_model/__init__.py

@@ -0,0 +1 @@
+"""Local model endpoints for the Core edition."""

server/local_model/app.py

@@ -0,0 +1,62 @@
+"""Minimal FastAPI application serving local embedding and generation endpoints."""
+from __future__ import annotations
+
+import hashlib
+from typing import List
+
+from fastapi import FastAPI
+from pydantic import BaseModel
+
+app = FastAPI(title="Compair Local Model", version="0.1.0")
+
+EMBED_DIMENSION = 384
+
+
+def _hash_embedding(text: str, dimension: int = EMBED_DIMENSION) -> List[float]:
+    if not text:
+        text = " "
+    digest = hashlib.sha256(text.encode("utf-8", "ignore")).digest()
+    vector: List[float] = []
+    while len(vector) < dimension:
+        for byte in digest:
+            vector.append((byte / 255.0) * 2 - 1)
+            if len(vector) == dimension:
+                break
+        digest = hashlib.sha256(digest).digest()
+    return vector
+
+
+class EmbedRequest(BaseModel):
+    text: str
+
+
+class EmbedResponse(BaseModel):
+    embedding: List[float]
+
+
+class GenerateRequest(BaseModel):
+    system: str | None = None
+    prompt: str
+    verbosity: str | None = None
+
+
+class GenerateResponse(BaseModel):
+    text: str
+
+
+@app.post("/embed", response_model=EmbedResponse)
+def embed(request: EmbedRequest) -> EmbedResponse:
+    return EmbedResponse(embedding=_hash_embedding(request.text))
+
+
+@app.post("/generate", response_model=GenerateResponse)
+def generate(request: GenerateRequest) -> GenerateResponse:
+    prompt = request.prompt.strip()
+    if not prompt:
+        return GenerateResponse(text="NONE")
+
+    first_sentence = prompt.split("\n", 1)[0][:200]
+    verbosity = request.verbosity or "default"
+    return GenerateResponse(
+        text=f"[local-{verbosity}] Key takeaway: {first_sentence}"
+    )

server/providers/console_mailer.py

@@ -0,0 +1,9 @@
+"""Console mailer used in Core builds to avoid delivering real email."""
+from __future__ import annotations
+
+from typing import Iterable
+
+
+class ConsoleMailer:
+    def send(self, subject: str, sender: str, receivers: Iterable[str], html: str) -> None:
+        print(f"[MAIL] {subject} -> {list(receivers)}")

server/providers/contracts.py

@@ -0,0 +1,66 @@
+"""Provider protocol definitions shared across editions."""
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import BinaryIO, Iterable, Mapping, Protocol
+
+
+@dataclass(slots=True)
+class BillingSession:
+    """Represents the result of creating a checkout session."""
+
+    id: str
+    url: str
+
+
+class StorageProvider(Protocol):
+    def put_file(self, key: str, fileobj: BinaryIO, content_type: str) -> str: ...
+
+    def get_file(self, key: str) -> tuple[BinaryIO, str]: ...
+
+    def build_url(self, key: str) -> str: ...
+
+
+class BillingProvider(Protocol):
+    def ensure_customer(self, *, user_email: str, user_id: str) -> str: ...
+
+    def create_checkout_session(
+        self,
+        *,
+        customer_id: str,
+        price_id: str,
+        qty: int,
+        success_url: str,
+        cancel_url: str,
+        metadata: Mapping[str, str] | None = None,
+    ) -> BillingSession: ...
+
+    def retrieve_session(self, session_id: str) -> BillingSession: ...
+
+    def get_checkout_url(self, session_id: str) -> str: ...
+
+    def create_customer_portal(self, *, customer_id: str, return_url: str) -> str: ...
+
+    def create_coupon(self, amount: int) -> str: ...
+
+    def apply_coupon(self, *, customer_id: str, coupon_id: str) -> None: ...
+
+    def construct_event(self, payload: bytes, signature: str | None) -> Mapping[str, object]: ...
+
+
+class OCRProvider(Protocol):
+    def submit(
+        self, *, user_id: str, filename: str, data: bytes, document_id: str | None
+    ) -> str: ...
+
+    def status(self, task_id: str) -> Mapping[str, object]: ...
+
+
+class Mailer(Protocol):
+    def send(self, subject: str, sender: str, receivers: Iterable[str], html: str) -> None: ...
+
+
+class Analytics(Protocol):
+    def track(
+        self, event_name: str, user_id: str, params: Mapping[str, object] | None = None
+    ) -> None: ...

server/providers/local_storage.py

@@ -0,0 +1,28 @@
+"""Local filesystem storage provider used in the Core edition."""
+from __future__ import annotations
+
+from pathlib import Path
+from typing import BinaryIO
+
+
+class LocalStorage:
+    def __init__(self, base_dir: str = "/data/uploads", base_url: str = "/uploads") -> None:
+        self.base_dir = Path(base_dir)
+        self.base_url = base_url.rstrip("/") or "/uploads"
+        self.base_dir.mkdir(parents=True, exist_ok=True)
+
+    def put_file(self, key: str, fileobj: BinaryIO, content_type: str) -> str:
+        destination = self.base_dir / key
+        destination.parent.mkdir(parents=True, exist_ok=True)
+        with destination.open("wb") as dest:
+            dest.write(fileobj.read())
+        return self.build_url(key)
+
+    def get_file(self, key: str) -> tuple[BinaryIO, str]:
+        path = self.base_dir / key
+        if not path.exists():
+            raise FileNotFoundError(key)
+        return path.open("rb"), "application/octet-stream"
+
+    def build_url(self, key: str) -> str:
+        return f"{self.base_url}/{key}".replace("//", "/")

server/providers/noop_analytics.py

@@ -0,0 +1,7 @@
+"""Analytics provider placeholder that records no events."""
+from __future__ import annotations
+
+
+class NoopAnalytics:
+    def track(self, *_: object, **__: object) -> None:
+        return None

server/providers/noop_billing.py

@@ -0,0 +1,30 @@
+"""Billing provider placeholder for the Core edition."""
+from __future__ import annotations
+
+from .contracts import BillingSession
+
+
+class NoopBilling:
+    def ensure_customer(self, **_: object) -> str:
+        raise NotImplementedError("Billing is not available in the Core edition.")
+
+    def create_checkout_session(self, **_: object) -> BillingSession:
+        raise NotImplementedError("Billing is not available in the Core edition.")
+
+    def retrieve_session(self, *_: object, **__: object) -> BillingSession:
+        raise NotImplementedError("Billing is not available in the Core edition.")
+
+    def get_checkout_url(self, *_: object, **__: object) -> str:
+        raise NotImplementedError("Billing is not available in the Core edition.")
+
+    def create_customer_portal(self, *_: object, **__: object) -> str:
+        raise NotImplementedError("Billing is not available in the Core edition.")
+
+    def create_coupon(self, *_: object, **__: object) -> str:
+        raise NotImplementedError("Billing is not available in the Core edition.")
+
+    def apply_coupon(self, *_: object, **__: object) -> None:
+        raise NotImplementedError("Billing is not available in the Core edition.")
+
+    def construct_event(self, *_: object, **__: object) -> dict[str, object]:
+        raise NotImplementedError("Billing is not available in the Core edition.")

server/providers/noop_ocr.py

@@ -0,0 +1,10 @@
+"""OCR provider placeholder for the Core edition."""
+from __future__ import annotations
+
+
+class NoopOCR:
+    def submit(self, **_: object) -> str:
+        raise NotImplementedError("OCR is not available in the Core edition.")
+
+    def status(self, task_id: str) -> dict[str, object]:
+        return {"status": "unknown", "task_id": task_id}

server/routers/capabilities.py

@@ -0,0 +1,38 @@
+"""Meta endpoints that describe edition capabilities for the CLI."""
+from __future__ import annotations
+
+from fastapi import APIRouter, Depends
+
+from ..settings import Settings, get_settings
+
+router = APIRouter(tags=["meta"])
+
+
+@router.get("/capabilities")
+def capabilities(settings: Settings = Depends(get_settings)) -> dict[str, object]:
+    edition = settings.edition.lower()
+    return {
+        "auth": {
+            "device_flow": edition == "cloud",
+            "password_login": True,
+        },
+        "inputs": {
+            "text": True,
+            "ocr": settings.ocr_enabled,
+            "repos": True,
+        },
+        "models": {
+            "premium": settings.premium_models,
+            "open": True,
+        },
+        "integrations": {
+            "slack": settings.integrations_enabled,
+            "github": settings.integrations_enabled,
+        },
+        "limits": {
+            "docs": None if edition == "core" else 100,
+            "feedback_per_day": None if edition == "core" else 50,
+        },
+        "server": "Compair Cloud" if edition == "cloud" else "Compair Core",
+        "version": settings.version,
+    }

server/settings.py

@@ -0,0 +1,51 @@
+"""Application settings and feature flag definitions."""
+from functools import lru_cache
+
+from pydantic_settings import BaseSettings
+
+
+class Settings(BaseSettings):
+    """Configuration injected via COMPAIR_ environment variables."""
+
+    # Edition metadata
+    edition: str = "core"  # core | cloud
+    version: str = "dev"
+
+    # Feature gates
+    ocr_enabled: bool = False
+    billing_enabled: bool = False
+    integrations_enabled: bool = False
+    premium_models: bool = False
+
+    # Core/local storage defaults
+    local_upload_dir: str = "/data/uploads"
+    local_upload_base_url: str = "/uploads"
+
+    # Cloud storage (R2/S3-compatible)
+    r2_bucket: str | None = None
+    r2_cdn_base: str | None = None
+    r2_access_key: str | None = None
+    r2_secret_key: str | None = None
+    r2_endpoint_url: str | None = None
+
+    # Optional cloud secrets
+    stripe_key: str | None = None
+    stripe_endpoint_secret: str | None = None
+    stripe_success_url: str = "https://compair.sh/home"
+    stripe_cancel_url: str = "https://compair.sh/home"
+    ga4_measurement_id: str | None = None
+    ga4_api_secret: str | None = None
+
+    # Local model endpoints
+    local_model_url: str = "http://local-model:9000"
+    local_embedding_route: str = "/embed"
+    local_generation_route: str = "/generate"
+
+    class Config:
+        env_prefix = "COMPAIR_"
+
+
+@lru_cache
+def get_settings() -> Settings:
+    """Cached settings instance for dependency injection."""
+    return Settings()

compair_core-0.3.1.dist-info/RECORD

@@ -1,5 +0,0 @@
-compair_core-0.3.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-compair_core-0.3.1.dist-info/METADATA,sha256=I7eh0b6GwaAqI3XASxC-qYVQDVVLCpdTXWadD3CZ7hI,4466
-compair_core-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-compair_core-0.3.1.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-compair_core-0.3.1.dist-info/RECORD,,

compair_core-0.3.1.dist-info/top_level.txt

@@ -1 +0,0 @@
-