fix: remove reference to a non-existent method in NvidiaRerankComponent (#5957)

Removing reference to nonexistent method

Co-authored-by: brian.ogrady@datastax.com <brian.ogrady@datastax.com>
This commit is contained in:
Gabriel Luiz Freitas Almeida 2025-01-27 12:53:05 -03:00 committed by GitHub
commit ea2685fb83
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -53,7 +53,7 @@ class NvidiaRerankComponent(LCVectorStoreComponent):
def update_build_config(self, build_config: dotdict, field_value: Any, field_name: str | None = None):
if field_name == "base_url" and field_value:
try:
build_model = self.build_model()
build_model = self.build_reranker()
ids = [model.id for model in build_model.available_models]
build_config["model"]["options"] = ids
build_config["model"]["value"] = ids[0]