diff --git a/mediapipe/tasks/python/components/containers/__init__.py b/mediapipe/tasks/python/components/containers/__init__.py
index 1a3f0c3395..457450b06f 100644
--- a/mediapipe/tasks/python/components/containers/__init__.py
+++ b/mediapipe/tasks/python/components/containers/__init__.py
@@ -23,6 +23,7 @@
 import mediapipe.tasks.python.components.containers.landmark
 import mediapipe.tasks.python.components.containers.landmark_detection_result
 import mediapipe.tasks.python.components.containers.rect
+import mediapipe.tasks.python.components.containers.keypoint
 
 AudioDataFormat = audio_data.AudioDataFormat
 AudioData = audio_data.AudioData
@@ -36,6 +37,7 @@
 EmbeddingResult = embedding_result.EmbeddingResult
 Landmark = landmark.Landmark
 NormalizedLandmark = landmark.NormalizedLandmark
+NormalizedKeypoint = keypoint.NormalizedKeypoint
 LandmarksDetectionResult = landmark_detection_result.LandmarksDetectionResult
 Rect = rect.Rect
 NormalizedRect = rect.NormalizedRect
@@ -51,3 +53,4 @@
 del landmark_detection_result
 del rect
 del mediapipe
+del keypoint
diff --git a/mediapipe/tasks/python/components/utils/__init__.py b/mediapipe/tasks/python/components/utils/__init__.py
index 3fbc1e8a7a..3aa94d157f 100644
--- a/mediapipe/tasks/python/components/utils/__init__.py
+++ b/mediapipe/tasks/python/components/utils/__init__.py
@@ -11,3 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from mediapipe.tasks.python.components.utils import cosine_similarity as _cosine_similarity_module
+
+cosine_similarity = _cosine_similarity_module.cosine_similarity
+
+del _cosine_similarity_module
\ No newline at end of file