Has budgets for end-users been implemented?

[lassib]

According to the docs it should be possible to apply budgets to end-users:
https://docs.litellm.ai/docs/proxy/customers#setting-customer-budgets

But when I try to apply it using the API endpoint of /customer/update I keep getting an error:

{
  "error": {
    "message": "Internal Server Error, Field: \"litellm_endusertable\" either does not exist or is not a relational field on the LiteLLM_BudgetTable model",
    "type": "internal_error",
    "param": "None",
    "code": "500"
  }
}

I can apply the budget to end-users by manipulating the DB table "LiteLLM_EndUserTable" and budget limit seems to apply and work properly. Now if I try to change the budged for that specific end-user using /customer/update endpoint it tries to change the name/id of the budget and not the actual budget.