groq_prompts_notebook / model_parsers.py
tanyarai's picture
Update model_parsers.py
7c25085 verified
raw
history blame
400 Bytes
from aiconfig_extension_groq import GroqParser
from aiconfig import ModelParserRegistry
import dotenv
def register_model_parsers() -> None:
groq_mixtral = GroqParser(model="mixtral-8x7b-32768")
ModelParserRegistry.register_model_parser(groq_mixtral)
groq_llama = GroqParser(model="llama2-70b-4096")
ModelParserRegistry.register_model_parser(groq_llama)
dotenv.load_dotenv()