model_parsers.py

from aiconfig_extension_groq import GroqParser
from aiconfig import ModelParserRegistry


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)