dont truncate in huggingface hub

Author: shellmayr

sentry_sdk/integrations/huggingface_hub.py

@@ -3,7 +3,7 @@
 
 import sentry_sdk
 from sentry_sdk.ai.monitoring import record_token_usage
-from sentry_sdk.ai.utils import set_data_normalized, truncate_and_serialize_messages
+from sentry_sdk.ai.utils import set_data_normalized
 from sentry_sdk.consts import OP, SPANDATA
 from sentry_sdk.integrations import DidNotEnable, Integration
 from sentry_sdk.scope import should_send_default_pii
@@ -103,11 +103,9 @@ def new_huggingface_task(*args, **kwargs):
 
         # Input attributes
         if should_send_default_pii() and integration.include_prompts:
-            # Convert prompt to message format if it's a string
-            messages = [prompt] if isinstance(prompt, str) else prompt
-            messages_data = truncate_and_serialize_messages(messages)
-            if messages_data is not None:
-                span.set_data(SPANDATA.GEN_AI_REQUEST_MESSAGES, messages_data)
+            set_data_normalized(
+                span, SPANDATA.GEN_AI_REQUEST_MESSAGES, prompt, unpack=False
+            )
 
         attribute_mapping = {
             "tools": SPANDATA.GEN_AI_REQUEST_AVAILABLE_TOOLS,

tests/integrations/huggingface_hub/test_huggingface_hub.py

@@ -1,18 +1,16 @@
-import re
-from typing import TYPE_CHECKING
 from unittest import mock
-
-import httpx
 import pytest
+import re
 import responses
+import httpx
+
 from huggingface_hub import InferenceClient
 
 import sentry_sdk
-from sentry_sdk._types import AnnotatedValue
-from sentry_sdk.ai.utils import MAX_GEN_AI_MESSAGE_BYTES
-from sentry_sdk.integrations.huggingface_hub import HuggingfaceHubIntegration
-from sentry_sdk.serializer import serialize
 from sentry_sdk.utils import package_version
+from sentry_sdk.integrations.huggingface_hub import HuggingfaceHubIntegration
+
+from typing import TYPE_CHECKING
 
 try:
     from huggingface_hub.utils._errors import HfHubHTTPError