### What problem does this PR solve? Type of change Bug Fix (Import necessary class for retrieval API ) New Feature (Add multi-KB search to retrieval API)