NVIDIA reranker undefined build_model method

[brian-ogrady]

Bug Description

The NVIDIA reranker into a flow and it will produce an error related to the undefined build_model method.

Reproduction

Simply drop the NVIDIA reranker into a flow and it will produce an error related to the undefined build_model method.

Expected behavior

NVIDIA reranker will cause an error in the flow.

Who can help?

No response

Operating System

Mac OS 15.1.1

Langflow Version

1.1

Python Version

3.10

Screenshot

No response

Flow File

No response

[dosubot]

Hey @brian-ogrady! I'm here to help you with bugs, questions, and becoming a contributor. Let's tackle this NVIDIA reranker issue together while we wait for a human maintainer.

To resolve the error related to the undefined build_model method when using the NVIDIA reranker, ensure that the build_model method is correctly defined in your NVIDIAModelComponent class. The method should be defined as follows:

def build_model(self) -> LanguageModel:  # type: ignore[type-var]
    try:
        from langchain_nvidia_ai_endpoints import ChatNVIDIA
    except ImportError as e:
        msg = "Please install langchain-nvidia-ai-endpoints to use the NVIDIA model."
        raise ImportError(msg) from e
    api_key = self.api_key
    temperature = self.temperature
    model_name: str = self.model_name
    max_tokens = self.max_tokens
    seed = self.seed
    return ChatNVIDIA(
        max_tokens=max_tokens or None,
        model=model_name,
        base_url=self.base_url,
        api_key=api_key,
        temperature=temperature or 0.1,
        seed=seed,
    )

Ensure that the langchain-nvidia-ai-endpoints package is installed, as the method imports ChatNVIDIA from this package. You can install it using:

pip install langchain-nvidia-ai-endpoints

If the method is still reported as undefined, verify that there are no typos or issues in the class definition or method calls ^[1].

To continue talking to Dosu, mention @dosu.

---

^{Help Dosu learn! Give it feedback: Great Response | Irrelevant Answer | Incorrect Sources | Too Verbose | Hallucination | Other}

[brian-ogrady]

Related PR