added exception import in groq

aiplanethub · Jul 1, 2024 · 6e760d7 · 6e760d7
1 parent fc9b3ac
commit 6e760d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/openagi/llms/groq.py b/src/openagi/llms/groq.py
@@ -1,5 +1,6 @@
 from typing import Any
 from langchain_core.messages import HumanMessage
+from openagi.exception import OpenAGIException
 from openagi.llms.base import LLMBaseModel, LLMConfigModel
 from openagi.utils.yamlParse import read_from_env
 
@@ -60,4 +61,4 @@ def load_from_env_config() -> GroqConfigModel:
             groq_api_key=read_from_env("GROQ_API_KEY", raise_exception=True),
             model_name = read_from_env("GROQ_MODEL",raise_exception=True),
             temperature=read_from_env("GROQ_TEMP",raise_exception=True)
-        )
+        )