mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-03-17 21:32:36 +01:00
870b59a1cc
* Update text embedding model to version 005 and enhance embedding retrieval process * re * Fix formatting issues * Add support for Bedrock reranking provider and AWS credentials handling * fix: improve AWS key format validation and error messages * Fix vertex embedding model crash * feat: add environment template for local development setup * Add display name for Claude 3.7 Sonnet model * Add display names for Gemini 2.0 models and update Claude 3.7 Sonnet entry * Fix ruff errors by ensuring lines are within 130 characters * revert to currently default onyx browser settings * add / fix boto requirements --------- Co-authored-by: ferdinand loesch <f.loesch@sportradar.com> Co-authored-by: Ferdinand Loesch <ferdinandloesch@me.com> Co-authored-by: Richard Kuo (Danswer) <rkuo@onyx.app>
22 lines
377 B
Python
22 lines
377 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"
|
|
BEDROCK = "bedrock"
|
|
|
|
|
|
class EmbedTextType(str, Enum):
|
|
QUERY = "query"
|
|
PASSAGE = "passage"
|