changed model to .env gemini-flash-2.0

gemini_langchain_agent.py

@@ -16,7 +16,7 @@ import datetime as dt
 matplotlib.use('Agg')
 
 load_dotenv()
-model_name = 'gemini-2.0-flash'  # Specify the model name
+model_name = os.getenv("GEMINI_LLM_MODEL")  # Specify the model name
 google_api_keys = os.getenv("GEMINI_API_KEYS").split(",")
 
 # Create pre-initialized LLM instances

gemini_report_generator.py

@@ -25,7 +25,7 @@ load_dotenv()
 
 
 API_KEYS = os.getenv("GEMINI_API_KEYS", "").split(",")[::-1]
-MODEL_NAME = 'gemini-2.0-flash'
+MODEL_NAME = os.getenv("GEMINI_LLM_MODEL") 
 
 class FileProps(BaseModel):
     fileName: str

orchestrator_agent.py

@@ -18,7 +18,7 @@ GEMINI_API_KEYS = os.getenv("GEMINI_API_KEYS", "").split(",")  # Expecting a com
 # Function to initialize the model with a specific API key
 def initialize_model(api_key: str) -> GeminiModel:
     return GeminiModel(
-        'gemini-2.0-flash',
+        os.getenv("GEMINI_LLM_MODEL"),
         provider=GoogleGLAProvider(api_key=api_key)
     )