def _get_cue()

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

• 9 lines of code
• 4 McCabe index (conditional complexity)

    def _get_cue(self, message: BaseMessage) -> Optional[str]:
        if isinstance(message, HumanRegularMessage):
            return self.cues["human"]
        elif isinstance(message, AIRegularMessage):
            return self.cues["ai"]
        elif isinstance(message, BaseMessage):
            return self.cues["system"]
        else:
            raise ValueError(f"Unknown message type: {message.type}")