| __all__ = ['GetBudget'] | |
| from src.manager.budget_manager import BudgetManager | |
| class GetBudget(): | |
| dependencies = [] | |
| inputSchema = { | |
| "name": "GetBudget", | |
| "description": "Retrieves the current budget available.", | |
| "parameters": { | |
| "type": "object", | |
| "properties":{}, | |
| "required": [], | |
| }, | |
| } | |
| def run(self, **kwargs): | |
| budget_manager = BudgetManager() | |
| total_resource_budget = budget_manager.get_total_resource_budget() | |
| current_resource_usage = budget_manager.get_current_resource_usage() | |
| current_remaining_resource_budget = budget_manager.get_current_remaining_resource_budget() | |
| total_expense_budget = budget_manager.get_total_expense_budget() | |
| current_expense = budget_manager.get_current_expense() | |
| current_remaining_expense_budget = budget_manager.get_total_expense_budget() - budget_manager.get_current_expense() | |
| return { | |
| "status": "success", | |
| "message": "Budget retrieved successfully", | |
| "output": { | |
| "total_resource_budget": total_resource_budget, | |
| "current_resource_usage": current_resource_usage, | |
| "current_remaining_resource_budget": current_remaining_resource_budget, | |
| "total_expense_budget": total_expense_budget, | |
| "current_expense": current_expense, | |
| "current_remaining_expense_budget": current_remaining_expense_budget, | |
| } | |
| } | |