Phoenix21's picture
Create models.py
3bbc9f7 verified
raw
history blame contribute delete
275 Bytes
# models.py
from google import genai
from langchain_groq import ChatGroq
from config import gemini_api_key, groq_api_key
client = genai.Client(api_key=gemini_api_key)
model_name = "gemini-2.0-flash-exp"
chat_model = ChatGroq(model="Gemma2-9b-It", groq_api_key=groq_api_key)