bing.py 857 B

1234567891011121314151617181920212223
  1. from typing import Any
  2. from core.tools.errors import ToolProviderCredentialValidationError
  3. from core.tools.provider.builtin.bing.tools.bing_web_search import BingSearchTool
  4. from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
  5. class BingProvider(BuiltinToolProviderController):
  6. def _validate_credentials(self, credentials: dict[str, Any]) -> None:
  7. try:
  8. BingSearchTool().fork_tool_runtime(
  9. runtime={
  10. "credentials": credentials,
  11. }
  12. ).validate_credentials(
  13. credentials=credentials,
  14. tool_parameters={
  15. "query": "test",
  16. "result_type": "link",
  17. },
  18. )
  19. except Exception as e:
  20. raise ToolProviderCredentialValidationError(str(e))