Error function, sqlalchemy

всем привет, гайс

Добавлю платежку кб и делаю добавление баланса в бд, выдаёт ошибку
TypeError: add_balance() takes 1 positional argument but 2 were given

await add_balance(amount,user_id)


User_id и amount не пустые, имеют значение

async def add_balance(amount, user_id):

async with async_session() as session:

current_balance = await session.scalar(select(User.balance).where(User.id == user_id))

amount = float(amount)



new_balance = current_balance + amount



await session.execute(

update(User)

.where(User.id == user_id)

.values(balance=new_balance)

)



await session.commit()



return new_balance

Python

async def add_balance(amount, user_id):



    async with async_session() as session:



        current_balance = await session.scalar(select(User.balance).where(User.id == user_id))



        amount = float(amount)







        new_balance = current_balance + amount



        



        await session.execute(



            update(User)



            .where(User.id == user_id)



            .values(balance=new_balance)



        )







        await session.commit()







        return new_balance