kubeflow / kubeflow/mcp-server

feat(hub): Model Registry client module

Open
#181 8 comments 0 reactions 1 assignee Claimed by @myselfkunal View on GitHub
area/hub kind/feature
Dominant language
Python
Stars
44
Forks
54
Avg merge
2d 18h
Merged PRs (30d)
29

Description

Implement the `hub` client module wrapping `kubeflow.hub.ModelRegistryClient` (SDK 0.4.x). Completes the train -> register -> serve lifecycle within a single MCP session.

## Context

- SDK 0.4.0 shipped `ModelRegistryClient` (v0.3.0+)
- Stub exists at `kubeflow_mcp/hub/`
- Without this, agents hit a dead end after "job complete"

## Proposed Tools

Read-only:
- `list_models`, `get_model`, `list_model_versions`, `get_model_version`, `compare_models`

Mutating (confirm gate):
- `register_model`, `promote_model`, `update_model_metadata`

## Acceptance Criteria

- Follows module contract (TOOLS, CLIENT_TOOL_DESCRIPTIONS, CLIENT_TOOL_ANNOTATIONS, INSTRUCTION_SECTIONS, CLIENT_RESOURCES, SECTION_ORDER)
- Confirm gate on mutations, persona filtering consistent with trainer
- Co-located tests, schema snapshot updated
- Resource: `hub://guides/model-lifecycle`

## References

- SDK source: `kubeflow/hub/api/model_registry_client.py`
- ROADMAP Phase 6
- Related: kubeflow/sdk#107

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.