Add import for marqo

zilliztech · Feb 24, 2024 · 054bca2 · 054bca2
1 parent d749ff2
commit 054bca2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gptcache/utils/__init__.py b/gptcache/utils/__init__.py
@@ -42,6 +42,7 @@
     "import_redis",
     "import_qdrant",
     "import_weaviate",
+    "import_marqo",
     ]
 
 import importlib.util
@@ -59,6 +60,8 @@ def _check_library(libname: str, prompt: bool = True, package: Optional[str] = N
         prompt_install(package if package else libname)
     return is_avail
 
+def import_marqo():
+    _check_library("marqo")
 
 def import_pymilvus():
     _check_library("pymilvus")