From d8d0300c2a8f6e3be7278b5c1994709cb468040a Mon Sep 17 00:00:00 2001 From: Carles Onielfa Date: Wed, 15 Jan 2025 10:36:17 +0100 Subject: [PATCH] Fixes creating a KB from an Import --- nucliadb/src/nucliadb/writer/api/v1/knowledgebox.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/nucliadb/src/nucliadb/writer/api/v1/knowledgebox.py b/nucliadb/src/nucliadb/writer/api/v1/knowledgebox.py index ef297a517e..46c7fdd5d9 100644 --- a/nucliadb/src/nucliadb/writer/api/v1/knowledgebox.py +++ b/nucliadb/src/nucliadb/writer/api/v1/knowledgebox.py @@ -96,6 +96,16 @@ async def create_kb(item: KnowledgeBoxConfig) -> tuple[str, str]: if "semantic_model" in user_learning_config: user_learning_config["semantic_models"] = [user_learning_config.pop("semantic_model")] + deprecated_keys = [ + "semantic_vector_similarity", + "semantic_vector_size", + "semantic_matryoshka_dimensions", + "semantic_model_configs", + "semantic_threshold", + ] + for key in deprecated_keys: + user_learning_config.pop(key, None) + # we rely on learning to return the updated configuration with defaults and # any other needed values (e.g. matryoshka settings if available) learning_config = await learning_proxy.set_configuration(kbid, config=user_learning_config)