E532 [MCP Integration] Phase 8: MLE workflow prompt by aniruddh-alt · Pull Request #2241 · oumi-ai/oumi · GitHub
[go: up one dir, main page]

Skip to content

[MCP Integration] Phase 8: MLE workflow prompt#2241

Open
aniruddh-alt wants to merge 28 commits intoani/mcp-integration-07-preflightfrom
ani/mcp-integration-08-prompt
Open

[MCP Integration] Phase 8: MLE workflow prompt#2241
aniruddh-alt wants to merge 28 commits intoani/mcp-integration-07-preflightfrom
ani/mcp-integration-08-prompt

Conversation

@aniruddh-alt
Copy link
Contributor

Description

Part of the MCP Integration PR chain (Phase 8 of 10) - Stage: prompt

What changed: Added mle_prompt.py containing the structured XML-tagged prompt that guides LLM agents through ML training workflows using Oumi MCP tools.

Why: This prompt is returned by get_started() and teaches agents the correct tool ordering, path rules, and config usage guidelines.

Related issues

Before submitting

  • Did you link the issue(s) related to this PR in the section above?
  • Did you add / update tests where needed?

@gitar-bot
Copy link
gitar-bot bot commented Mar 3, 2026

Important

Upgrade your plan to unlock code review, CI analysis, custom rules, and more.

@aniruddh-alt aniruddh-alt force-pushed the ani/mcp-integration-07-preflight branch from 3dab445 to 904e085 Compare March 4, 2026 00:04
@aniruddh-alt aniruddh-alt force-pushed the ani/mcp-integration-08-prompt branch from 0a68081 to a7df0a6 Compare March 4, 2026 00:05
@aniruddh-alt aniruddh-alt force-pushed the ani/mcp-integration-07-preflight branch from 904e085 to 5a719a8 Compare March 4, 2026 00:47
Aniruddhan Ramesh and others added 4 commits March 3, 2026 16:50
- Fix dead code: list literal always truthy, fallback never triggers
- Add CUDA version detection via torch
- Fix hf_token=True wrong semantic (use actual token)
Add mle_prompt.py containing the structured XML-tagged prompt that
guides LLM agents through ML training workflows using Oumi MCP tools.
Covers tool ordering, path rules, and config usage guidelines.

Part of the MCP integration PR chain (Phase 8 of 10).

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
- Fix all parameter names to match server.py function signatures
- Remove references to non-existent tools
- Ensure every example uses correct parameter names
@aniruddh-alt aniruddh-alt force-pushed the ani/mcp-integration-08-prompt branch from a7df0a6 to c28a572 Compare March 4, 2026 00:50
@aniruddh-alt aniruddh-alt force-pushed the ani/mcp-integration-07-preflight branch from 5a719a8 to 29015a1 Compare March 4, 2026 00:50
aniruddh-alt and others added 20 commits March 4, 2026 16:25
Co-authored-by: Aniruddhan Ramesh <aniruddhanramesh@Aniruddhans-MacBook-Pro.local>
- Remove huggingface-hub and pyyaml (already transitive core deps)
- Add upper bound to mcp dependency (<2) for consistency
- Tighten ruff per-file-ignores to only ASYNC2xx rules
- Add mcp to ci_cpu target so CI tests MCP code

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Pydantic requires typing_extensions.TypedDict (not typing.TypedDict)
on Python < 3.12 when NotRequired fields are present.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Aniruddhan Ramesh and others added 4 commits March 4, 2026 16:28
The license header test requires files to start with the copyright
block. Moved the `# ruff: noqa: E501` directive below the header.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@aniruddh-alt aniruddh-alt marked this pull request as ready for review March 8, 2026 16:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

0