bsiddhharth commited on
Commit
d4c2a41
·
1 Parent(s): 86e9d14

implemented groq_api_key in try except for extraction.py only

Browse files
Files changed (1) hide show
  1. 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
- groq_api_key = st.secrets["GROQ_API_KEY"]
 
 
 
 
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")