nominatim.py 851 B

1234567891011121314151617181920212223
  1. from typing import Any
  2. from core.tools.errors import ToolProviderCredentialValidationError
  3. from core.tools.provider.builtin.nominatim.tools.nominatim_search import NominatimSearchTool
  4. from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
  5. class NominatimProvider(BuiltinToolProviderController):
  6. def _validate_credentials(self, credentials: dict[str, Any]) -> None:
  7. try:
  8. result = NominatimSearchTool().fork_tool_runtime(
  9. runtime={
  10. "credentials": credentials,
  11. }
  12. ).invoke(
  13. user_id='',
  14. tool_parameters={
  15. 'query': 'London',
  16. 'limit': 1,
  17. },
  18. )
  19. except Exception as e:
  20. raise ToolProviderCredentialValidationError(str(e))