src/model.py

from langchain_google_genai import ChatGoogleGenerativeAI

class GoogleModel:
    def __init__(
        self,
        model: str = "gemini-2.0-flash",
        temperature: int = 0.2,
        tools: list = None,
    ):  
        """Initialize the GoogleModel with the specified model name and temperature.
        """
        
        if tools:
            self.llm = ChatGoogleGenerativeAI(
                model=model,
                temperature=temperature,
                max_tokens=None,
                timeout=None,
                max_retries=1,
            ).bind_tools(tools=tools)
        else:
            self.llm = ChatGoogleGenerativeAI(
                model=model,
                temperature=temperature,
                max_tokens=None,
                timeout=None,
                max_retries=1,
            )