Update langchain_logic/agent_setup.py

langchain_logic/agent_setup.py

@@ -6,6 +6,8 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_logic.tools import schedule_appointment, search_for_appointments, delete_appointment_records, update_appointment_record, list_all_appointments
 from datetime import datetime
 
+google_api_key=os.getenv("GOOGLE_API_KEY")
+
 def create_agent_executor():
     """Creates the LangChain agent and executor."""
     # Add the new tool to the list
@@ -16,8 +18,9 @@ def create_agent_executor():
         delete_appointment_records,
         update_appointment_record
     ]
+    
 
-    llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-lite-preview-06-17", temperature=0, google_api_key=os.getenv("GOOGLE_API_KEY"))
+    llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-lite-preview-06-17", temperature=0, google_api_key=google_api_key)
 
     # You can optionally update the prompt to mention the new capability
     prompt = ChatPromptTemplate.from_messages([