Method check_content

datastudio/models/base.py:139–158

Check input content type. Args: msgs: Raw input messages. Returns: str: Message type (str/dict/liststr/listdict).

(self, msgs)

            return self.encode_image_directly(img)

    def check_content(self, msgs):
        """Check input content type.

        Args:
            msgs: Raw input messages.

        Returns:
            str: Message type (str/dict/liststr/listdict).
        """
        if isinstance(msgs, str):
            return "str"
        if isinstance(msgs, dict):
            return "dict"
        if isinstance(msgs, list):
            types = [self.check_content(m) for m in msgs]
            if all(t == "str" for t in types):
                return "liststr"
            if all(t == "dict" for t in types):
                return "listdict"
        return "unknown"

    def preproc_content(self, inputs):
        """Convert raw input messages to unified dict list format.