diff --git a/requirements_generator/ui/prompts.py b/requirements_generator/ui/prompts.py index c8cbc2c..7545fa7 100644 --- a/requirements_generator/ui/prompts.py +++ b/requirements_generator/ui/prompts.py @@ -194,7 +194,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 @@ -358,7 +358,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