Spaces:
Sleeping
Sleeping
bsiddhharth
commited on
Commit
·
d4c2a41
1
Parent(s):
86e9d14
implemented groq_api_key in try except for extraction.py only
Browse files- extraction.py +5 -1
extraction.py
CHANGED
@@ -55,7 +55,11 @@ def initialize_llm() -> ChatGroq:
|
|
55 |
|
56 |
# os.environ['GROQ_API_KEY'] = os.getenv("GROQ_API_KEY")
|
57 |
# groq_api_key = os.getenv("GROQ_API_KEY")
|
58 |
-
|
|
|
|
|
|
|
|
|
59 |
|
60 |
if not groq_api_key:
|
61 |
logger.error("GROQ_API_KEY is not set")
|
|
|
55 |
|
56 |
# os.environ['GROQ_API_KEY'] = os.getenv("GROQ_API_KEY")
|
57 |
# groq_api_key = os.getenv("GROQ_API_KEY")
|
58 |
+
try:
|
59 |
+
groq_api_key = st.secrets["GROQ_API_KEY"]
|
60 |
+
|
61 |
+
except:
|
62 |
+
groq_api_key = os.getenv("GROQ_API_KEY")
|
63 |
|
64 |
if not groq_api_key:
|
65 |
logger.error("GROQ_API_KEY is not set")
|