Zeggai Abdellah commited on
Commit
8355f0c
·
1 Parent(s): 744fb55

gemini env key

Browse files
Files changed (2) hide show
  1. prepare_env.py +1 -1
  2. rag_pipeline.py +1 -1
prepare_env.py CHANGED
@@ -76,7 +76,7 @@ def setup_models():
76
  )
77
 
78
  # Initialize LLM
79
- genai_api_key = os.getenv('GOOGLE_API_KEY', 'AIzaSyBho3W4W9fR7wHUJbX18JKH-12wDSD7pWg')
80
  llm = ChatGoogleGenerativeAI(
81
  model="gemini-2.0-flash",
82
  google_api_key=genai_api_key
 
76
  )
77
 
78
  # Initialize LLM
79
+ genai_api_key = os.getenv('GOOGLE_API_KEY')
80
  llm = ChatGoogleGenerativeAI(
81
  model="gemini-2.0-flash",
82
  google_api_key=genai_api_key
rag_pipeline.py CHANGED
@@ -223,7 +223,7 @@ def initialize_rag_pipeline(tools):
223
  # Initialize LlamaIndex LLM
224
  llama_index_llm = GoogleGenAI(
225
  model="models/gemini-2.0-flash",
226
- api_key=os.getenv('GOOGLE_API_KEY', 'AIzaSyDsbC8H6e08TKDwa5WPE3SiBA39e20K4co'),
227
  )
228
 
229
  # Create agent
 
223
  # Initialize LlamaIndex LLM
224
  llama_index_llm = GoogleGenAI(
225
  model="models/gemini-2.0-flash",
226
+ api_key=os.getenv('GOOGLE_API_KEY'),
227
  )
228
 
229
  # Create agent