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：項目配置和依賴項。