AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 19:08:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

UPDATE: Twilio tool crdential verification (#2741)

Browse Source
This commit is contained in:
Yash Parmar 2024-03-07 15:38:52 +05:30 committed by GitHub
parent b75ff5fa03
commit c0b82f8e58
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
api/core/tools/provider/builtin/twilio/twilio.py
View File

@ -1,5 +1,8 @@
from typing import Any from typing import Any
from twilio.base.exceptions import TwilioRestException
from twilio.rest import Client
from core.tools.errors import ToolProviderCredentialValidationError from core.tools.errors import ToolProviderCredentialValidationError
from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@ -7,19 +10,20 @@ from core.tools.provider.builtin_tool_provider import BuiltinToolProviderControl
class TwilioProvider(BuiltinToolProviderController): class TwilioProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None: def _validate_credentials(self, credentials: dict[str, Any]) -> None:
try: try:
""" # Extract credentials
SendMessageTool().fork_tool_runtime( account_sid = credentials["account_sid"]
meta={ auth_token = credentials["auth_token"]
"credentials": credentials, from_number = credentials["from_number"]
}
).invoke( # Initialize twilio client
user_id="", client = Client(account_sid, auth_token)
tool_parameters={
"message": "Credential validation message", # fetch account
"to_number": "+14846624384", client.api.accounts(account_sid).fetch()
},
) except TwilioRestException as e:
""" raise ToolProviderCredentialValidationError(f"Twilio API error: {e.msg}") from e
pass except KeyError as e:
raise ToolProviderCredentialValidationError(f"Missing required credential: {e}") from e
except Exception as e: except Exception as e:
raise ToolProviderCredentialValidationError(str(e)) raise ToolProviderCredentialValidationError(str(e))