File size: 507 Bytes
73cddce 3fa0e2d 73cddce 3fa0e2d 73cddce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from langchain_google_genai import ChatGoogleGenerativeAI
from .config import GOOGLE_API_KEY
import warnings
warnings.filterwarnings("ignore", category=UserWarning, module="langchain_google_genai")
def get_llm():
try:
llm = ChatGoogleGenerativeAI(
model="gemini-2.0-flash",
google_api_key=GOOGLE_API_KEY,
temperature=0.1,
convert_system_message_to_human=True
)
return llm
except Exception as e:
raise
llm = get_llm() |