wehub-resource-sync
c56bef871b
Sync docs with Docusaurus / sync (push) Waiting to run
Tests / Check if changed (push) Waiting to run
Tests / format (push) Blocked by required conditions
Tests / check-imports (push) Blocked by required conditions
Tests / Unit / macos-latest (push) Blocked by required conditions
Tests / Unit / ubuntu-latest (push) Blocked by required conditions
Tests / Unit / windows-latest (push) Blocked by required conditions
Tests / mypy (push) Blocked by required conditions
Tests / Integration / ubuntu-latest (push) Blocked by required conditions
Tests / Integration / macos-latest (push) Blocked by required conditions
Tests / Integration / windows-latest (push) Blocked by required conditions
Tests / notify-slack-on-failure (push) Blocked by required conditions
Tests / Mark tests as completed (push) Blocked by required conditions
Docker image release / Build base image (push) Waiting to run
CodeQL / Analyze (python) (push) Has been cancelled
Update Platform Components Table / update (push) Has been cancelled
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import json
|
|
from typing import Any
|
|
|
|
from haystack import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
PRIMITIVE_TYPES = (bool, str, int, float)
|
|
|
|
|
|
def coerce_tag_value(value: Any) -> bool | str | int | float:
|
|
"""
|
|
Coerces span tag values to compatible types for the tracing backend.
|
|
|
|
Most tracing libraries don't support sending complex types to the backend. Hence, we need to convert them to
|
|
compatible types.
|
|
|
|
:param value: an arbitrary value which should be coerced to a compatible type
|
|
:return: the value coerced to a compatible type
|
|
"""
|
|
if isinstance(value, PRIMITIVE_TYPES):
|
|
return value
|
|
|
|
if value is None:
|
|
return ""
|
|
|
|
try:
|
|
# do that with-in try-except because who knows what kind of objects are being passed
|
|
serializable = _serializable_value(value=value, use_placeholders=True)
|
|
return json.dumps(serializable)
|
|
except Exception as error:
|
|
logger.debug("Failed to coerce tag value to string: {error}", error=error)
|
|
|
|
# Our last resort is to convert the value to a string
|
|
return str(value)
|
|
|
|
|
|
def _serializable_value(value: Any, use_placeholders: bool = True) -> Any:
|
|
"""
|
|
Serializes a value into a format suitable for tracing.
|
|
|
|
One-way only: this is never deserialized, so it's allowed to lose data (e.g. replace large
|
|
objects with a placeholder). Don't swap it for `base_serialization`'s schema serializer, which
|
|
is built to round-trip exactly.
|
|
|
|
:param value:
|
|
The value to serialize.
|
|
:param use_placeholders:
|
|
Whether to use string placeholders for large objects like ByteStream and ImageContent.
|
|
:returns:
|
|
The serialized value.
|
|
"""
|
|
if isinstance(value, list):
|
|
return [_serializable_value(value=v, use_placeholders=use_placeholders) for v in value]
|
|
|
|
if isinstance(value, dict):
|
|
return {k: _serializable_value(value=v, use_placeholders=use_placeholders) for k, v in value.items()}
|
|
|
|
if use_placeholders and getattr(value, "_to_trace_dict", None):
|
|
return _serializable_value(value._to_trace_dict())
|
|
|
|
if getattr(value, "to_dict", None):
|
|
return _serializable_value(value.to_dict())
|
|
|
|
return value
|