Update to use meta struct for the rules

Author: manuel-alvarez-alvarez

ddtrace/appsec/ai_guard/_api_client.py

@@ -211,7 +211,8 @@ def evaluate(self, messages: List[Message], options: Optional[Options] = None) -
                     span.set_tag(AI_GUARD.TOOL_NAME_TAG, tool_name)
                 else:
                     span.set_tag(AI_GUARD.TARGET_TAG, "prompt")
-                span.set_struct_tag(AI_GUARD.STRUCT, {"messages": self._messages_for_meta_struct(messages)})
+
+                span._set_struct_tag(AI_GUARD.STRUCT, {"messages": self._messages_for_meta_struct(messages)})
 
                 try:
                     response = self._execute_request(f"{self._endpoint}/evaluate", payload)
@@ -241,7 +242,8 @@ def evaluate(self, messages: List[Message], options: Optional[Options] = None) -
 
                     span.set_tag(AI_GUARD.ACTION_TAG, action)
                     if len(tags) > 0:
-                        span.set_tag(AI_GUARD.TAG + ".matching_rules", json.dumps(tags))
+                        meta_struct = span._get_struct_tag(AI_GUARD.STRUCT)
+                        meta_struct.update({"matching_rules": tags})
                     if reason:
                         span.set_tag(AI_GUARD.REASON_TAG, reason)
                 else:

tests/appsec/ai_guard/api/test_api_client.py

@@ -1,4 +1,3 @@
-import json
 from itertools import product
 from unittest.mock import Mock
 from unittest.mock import patch
@@ -115,12 +114,13 @@ def test_evaluate_method(
         expected_tags.update({"ai_guard.tool_name": "calc"})
     if action != "ALLOW" and blocking:
         expected_tags.update({"ai_guard.blocked": "true"})
+    expected_meta_struct = {"messages": messages}
     if len(tags) > 0:
-        expected_tags.update({"ai_guard.matching_rules": json.dumps(tags)})
+        expected_meta_struct.update({"matching_rules": tags})
     assert_ai_guard_span(
         tracer,
-        messages,
         expected_tags,
+        expected_meta_struct,
     )
     assert_telemetry(
         telemetry_mock,

tests/appsec/ai_guard/utils.py

@@ -29,13 +29,19 @@ def find_ai_guard_span(tracer: DummyTracer) -> Span:
     return span
 
 
-def assert_ai_guard_span(tracer: DummyTracer, messages: List[Message], tags: Dict[str, Any]) -> None:
+def assert_ai_guard_span(
+    tracer: DummyTracer,
+    tags: Dict[str, Any],
+    meta_struct: Dict[str, Any],
+) -> None:
     span = find_ai_guard_span(tracer)
     for tag, value in tags.items():
         assert tag in span.get_tags(), f"Missing {tag} from spans tags"
         assert span.get_tag(tag) == value, f"Wrong value {span.get_tag(tag)}, expected {value}"
     struct = span._get_struct_tag(AI_GUARD.TAG)
-    assert struct["messages"] == messages
+    for meta, value in meta_struct.items():
+        assert meta in struct.keys(), f"Missing {meta} from meta_struct keys"
+        assert struct[meta] == value, f"Wrong value {struct[meta]}, expected {value}"
 
 
 def mock_evaluate_response(action: str, reason: str = "", tags: List[str] = None, block: bool = True) -> Mock: