from __future__ import annotations

from enum import Enum


class LucosSourceType(str, Enum):
    """Daemon upload source types (lucos-daemon-data-flow.md)."""

    REPO_CODE = "repo_code"
    REPO_DOC = "repo_doc"


class EmbeddingSourceType(str, Enum):
    """embedding-engine normalized source types."""

    CODE = "code"
    DOCS = "docs"


def to_embedding_source_type(source_type: LucosSourceType) -> EmbeddingSourceType:
    if source_type == LucosSourceType.REPO_DOC:
        return EmbeddingSourceType.DOCS
    return EmbeddingSourceType.CODE