Fix broken pyproj version lookup (#654)

Author: Sushisource

cmd/run_python.go

@@ -19,27 +19,30 @@ func (p *Preparer) BuildPythonProgram(ctx context.Context) (sdkbuild.Program, er
 
 	// Get version from pyproject.toml if not present
 	version := p.config.Version
+	versionFromPyProj := ""
 	if version == "" {
 		b, err := os.ReadFile(filepath.Join(p.rootDir, "pyproject.toml"))
 		if err != nil {
 			return nil, fmt.Errorf("failed reading pyproject.toml: %w", err)
 		}
 		for _, line := range strings.Split(string(b), "\n") {
 			line = strings.TrimSpace(line)
-			if strings.HasPrefix(line, "temporalio = ") {
+			if strings.Contains(line, "temporalio") {
 				version = line[strings.Index(line, `"`)+1 : strings.LastIndex(line, `"`)]
 				break
 			}
 		}
 		if version == "" {
 			return nil, fmt.Errorf("version not found in pyproject.toml")
 		}
+		versionFromPyProj = version
 	}
 
 	prog, err := sdkbuild.BuildPythonProgram(ctx, sdkbuild.BuildPythonProgramOptions{
-		BaseDir: p.rootDir,
-		DirName: p.config.DirName,
-		Version: version,
+		BaseDir:           p.rootDir,
+		DirName:           p.config.DirName,
+		Version:           version,
+		VersionFromPyProj: versionFromPyProj,
 	})
 	if err != nil {
 		return nil, fmt.Errorf("failed preparing: %w", err)

sdkbuild/python.go

@@ -19,6 +19,9 @@ type BuildPythonProgramOptions struct {
 	// a single wheel in the dist directory. Otherwise it is a specific version
 	// (with leading "v" is trimmed if present).
 	Version string
+	// If specified, takes precedence over Version. Is a PEP 508 requirement string, like
+	// `temporalio>=1.13.0,<2`.
+	VersionFromPyProj string
 	// If present, this directory is expected to exist beneath base dir. Otherwise
 	// a temporary dir is created.
 	DirName string
@@ -89,7 +92,9 @@ requires-python = "~=3.9"
 		return nil, fmt.Errorf("failed writing pyproject.toml: %w", err)
 	}
 
-	if strings.ContainsAny(options.Version, `/\`) {
+	if options.VersionFromPyProj != "" {
+		executeCommand("uv", "add", options.VersionFromPyProj)
+	} else if strings.ContainsAny(options.Version, `/\`) {
 		// It's a path; install from wheel
 		wheel, err := getWheel(ctx, options.Version)
 		if err != nil {