Skip to content

[MCP Integration] Phase 1: Package scaffold and dependencies#2234

Open
aniruddh-alt wants to merge 8 commits intomainfrom
ani/mcp-integration-01-scaffold
Open

[MCP Integration] Phase 1: Package scaffold and dependencies#2234
aniruddh-alt wants to merge 8 commits intomainfrom
ani/mcp-integration-01-scaffold

Conversation

@aniruddh-alt
Copy link
Contributor

Description

Part of the MCP Integration PR chain (Phase 1 of 10) - Stage: scaffold

What changed: Added the oumi.mcp package skeleton with optional dependencies (fastmcp, httpx, huggingface-hub, mcp, pyyaml), the oumi-mcp console entry point, and ruff lint ignores.

Why: This establishes the package structure and dependency declarations needed for the MCP server integration.

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?

Add the oumi.mcp package skeleton with optional dependencies (fastmcp,
httpx, huggingface-hub, mcp, pyyaml), the oumi-mcp console entry point,
and ruff lint ignores for the MCP server code.

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

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@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.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@aniruddh-alt aniruddh-alt force-pushed the ani/mcp-integration-01-scaffold branch from 70e68d6 to b35dc2d Compare March 4, 2026 00:31
- Remove unnecessary mcp[cli] extra (only fastmcp is used)
- Add upper-bound pins to httpx and huggingface-hub deps
- Narrow ASYNC ruff suppression to specific rules (210, 220, 230, 240)
@aniruddh-alt aniruddh-alt force-pushed the ani/mcp-integration-01-scaffold branch from b35dc2d to 79de029 Compare March 4, 2026 00:41
aniruddh-alt and others added 4 commits March 4, 2026 09:39
- 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>
@@ -0,0 +1,22 @@
# Copyright 2025 - Oumi
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

2026*

@aniruddh-alt aniruddh-alt marked this pull request as ready for review March 8, 2026 16:51

[project.scripts]
oumi = "oumi.cli.main:run"
oumi-mcp = "oumi.mcp.server:main"
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Bug: The oumi-mcp entry point in pyproject.toml points to oumi.mcp.server:main, but this module and function do not exist, causing a ModuleNotFoundError at runtime.
Severity: HIGH

Suggested Fix

Either remove the oumi-mcp entry point from pyproject.toml and add it in a future PR that implements the oumi.mcp.server:main function, or create a placeholder server.py file with a main function that provides a meaningful error message.

Prompt for AI Agent
Review the code at the location below. A potential bug has been identified by an AI
agent.
Verify if this is a real issue. If it is, propose a fix; if not, explain why it's not
valid.

Location: pyproject.toml#L215

Potential issue: The `pyproject.toml` file defines a command-line entry point `oumi-mcp`
which is configured to execute the `main` function within the `oumi.mcp.server` module.
However, the `oumi.mcp.server` module does not exist in the codebase. As a result, any
user who installs the package with the `mcp` extra and attempts to run the `oumi-mcp`
command will encounter an immediate `ModuleNotFoundError` because the specified module
cannot be found, making the feature unusable.

Did we get this right? 👍 / 👎 to inform future reviews.

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