Skip to content

Toolkits for filesystem, bash, and notebook #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Zsailer merged 9 commits into from
Jun 26, 2025
Merged

Toolkits for filesystem, bash, and notebook #3

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
5c8e441
WIP: Updates to match toolkit changes
3coins Jun 19, 2025
89ca3af
Fixed register tool
3coins Jun 20, 2025
a0208c3
WIP: Toolkits for Jupyter
3coins Jun 25, 2025
288c291
Removed nbconvert
3coins Jun 26, 2025
950ed4b
Working version for file_system toolkit
3coins Jun 26, 2025
a4659b4
Updated jupyter ai version
3coins Jun 26, 2025
c8dd607
Fixed lint errors
3coins Jun 26, 2025
54c096f
lint and fmt
3coins Jun 26, 2025
db1ab46
fmt
3coins Jun 26, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 19 additions & 3 deletions jupyter_ai_tools/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from .extension import jupyter_server_extension_tools

__all__ = ["jupyter_server_extension_tools"]
from .extension import get_git_tools, get_notebook_tools
from jupyter_server_ai_tools.models import ToolSet, Toolkit

__version__ = "0.1.2"

Expand All @@ -11,3 +10,20 @@ def _jupyter_server_extension_points():

def _load_jupyter_server_extension(serverapp):
serverapp.log.info("✅ jupyter_ai_tools extension loaded.")

async def _start_jupyter_server_extension(serverapp):
registry = serverapp.web_app.settings["toolkit_registry"]
if registry:
notebook_tools = ToolSet(get_notebook_tools())
registry.register_toolkit(
Toolkit(
name="notebook_toolkit", tools=notebook_tools
)
)

git_tools = ToolSet(get_git_tools())
registry.register_toolkit(
Toolkit(
name="git_toolkit", tools=git_tools
)
)
259 changes: 22 additions & 237 deletions jupyter_ai_tools/extension.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,241 +1,26 @@
from jupyter_server_ai_tools.models import ToolDefinition
from jupyter_server_ai_tools.models import Tool

from . import git_tools, ynotebook_tools


def jupyter_server_extension_tools():
return [
ToolDefinition(
callable=ynotebook_tools.delete_cell,
metadata={
"name": "delete_cell",
"description": "Remove the cell at the specified index and return its contents.",
"inputSchema": {
"type": "object",
"properties": {
"index": {
"type": "integer",
"description": "The index of the cell to delete",
}
},
"required": ["index"],
},
},
),
ToolDefinition(
callable=ynotebook_tools.add_cell,
metadata={
"name": "add_cell",
"description": "Insert a blank cell at the specified index.",
"inputSchema": {
"type": "object",
"properties": {
"index": {"type": "integer", "description": "The index to insert at"},
"cell_type": {
"type": "string",
"description": "The type of cell: 'code' or 'markdown' ",
"default": "code",
},
},
"required": ["index"],
},
},
),
ToolDefinition(
callable=ynotebook_tools.write_to_cell,
metadata={
"name": "write_to_cell",
"description": "Overwrite the source of a cell with content at the given index "
"in the notebook.",
"inputSchema": {
"type": "object",
"properties": {
"index": {"type": "integer", "description": "The index to write at"},
"content": {
"type": "string",
"description": "The content to write into the cell, either python "
"code or markdown",
},
},
"required": ["index", "content"],
},
},
),
ToolDefinition(
callable=ynotebook_tools.get_max_cell_index,
metadata={
"name": "get_max_cell_index",
"description": "Return the highest valid cell index in the current notebook.",
"inputSchema": {"type": "object", "properties": {}},
},
),
ToolDefinition(
callable=ynotebook_tools.read_cell,
metadata={
"name": "read_cell",
"description": "Read the full content of a specific cell, including outputs, "
"source, and metadata.",
"inputSchema": {
"type": "object",
"properties": {
"index": {"type": "integer", "description": "The index of the cell to read"}
},
"required": ["index"],
},
},
),
ToolDefinition(
callable=ynotebook_tools.read_notebook,
metadata={
"name": "read_notebook",
"description": "Return all cells in the notebook as a JSON-formatted list.",
"inputSchema": {"type": "object", "properties": {}},
},
),
ToolDefinition(
callable=git_tools.git_clone,
metadata={
"name": "git_clone",
"description": "Clone a Git repo into the specified path.",
"inputSchema": {
"type": "object",
"properties": {
"path": {"type": "string", "description": "Target path"},
"url": {"type": "string", "description": "Repository URL"},
},
"required": ["path", "url"],
},
},
),
ToolDefinition(
callable=git_tools.git_status,
metadata={
"name": "git_status",
"description": "Get the current Git status in the specified path.",
"inputSchema": {
"type": "object",
"properties": {
"path": {
"type": "string",
"description": "Path to the Git repository root directory",
}
},
"required": ["path"],
},
},
),
ToolDefinition(
callable=git_tools.git_log,
metadata={
"name": "git_log",
"description": "Get the last N Git commits.",
"inputSchema": {
"type": "object",
"properties": {
"path": {
"type": "string",
"description": "Path to the Git repository root directory",
},
"history_count": {
"type": "integer",
"description": "Number of commits",
"default": 10,
},
},
"required": ["path"],
},
},
),
ToolDefinition(
callable=git_tools.git_pull,
metadata={
"name": "git_pull",
"description": "Pull the latest changes from the remote.",
"inputSchema": {
"type": "object",
"properties": {
"path": {
"type": "string",
"description": "Path to the Git repository root directory",
}
},
"required": ["path"],
},
},
),
ToolDefinition(
callable=git_tools.git_push,
metadata={
"name": "git_push",
"description": "Push local changes to the remote.",
"inputSchema": {
"type": "object",
"properties": {
"path": {
"type": "string",
"description": "Path to the Git repository root directory",
},
"branch": {"type": "string", "description": "Repo branch"},
},
"required": ["path"],
},
},
),
ToolDefinition(
callable=git_tools.git_commit,
metadata={
"name": "git_commit",
"description": "Commit staged changes with a message.",
"inputSchema": {
"type": "object",
"properties": {
"path": {
"type": "string",
"description": "Path to the Git repository root directory",
},
"message": {"type": "string", "description": "Commit message"},
},
"required": ["path", "message"],
},
},
),
ToolDefinition(
callable=git_tools.git_add,
metadata={
"name": "git_add",
"description": "Stage files for commit. Optionally add all files.",
"inputSchema": {
"type": "object",
"properties": {
"path": {
"type": "string",
"description": "Path to the Git repository root directory",
},
"add_all": {
"type": "boolean",
"default": True,
"description": "Stage all files",
},
"filename": {
"type": "string",
"description": "File to add (used if add_all is false)",
},
},
"required": ["path"],
},
},
),
ToolDefinition(
callable=git_tools.git_get_repo_root,
metadata={
"name": "git_get_repo_root_from_notebookpath",
"description": "Given the path of a file, return the path to the Repo root"
" if any.",
"inputSchema": {
"type": "object",
"properties": {"path": {"type": "string", "description": "the path to a file"}},
"required": ["path"],
},
},
),
]
def get_notebook_tools():
return {
Tool(callable=ynotebook_tools.delete_cell, delete=True),
Tool(callable=ynotebook_tools.add_cell, write=True),
Tool(callable=ynotebook_tools.write_to_cell, read=True, write=True),
Tool(callable=ynotebook_tools.get_max_cell_index, read=True),
Tool(callable=ynotebook_tools.read_cell, read=True),
Tool(callable=ynotebook_tools.read_notebook, read=True)
}

def get_git_tools():
return {
Tool(callable=git_tools.git_clone, write=True),
Tool(callable=git_tools.git_status, read=True),
Tool(callable=git_tools.git_log, read=True),
Tool(callable=git_tools.git_pull, read=True, write=True),
Tool(callable=git_tools.git_push, read=True, write=True),
Tool(callable=git_tools.git_commit, write=True),
Tool(callable=git_tools.git_add, write=True),
Tool(callable=git_tools.git_get_repo_root, read=True)
}
Loading