Skip to content

feat: Support token, uri and db_name in the Milvus online store - #6895

Merged
ntkathole merged 1 commit into
feast-dev:masterfrom
simonhearne:feat/milvus-token-uri-db
Oct 1, 2026
Merged

ntkathole merged 1 commit into
feast-dev:masterfrom
simonhearne:feat/milvus-token-uri-db

Conversation

@simonhearne

Copy link
Copy Markdown
Contributor

What this PR does / why we need it:

The Milvus client was always built as MilvusClient(uri=f"{host}:{port}", token=f"{username}:{password}").
API keys couldn't be used, and there was no way to choose a database.

This PR adds three new optional config parameters:

Field Behaviour
uri Full endpoint. Takes precedence over host/port, and over a Milvus Lite path.
token API key or username:password. Takes precedence over username/password.
db_name Database to use; must already exist.

Existing configs build exactly the same client arguments (covered by a test). The docs add a Zilliz
Cloud example that reads the token from an environment variable.

Which issue(s) this PR fixes:

N/A

Checks

  • I've made sure the tests are passing.
  • My commits are signed off (git commit -s)
  • My PR title follows conventional commits format

Testing Strategy

  • Unit tests

  • Integration tests

  • Manual tests

  • Testing is not required for this change

  • Unit: parametrized client arguments for existing configs, token/uri precedence and db_name;
    uri beats path; path still selects Milvus Lite.

  • Server: collections are created in db_name and not in default. Token authentication was
    exercised against Zilliz Cloud with an API key.

Unit tests run on Milvus Lite 3.2.1 (pymilvus 3.0.2). Server tests in sdk/python/tests/integration/online_store/test_milvus_remote.py are marked integration and skip unless ZILLIZ_URI and ZILLIZ_TOKEN are set; they passed against a local Milvus 2.6.0 server and against Zilliz Cloud. The existing Milvus unit and universal integration tests pass unchanged.

@simonhearne
simonhearne requested a review from a team as a code owner September 29, 2026 16:53
@codecov-commenter

codecov-commenter commented Sep 29, 2026 •

Copy link
Copy Markdown

⚠️ Please install the 'codecov app svg image' to ensure uploads and comments are reliably processed by Codecov.

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 48.07%. Comparing base (b8989cb) to head (6478647).
⚠️ Report is 2 commits behind head on master.
❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##           master    #6895      +/-   ##
==========================================
+ Coverage   48.06%   48.07%   +0.01%     
==========================================
  Files         427      427              
  Lines       53591    53605      +14     
  Branches     7799     7802       +3     
==========================================
+ Hits        25758    25772      +14     
  Misses      25980    25980              
  Partials     1853     1853              
Flag Coverage Δ
go-feature-server 30.58% <ø> (ø)
python-unit 49.41% <100.00%> (+0.01%) ⬆️
Files with missing lines Coverage Δ
.../infra/online_stores/milvus_online_store/milvus.py 71.24% <100.00%> (+0.90%) ⬆️

Continue to review full report in Codecov by Harness.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update b8989cb...6478647. Read the comment docs.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
  • 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

The Milvus client was always built from host:port with a
username:password token, so Zilliz Cloud API keys could not be used
and every collection landed in the default database.

Adds three optional config fields:
- token: an API key or username:password; takes precedence over
  username/password.
- uri: a full endpoint; takes precedence over host/port (and over a
  Milvus Lite path).
- db_name: the Milvus database to use.

Existing configurations build exactly the same client arguments.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Signed-off-by: Simon Hearne <simon.hearne@gmail.com>
@ntkathole
ntkathole force-pushed the feat/milvus-token-uri-db branch from 8c75f09 to 6478647 Compare October 1, 2026 04:12
@ntkathole
ntkathole merged commit c5efecd into feast-dev:master Oct 1, 2026
20 of 23 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants