DocumentChunkLocation dataclass

Specifies a chunk-level location within a document, providing positioning information for cited content using logical document segments or chunks.

Source code in src/aws_sdk_bedrock_runtime/models.py

@dataclass(kw_only=True)
class DocumentChunkLocation:
    """Specifies a chunk-level location within a document, providing
    positioning information for cited content using logical document
    segments or chunks.
    """

    document_index: int | None = None
    """The index of the document within the array of documents provided in the
    request.
    """

    start: int | None = None
    """The starting chunk identifier or index of the cited content within the
    document.
    """

    end: int | None = None
    """The ending chunk identifier or index of the cited content within the
    document.
    """

    def serialize(self, serializer: ShapeSerializer):
        serializer.write_struct(_SCHEMA_DOCUMENT_CHUNK_LOCATION, self)

    def serialize_members(self, serializer: ShapeSerializer):
        if self.document_index is not None:
            serializer.write_integer(
                _SCHEMA_DOCUMENT_CHUNK_LOCATION.members["documentIndex"],
                self.document_index,
            )

        if self.start is not None:
            serializer.write_integer(
                _SCHEMA_DOCUMENT_CHUNK_LOCATION.members["start"], self.start
            )

        if self.end is not None:
            serializer.write_integer(
                _SCHEMA_DOCUMENT_CHUNK_LOCATION.members["end"], self.end
            )

    @classmethod
    def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
        return cls(**cls.deserialize_kwargs(deserializer))

    @classmethod
    def deserialize_kwargs(cls, deserializer: ShapeDeserializer) -> dict[str, Any]:
        kwargs: dict[str, Any] = {}

        def _consumer(schema: Schema, de: ShapeDeserializer) -> None:
            match schema.expect_member_index():
                case 0:
                    kwargs["document_index"] = de.read_integer(
                        _SCHEMA_DOCUMENT_CHUNK_LOCATION.members["documentIndex"]
                    )

                case 1:
                    kwargs["start"] = de.read_integer(
                        _SCHEMA_DOCUMENT_CHUNK_LOCATION.members["start"]
                    )

                case 2:
                    kwargs["end"] = de.read_integer(
                        _SCHEMA_DOCUMENT_CHUNK_LOCATION.members["end"]
                    )

                case _:
                    logger.debug("Unexpected member schema: %s", schema)

        deserializer.read_struct(_SCHEMA_DOCUMENT_CHUNK_LOCATION, consumer=_consumer)
        return kwargs

Attributes

document_index class-attribute instance-attribute

document_index: int | None = None

The index of the document within the array of documents provided in the request.

end class-attribute instance-attribute

end: int | None = None

The ending chunk identifier or index of the cited content within the document.

start class-attribute instance-attribute

start: int | None = None

The starting chunk identifier or index of the cited content within the document.