def display()

in aidial_adapter_bedrock/llm/chat_emulator.py [0:0]

• 21 lines of code
• 8 McCabe index (conditional complexity)

    def display(self, messages: List[BaseMessage]) -> Tuple[str, List[str]]:
        if (
            self.fallback_to_completion
            and len(messages) == 1
            and isinstance(messages[0], HumanRegularMessage)
        ):
            return messages[0].text_content, []

        ret: List[str] = []

        if self._prelude is not None:
            ret.append(self._prelude)

        for message in messages:
            ret.append(self._format_message(message, len(ret)))

        if self.add_invitation_cue:
            ret.append(
                self._format_message(AIRegularMessage(content=""), len(ret))
            )

        stop_sequences: List[str] = []
        human_role = self.cues["human"]
        if human_role is not None:
            stop_sequences = [self.separator + human_role]

        return self.separator.join(ret), stop_sequences