python-tools

import { type PluginContext } from "@lmstudio/sdk";

import { checkForBugsTool } from "./tools/checkForBugs";
import { checkForBugsInFileTool } from "./tools/checkForBugsInFile";
import { editPythonFileTool } from "./tools/editPythonFile";
import { editPythonFileByLineTool } from "./tools/editPythonFileByLine";
import { installModuleTool } from "./tools/installModule";
import { listDirectoryTool } from "./tools/listDirectory";
import { listToolsTool } from "./tools/listTools";
import { readPythonFileTool } from "./tools/readPythonFile";
import { runPythonTool } from "./tools/runPython";
import { runPythonFileTool } from "./tools/runPythonFile";
import { runPythonFileInteractiveTool } from "./tools/runPythonFileInteractive";
import { runPythonInteractiveTool } from "./tools/runPythonInteractive";
import { savePythonFileTool } from "./tools/savePythonFile";
import { switchPythonVersionTool } from "./tools/switchPythonVersion";
import { uninstallModuleTool } from "./tools/uninstallModule";

export async function main(context: PluginContext) {
  context.withToolsProvider(async () => {
    return [
      runPythonTool,
      runPythonFileTool,
      runPythonInteractiveTool,
      runPythonFileInteractiveTool,
      installModuleTool,
      uninstallModuleTool,
      switchPythonVersionTool,
      savePythonFileTool,
      readPythonFileTool,
      listDirectoryTool,
      editPythonFileTool,
      editPythonFileByLineTool,
      checkForBugsTool,
      checkForBugsInFileTool,
      listToolsTool,
    ];
  });
}