提交with_search

with_search.py

@@ -9,14 +9,8 @@ from flask import Flask, abort, request
 from bs4 import BeautifulSoup
 import markdown
 
-from google import genai
-from google.genai import types # 加入system prompot所需的types模組
 from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
 
-google_search_tool = Tool(
-    google_search = GoogleSearch()
-)
-
 from linebot.v3 import WebhookHandler
 from linebot.v3.exceptions import InvalidSignatureError
 from linebot.v3.messaging import (
@@ -54,8 +48,11 @@ handler = WebhookHandler(channel_secret)
 
 def query(payload: str) -> str:
     """Send a prompt to Gemini and return the response text."""
-    response = chat.send_message(message=payload)
-    return response.text
+    try:
+        response = chat.send_message(message=payload)
+        return response.text
+    except Exception as e:
+        return f"發生錯誤：{e}"
 
 
 @app.route("/", methods=["GET"])