26 lines
977 B
Python
26 lines
977 B
Python
import json
|
|
from typing import Optional
|
|
|
|
from arcade.sdk.errors import RetryableToolError
|
|
|
|
from arcade_search.google_data import COUNTRY_CODES, LANGUAGE_CODES
|
|
|
|
|
|
class GoogleRetryableError(RetryableToolError):
|
|
pass
|
|
|
|
|
|
class CountryNotFoundError(GoogleRetryableError):
|
|
def __init__(self, country: Optional[str]) -> None:
|
|
valid_countries = json.dumps(COUNTRY_CODES, default=str)
|
|
message = f"Country not found: '{country}'."
|
|
additional_message = f"Valid countries are: {valid_countries}"
|
|
super().__init__(message, additional_prompt_content=additional_message)
|
|
|
|
|
|
class LanguageNotFoundError(GoogleRetryableError):
|
|
def __init__(self, language: Optional[str]) -> None:
|
|
valid_languages = json.dumps(LANGUAGE_CODES, default=str)
|
|
message = f"Language not found: '{language}'."
|
|
additional_message = f"Valid languages are: {valid_languages}"
|
|
super().__init__(message, additional_prompt_content=additional_message)
|