🚧 added additional tests to increase code coverage

Author: m-misiura

tests/test_embedding_topic_detector.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 
 import asyncio
+from types import SimpleNamespace
 
 from nemoguardrails import RailsConfig
 from tests.utils import TestChat
@@ -81,3 +82,72 @@ def test_detector_logic():
     off_topic = asyncio.run(detector.detect("Who won the Super Bowl?"))
     assert off_topic["on_topic"] is False
     assert off_topic["confidence"] < 0.5
+
+
+def test_empty_query_handling():
+    """Test that empty queries are handled gracefully."""
+    from nemoguardrails.library.embedding_topic_detector.actions import _check
+
+    llm_task_manager = SimpleNamespace(
+        config=SimpleNamespace(
+            rails=SimpleNamespace(
+                config=SimpleNamespace(
+                    embedding_topic_detector={
+                        "embedding_model": "BAAI/bge-small-en-v1.5",
+                        "embedding_engine": "FastEmbed",
+                        "examples": {"coffee": ["espresso"]},
+                        "threshold": 0.5,
+                        "top_k": 3,
+                    }
+                )
+            )
+        )
+    )
+
+    # Test with None context
+    result = asyncio.run(_check(None, llm_task_manager, "user_message"))
+    assert result == {
+        "on_topic": True,
+        "confidence": 0.0,
+        "top_category": None,
+        "category_scores": {},
+    }
+
+    # Test with empty message in context
+    result = asyncio.run(_check({}, llm_task_manager, "user_message"))
+    assert result["on_topic"] is True
+    assert result["confidence"] == 0.0
+
+
+def test_output_rail():
+    """Test that output rail (bot message checking) works."""
+    yaml_with_output = """
+models:
+  - type: main
+    engine: fake
+    model: test
+
+rails:
+  config:
+    embedding_topic_detector:
+      embedding_model: "BAAI/bge-small-en-v1.5"
+      embedding_engine: "FastEmbed"
+      threshold: 0.5
+      top_k: 3
+      examples:
+        coffee:
+          - "how to brew coffee"
+          - "espresso tips"
+
+  output:
+    flows:
+      - embedding topic check output
+"""
+
+    config = RailsConfig.from_content(
+        colang_content=COLANG_CONFIG, yaml_content=yaml_with_output
+    )
+    chat = TestChat(config, llm_completions=["Who won the Super Bowl yesterday?"])
+
+    chat >> "Hello"
+    chat << "I'm sorry, I can't respond to that."