aa358df28e
- Introduction of retrievers - Ensuring processing information is collected from Catalog iso Tenant - Introduction of a generic Form class to enable dynamic fields based on a configuration - Realisation of Retriever functionality to support dynamic fields
31 lines
1003 B
Python
31 lines
1003 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,
|
|
},
|
|
"rag_tuning": {
|
|
"name": "rag_tuning",
|
|
"type": "boolean",
|
|
"description": "Whether to do tuning logging or not.",
|
|
"required": False,
|
|
"default": False,
|
|
}
|
|
}
|
|
}
|
|
}
|