mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-04-08 20:08:36 +02:00
* functional but janky * nit * adapt for azure * nit * minor updates * nits * nit * nit * ensure access to litellm * k
21 lines
353 B
Python
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"
|