from model.multimodal_embedding import Embedding
from PIL import Image
from utils.decorators.langfuse_trace_embedding import langfuse_trace_embedding

class TrackedMultiEmbeddings(Embedding):
    
    @langfuse_trace_embedding(name="multi-embed-query")
    def embed_query(self, text: str):
        return super().get_text_embedding(text)

    @langfuse_trace_embedding(name="multi-embed-documents")
    def embed_documents(self, texts: list[str]):
        return super().get_texts_embedding(texts)

    @langfuse_trace_embedding(name="multi-embed-image")
    def embed_image(self, image: Image):
        return super().get_image_embedding(image)

    @langfuse_trace_embedding(name="multi-embed-multimodal")
    def embed_multimodal(self, text: str, image: Image):
        return super().get_multimodal_embedding(text, image)