Method pre_process

datastudio/models/base.py:398–427

Validate and preprocess raw input messages into API-ready format. Args: msg: Raw input message (str, dict, or list). Returns: list: Preprocessed message list in OpenAI format. Raises: AssertionError: If input type is unsupported.

(self, msg)

        return self.fail_msg

    def pre_process(self, msg):
        """Validate and preprocess raw input messages into API-ready format.

        Args:
            msg: Raw input message (str, dict, or list).

        Returns:
            list: Preprocessed message list in OpenAI format.

        Raises:
            AssertionError: If input type is unsupported.
        """
        assert self.check_content(msg) in [
            "str",
            "dict",
            "liststr",
            "listdict",
        ], f"Unsupported input type: {msg}"

        msg = self.preproc_content(msg)
        assert msg is not None and self.check_content(msg) == "listdict"

        for item in msg:
            assert (
                item["type"] in self.allowed_types
            ), f'Unsupported input type: {item["type"]}'

        # Prepare input messages
        processed_msg, _ = self.prepare_inputs(msg)
        return processed_msg

    def _pre_encode_images(self, messages):
        """Pre-encode all PIL images to base64 strings in parallel.