Skip to content

Integrate encode_query/encode_document from SentenceTransformers v5 #2871

@Samoed

Description

@Samoed

Description of the feature

For now we're using same prompts as SentneceTransformers, but in future there can be different models that would use Router. I think we can simply pass task type based on PromptType to encode without any other changes

https://github.com/UKPLab/sentence-transformers/blob/cfb492fce85e142bca6a2cdeca6492d5548fe1c5/sentence_transformers/SentenceTransformer.py#L416

Doc ref https://www.sbert.net/docs/package_reference/sentence_transformer/SentenceTransformer.html#sentence_transformers.SentenceTransformer.encode_query

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions