def _to_gemini_role()

in aidial_adapter_vertexai/chat/gemini/conversation_factory.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

    def _to_gemini_role(role: Role) -> str:
        match role:
            case Role.SYSTEM:
                raise ValidationError(
                    "System messages other than the first system message are not allowed"
                )
            case Role.USER | Role.FUNCTION | Role.TOOL:
                return ChatSession._USER_ROLE
            case Role.ASSISTANT:
                return ChatSession._MODEL_ROLE
            case _:
                assert_never(role)