Spaces:

JasonFinley0821
/

cookbook

Runtime error

cookbook / models /model_wrapper.py

feat : upgrade LLM model

15f14e8 2 months ago

264 Bytes

import os
from dotenv import load_dotenv
from langchain_google_genai import ChatGoogleGenerativeAI

GENAI_MODEL = "gemini-3.6-flash"

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

	import os
	from dotenv import load_dotenv
	from langchain_google_genai import ChatGoogleGenerativeAI

	GENAI_MODEL = "gemini-3.6-flash"

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