1807435339
- Introduction of dynamic Processors - Introduction of caching system - Introduction of a better template manager - Adaptation of ModelVariables to support dynamic Processors / Retrievers / Specialists - Start adaptation of chat client
32 lines
993 B
Python
32 lines
993 B
Python
# Retriever Types
|
|
RETRIEVER_TYPES = {
|
|
"STANDARD_RAG": {
|
|
"name": "Standard RAG Retriever",
|
|
"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,
|
|
},
|
|
},
|
|
"arguments": {
|
|
"query": {
|
|
"name": "query",
|
|
"type": "str",
|
|
"description": "Query to retrieve embeddings",
|
|
"required": True,
|
|
},
|
|
}
|
|
}
|
|
}
|