File size: 493 Bytes
22ca910 61e31d6 22ca910 2fb3630 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from langchain_google_genai import ChatGoogleGenerativeAI
import os
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
def get_gemma(temperature=0, top_k=40, top_p=0.95):
gemma_model = ChatGoogleGenerativeAI(
model="gemma-3-12b-it",
temperature=temperature,
max_tokens=2048,
timeout=None,
max_retries=2,
google_api_key=GOOGLE_API_KEY,
top_k=top_k,
top_p=top_p
)
return gemma_model |