Add to root member

crates/uv-workspace/src/workspace.rs

@@ -674,6 +674,17 @@ impl ProjectWorkspace {
         &self.workspace().packages[&self.project_name]
     }
 
+    /// Returns the workspace root member as a [`WorkspaceMember`].
+    pub fn root_member(&self) -> &WorkspaceMember {
+        self.workspace().packages.iter().find_map(|(_, member)| {
+            if member.root() == self.workspace().install_path() {
+                Some(member)
+            } else {
+                None
+            }
+        }).expect("workspace root member must exist")
+    }
+
     /// Find the workspace for a project.
     pub async fn from_project(
         install_path: &Path,

crates/uv/src/commands/project/init.rs

@@ -111,23 +111,21 @@ pub(crate) async fn init(
 
     if let Some(workspace) = workspace {
         // Add the package to the workspace.
-        let mut pyproject =
-            PyProjectTomlMut::from_toml(workspace.current_project().pyproject_toml())?;
-        pyproject.add_workspace(path.strip_prefix(workspace.project_root())?)?;
+        let root_member = workspace.root_member();
+        let mut pyproject = PyProjectTomlMut::from_toml(root_member.pyproject_toml())?;
+        pyproject.add_workspace(path.strip_prefix(root_member.root())?)?;
 
         // Save the modified `pyproject.toml`.
         fs_err::write(
-            workspace.current_project().root().join("pyproject.toml"),
+            root_member.root().join("pyproject.toml"),
             pyproject.to_string(),
         )?;
 
         writeln!(
             printer.stderr(),
             "Adding {} as member of workspace {}",
             name.cyan(),
-            workspace
-                .workspace()
-                .install_path()
+            root_member.root()
                 .simplified_display()
                 .cyan()
         )?;