fix(collect_info): parse package names safely from requirements constraints (#1313)

* fix(collect_info): parse package names safely from requirements constraints

* chore(collect_info): replace custom requirement parser with packaging.Requirement

* chore(collect_info): improve variable naming when parsing package requirements
This commit is contained in:
Linlang
2025-12-09 17:54:47 +08:00
committed by GitHub
parent d603d5a5aa
commit 99a71bf533
+4 -4
View File
@@ -5,6 +5,7 @@ from pathlib import Path
import docker
import requests
from packaging.requirements import Requirement
from setuptools_scm import get_version
from rdagent.log import rdagent_logger as logger
@@ -69,10 +70,9 @@ def rdagent_info():
]
package_version_list = []
for package in package_list:
if package == "typer[all]":
package = "typer"
version = importlib.metadata.version(package)
package_version_list.append(f"{package}=={version}")
pkg = Requirement(package)
version = importlib.metadata.version(pkg.name)
package_version_list.append(f"{pkg.name}=={version}")
logger.info(f"Package version: {package_version_list}")
return None