types

const (
	// MediaTypeModelConfigV01 is the media type for the model config json.
	MediaTypeModelConfigV01 = types.MediaType("application/vnd.docker.ai.model.config.v0.1+json")

	// MediaTypeGGUF indicates a file in GGUF version 3 format, containing a tensor model.
	MediaTypeGGUF = types.MediaType("application/vnd.docker.ai.gguf.v3")

	// MediaTypeSafetensors indicates a file in safetensors format, containing model weights.
	MediaTypeSafetensors = types.MediaType("application/vnd.docker.ai.safetensors")

	// MediaTypeVLLMConfigArchive indicates a tar archive containing vLLM-specific config files.
	MediaTypeVLLMConfigArchive = types.MediaType("application/vnd.docker.ai.vllm.config.tar")

	// MediaTypeDirTar indicates a tar archive containing a directory with its structure preserved.
	MediaTypeDirTar = types.MediaType("application/vnd.docker.ai.dir.tar")

	// MediaTypeLicense indicates a plain text file containing a license
	MediaTypeLicense = types.MediaType("application/vnd.docker.ai.license")

	// MediaTypeMultimodalProjector indicates a Multimodal projector file
	MediaTypeMultimodalProjector = types.MediaType("application/vnd.docker.ai.mmproj")

	// MediaTypeChatTemplate indicates a Jinja chat template
	MediaTypeChatTemplate = types.MediaType("application/vnd.docker.ai.chat.template.jinja")

	FormatGGUF        = Format("gguf")
	FormatSafetensors = Format("safetensors")

	// AnnotationFilePath specifies the file path of the layer (string)
	AnnotationFilePath = "org.cncf.model.filepath"

	// AnnotationFileMetadata specifies the metadata of the file (string), value is the JSON string of FileMetadata
	AnnotationFileMetadata = "org.cncf.model.file.metadata+json"

	// AnnotationMediaTypeUntested indicates whether the media type classification of files in the layer is untested (string)
	// Valid values are "true" or "false". When set to "true", it signals that the model packager has not verified
	// the media type classification and the type is inferred or assumed based on some heuristics.
	AnnotationMediaTypeUntested = "org.cncf.model.file.mediatype.untested"
)