code: def get_total_budget(self, user: str):
      return self.user_dict[user]["total_budget"]
  
imports: import os, json, time
  import litellm
  from litellm.utils import ModelResponse
  import requests, threading
  from typing import Optional, Union, Literal
label_desc: Use image node_id 6 for calling the BudgetManager obj's underlying member method code with example usage: obj.get_total_budget(user) and returns: self