from dataclasses import dataclass


@dataclass
class EmbeddingDbConnection:
    connection: str
    database: str
    collection: str
    index: str


class Embedding:
    def embedd(self, chunks, metadata: list[dict] = None):
        pass

    def get_vector_store(self):
        pass

    def search(self, query, search_type, doc_count):
        pass