dify/api/core/tools/provider/builtin/gaode/gaode.py

import urllib.parse

import requests

from core.tools.errors import ToolProviderCredentialValidationError
from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController


class GaodeProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            if "api_key" not in credentials or not credentials.get("api_key"):
                raise ToolProviderCredentialValidationError("Gaode API key is required.")

            try:
                response = requests.get(
                    url="https://restapi.amap.com/v3/geocode/geo?address={address}&key={apikey}".format(
                        address=urllib.parse.quote("广东省广州市天河区广州塔"), apikey=credentials.get("api_key")
                    )
                )
                if response.status_code == 200 and (response.json()).get("info") == "OK":
                    pass
                else:
                    raise ToolProviderCredentialValidationError((response.json()).get("info"))
            except Exception as e:
                raise ToolProviderCredentialValidationError("Gaode API Key is invalid. {}".format(e))
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))