ultralytics-mcp 0.1.0 → 0.1.1

package/README.md

@@ -1,9 +1,28 @@
 # Ultralytics Platform MCP
 
-MCP server for the Ultralytics Platform REST API.
+MCP server for the [Ultralytics Platform](https://platform.ultralytics.com).
 
-Current milestone: protocol scaffold with CLI entrypoint and protocol-level
-tests. Tooling and API features land incrementally from here.
+> Independent community project. Not affiliated with or endorsed by Ultralytics.
+
+Current milestone: read, monitor, predict, export, and initial project and
+dataset lifecycle tools are available. Additional resource-management tools
+land incrementally from here.
+
+## Tools (18)
+
+| Tool | Description |
+| --- | --- |
+| `projects_list` / `projects_get` | Browse projects |
+| `projects_create` / `projects_delete` | Create / soft-delete projects |
+| `datasets_list` / `datasets_get` / `datasets_create` / `datasets_delete` | Browse / create / soft-delete datasets |
+| `models_list` / `models_get` | Browse trained models and metrics |
+| `training_monitor` | Status, progress, and latest metrics |
+| `model_predict` | Run inference on an image URL or base64 source |
+| `model_download` | Download a model weight file to a local path |
+| `gpu_availability` | Cloud GPU stock status |
+| `exports_list` / `export_status` | List / check export jobs |
+| `export_create` | Create an export job — **requires `confirm_cost: true`** |
+| `training_start` | Start cloud training — **requires `confirm_cost: true`** |
 
 ## Development
 

package/dist/client.js

@@ -63,6 +63,10 @@ export class UltralyticsClient {
             retryOn429: options.retryOn429 ?? false,
         });
     }
+    /** DELETE requests are state-changing and do not retry 429 responses. */
+    async delete(path) {
+        return this.request("DELETE", path, { retryOn429: false });
+    }
     /** Download bytes from a signed URL WITHOUT forwarding API credentials. */
     async downloadBytes(url) {
         let attempt = 0;

package/dist/tools/datasets.js

@@ -1,6 +1,18 @@
 /** Read-only dataset tools. */
 import { resolveDataset } from "../resolve.js";
 import { asRecord, listField, pyCount, pyField } from "./shared.js";
+const DATASET_TASKS = new Set([
+    "detect",
+    "segment",
+    "semantic",
+    "classify",
+    "pose",
+    "obb",
+]);
+function resourceId(item, fallback) {
+    const value = item._id ?? item.id ?? item.projectId ?? item.datasetId;
+    return String(value ?? fallback ?? "None");
+}
 /** List datasets in the workspace, optionally filtered by username. */
 export async function datasetsList(client, username) {
     const data = await client.get("/datasets", username ? { username } : undefined);
@@ -28,3 +40,46 @@ export async function datasetsGet(client, dataset) {
         data: item,
     };
 }
+/** Create a dataset. */
+export async function datasetsCreate(client, options) {
+    if (!DATASET_TASKS.has(options.task)) {
+        const allowed = Array.from(DATASET_TASKS).sort().join(", ");
+        throw new Error(`Unsupported dataset task '${options.task}'. Expected one of: ${allowed}.`);
+    }
+    if (!options.slug.trim()) {
+        throw new Error("`slug` is required.");
+    }
+    const payload = {
+        name: options.name,
+        task: options.task,
+        slug: options.slug,
+    };
+    if (options.description !== undefined) {
+        payload.description = options.description;
+    }
+    if (options.visibility !== undefined) {
+        payload.visibility = options.visibility;
+    }
+    if (options.classNames !== undefined) {
+        payload.classNames = options.classNames;
+    }
+    const data = await client.postJson("/datasets", payload);
+    const record = asRecord(data);
+    const item = asRecord("dataset" in record ? record.dataset : data);
+    const id = resourceId(item);
+    const slug = item.slug ?? options.slug;
+    const task = item.task ?? options.task;
+    return {
+        summary: `Created dataset ${id} slug=${String(slug)} task=${String(task)}.`,
+        data: item,
+    };
+}
+/** Soft-delete a dataset by id, slug, username/slug, or dataset ul:// URI. */
+export async function datasetsDelete(client, dataset) {
+    const datasetId = await resolveDataset(client, dataset);
+    const data = await client.delete(`/datasets/${datasetId}`);
+    return {
+        summary: `Deleted dataset ${datasetId} (soft delete).`,
+        data: { id: datasetId, response: data },
+    };
+}

package/dist/tools/index.js

@@ -7,28 +7,32 @@
  */
 import { z } from "zod";
 import { toMcpTextResult } from "../tool-result.js";
-import { datasetsGet, datasetsList } from "./datasets.js";
+import { datasetsCreate, datasetsDelete, datasetsGet, datasetsList, } from "./datasets.js";
 import { modelDownload } from "./downloads.js";
 import { exportCreate, exportStatus, exportsList } from "./exports.js";
 import { gpuAvailability } from "./gpu.js";
 import { modelsGet, modelsList } from "./models.js";
 import { modelPredict } from "./predict.js";
-import { projectsGet, projectsList } from "./projects.js";
+import { projectsCreate, projectsDelete, projectsGet, projectsList, } from "./projects.js";
 import { trainingMonitor, trainingStart } from "./training.js";
-export { datasetsGet, datasetsList } from "./datasets.js";
+export { datasetsCreate, datasetsDelete, datasetsGet, datasetsList, } from "./datasets.js";
 export { modelDownload } from "./downloads.js";
 export { exportCreate, exportStatus, exportsList } from "./exports.js";
 export { gpuAvailability } from "./gpu.js";
 export { modelsGet, modelsList } from "./models.js";
 export { modelPredict } from "./predict.js";
-export { projectsGet, projectsList } from "./projects.js";
+export { projectsCreate, projectsDelete, projectsGet, projectsList, } from "./projects.js";
 export { trainingMonitor, trainingStart } from "./training.js";
 /** Names of the read-only tools registered by `registerReadTools`. */
 export const READ_TOOL_NAMES = [
     "projects_list",
     "projects_get",
+    "projects_create",
+    "projects_delete",
     "datasets_list",
     "datasets_get",
+    "datasets_create",
+    "datasets_delete",
     "models_list",
     "models_get",
     "gpu_availability",
@@ -43,6 +47,18 @@ export function registerReadTools(server, getClient) {
         description: "Get details for one project by id, slug, username/slug, or project ul:// URI.",
         inputSchema: { project: z.string() },
     }, async ({ project }) => toMcpTextResult(await projectsGet(getClient(), project)));
+    server.registerTool("projects_create", {
+        description: "Create a project in your Ultralytics workspace.",
+        inputSchema: {
+            name: z.string(),
+            slug: z.string().optional(),
+            description: z.string().optional(),
+        },
+    }, async ({ name, slug, description }) => toMcpTextResult(await projectsCreate(getClient(), { name, slug, description })));
+    server.registerTool("projects_delete", {
+        description: "Soft-delete a project by id, slug, username/slug, or project ul:// URI.",
+        inputSchema: { project: z.string() },
+    }, async ({ project }) => toMcpTextResult(await projectsDelete(getClient(), project)));
     server.registerTool("datasets_list", {
         description: "List datasets in your Ultralytics workspace.",
         inputSchema: { username: z.string().optional() },
@@ -51,6 +67,28 @@ export function registerReadTools(server, getClient) {
         description: "Get details for one dataset by id, slug, username/slug, or dataset ul:// URI.",
         inputSchema: { dataset: z.string() },
     }, async ({ dataset }) => toMcpTextResult(await datasetsGet(getClient(), dataset)));
+    server.registerTool("datasets_create", {
+        description: "Create a dataset in your Ultralytics workspace.",
+        inputSchema: {
+            name: z.string(),
+            task: z.string(),
+            slug: z.string(),
+            description: z.string().optional(),
+            visibility: z.string().optional(),
+            classNames: z.array(z.string()).optional(),
+        },
+    }, async ({ name, task, slug, description, visibility, classNames }) => toMcpTextResult(await datasetsCreate(getClient(), {
+        name,
+        task,
+        slug,
+        description,
+        visibility,
+        classNames,
+    })));
+    server.registerTool("datasets_delete", {
+        description: "Soft-delete a dataset by id, slug, username/slug, or dataset ul:// URI.",
+        inputSchema: { dataset: z.string() },
+    }, async ({ dataset }) => toMcpTextResult(await datasetsDelete(getClient(), dataset)));
     server.registerTool("models_list", {
         description: "List models in a project by project id, slug, username/slug, or project ul:// URI.",
         inputSchema: { project: z.string() },

package/dist/tools/projects.js

@@ -1,6 +1,10 @@
 /** Read-only project tools. */
 import { resolveProject } from "../resolve.js";
 import { asRecord, listField, pyCount, pyField } from "./shared.js";
+function resourceId(item, fallback) {
+    const value = item._id ?? item.id ?? item.projectId ?? item.datasetId;
+    return String(value ?? fallback ?? "None");
+}
 /** List projects in the workspace, optionally filtered by username. */
 export async function projectsList(client, username) {
     const data = await client.get("/projects", username ? { username } : undefined);
@@ -27,3 +31,31 @@ export async function projectsGet(client, project) {
         data: item,
     };
 }
+/** Create a project. */
+export async function projectsCreate(client, options) {
+    const payload = { name: options.name };
+    if (options.slug !== undefined) {
+        payload.slug = options.slug;
+    }
+    if (options.description !== undefined) {
+        payload.description = options.description;
+    }
+    const data = await client.postJson("/projects", payload);
+    const record = asRecord(data);
+    const item = asRecord("project" in record ? record.project : data);
+    const id = resourceId(item);
+    const slug = item.slug ?? options.slug ?? "None";
+    return {
+        summary: `Created project ${id} slug=${String(slug)}.`,
+        data: item,
+    };
+}
+/** Soft-delete a project by id, slug, username/slug, or project ul:// URI. */
+export async function projectsDelete(client, project) {
+    const projectId = await resolveProject(client, project);
+    const data = await client.delete(`/projects/${projectId}`);
+    return {
+        summary: `Deleted project ${projectId} (soft delete).`,
+        data: { id: projectId, response: data },
+    };
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ultralytics-mcp",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "TypeScript MCP server for the Ultralytics Platform REST API.",
   "type": "module",
   "bin": {
@@ -20,7 +20,8 @@
     "url": "https://github.com/amanharshx/ultralytics-mcp/issues"
   },
   "publishConfig": {
-    "access": "public"
+    "access": "public",
+    "provenance": true
   },
   "engines": {
     "node": ">=20"