Documentation
¶
Overview ¶
Package modelregistry markdown.go provides human-readable Markdown formatters for model registry tools.
Package modelregistry implements MCP tools for GitLab ML model registry, providing download access to machine learning model package files.
Package modelregistry register.go wires ML model registry MCP tools to the MCP server.
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func FormatDownloadMarkdown ¶
func FormatDownloadMarkdown(o DownloadOutput) string
FormatDownloadMarkdown formats a downloaded ML model package file as Markdown.
func RegisterTools ¶
func RegisterTools(server *mcp.Server, client *gitlabclient.Client)
RegisterTools registers MCP tools for GitLab ML model registry.
Types ¶
type DownloadInput ¶
type DownloadInput struct {
ProjectID toolutil.StringOrInt `json:"project_id" jsonschema:"Project ID or URL-encoded path,required"`
ModelVersionID toolutil.StringOrInt `json:"model_version_id" jsonschema:"Model version ID (numeric or string like candidate:5),required"`
Path string `json:"path" jsonschema:"Path within the model package,required"`
Filename string `json:"filename" jsonschema:"Name of the file to download,required"`
}
DownloadInput holds parameters for downloading a ML model package file.
type DownloadOutput ¶
type DownloadOutput struct {
toolutil.HintableOutput
ProjectID string `json:"project_id"`
ModelVersionID string `json:"model_version_id"`
Path string `json:"path"`
Filename string `json:"filename"`
ContentBase64 string `json:"content_base64"`
SizeBytes int `json:"size_bytes"`
}
DownloadOutput represents the downloaded ML model package file.
func Download ¶
func Download(ctx context.Context, client *gitlabclient.Client, in DownloadInput) (DownloadOutput, error)
Download retrieves a machine learning model package file.
Source Files
Click to show internal directories.
Click to hide internal directories.