Update optimum_encoder.py

optimum_encoder.py

@@ -4,7 +4,7 @@ import os
 import numpy as np
 from typing import Any, List, Optional, Dict
 from pydantic.v1 import PrivateAttr
-
+from fastembed.common.utils import normalize
 from semantic_router.encoders import BaseEncoder
 from semantic_router.utils.logger import logger
 
@@ -137,7 +137,7 @@ class OptimumEncoder(BaseEncoder):
 
             if normalize_embeddings:
                 if convert_to_numpy:
-                    embeddings = embeddings / np.linalg.norm(embeddings, axis=1, keepdims=True)
+                    embeddings = normalize(embeddings[:, 0]).astype(np.float32)
                 else:
                     embeddings = self._torch.nn.functional.normalize(embeddings, p=2, dim=1).detach().cpu().tolist()