TS 怎么或者可以不可以实现类似的函数(签名)，根据调用时传入的泛型类型，决定返回值的类型

interface A {
    A: string;
};
interface B {
    B: string;
}
function returnTwoType<T extends A | B>(type: 'A' | 'B'): T {
    if (type === 'A') {
        const res = { A: '类型 A' };
        // Type '{ A: string; }' is not assignable to type 'T'.
        // '{ A: string; }' is assignable to the constraint of type 'T', but 'T' could be instantiated with a different subtype of constraint 'A | B'.
        return res
    }
    else {
        const res = {
            B: '类型 B'
        };
        // Type '{ B: string; }' is not assignable to type 'T'.
        // '{ B: string; }' is assignable to the constraint of type 'T', but 'T' could be instantiated with a different subtype of constraint 'A | B'.
        return res
    }
}

const a = returnTwoType<A>('A')
const b = returnTwoType<B>('B')