entropy25 commited on
Commit
7ee1f4e
·
verified ·
1 Parent(s): d1ad851

Update ai_engine.py

Browse files
Files changed (1) hide show
  1. ai_engine.py +4 -2
ai_engine.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
  import pandas as pd
3
  from typing import Optional, Dict
4
- from groq import Groq
5
 
6
  from config import get_settings
7
 
@@ -14,7 +14,9 @@ def init_ai_model():
14
  return None
15
 
16
  try:
17
- return Groq(api_key=api_key)
 
 
18
  except Exception as e:
19
  st.error(f"AI configuration failed: {str(e)}")
20
  return None
 
1
  import streamlit as st
2
  import pandas as pd
3
  from typing import Optional, Dict
4
+ import os
5
 
6
  from config import get_settings
7
 
 
14
  return None
15
 
16
  try:
17
+ from groq import Groq
18
+ os.environ["GROQ_API_KEY"] = api_key
19
+ return Groq()
20
  except Exception as e:
21
  st.error(f"AI configuration failed: {str(e)}")
22
  return None