Osmosis Apply 1.7b MCP

🚀 CodeMerge

CodeMerge 是一个模型上下文协议（MCP）服务器，它借助 Osmosis-Apply-1.7B 模型来合并代码编辑。其主要价值在于，它能帮助开发者更高效地处理代码合并工作，同时可集成到现有的 AI IDE 解决方案中，提升开发流程的自动化程度。

🚀 快速开始

CodeMerge 可以轻松地集成到你的开发环境中，无论是作为 MCP 服务器使用，还是直接在 Python 中调用，都能为你提供强大的代码合并功能。

✨ 主要特性

• 代码合并：利用 Osmosis/Osmosis-Apply-1.7B 模型应用代码编辑。
• MCP 集成：作为模型上下文协议服务器，可集成到现有的 AI IDE 解决方案中。
• 文件操作：可以直接对文件进行原地编辑，也可以返回编辑后的代码。

📦 安装指南

前提条件

• Python 3.10 或更高版本
• 安装并运行 Ollama
• 拉取 Osmosis 模型：ollama pull Osmosis/Osmosis-Apply-1.7B

安装步骤

1. 克隆仓库：

git clone <repository-url>
cd codemerge

2. 使用 uv 安装依赖：

uv sync

3. 运行：

uv run python codemerge.py

💻 使用示例

作为 MCP 服务器

CodeMerge 可以作为 MCP 服务器使用。在你的 MCP 客户端中进行配置：

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

直接使用

你也可以在 Python 中直接使用 CodeMerge：

from call_snippet import apply_code_edit

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

# 要应用的编辑
edit_snippet = """def hello():
    print('Hello, Universe!')
    print('Some code')
    # ... existing code ...
    return 'success'"""

# 应用编辑
result = apply_code_edit(original_code, edit_snippet)
print(result)

MCP 工具使用

MCP 服务器提供了一个 edit_snippet 工具，具有以下参数：

• original_code：要编辑的原始代码。
• edit_snippet：要应用的编辑，使用 // ... existing code ... 标记。
• file_path：要原地更新的文件的绝对路径。

工具调用示例：

{
  "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"
  }
}

📚 详细文档

编辑格式

CodeMerge 使用特定的编辑格式，在保持简洁的同时保留上下文：

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

// ... existing code ... 标记表示应保留原始代码的位置。这允许你仅指定要更改的部分，同时保持原始代码的精确结构和格式。

🔧 技术细节

依赖项

• httpx>=0.28.1：用于 API 调用的 HTTP 客户端。
• mcp[cli]>=1.10.1：模型上下文协议实现。
• ollama：Ollama 的 Python 客户端（需单独安装）。

架构

CodeMerge 由以下几个组件组成：

• codemerge.py：主要的 MCP 服务器实现。
• call_snippet.py：支持流式处理的直接使用接口。
• test_codemerge.py：全面的测试套件。
• pyproject.toml：项目配置和依赖项。