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")
|