diff --git a/modules/models/Groq.py b/modules/models/Groq.py index 61648233..6e95ba43 100644 --- a/modules/models/Groq.py +++ b/modules/models/Groq.py @@ -18,7 +18,13 @@ class Groq_Client(BaseLLMModel): def __init__(self, model_name, api_key, user_name="") -> None: - super().__init__(model_name=model_name, user=user_name, api_key=api_key) + super().__init__( + model_name=model_name, + user=user_name, + config={ + "api_key": api_key + } + ) self.client = Groq( api_key=os.environ.get("GROQ_API_KEY"), base_url=self.api_host,