pablodanswer 299cb5035c
Add litellm proxy embeddings (#2291)
* 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
2024-09-02 09:08:35 -07:00

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"