diff --git a/.env.template b/.env.template index ff591c0a7..75a57de4d 100644 --- a/.env.template +++ b/.env.template @@ -14,7 +14,7 @@ GRAPH_DATABASE_URL= GRAPH_DATABASE_USERNAME= GRAPH_DATABASE_PASSWORD= -# "qdrant", "pgvector", "weaviate" or "lancedb" +# "qdrant", "pgvector", "weaviate", "milvus" or "lancedb" VECTOR_DB_PROVIDER="lancedb" # Not needed if using "lancedb" or "pgvector" VECTOR_DB_URL= diff --git a/README.md b/README.md index 2b29f1448..efb6e23ba 100644 --- a/README.md +++ b/README.md @@ -53,6 +53,12 @@ pip install 'cognee[qdrant]' pip install 'cognee[neo4j]' ``` +### With pip with Milvus support + +```bash +pip install 'cognee[milvus]' +``` + ### With poetry ```bash @@ -83,6 +89,12 @@ poetry add cognee -E qdrant poetry add cognee -E neo4j ``` +### With poetry with Milvus support + +```bash +poetry add cognee -E milvus +``` + ## 💻 Basic Usage diff --git a/pyproject.toml b/pyproject.toml index 44ca875a6..c66b23c89 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -70,7 +70,7 @@ asyncpg = {version = "0.30.0", optional = true} pgvector = {version = "^0.3.5", optional = true} psycopg2 = {version = "^2.9.10", optional = true} llama-index-core = {version = "^0.11.22", optional = true} -pymilvus = "^2.5.0" +pymilvus = {version = "^2.5.0", optional = true} [tool.poetry.extras] filesystem = ["s3fs", "botocore"] @@ -85,7 +85,7 @@ posthog = ["posthog"] falkordb = ["falkordb"] groq = ["groq"] langfuse = ["langfuse"] - +milvus = ["pymilvus"] [tool.poetry.group.dev.dependencies] pytest = "^7.4.0"