Hi Brijesh,
Do you want to capture the cost of repair/rework?
If not, how about using Change of Stock/Physical Inventory adjustments to denote affected pieces as restricted or, even as separate product? Once re-work or repair is done, re adjust to original stock type/product("A", in your example)?
If yes, then its more complex, I think. Change this to a separate product("B" having same valuation) which has a repair/correction Production model and whose output is "A". Capture your repair costs(via service or repair) on this.
Problem I see is that repair costs will be captured on a separate Production Lot rather than the original. That is why I think it makes sense to use integrated Quality check(Inspection task as last task) within Production Model so, that the original Production Lot is not closed till actual quantity is of full quantity and correct quality and all costs are captured together.
Regards,
Harsh Vardhan Shrotriya