0917

app.py

@@ -20,7 +20,7 @@ from linebot.v3.webhook import WebhookParser
 from linebot.v3.exceptions import InvalidSignatureError
 # --------------------------------------------------------------
 from openai import OpenAI
-from ddgs import DDGS
+from tavily import TavilyClient  # 新增 Tavily 客戶端
 
 # ==== CONFIG (從環境變數載入，或使用預設值) ====
 def _require_env(var: str) -> str:
@@ -33,6 +33,9 @@ def _require_env(var: str) -> str:
 CHANNEL_SECRET = _require_env("CHANNEL_SECRET")
 CHANNEL_ACCESS_TOKEN = _require_env("CHANNEL_ACCESS_TOKEN")
 
+# 新增 Tavily API Key (免費版使用)
+TAVILY_API_KEY = "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"
+
 # LLM API 設定
 LLM_API_CONFIG = {
     "base_url": os.getenv("LLM_BASE_URL", "https://litellm-ekkks8gsocw.dgx-coolify.apmic.ai/"),
@@ -76,22 +79,23 @@ def split_text_for_line(text: str, max_length: int = 4800) -> List[str]:
         text = text[split_pos:].lstrip()
     return chunks
 
-# ---------- 網路搜尋函數（新增 log/print 檢索過程） ----------
+# ---------- 網路搜尋函數（改用 Tavily API，支援免費版） ----------
 def perform_web_search(query: str, max_results: int = 5) -> str:
-    """使用 DuckDuckGo 進行網路搜尋，並返回摘要結果。同時 log/print 檢索過程。"""
+    """使用 Tavily 進行網路搜尋，並返回摘要結果。同時 log/print 檢索過程。"""
     print(f"開始網路搜尋：查詢詞 = '{query}'，最大結果數 = {max_results}")
     try:
-        with DDGS() as ddgs:
-            results = [r for r in ddgs.text(query, max_results=max_results)]
-        if not results:
+        client = TavilyClient(api_key=TAVILY_API_KEY)
+        response = client.search(query, max_results=max_results, include_answer=True)
+        if not response['results']:
             print("搜尋完成：沒有找到相關結果。")
             return "沒有找到相關的網路搜尋結果。"
         
         search_summary = "以下是相關的網路搜尋結果摘要：\n"
-        for i, result in enumerate(results, 1):
-            print(f"結果 {i}: 標題 = '{result['title']}'，摘要 = '{result['body'][:200]}...'，來源 = '{result['href']}'")
-            search_summary += f"{i}. {result['title']}: {result['body'][:200]}... (來源: {result['href']})\n"
-        print("搜尋完成：總結果數 = {len(results)}")
+        search_summary += f"AI總結：{response.get('answer', '無總結可用')}\n\n"
+        for i, result in enumerate(response['results'], 1):
+            print(f"結果 {i}: 標題 = '{result['title']}'，內容 = '{result['content'][:200]}...'，來源 = '{result['url']}'")
+            search_summary += f"{i}. {result['title']}: {result['content'][:200]}... (來源: {result['url']})\n"
+        print(f"搜尋完成：總結果數 = {len(response['results'])}")
         return search_summary
     except Exception as e:
         print(f"網路搜尋錯誤：{e}")

requirements.txt

@@ -2,4 +2,4 @@ fastapi
 uvicorn
 line-bot-sdk
 openai
-ddgs
+tavily-python