What happened?

I created a budget (via /budget/new) and a customer with this budget (via /customer/new). I can see both the customer and the budget in the database, but the /customer/info method always returns an object without litellm_budget_table ({"user_id": "...", ... , "litellm_budget_table": null}).

I think this happens because prisma_client call in this method doesn't use the include parameter. In the /organization/info method this parameter is used (include={"litellm_budget_table": True}) and litellm_budget_table returns successfully.

I can make a PR for these changes, but I don't understand all places where this parameter should be used, so I can add them only for methods that return LiteLLM_EndUserTable.

Relevant log output

No response

Twitter / LinkedIn details

No response