Create models.py
Browse files
models.py
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# models.py
|
2 |
+
from google import genai
|
3 |
+
from langchain_groq import ChatGroq
|
4 |
+
from config import gemini_api_key, groq_api_key
|
5 |
+
|
6 |
+
client = genai.Client(api_key=gemini_api_key)
|
7 |
+
model_name = "gemini-2.0-flash-exp"
|
8 |
+
chat_model = ChatGroq(model="Gemma2-9b-It", groq_api_key=groq_api_key)
|