File size: 264 Bytes
a9e6507
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
import os
from dotenv import load_dotenv
from langchain_google_genai import ChatGoogleGenerativeAI

GENAI_MODEL = "gemini-2.0-flash"

def get_llm():
    # LangChain 的 Google Generative API 封裝
    llm = ChatGoogleGenerativeAI(model=GENAI_MODEL)
    return llm