INNER CODE UNIT · Python

__init__

mfmezger/conversational-agent-langchain · src/agent/backend/graph.py:28

    def __init__(self) -> None:
        """Initialize the Graph."""
        self.cfg = settings

        # define models
        self.llm = ChatLiteLLM(model_name=self.cfg.model_name, streaming=True)

        # Initialize Cohere model for grounded generation (if API key is available)
        self.cohere_llm: ChatCohere | None = None
        if self.cfg.cohere_api_key:
            self.cohere_llm = ChatCohere(
                cohere_api_key=self.cfg.cohere_api_key,
                model=self.cfg.cohere_model_name,
                streaming=True,
            )

        # Initialize reranker
        self.reranker = get_reranker(

View source record →