def to_file_data()

in google/generativeai/types/file_types.py [0:0]

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

def to_file_data(file_data: FileDataType):
    if isinstance(file_data, dict):
        if "file_uri" in file_data:
            file_data = protos.FileData(file_data)
        else:
            file_data = protos.File(file_data)

    if isinstance(file_data, File):
        file_data = file_data.to_proto()

    if isinstance(file_data, protos.File):
        file_data = protos.FileData(
            mime_type=file_data.mime_type,
            file_uri=file_data.uri,
        )

    if isinstance(file_data, protos.FileData):
        return file_data
    else:
        raise TypeError(
            f"Invalid input type. Failed to convert input to `FileData`.\n"
            f"Received an object of type: {type(file_data)}.\n"
            f"Object Value: {file_data}"
        )