upload

backend/tests/test_semantic_selection.py

@@ -0,0 +1,41 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+
+from app.services.semantic_selection import SemanticSelectionService
+
+
+def test_score_maps_ru_users_query_to_en_capability_tokens():
+    service = SemanticSelectionService()
+    query_tokens = service._tokenize("Хочу получить пользователей")
+    query_tokens_expanded = service._expand_tokens(query_tokens)
+    capability = SimpleNamespace(
+        name="get_users",
+        description="Get users list",
+    )
+
+    score = service._score_capability(query_tokens, query_tokens_expanded, capability)
+
+    assert score >= 0.45
+
+
+def test_score_uses_capability_action_context_tokens():
+    service = SemanticSelectionService()
+    query_tokens = service._tokenize("Отправь email по кампании")
+    query_tokens_expanded = service._expand_tokens(query_tokens)
+    capability = SimpleNamespace(
+        name="execute_action",
+        description="General API action",
+        llm_payload={
+            "action_context_brief": {
+                "method": "POST",
+                "path": "/v1/campaigns/emails/send",
+                "tags": ["campaign", "email"],
+                "summary": "Send campaign emails",
+            }
+        },
+    )
+
+    score = service._score_capability(query_tokens, query_tokens_expanded, capability)
+
+    assert score > 0.0