Skip to content

Add RerankRequestChunker #130485

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from

Conversation

dan-rubinstein
Copy link
Member

@dan-rubinstein dan-rubinstein commented Jul 2, 2025

Issue - #121567

This is a POC for testing the proposed process. The issue above has some decisions that still need to be made in order to properly complete the implementation.

Questions remaining:

  1. Is there a limit to how many chunks we can send to the model?

@dan-rubinstein dan-rubinstein added :ml Machine learning Team:ML Meta label for the ML team v9.2.0 labels Jul 2, 2025
@dan-rubinstein
Copy link
Member Author

@elasticmachine merge upstream

@@ -74,6 +74,7 @@ protected void doInference(
InferenceService service,
ActionListener<InferenceServiceResults> listener
) {

Copy link
Member Author

Choose a reason for hiding this comment

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

This can be removed.

@@ -156,6 +156,7 @@ protected RankFeatureDoc[] preprocess(RankFeatureDoc[] originalDocs, boolean rer
}

protected InferenceAction.Request generateRequest(List<String> docFeatures) {
// TODO: Try running the RerankRequestChunker here.
Copy link
Member Author

Choose a reason for hiding this comment

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

This can be removed as we're calling this in the service now.

@davidkyle davidkyle added the cloud-deploy Publish cloud docker image for Cloud-First-Testing label Jul 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cloud-deploy Publish cloud docker image for Cloud-First-Testing :ml Machine learning Team:ML Meta label for the ML team v9.2.0
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants