將設定移動到建立chat的地方

with_search.py

@@ -32,7 +32,12 @@ from linebot.v3.webhooks import MessageEvent, TextMessageContent
 # Initialize Google Gemini
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 client = genai.Client(api_key=GOOGLE_API_KEY)
-chat = client.chats.create(model="gemini-2.0-flash")
+chat = client.chats.create(model="gemini-2.0-flash",
+    config=GenerateContentConfig(
+        system_instruction="你是一個中文的AI助手，請用繁體中文回答",
+        tools=[google_search_tool],
+        response_modalities=["TEXT"], 
+    ))
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -49,13 +54,7 @@ handler = WebhookHandler(channel_secret)
 
 def query(payload: str) -> str:
     """Send a prompt to Gemini and return the response text."""
-    response = chat.send_message(
-        config=GenerateContentConfig(
-            system_instruction="你是一個中文的AI助手，請用繁體中文回答"),
-            tools=[google_search_tool],
-            response_modalities=["TEXT"],
-        message=payload
-    )
+    response = chat.send_message(message=payload)
     return response.text