Jupyter Notebook Agent

機械学習タスクのGitHub上の実装のJupyter Notebookを作成してくれるエージェントです。 Notebookを自動で実行して、エラーが発生した場合は繰り返し修正してくれます。 Claude Agent SDK for Pythonを使用して、Claude Code経由でClaudeとやり取りすることを前提としています。

必要な環境

• Python 3.10以上
• Node.js
• Claude Code 2.0.0以上

インストール

# Claude Code のインストール
npm install -g @anthropic-ai/claude-code

# uvのインストール
curl -LsSf https://astral.sh/uv/install.sh | sh

# Python パッケージのインストール
uv sync

使用方法

# 基本的な使い方
usage: notebook_agent.py [-h] [--max-attempts MAX_ATTEMPTS] yaml_file

YAMLファイルに記載されたGitHubリポジトリをJupyter Notebookに実装する

positional arguments:
  yaml_file             リポジトリURLリストが記載されたYAMLファイルのパス

options:
  -h, --help            show this help message and exit
  --max-attempts MAX_ATTEMPTS
                        Notebookのエラー修正最大回数

# 例：semantic segmentationのawesome listsを実装してもらう
uv run notebook_agent.py repos.yaml --max-attempts 3

参考リンク

• Claude Agent SDK Documentation
• Claude Code Documentation