Skip to content

Add Milvus vector store driver#2235

Open
zc277584121 wants to merge 1 commit into
griptape-ai:mainfrom
zc277584121:add-milvus-vector-store
Open

Add Milvus vector store driver#2235
zc277584121 wants to merge 1 commit into
griptape-ai:mainfrom
zc277584121:add-milvus-vector-store

Conversation

@zc277584121

@zc277584121 zc277584121 commented Jul 9, 2026

Copy link
Copy Markdown

Summary

  • Add MilvusVectorStoreDriver with Milvus Lite default storage at ./milvus.db plus configurable uri, token, db_name, and consistency_level.
  • Create collections with explicit id, vector, text, namespace, and metadata fields, AUTOINDEX vector indexing, and validation for existing collection schema and vector dimensions.
  • Support upsert, load, query, delete, namespace filtering, and scalar metadata filters with escaped filter expressions.
  • Add the drivers-vector-milvus extra, vector store docs, and a Milvus Lite example.

Tests

  • uv run --python 3.12 --no-default-groups --with ruff ruff check griptape/drivers/vector/milvus_vector_store_driver.py griptape/drivers/vector/milvus/__init__.py tests/unit/drivers/vector/test_milvus_vector_store_driver.py tests/integration/drivers/vector/test_milvus_vector_store_driver.py tests/integration/test_code_blocks.py docs/griptape-framework/drivers/src/vector_store_drivers_13.py
  • uv run --python 3.12 --no-default-groups --extra drivers-vector-milvus --with pytest --with pytest-mock pytest tests/unit/drivers/vector/test_milvus_vector_store_driver.py tests/integration/drivers/vector/test_milvus_vector_store_driver.py
  • uv run --python 3.12 --no-default-groups --extra drivers-vector-milvus --with pyright pyright griptape/drivers/vector/milvus_vector_store_driver.py docs/griptape-framework/drivers/src/vector_store_drivers_13.py
  • uv run --python 3.12 --no-default-groups --with pytest pytest tests/integration/test_code_blocks.py -k vector_store_drivers_13
  • uv run --no-project --python 3.12 --with mdformat --with mdformat-gfm --with mdformat-frontmatter --with mdformat-footnote --with mdformat-mkdocs mdformat --check docs/griptape-framework/drivers/vector-store-drivers.md
  • uv run --no-project --python 3.12 --with typos typos griptape/drivers/vector/milvus_vector_store_driver.py tests/unit/drivers/vector/test_milvus_vector_store_driver.py tests/integration/drivers/vector/test_milvus_vector_store_driver.py docs/griptape-framework/drivers/vector-store-drivers.md docs/griptape-framework/drivers/src/vector_store_drivers_13.py docs/griptape-framework/drivers/logs/vector_store_drivers_13.txt pyproject.toml
  • uv run --isolated --python 3.12 --no-default-groups --with-editable . python -c "from griptape.drivers.vector.milvus import MilvusVectorStoreDriver; print(MilvusVectorStoreDriver.__name__)"

📚 Documentation preview 📚: https://griptape--2235.org.readthedocs.build//2235/

@codecov

codecov Bot commented Jul 9, 2026

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 95.38462% with 9 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
...ptape/drivers/vector/milvus_vector_store_driver.py 95.33% 4 Missing and 5 partials ⚠️

📢 Thoughts on this report? Let us know!

@zc277584121
zc277584121 force-pushed the add-milvus-vector-store branch from 1f428b8 to 4ac8076 Compare July 9, 2026 01:18
@zc277584121

Copy link
Copy Markdown
Author

Hi @collindutter @shhlife, friendly ping for review when you have a chance.

This adds a Milvus vector store driver using the existing driver pattern, with optional dependencies, docs, unit coverage, and Milvus Lite integration coverage. The current checks are green; happy to adjust naming or configuration details to fit the project conventions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant