CountTokensInput module-attribute

CountTokensInput = Union[CountTokensInputInvokeModel | CountTokensInputConverse | CountTokensInputUnknown]

The input value for token counting. The value should be either an InvokeModel or Converse request body.

Union Members

CountTokensInputInvokeModel dataclass

An InvokeModel request for which to count tokens. Use this field when you want to count tokens for a raw text input that would be sent to the InvokeModel operation.

Source code in src/aws_sdk_bedrock_runtime/models.py

@dataclass
class CountTokensInputInvokeModel:
    """An `InvokeModel` request for which to count tokens. Use this field when
    you want to count tokens for a raw text input that would be sent to the
    `InvokeModel` operation.
    """

    value: InvokeModelTokensRequest

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

    def serialize_members(self, serializer: ShapeSerializer):
        serializer.write_struct(
            _SCHEMA_COUNT_TOKENS_INPUT.members["invokeModel"], self.value
        )

    @classmethod
    def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
        return cls(value=InvokeModelTokensRequest.deserialize(deserializer))

CountTokensInputConverse dataclass

A Converse request for which to count tokens. Use this field when you want to count tokens for a conversation-based input that would be sent to the Converse operation.

Source code in src/aws_sdk_bedrock_runtime/models.py

@dataclass
class CountTokensInputConverse:
    """A `Converse` request for which to count tokens. Use this field when you
    want to count tokens for a conversation-based input that would be sent
    to the `Converse` operation.
    """

    value: ConverseTokensRequest

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

    def serialize_members(self, serializer: ShapeSerializer):
        serializer.write_struct(
            _SCHEMA_COUNT_TOKENS_INPUT.members["converse"], self.value
        )

    @classmethod
    def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
        return cls(value=ConverseTokensRequest.deserialize(deserializer))

CountTokensInputUnknown dataclass

Represents an unknown variant.

If you receive this value, you will need to update your library to receive the parsed value.

This value may not be deliberately sent.

Source code in src/aws_sdk_bedrock_runtime/models.py

@dataclass
class CountTokensInputUnknown:
    """Represents an unknown variant.

    If you receive this value, you will need to update your library to receive the
    parsed value.

    This value may not be deliberately sent.
    """

    tag: str

    def serialize(self, serializer: ShapeSerializer):
        raise SerializationError("Unknown union variants may not be serialized.")

    def serialize_members(self, serializer: ShapeSerializer):
        raise SerializationError("Unknown union variants may not be serialized.")

    @classmethod
    def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
        raise NotImplementedError()