refactor: accept flat text as input for token classification mapper (#…

…1686)

Closes #482

(cherry picked from commit 379ac56)

src/rubrix/monitoring/asgi.py

@@ -46,7 +46,12 @@
 
 
 def token_classification_mapper(inputs, outputs):
-    text = inputs.get("text", "")
+    if isinstance(inputs, str):
+        text = inputs
+    elif isinstance(inputs, dict):
+        text = inputs.get("text", "")
+    else:
+        text = ""
     tokens = outputs.get("tokens") if isinstance(outputs, dict) else None
     return TokenClassificationRecord(
         text=text,

tests/client/test_asgi.py

@@ -123,7 +123,7 @@ def __call__(self, records, name: str, **kwargs):
 
     mock.post(
         expected_endpoint,
-        json=[{"text": "The main text data"}, {"text": "The main text data"}],
+        json=[{"text": "The main text data"}, "The main text data"],
     )
     time.sleep(0.2)
     assert mock_log.was_called