From 86ac7cd63e8f3bc9e9dc2c6557e64434c8575d7e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Pedersen?= <andrped94@gmail.com>
Date: Sat, 21 Sep 2024 11:31:11 +0200
Subject: [PATCH] Added missing import tests

---
 tests/test_imports.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/tests/test_imports.py b/tests/test_imports.py
index 0efd180f..91a18cca 100644
--- a/tests/test_imports.py
+++ b/tests/test_imports.py
@@ -2,8 +2,13 @@
 
 def test_regular_imports():
     from vanna.anthropic.anthropic_chat import Anthropic_Chat
+    from vanna.azuresearch.azuresearch_vector import AzureAISearch_VectorStore
     from vanna.base.base import VannaBase
+    from vanna.bedrock.bedrock_converse import Bedrock_Converse
     from vanna.chromadb.chromadb_vector import ChromaDB_VectorStore
+    from vanna.faiss.faiss import FAISS
+    from vanna.google.bigquery_vector import BigQuery_VectorStore
+    from vanna.google.gemini_chat import GoogleGeminiChat
     from vanna.hf.hf import Hf
     from vanna.local import LocalContext_OpenAI
     from vanna.marqo.marqo import Marqo_VectorStore
@@ -14,17 +19,23 @@ def test_regular_imports():
     from vanna.openai.openai_embeddings import OpenAI_Embeddings
     from vanna.opensearch.opensearch_vector import OpenSearch_VectorStore
     from vanna.pinecone.pinecone_vector import PineconeDB_VectorStore
+    from vanna.qdrant.qdrant import Qdrant_VectorStore
+    from vanna.qianfan.Qianfan_Chat import Qianfan_Chat
+    from vanna.qianfan.Qianfan_embeddings import Qianfan_Embeddings
+    from vanna.qianwen.QianwenAI_chat import QianWenAI_Chat
+    from vanna.qianwen.QianwenAI_embeddings import QianWenAI_Embeddings
     from vanna.remote import VannaDefault
     from vanna.vannadb.vannadb_vector import VannaDB_VectorStore
     from vanna.weaviate.weaviate_vector import WeaviateDatabase
     from vanna.ZhipuAI.ZhipuAI_Chat import ZhipuAI_Chat
     from vanna.ZhipuAI.ZhipuAI_embeddings import ZhipuAI_Embeddings
-    from vanna.azuresearch.azuresearch_vector import AzureAISearch_VectorStore
 
 def test_shortcut_imports():
     from vanna.anthropic import Anthropic_Chat
+    from vanna.azuresearch import AzureAISearch_VectorStore
     from vanna.base import VannaBase
     from vanna.chromadb import ChromaDB_VectorStore
+    from vanna.faiss import FAISS
     from vanna.hf import Hf
     from vanna.marqo import Marqo_VectorStore
     from vanna.milvus import Milvus_VectorStore
@@ -33,8 +44,10 @@ def test_shortcut_imports():
     from vanna.openai import OpenAI_Chat, OpenAI_Embeddings
     from vanna.opensearch import OpenSearch_VectorStore
     from vanna.pinecone import PineconeDB_VectorStore
+    from vanna.qdrant import Qdrant_VectorStore
+    from vanna.qianfan import Qianfan_Chat, Qianfan_Embeddings
+    from vanna.qianwen import QianWenAI_Chat, QianWenAI_Embeddings
     from vanna.vannadb import VannaDB_VectorStore
     from vanna.vllm import Vllm
     from vanna.weaviate import WeaviateDatabase
     from vanna.ZhipuAI import ZhipuAI_Chat, ZhipuAI_Embeddings
-    from vanna.azuresearch import AzureAISearch_VectorStore
\ No newline at end of file