Update langchain_logic/agent_setup.py

langchain_logic/agent_setup.py

@@ -1,3 +1,4 @@
+import os
 from langchain.agents import AgentExecutor, create_tool_calling_agent
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_google_genai import ChatGoogleGenerativeAI
@@ -16,7 +17,7 @@ def create_agent_executor():
         update_appointment_record
     ]
 
-    llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-lite-preview-06-17", temperature=0)
+    llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-lite-preview-06-17", temperature=0, google_api_key=os.getenv("GOOGLE_API_KEY"))
 
     # You can optionally update the prompt to mention the new capability
     prompt = ChatPromptTemplate.from_messages([