chore(api/core): apply ruff reformatting (#7624)

This commit is contained in:
Bowen Liang
2024-09-10 17:00:20 +08:00
committed by GitHub
parent 178730266d
commit 2cf1187b32
724 changed files with 21180 additions and 21123 deletions

View File

@@ -8,60 +8,67 @@ from core.tools.tool.builtin_tool import BuiltinTool
class YoutubeVideosAnalyticsTool(BuiltinTool):
def _invoke(self, user_id: str, tool_parameters: dict[str, Any]) \
-> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
def _invoke(
self, user_id: str, tool_parameters: dict[str, Any]
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
"""
invoke tools
invoke tools
"""
channel = tool_parameters.get('channel', '')
channel = tool_parameters.get("channel", "")
if not channel:
return self.create_text_message('Please input symbol')
return self.create_text_message("Please input symbol")
time_range = [None, None]
start_date = tool_parameters.get('start_date', '')
start_date = tool_parameters.get("start_date", "")
if start_date:
time_range[0] = start_date
else:
time_range[0] = '1800-01-01'
time_range[0] = "1800-01-01"
end_date = tool_parameters.get('end_date', '')
end_date = tool_parameters.get("end_date", "")
if end_date:
time_range[1] = end_date
else:
time_range[1] = datetime.now().strftime('%Y-%m-%d')
time_range[1] = datetime.now().strftime("%Y-%m-%d")
if 'google_api_key' not in self.runtime.credentials or not self.runtime.credentials['google_api_key']:
return self.create_text_message('Please input api key')
if "google_api_key" not in self.runtime.credentials or not self.runtime.credentials["google_api_key"]:
return self.create_text_message("Please input api key")
youtube = build('youtube', 'v3', developerKey=self.runtime.credentials['google_api_key'])
youtube = build("youtube", "v3", developerKey=self.runtime.credentials["google_api_key"])
# try to get channel id
search_results = youtube.search().list(q=channel, type='channel', order='relevance', part='id').execute()
channel_id = search_results['items'][0]['id']['channelId']
search_results = youtube.search().list(q=channel, type="channel", order="relevance", part="id").execute()
channel_id = search_results["items"][0]["id"]["channelId"]
start_date, end_date = time_range
start_date = datetime.strptime(start_date, '%Y-%m-%d').strftime('%Y-%m-%dT%H:%M:%SZ')
end_date = datetime.strptime(end_date, '%Y-%m-%d').strftime('%Y-%m-%dT%H:%M:%SZ')
start_date = datetime.strptime(start_date, "%Y-%m-%d").strftime("%Y-%m-%dT%H:%M:%SZ")
end_date = datetime.strptime(end_date, "%Y-%m-%d").strftime("%Y-%m-%dT%H:%M:%SZ")
# get videos
time_range_videos = youtube.search().list(
part='snippet', channelId=channel_id, order='date', type='video',
publishedAfter=start_date,
publishedBefore=end_date
).execute()
time_range_videos = (
youtube.search()
.list(
part="snippet",
channelId=channel_id,
order="date",
type="video",
publishedAfter=start_date,
publishedBefore=end_date,
)
.execute()
)
def extract_video_data(video_list):
data = []
for video in video_list['items']:
video_id = video['id']['videoId']
video_info = youtube.videos().list(part='snippet,statistics', id=video_id).execute()
title = video_info['items'][0]['snippet']['title']
views = video_info['items'][0]['statistics']['viewCount']
data.append({'Title': title, 'Views': views})
for video in video_list["items"]:
video_id = video["id"]["videoId"]
video_info = youtube.videos().list(part="snippet,statistics", id=video_id).execute()
title = video_info["items"][0]["snippet"]["title"]
views = video_info["items"][0]["statistics"]["viewCount"]
data.append({"Title": title, "Views": views})
return data
summary = extract_video_data(time_range_videos)
return self.create_text_message(str(summary))

View File

@@ -11,7 +11,7 @@ class YahooFinanceProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
user_id="",
tool_parameters={
"channel": "TOKYO GIRLS COLLECTION",
"start_date": "2020-01-01",
@@ -20,4 +20,3 @@ class YahooFinanceProvider(BuiltinToolProviderController):
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))