File size: 733 Bytes
f45999f 38ccc4e f45999f 38ccc4e f45999f 38ccc4e f45999f 38ccc4e f45999f 8fe9ff0 38ccc4e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from langchain.chat_models import init_chat_model
from langchain_groq import ChatGroq
from langchain_google_genai import ChatGoogleGenerativeAI
from core.config import get_settings
settings = get_settings()
main_model = (
ChatGoogleGenerativeAI(model=settings.DEFAULT_MODEL, google_api_key=settings.GOOGLE_API_KEY)
.with_fallbacks([
ChatGoogleGenerativeAI(model=settings.FALLBACK_MODEL, google_api_key=settings.GOOGLE_API_KEY),
])
)
output_formatter_model = ChatGroq(
model=settings.OUTPUT_FORMATTER_MODEL,
api_key=settings.GROQ_API_KEY,
temperature=0
)
title_generator_model = ChatGoogleGenerativeAI(model=settings.TITLE_GENERATOR_MODEL, google_api_key=settings.GOOGLE_API_KEY, temperature=0.2) |