mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-30 09:40:35 +02:00
* add litellm proxy * formatting * move `api_url` to cloud provider + nits * remove log * typing * quick tuyping fix * update LiteLLM selection logic * remove logs + validate functionality * rename proxy var * update path casing * remove pricing for custom models * functional values
19 lines
309 B
Python
19 lines
309 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"
|
|
|
|
|
|
class EmbedTextType(str, Enum):
|
|
QUERY = "query"
|
|
PASSAGE = "passage"
|