🐛 addressing greptile feedback

Author: m-misiura

nemoguardrails/library/embedding_topic_detector/actions.py

@@ -51,10 +51,8 @@ async def detect(self, query: str) -> Dict:
             [
                 (
                     cat,
-                    float(
-                        np.dot(query_emb, emb)
-                        / (np.linalg.norm(query_emb) * np.linalg.norm(emb) or 1)
-                    ),
+                    np.dot(query_emb, emb)
+                    / ((np.linalg.norm(query_emb) * np.linalg.norm(emb)) or 1e-10),
                 )
                 for cat, embs in self.embeddings.items()
                 for emb in embs
@@ -79,7 +77,7 @@ async def detect(self, query: str) -> Dict:
 
 async def _check(context: Optional[dict], llm_task_manager, message_key: str) -> dict:
     config = llm_task_manager.config.rails.config.embedding_topic_detector
-    cache_key = f"{config['embedding_model']}_{config['embedding_engine']}_{config.get('threshold', 0.75)}"
+    cache_key = f"{config['embedding_model']}_{config['embedding_engine']}_{config.get('threshold', 0.75)}_{config.get('top_k', 3)}"
 
     if cache_key not in _detector_cache:
         _detector_cache[cache_key] = EmbeddingTopicDetector(