5841525b4c
- Tuning moved to Retriever iso in the configuration, as this is an attribute that should be available for all types of Retrievers
24 lines
748 B
Python
24 lines
748 B
Python
# Retriever Types
|
|
RETRIEVER_TYPES = {
|
|
"DEFAULT_RAG": {
|
|
"name": "Default RAG",
|
|
"description": "Retrieving all embeddings conform the query",
|
|
"configuration": {
|
|
"es_k": {
|
|
"name": "es_k",
|
|
"type": "int",
|
|
"description": "K-value to retrieve embeddings (max embeddings retrieved)",
|
|
"required": True,
|
|
"default": 8,
|
|
},
|
|
"es_similarity_threshold": {
|
|
"name": "es_similarity_threshold",
|
|
"type": "float",
|
|
"description": "Similarity threshold for retrieving embeddings",
|
|
"required": True,
|
|
"default": 0.3,
|
|
},
|
|
}
|
|
}
|
|
}
|