# 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,
            }
        }
    }
}