Osmosis Apply 1.7b MCP

🚀 CodeMerge

A Model Context Protocol (MCP) server that uses the Osmosis-Apply-1.7B model to merge code edits.

CodeMerge leverages the Osmosis/Osmosis-Apply-1.7B model to seamlessly merge code edits. It integrates with the Model Context Protocol, enabling smooth incorporation into existing AI IDE solutions. Additionally, it supports direct file editing and can return edited code.

🚀 Quick Start

CodeMerge is a powerful tool for code merging. Below are the steps to get started with it.

✨ Features

• Code Merging: Utilizes the Osmosis/Osmosis-Apply-1.7B model to apply edits.
• MCP Integration: Functions as a Model Context Protocol server for integration into existing AI IDE solutions.
• File Operations: Capable of directly editing files in place or returning edited code.

📦 Installation

Prerequisites

• Python 3.10 or higher
• Ollama installed and running
• The Osmosis model pulled: ollama pull Osmosis/Osmosis-Apply-1.7B

Setup

1. Clone the repository:

git clone <repository-url>
cd codemerge

2. Install dependencies using uv:

uv sync

3. Run:

uv run python codemerge.py

💻 Usage Examples

As an MCP Server

CodeMerge can serve as an MCP server. Configure it in your MCP client:

{
  "mcpServers": {
    "codemerge": {
      "command": "uv",
      "args": [
        "--directory",
        "/path/to/codemerge",
        "run",
        "codemerge.py"
      ]
    }
  }
}

Direct Usage

You can also use CodeMerge directly in Python:

from call_snippet import apply_code_edit

# Original code
original_code = """def hello():
    print('Hello, World!')
    print('Some code')
    print('Some more code')
    return 'success'"""

# Edit to apply
edit_snippet = """def hello():
    print('Hello, Universe!')
    print('Some code')
    # ... existing code ...
    return 'success'"""

# Apply the edit
result = apply_code_edit(original_code, edit_snippet)
print(result)

MCP Tool Usage

The MCP server offers an edit_snippet tool with the following parameters:

• original_code: The exact original code to be edited.
• edit_snippet: The edit to apply, using // ... existing code ... markers.
• file_path: Absolute path to a file to update in place.

Example tool call:

{
  "name": "edit_snippet",
  "arguments": {
    "original_code": "def hello():\n    print('Hello, World!')",
    "edit_snippet": "def hello():\n    print('Hello, Universe!')\n    # ... existing code ...",
    "file_path": "/path/to/your/file.py"
  }
}

📚 Documentation

Edit Format

CodeMerge employs a specific format for edits that preserves context while being concise:

// ... existing code ...
FIRST_EDIT
// ... existing code ...
SECOND_EDIT
// ... existing code ...
THIRD_EDIT
// ... existing code ...

The // ... existing code ... markers indicate where the original code should be preserved. This allows you to specify only the parts you want to change while maintaining the exact structure and formatting of the original code.

Dependencies

• httpx>=0.28.1: HTTP client for API calls.
• mcp[cli]>=1.10.1: Model Context Protocol implementation.
• ollama: Python client for Ollama (installed separately).

Architecture

CodeMerge consists of several components:

• codemerge.py: Main MCP server implementation.
• call_snippet.py: Direct usage interface with streaming support.
• test_codemerge.py: Comprehensive test suite.
• pyproject.toml: Project configuration and dependencies.