Skip to content

Commit 4dd2a20

Browse files
committed
fix: correct pyyaml module name in dependency check
1 parent d67fa75 commit 4dd2a20

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

check_production_ready.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -24,19 +24,19 @@ def check_dependencies():
2424
"""检查依赖包"""
2525
print("\n📦 Checking dependencies...")
2626
required_packages = [
27-
'litellm',
28-
'pyyaml',
29-
'asyncio',
27+
('litellm', 'litellm'),
28+
('pyyaml', 'yaml'), # pyyaml 的模块名是 yaml
29+
('asyncio', 'asyncio'),
3030
]
3131

3232
missing = []
33-
for package in required_packages:
33+
for package_name, module_name in required_packages:
3434
try:
35-
__import__(package.replace('-', '_'))
36-
print(f" ✅ {package}")
35+
__import__(module_name)
36+
print(f" ✅ {package_name}")
3737
except ImportError:
38-
print(f" ❌ {package} (missing)")
39-
missing.append(package)
38+
print(f" ❌ {package_name} (missing)")
39+
missing.append(package_name)
4040

4141
if missing:
4242
print(f"\n Install missing packages: pip install {' '.join(missing)}")

0 commit comments

Comments
 (0)