From 054bca22a1e8cd74b4832ffdb8f3de96dffd565a Mon Sep 17 00:00:00 2001
From: Aakash Thatte <aathatte2002@gmail.com>
Date: Sun, 25 Feb 2024 01:40:01 +0530
Subject: [PATCH] Add import for marqo

---
 gptcache/utils/__init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gptcache/utils/__init__.py b/gptcache/utils/__init__.py
index 093fd354..c588c215 100644
--- 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")