utils/mapper.py

from utils.common import *
from utils.AccountInformation import account_information_tool, AccountInformation
from utils.GetBookingsForSpecificDate import get_booking_for_specific_date_tool, GetBookingsForSpecificDate
from utils.GetBookingSummary import get_booking_summary_tool, GetBookingSummary
from utils.GetLeadsInfo import get_leads_info_tool, GetLeadsInfo
from utils.SpecificLocationInformation import specific_location_information_tool, SpecificLocationInformation


AVAILABLE_FUNCTIONS = {
    "AccountInformation": AccountInformation,
    "GetBookingsForSpecificDate": GetBookingsForSpecificDate,
    "GetBookingSummary": GetBookingSummary,
    "GetLeadsInfo": GetLeadsInfo,
    "SpecificLocationInformation": SpecificLocationInformation
}

TOOLS = [
    account_information_tool,
    get_booking_for_specific_date_tool,
    get_booking_summary_tool,
    get_leads_info_tool,
    specific_location_information_tool
]


COMMON_VARS = {
    "account_id": ACCOUNT_ID,
    "location_id": LOCATION_ID,
    "date": DATE,
    "date_from": DATE_FROM,
    "date_to": DATE_TO
}