From 307bde9c1ce4b235d6e31c4cf4a878f15efbf173 Mon Sep 17 00:00:00 2001 From: Aarthi Iyer Date: Tue, 3 Dec 2024 19:35:24 +0000 Subject: [PATCH] reformatted --- weaviate_cli/commands/create.py | 10 +++++++++- weaviate_cli/managers/collection_manager.py | 2 +- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/weaviate_cli/commands/create.py b/weaviate_cli/commands/create.py index 8d0cc5a..d8025f9 100644 --- a/weaviate_cli/commands/create.py +++ b/weaviate_cli/commands/create.py @@ -92,7 +92,15 @@ def create() -> None: "--vectorizer", default=CreateCollectionDefaults.vectorizer, type=click.Choice( - ["contextionary", "transformers", "openai", "ollama", "cohere", "jinaai", "weaviate"] + [ + "contextionary", + "transformers", + "openai", + "ollama", + "cohere", + "jinaai", + "weaviate", + ] ), help="Vectorizer to use.", ) diff --git a/weaviate_cli/managers/collection_manager.py b/weaviate_cli/managers/collection_manager.py index a2ae06b..29ec8c5 100644 --- a/weaviate_cli/managers/collection_manager.py +++ b/weaviate_cli/managers/collection_manager.py @@ -126,7 +126,7 @@ def create_collection( ), "cohere": wvc.Configure.Vectorizer.text2vec_cohere(), "jinaai": wvc.Configure.Vectorizer.text2vec_jinaai(), - "weaviate": wvc.Configure.Vectorizer.text2vec_weaviate() + "weaviate": wvc.Configure.Vectorizer.text2vec_weaviate(), } inverted_index_map: Dict[str, wvc.InvertedIndexConfig] = {