models.py

import os
from dotenv import load_dotenv
from langchain_google_genai import ChatGoogleGenerativeAI
from langfuse.langchain import CallbackHandler


load_dotenv()

api_key = os.getenv("GEMINI_API_KEY")

# Initialize Langfuse CallbackHandler for LangGraph/Langchain (tracing)
langfuse_handler = CallbackHandler()

# Create LLM class
vlm = ChatGoogleGenerativeAI(
    model= "gemini-2.5-flash",
    temperature=0,
    max_retries=2,
    google_api_key=api_key
)

llm = ChatGoogleGenerativeAI(
    model= "gemini-2.5-flash", #pro,flash-lite-preview-06-17
    temperature=0,
    max_retries=2,
    google_api_key=api_key#,
    #thinking_budget=0
)