__init__.py

from typing import Optional

from dotenv import load_dotenv
from semantic_router.encoders import BaseEncoder
from semantic_router.encoders.openai import OpenAIEncoder

from models.vector_database import VectorDatabase
from vectordbs.astra import AstraService
from vectordbs.base import BaseVectorDatabase
from vectordbs.pinecone import PineconeService
from vectordbs.qdrant import QdrantService
from vectordbs.weaviate import WeaviateService
from vectordbs.pgvector import PGVectorService

load_dotenv()


def get_vector_service(
    *,
    index_name: str,
    credentials: VectorDatabase,
    encoder: BaseEncoder = OpenAIEncoder(),
    dimensions: Optional[int] = 384,
) -> BaseVectorDatabase:
    services = {
        "pinecone": PineconeService,
        "qdrant": QdrantService,
        "weaviate": WeaviateService,
        "astra": AstraService,
        "pgvector": PGVectorService,
        # Add other providers here
        # e.g "weaviate": WeaviateVectorService,
    }
    service = services.get(credentials.type.value)

    if service is None:
        raise ValueError(f"Unsupported provider: {credentials.type.value}")

    return service(
        index_name=index_name,
        dimension=dimensions,
        credentials=dict(credentials.config),
        encoder=encoder,
    )