hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-22 01:06:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

17
api/core/tools/provider/builtin/trello/tools/create_board.py
View File

@@ -22,19 +22,15 @@ class CreateBoardTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_name = tool_parameters.get('name')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_name = tool_parameters.get("name")
if not (api_key and token and board_name):
return self.create_text_message("Missing required parameters: API key, token, or board name.")
url = "https://api.trello.com/1/boards/"
query_params = {
'name': board_name,
'key': api_key,
'token': token
}
query_params = {"name": board_name, "key": api_key, "token": token}
try:
response = requests.post(url, params=query_params)
@@ -43,5 +39,6 @@ class CreateBoardTool(BuiltinTool):
return self.create_text_message("Failed to create board")
board = response.json()
return self.create_text_message(text=f"Board created successfully! Board name: {board['name']}, ID: {board['id']}")
return self.create_text_message(
text=f"Board created successfully! Board name: {board['name']}, ID: {board['id']}"
)

19
api/core/tools/provider/builtin/trello/tools/create_list_on_board.py
View File

@@ -22,20 +22,16 @@ class CreateListOnBoardTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.get('id')
list_name = tool_parameters.get('name')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.get("id")
list_name = tool_parameters.get("name")
if not (api_key and token and board_id and list_name):
return self.create_text_message("Missing required parameters: API key, token, board ID, or list name.")
url = f"https://api.trello.com/1/boards/{board_id}/lists"
params = {
'name': list_name,
'key': api_key,
'token': token
}
params = {"name": list_name, "key": api_key, "token": token}
try:
response = requests.post(url, params=params)
@@ -44,5 +40,6 @@ class CreateListOnBoardTool(BuiltinTool):
return self.create_text_message("Failed to create list")
new_list = response.json()
return self.create_text_message(text=f"List '{new_list['name']}' created successfully with Id {new_list['id']} on board {board_id}.")
return self.create_text_message(
text=f"List '{new_list['name']}' created successfully with Id {new_list['id']} on board {board_id}."
)

13
api/core/tools/provider/builtin/trello/tools/create_new_card_on_board.py
View File

@@ -22,15 +22,15 @@ class CreateNewCardOnBoardTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
# Ensure required parameters are present
if 'name' not in tool_parameters or 'idList' not in tool_parameters:
if "name" not in tool_parameters or "idList" not in tool_parameters:
return self.create_text_message("Missing required parameters: name or idList.")
url = "https://api.trello.com/1/cards"
params = {**tool_parameters, 'key': api_key, 'token': token}
params = {**tool_parameters, "key": api_key, "token": token}
try:
response = requests.post(url, params=params)
@@ -39,5 +39,6 @@ class CreateNewCardOnBoardTool(BuiltinTool):
except requests.exceptions.RequestException as e:
return self.create_text_message("Failed to create card")
return self.create_text_message(text=f"New card '{new_card['name']}' created successfully with ID {new_card['id']}.")
return self.create_text_message(
text=f"New card '{new_card['name']}' created successfully with ID {new_card['id']}."
)

7
api/core/tools/provider/builtin/trello/tools/delete_board.py
View File

@@ -22,9 +22,9 @@ class DeleteBoardTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.get('boardId')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.get("boardId")
if not (api_key and token and board_id):
return self.create_text_message("Missing required parameters: API key, token, or board ID.")
@@ -38,4 +38,3 @@ class DeleteBoardTool(BuiltinTool):
return self.create_text_message("Failed to delete board")
return self.create_text_message(text=f"Board with ID {board_id} deleted successfully.")

7
api/core/tools/provider/builtin/trello/tools/delete_card.py
View File

@@ -22,9 +22,9 @@ class DeleteCardByIdTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
card_id = tool_parameters.get('id')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
card_id = tool_parameters.get("id")
if not (api_key and token and card_id):
return self.create_text_message("Missing required parameters: API key, token, or card ID.")
@@ -38,4 +38,3 @@ class DeleteCardByIdTool(BuiltinTool):
return self.create_text_message("Failed to delete card")
return self.create_text_message(text=f"Card with ID {card_id} has been successfully deleted.")

8
api/core/tools/provider/builtin/trello/tools/fetch_all_boards.py
View File

@@ -28,9 +28,7 @@ class FetchAllBoardsTool(BuiltinTool):
token = self.runtime.credentials.get("trello_api_token")
if not (api_key and token):
return self.create_text_message(
"Missing Trello API key or token in credentials."
)
return self.create_text_message("Missing Trello API key or token in credentials.")
# Including board filter in the request if provided
board_filter = tool_parameters.get("boards", "open")
@@ -48,7 +46,5 @@ class FetchAllBoardsTool(BuiltinTool):
return self.create_text_message("No boards found in Trello.")
# Creating a string with both board names and IDs
boards_info = ", ".join(
[f"{board['name']} (ID: {board['id']})" for board in boards]
)
boards_info = ", ".join([f"{board['name']} (ID: {board['id']})" for board in boards])
return self.create_text_message(text=f"Boards: {boards_info}")

11
api/core/tools/provider/builtin/trello/tools/get_board_actions.py
View File

@@ -22,9 +22,9 @@ class GetBoardActionsTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.get('boardId')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.get("boardId")
if not (api_key and token and board_id):
return self.create_text_message("Missing required parameters: API key, token, or board ID.")
@@ -38,6 +38,7 @@ class GetBoardActionsTool(BuiltinTool):
except requests.exceptions.RequestException as e:
return self.create_text_message("Failed to retrieve board actions")
actions_summary = "\n".join([f"{action['type']}: {action.get('data', {}).get('text', 'No details available')}" for action in actions])
actions_summary = "\n".join(
[f"{action['type']}: {action.get('data', {}).get('text', 'No details available')}" for action in actions]
)
return self.create_text_message(text=f"Actions for Board ID {board_id}:\n{actions_summary}")

7
api/core/tools/provider/builtin/trello/tools/get_board_by_id.py
View File

@@ -22,9 +22,9 @@ class GetBoardByIdTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.get('boardId')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.get("boardId")
if not (api_key and token and board_id):
return self.create_text_message("Missing required parameters: API key, token, or board ID.")
@@ -63,4 +63,3 @@ class GetBoardByIdTool(BuiltinTool):
f"Background Color: {board['prefs']['backgroundColor']}"
)
return details

7
api/core/tools/provider/builtin/trello/tools/get_board_cards.py
View File

@@ -22,9 +22,9 @@ class GetBoardCardsTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.get('boardId')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.get("boardId")
if not (api_key and token and board_id):
return self.create_text_message("Missing required parameters: API key, token, or board ID.")
@@ -40,4 +40,3 @@ class GetBoardCardsTool(BuiltinTool):
cards_summary = "\n".join([f"{card['name']} (ID: {card['id']})" for card in cards])
return self.create_text_message(text=f"Cards for Board ID {board_id}:\n{cards_summary}")

13
api/core/tools/provider/builtin/trello/tools/get_filterd_board_cards.py
View File

@@ -22,10 +22,10 @@ class GetFilteredBoardCardsTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.get('boardId')
filter = tool_parameters.get('filter')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.get("boardId")
filter = tool_parameters.get("filter")
if not (api_key and token and board_id and filter):
return self.create_text_message("Missing required parameters: API key, token, board ID, or filter.")
@@ -40,5 +40,6 @@ class GetFilteredBoardCardsTool(BuiltinTool):
return self.create_text_message("Failed to retrieve filtered cards")
card_details = "\n".join([f"{card['name']} (ID: {card['id']})" for card in filtered_cards])
return self.create_text_message(text=f"Filtered Cards for Board ID {board_id} with Filter '{filter}':\n{card_details}")
return self.create_text_message(
text=f"Filtered Cards for Board ID {board_id} with Filter '{filter}':\n{card_details}"
)

7
api/core/tools/provider/builtin/trello/tools/get_lists_on_board.py
View File

@@ -22,9 +22,9 @@ class GetListsFromBoardTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.get('boardId')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.get("boardId")
if not (api_key and token and board_id):
return self.create_text_message("Missing required parameters: API key, token, or board ID.")
@@ -40,4 +40,3 @@ class GetListsFromBoardTool(BuiltinTool):
lists_info = "\n".join([f"{list['name']} (ID: {list['id']})" for list in lists])
return self.create_text_message(text=f"Lists on Board ID {board_id}:\n{lists_info}")

11
api/core/tools/provider/builtin/trello/tools/update_board.py
View File

@@ -22,9 +22,9 @@ class UpdateBoardByIdTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
board_id = tool_parameters.pop('boardId', None)
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
board_id = tool_parameters.pop("boardId", None)
if not (api_key and token and board_id):
return self.create_text_message("Missing required parameters: API key, token, or board ID.")
@@ -33,8 +33,8 @@ class UpdateBoardByIdTool(BuiltinTool):
# Removing parameters not intended for update action or with None value
params = {k: v for k, v in tool_parameters.items() if v is not None}
params['key'] = api_key
params['token'] = token
params["key"] = api_key
params["token"] = token
try:
response = requests.put(url, params=params)
@@ -44,4 +44,3 @@ class UpdateBoardByIdTool(BuiltinTool):
updated_board = response.json()
return self.create_text_message(text=f"Board '{updated_board['name']}' updated successfully.")

10
api/core/tools/provider/builtin/trello/tools/update_card.py
View File

@@ -22,17 +22,17 @@ class UpdateCardByIdTool(BuiltinTool):
Returns:
ToolInvokeMessage: The result of the tool invocation.
"""
api_key = self.runtime.credentials.get('trello_api_key')
token = self.runtime.credentials.get('trello_api_token')
card_id = tool_parameters.get('id')
api_key = self.runtime.credentials.get("trello_api_key")
token = self.runtime.credentials.get("trello_api_token")
card_id = tool_parameters.get("id")
if not (api_key and token and card_id):
return self.create_text_message("Missing required parameters: API key, token, or card ID.")
# Constructing the URL and the payload for the PUT request
url = f"https://api.trello.com/1/cards/{card_id}"
params = {k: v for k, v in tool_parameters.items() if v is not None and k != 'id'}
params.update({'key': api_key, 'token': token})
params = {k: v for k, v in tool_parameters.items() if v is not None and k != "id"}
params.update({"key": api_key, "token": token})
try:
response = requests.put(url, params=params)

7
api/core/tools/provider/builtin/trello/trello.py
View File

@@ -9,17 +9,17 @@ from core.tools.provider.builtin_tool_provider import BuiltinToolProviderControl
class TrelloProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None:
"""Validate Trello API credentials by making a test API call.
Args:
credentials (dict[str, Any]): The Trello API credentials to validate.
Raises:
ToolProviderCredentialValidationError: If the credentials are invalid.
"""
api_key = credentials.get("trello_api_key")
token = credentials.get("trello_api_token")
url = f"https://api.trello.com/1/members/me?key={api_key}&token={token}"
try:
response = requests.get(url)
response.raise_for_status() # Raises an HTTPError for bad responses
@@ -32,4 +32,3 @@ class TrelloProvider(BuiltinToolProviderController):
except requests.exceptions.RequestException as e:
# Handle other exceptions, such as connection errors
raise ToolProviderCredentialValidationError("Error validating Trello credentials")