---@class with function return question

so I have this (as a mockup):

do
    ---@class a
    local a = {}

    ---@generic T
    ---@param self a
    ---@param new T
    ---@return T
    a.extend = function(self, new)
        --...
        return new
    end
end

and it's supposed to be inheritance for prototypes, right, but I'm trying to make the LS think that it returns the second parameter new and it does that and the LS thinks that, so far so good. So this

    local b = a:extend{
        n = 1
    }

will make it like this:

but the problem is, that if I use ---@class b here, then it will ignore what the function returns and I get this:

am I missing something or is it just not possible? Though, if I separate it into two statements like this,

    ---@class b
    local b = {
        n = 1
    }

    b = a:extend(b)

then I get what I want:

but it's not ideal, since it's two statements now and I can't use the fun{} syntactic sugar this way, which would be the whole reason, right. Is there a way to make it work? Also I'm unsure about how to properly word the title here

Không có tệp mã nguồn hoặc bài kiểm thử nào được nêu tên. Hãy tái hiện hai đoạn mã Lua một dòng và hai đoạn mã Lua hai câu lệnh, sau đó theo dõi quá trình suy luận kiểu của LuaLS đối với các khai báo cục bộ, chú thích lớp, generic và giá trị trả về của hàm. Được xem là hoàn tất khi hành vi được hỗ trợ đã được triển khai và bao phủ bằng một bài kiểm thử hồi quy, hoặc khi giới hạn này được ghi rõ trong tài liệu.