Project Files

.pydev-tools

_ast

describe.py

do_work.py

docstrings.py

extract_function.py

extract.py

legacy_handler_2.py

legacy_handler_3.py

legacy_handler.py

migration.py

reference.py

rename.py

workspace_description.md

prompt_examples

dependency_version_migration.txt

hotfix_rework.txt

new_project.txt

performance_optimization.txt

refactory.txt

security_hardening.txt

source_tree_analysis.txt

test_coverage_expansion.txt

scripts

e2e_runPytest_test.cjs

setupVenv.mjs

smokeNew6Tools.ts

smokeNewTools.ts

verify_testfile_fix.ts

verifySetupVenv.ts

skills

pydev-mcp.md

src

tools

_profiling_worker_main.py

analyzeImportsAndDependencies.ts

buildPythonPackage.ts

checkForBugs.ts

checkForBugsInFile.ts

createProjectStructure.ts

createTestFileTool.ts

editTextFile.ts

editTextFileByLine.ts

findFiles.ts

generateRequirementsTxtTool.ts

getSetupStatusTool.ts

getWorkspaceRootTool.ts

inspectEnvironmentTool.ts

installModule.ts

listDirectory.ts

listTools.ts

profileProject.ts

pydevAuditDocstringsTool.ts

pydevDescribeWorkspaceTool.ts

pydevExtractFunctionTool.ts

pydevGenerateReferenceTool.ts

pydevMigrationAuditTool.ts

pydevSafeRenameTool.ts

readTextFile.ts

runLinterAndFormatter.ts

runLinterAndFormatterInFile.ts

runProjectCoverage.ts

runPython.ts

runPythonFile.ts

runPythonFileInteractive.ts

runPythonInteractive.ts

runRepl.ts

runTests.ts

runWithDebuggerTool.ts

saveTextFile.ts

scanProjectSecurity.ts

searchDirectory.ts

setupVenvTool.ts

switchPythonVersion.ts

typeCheckProjectTool.ts

uninstallModule.ts

config.ts

index.ts

test

run_tests.cjs

tools.test.cjs

manifest.json

package-lock.json

package.json

professor_headache.jpg

README.md

setup-venv.sh

tsconfig.json

src / tools / setupVenvTool.ts

import { tool } from "@lmstudio/sdk";
import { z } from "zod";

import { getToolDefinition } from "../utils/toolRegistry";
import { setupVenv, type VenvSetupResult } from "../utils/pythonVenv";
import { getWorkspaceRoot } from "../utils/safePaths";

const TOOL_DEFINITION = getToolDefinition("pydev_setup_venv")!;

export const setupVenvTool = tool({
  name: TOOL_DEFINITION!.name,
  description: TOOL_DEFINITION!.description,
  parameters: {
    projectPath: z
      .string()
      .min(1)
      .max(500)
      .optional()
      .describe(
        "Deprecated / ignored. Per the single-workspace model, the venv is always created at the " +
          "workspace root's .venv so that every project scaffolded by pydev_create_project_structure " +
          "shares one environment.",
      ),
  },
  implementation: async () => {
    // The venv is ALWAYS installed at <workspaceRoot>/.venv. getWorkspaceRoot() honors the
    // user-set path → persisted default (.pydev-mcp-config.json) → env var → process.cwd().
    const workspaceRoot = getWorkspaceRoot();

    try {
      return await setupVenv(workspaceRoot);
    } catch (err) {
      const result: VenvSetupResult = {
        success: false,
        error: err instanceof Error ? err.message : String(err),
        venvPath: null,
      };
      return result;
    }
  },
});