Merge pull request #101 from weaviate/text2vec-weaviate

support text2vec_weaviate vectorizer
weaviate · Dec 3, 2024 · 69b7b31 · 69b7b31
2 parents 973c3e4 + 307bde9
commit 69b7b31
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -37,7 +37,7 @@ classifiers =
 include_package_data = True
 python_requires = >=3.9
 install_requires =
-    weaviate-client>=4.5.0
+    weaviate-client>=4.9.5
     click==8.1.7
     semver>=3.0.2
     numpy>=1.24.0

diff --git 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"]
+        [
+            "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
@@ -126,6 +126,7 @@ def create_collection(
             ),
             "cohere": wvc.Configure.Vectorizer.text2vec_cohere(),
             "jinaai": wvc.Configure.Vectorizer.text2vec_jinaai(),
+            "weaviate": wvc.Configure.Vectorizer.text2vec_weaviate(),
         }
 
         inverted_index_map: Dict[str, wvc.InvertedIndexConfig] = {