File size: 3,396 Bytes
800d064 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 |
import os
from pytrends.request import TrendReq
from pytrends.request import TrendReq
from googleapiclient.discovery import build
# 定義取得trending searches的function
def get_trending_searches(country_name, hl, tz=None):
"""
:param country_name: Country name (e.g. "taiwan", "south_korea", "czech_republic", "united_states")
:param language: Language code (e.g. "zh-TW", "zh-CN", "en-US", "en-GB", "ja-JP", "ko-KR")
:param timezone: Timezone offset (optional, timezone offset in minutes, e.g. 0 for UTC, -300 for eastern US, 480 for Taiwan)
:return: Trending searches
"""
try:
pytrends = TrendReq(hl=hl, tz=tz)
trending_df = pytrends.trending_searches(pn=country_name)
return trending_df[0].astype(str).tolist()[:10]
except Exception as e:
return f"Error fetching trends: {str(e)}"
# 定義google搜尋並製作提供給LLM的內容的function
def google_res(search_keyword):
GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
GOOGLE_ENGINE_ID = os.getenv('GOOGLE_ENGINE_ID')
# 初始化服務
service = build("customsearch", "v1", developerKey=GOOGLE_API_KEY)
# 執行搜索並獲取標題和摘要
res = service.cse().list(q=search_keyword, cx=GOOGLE_ENGINE_ID).execute()
results = []
for item in res.get('items', []):
title = item.get('title')
snippet = item.get('snippet')
results.append({'title': title, 'snippet': snippet})
# 做成提供給LLM的內容
content = "以下為已發生的事實：\n"
for result in results:
content += f"標題：{result['title']}\n摘要：{result['snippet']}\n\n"
return content
# 定義function call可用的function清單。在此放兩個function，看LLM是否能正確選擇需要的function
FUNCTIONS = [{
"type":"function",
"function": {
"name": "get_trending_searches", # 函式名稱
"description": "Fetch trending searches", # 函式說明
"parameters": {
"type": "object",
"properties": {
"country_name": { # 參數名稱
"type": "string", # 資料型別
"description": 'Country name(e.g. "taiwan", "south_korea", "czech_republic", "united_states")', # 參數說明
},
"hl": { # 參數名稱
"type": "string", # 資料型別
"description": 'Language code(e.g. "zh-TW", "zh-CN", "en-US", "en-GB", "ja-JP")', # 參數說明
},
"tz": { # 參數名稱
"type": "string", # 資料型別
"description": 'Timezone offset(e.g. 0 for UTC, -300 for eastern US, 480 for Taiwan)', # 參數說明
},
},
"required": ["country_name", "hl", "tz"], # 必要參數
},
}
},
{
"type":"function",
"function": {
"name": "google_res", # 函式名稱
"description": "Google search", # 函式說明
"parameters": {
"type": "object",
"properties": {
"search_keyword": { # 參數名稱
"type": "string", # 資料型別
"description": 'keyowrd to search', # 參數說明
}
},
"required": ["search_keyword"], # 必要參數
},
}
}]
|