diff --git a/pyproject.toml b/pyproject.toml index 25609a00..7f439c1e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,7 +33,7 @@ bigquery = ["google-cloud-bigquery"] snowflake = ["snowflake-connector-python"] duckdb = ["duckdb"] google = ["google-generativeai", "google-cloud-aiplatform"] -all = ["psycopg2-binary", "db-dtypes", "PyMySQL", "google-cloud-bigquery", "snowflake-connector-python", "duckdb", "openai", "qianfan", "mistralai>=1.0.0", "chromadb", "anthropic", "zhipuai", "marqo", "google-generativeai", "google-cloud-aiplatform", "qdrant-client", "fastembed", "ollama", "httpx", "opensearch-py", "opensearch-dsl", "transformers", "pinecone-client", "pymilvus[model]","weaviate-client", "azure-search-documents", "azure-identity", "azure-common"] +all = ["psycopg2-binary", "db-dtypes", "PyMySQL", "google-cloud-bigquery", "snowflake-connector-python", "duckdb", "openai", "qianfan", "mistralai>=1.0.0", "chromadb", "anthropic", "zhipuai", "marqo", "google-generativeai", "google-cloud-aiplatform", "qdrant-client", "fastembed", "ollama", "httpx", "opensearch-py", "opensearch-dsl", "transformers", "pinecone-client", "pymilvus[model]","weaviate-client", "azure-search-documents", "azure-identity", "azure-common", "faiss-cpu", "boto", "botocore"] test = ["tox"] chromadb = ["chromadb"] openai = ["openai"] @@ -53,3 +53,5 @@ milvus = ["pymilvus[model]"] bedrock = ["boto3", "botocore"] weaviate = ["weaviate-client"] azuresearch = ["azure-search-documents", "azure-identity", "azure-common", "fastembed"] +faiss-cpu = ["faiss-cpu"] +faiss-gpu = ["faiss-gpu"] 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