mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-03-29 03:01:59 +01:00
3a9b964d5c
* add ability ot set reranking litellm proxy * add fully functional rerank litellm cards * minor formatting enforcement * remove logs
20 lines
333 B
Python
20 lines
333 B
Python
from enum import Enum
|
|
|
|
|
|
class EmbeddingProvider(str, Enum):
|
|
OPENAI = "openai"
|
|
COHERE = "cohere"
|
|
VOYAGE = "voyage"
|
|
GOOGLE = "google"
|
|
LITELLM = "litellm"
|
|
|
|
|
|
class RerankerProvider(str, Enum):
|
|
COHERE = "cohere"
|
|
LITELLM = "litellm"
|
|
|
|
|
|
class EmbedTextType(str, Enum):
|
|
QUERY = "query"
|
|
PASSAGE = "passage"
|