Py.typed

python/uv/__init__.py

@@ -0,0 +1,8 @@
+"""
+The Python API of `uv` is not guaranteed to be stable and may chance at any time. It is 
+strongly discouraged to import from our `uv` Python module, but you can do so at your 
+own risk.
+"""
+from uv.__main__ import detect_virtualenv, find_uv_bin
+
+__all__ = ["detect_virtualenv", "find_uv_bin"]

python/uv/__main__.py

@@ -1,35 +1,17 @@
 import os
 import sys
 import sysconfig
-
-
-def detect_virtualenv() -> str:
-    """
-    Find the virtual environment path for the current Python executable.
-    """
-
-    # If it's already set, then just use it
-    value = os.getenv("VIRTUAL_ENV")
-    if value:
-        return value
-
-    # Otherwise, check if we're in a venv
-    venv_marker = os.path.join(sys.prefix, "pyvenv.cfg")
-
-    if os.path.exists(venv_marker):
-        return sys.prefix
-
-    return ""
+from pathlib import Path
 
 
 def find_uv_bin() -> str:
     """Return the uv binary path."""
 
     uv_exe = "uv" + sysconfig.get_config_var("EXE")
 
-    path = os.path.join(sysconfig.get_path("scripts"), uv_exe)
-    if os.path.isfile(path):
-        return path
+    path = Path(sysconfig.get_path("scripts")) / uv_exe
+    if path.is_file():
+        return str(path)
 
     if sys.version_info >= (3, 10):
         user_scheme = sysconfig.get_preferred_scheme("user")
@@ -40,11 +22,31 @@ def find_uv_bin() -> str:
     else:
         user_scheme = "posix_user"
 
-    path = os.path.join(sysconfig.get_path("scripts", scheme=user_scheme), uv_exe)
-    if os.path.isfile(path):
-        return path
+    path = Path(sysconfig.get_path("scripts", scheme=user_scheme)) / uv_exe
+    if path.is_file():
+        return str(path)
 
-    raise FileNotFoundError(path)
+    msg = str(path)
+    raise FileNotFoundError(msg)
+
+
+def detect_virtualenv() -> str:
+    """
+    Find the virtual environment path for the current Python executable.
+    """
+
+    # If it's already set, then just use it
+    value = os.getenv("VIRTUAL_ENV")
+    if value:
+        return value
+
+    # Otherwise, check if we're in a venv
+    venv_marker = Path(sys.prefix) / "pyvenv.cfg"
+
+    if venv_marker.exists():
+        return sys.prefix
+
+    return ""
 
 
 if __name__ == "__main__":
@@ -58,7 +60,7 @@ def find_uv_bin() -> str:
     if sys.platform == "win32":
         import subprocess
 
-        completed_process = subprocess.run([uv, *sys.argv[1:]], env=env)
+        completed_process = subprocess.run([uv, *sys.argv[1:]], env=env, check=False)
         sys.exit(completed_process.returncode)
     else:
         os.execvpe(uv, [uv, *sys.argv[1:]], env=env)

python/uv/py.typed

@@ -0,0 +1 @@
+