| 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 |