File size: 362 Bytes
5a93166
 
cda0f85
 
 
5a93166
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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)