RAGchain.utils.embed package

Submodules

RAGchain.utils.embed.embeddingfactory module

class RAGchain.utils.embed.embeddingfactory.EmbeddingFactory(embed_type: str, device_type: str = 'cuda')

Bases: object

EmbeddingFactory is a factory class that returns the embedding class according to the embedding type. You can create embedding class easily by using this class.

get()

Returns the embedding class according to the embedding type.

class RAGchain.utils.embed.embeddingfactory.EmbeddingType(value)

Bases: Enum

An enumeration.

CONTRIEVER = 'contriever'

KOSIMCSE = 'kosimcse'

KO_SROBERTA_MULTITASK = 'ko-sroberta-multitask'

MULTILINGUAL_E5 = 'multilingual-e5'

OPENAI = 'openai'

Module contents