Update main.py

main.py

@@ -31,8 +31,7 @@ from langchain.agents import create_tool_calling_agent, AgentExecutor
 # ==========================
 
 # 設置 Google AI API 金鑰
-os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
-
+google_api = os.environ["GOOGLE_API_KEY"]
 # 設置 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
 line_handler = WebhookHandler(os.environ["CHANNEL_SECRET"])
@@ -102,7 +101,7 @@ def generate_and_upload_image(prompt: str) -> str:
         生成圖片的 URL。
     """
     try:
-        genai_client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
+        genai_client = genai.Client(api_key=google_api)
         response = genai_client.models.generate_content(
             model="gemini-2.0-flash-exp",
             contents=prompt,
@@ -174,7 +173,7 @@ def analyze_image_with_text(image_path: str, user_text: str) -> str:
 tools = [generate_and_upload_image, analyze_image_with_text]
 
 # 建立 LLM 模型實例
-llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro", temperature=0.5)
+llm = ChatGoogleGenerativeAI(google_api_key=google_api, model="gemini-2.0-flash", temperature=0.2)
 
 # 建立提示模板
 prompt_template = ChatPromptTemplate.from_messages([