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