def to_gemini_genai_role()

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

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

    def to_gemini_genai_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 "user"
            case Role.ASSISTANT:
                return "model"
            case _:
                assert_never(role)