Skip to content

Conversation

@jperez999
Copy link
Collaborator

@jperez999 jperez999 commented Apr 22, 2025

Description

This PR changes how streaming is auto activated. It now checks the count of how many elements (chunks) are processed. The threshold has been increased to 50 given this change. And we also plumbed the stream parameter so you can activate streaming insert to milvus regardless of the number of elements.

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.
  • If adjusting docker-compose.yaml environment variables have you ensured those are mimicked in the Helm values.yaml file.

@jperez999 jperez999 added the bug Something isn't working label Apr 22, 2025
@jperez999 jperez999 self-assigned this Apr 22, 2025
@jperez999 jperez999 requested a review from a team as a code owner April 22, 2025 13:46
@jperez999 jperez999 requested review from edknv and removed request for a team April 22, 2025 13:46
@copy-pr-bot
Copy link

copy-pr-bot bot commented Apr 22, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@jperez999 jperez999 requested review from a team, ChrisJar, drobison00 and jdye64 April 22, 2025 14:07
secret_key: str = "minioadmin",
bucket_name: str = "a-bucket",
threshold: int = 10,
threshold: int = 50,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
threshold: int = 50,
threshold: int = 1000,

when testing w/ bo20, there are 812 chunks. with stream=True, total runtime goes from 114 to 76s

Seems like the threshold should be higher, which gives a nice perf lift for smaller ingest jobs

Copy link
Collaborator

@edknv edknv left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Changes look good to me. I think we can ignore the failing integration test; I don't see how this PR fails the integration test, and it's probably due to some unrelated issues with the build endpoint. However, there is one unit test that is failing and needs updating before merging.

@jperez999 jperez999 requested a review from randerzander April 22, 2025 17:47
@jperez999 jperez999 merged commit b591775 into NVIDIA:main Apr 22, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants