pablodanswer e022e77b6d
Simpler azure embedding (#2751)
* functional but janky

* nit

* adapt for azure

* nit

* minor updates

* nits

* nit

* nit

* ensure access to litellm

* k
2024-10-15 23:23:11 +00:00

21 lines
353 B
Python

from enum import Enum
class EmbeddingProvider(str, Enum):
OPENAI = "openai"
COHERE = "cohere"
VOYAGE = "voyage"
GOOGLE = "google"
LITELLM = "litellm"
AZURE = "azure"
class RerankerProvider(str, Enum):
COHERE = "cohere"
LITELLM = "litellm"
class EmbedTextType(str, Enum):
QUERY = "query"
PASSAGE = "passage"