nodes/web_search.py

from langchain_perplexity import ChatPerplexity

from talentum_score.configs import talentum_score_config
from talentum_score.nodes.base import BaseLLMNode


class WebSearchNode(BaseLLMNode):
    name: str = "WebSearchNode"
    description: str = "Generic web search node"

    def get_llm(self, **kwargs) -> ChatPerplexity:
        return ChatPerplexity(
            model=self.model,
            pplx_api_key=talentum_score_config.perplexity_api_key,
            **kwargs,
        )