From fcfeac57c53b16c149e2da038a0b212b425f029b Mon Sep 17 00:00:00 2001
From: Andre van Tonder <andre@drevan.me>
Date: Sun, 31 Aug 2025 14:53:03 +1000
Subject: [PATCH] fix: resolve virtual envs for python LSP (#2155)

Co-authored-by: rekram1-node <aidenpcline@gmail.com>
---
 packages/opencode/src/lsp/client.ts |  1 +
 packages/opencode/src/lsp/server.ts | 18 ++++++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/packages/opencode/src/lsp/client.ts b/packages/opencode/src/lsp/client.ts
index a03a2651..c6ccfbb0 100644
--- a/packages/opencode/src/lsp/client.ts
+++ b/packages/opencode/src/lsp/client.ts
@@ -60,6 +60,7 @@ export namespace LSPClient {
       return null
     })
     connection.onRequest("workspace/configuration", async () => {
+      // Return server initialization options
       return [input.server.initialization ?? {}]
     })
     connection.listen()
diff --git a/packages/opencode/src/lsp/server.ts b/packages/opencode/src/lsp/server.ts
index da743628..9861a579 100644
--- a/packages/opencode/src/lsp/server.ts
+++ b/packages/opencode/src/lsp/server.ts
@@ -298,6 +298,23 @@ export namespace LSPServer {
         args.push(...["run", js])
       }
       args.push("--stdio")
+
+      const initialization: Record<string, string> = {}
+
+      const potentialVenvPaths = [process.env["VIRTUAL_ENV"], path.join(root, ".venv"), path.join(root, "venv")].filter(
+        (p): p is string => p !== undefined,
+      )
+      for (const venvPath of potentialVenvPaths) {
+        const isWindows = process.platform === "win32"
+        const potentialPythonPath = isWindows
+          ? path.join(venvPath, "Scripts", "python.exe")
+          : path.join(venvPath, "bin", "python")
+        if (await Bun.file(potentialPythonPath).exists()) {
+          initialization["pythonPath"] = potentialPythonPath
+          break
+        }
+      }
+
       const proc = spawn(binary, args, {
         cwd: root,
         env: {
@@ -307,6 +324,7 @@ export namespace LSPServer {
       })
       return {
         process: proc,
+        initialization,
       }
     },
   }