diff --git a/requirements_generator/ui/prompts.py b/requirements_generator/ui/prompts.py index 9ca3c04..c8cbc2c 100644 --- a/requirements_generator/ui/prompts.py +++ b/requirements_generator/ui/prompts.py @@ -30,7 +30,7 @@ def select_project(prompt_text: str = "请选择项目 ID") -> Optional[Project] except ValueError: console.print("[red]ID 必须为整数[/red]") return None - project = db.get_project(pid) + project = db.get_project_by_id(pid) if not project: console.print(f"[red]项目 ID={pid} 不存在[/red]") return None