mirror of
https://github.com/Pythagora-io/gpt-pilot.git
synced 2026-01-10 13:37:55 -05:00
Fix re-entering database session
This commit is contained in:
@@ -84,20 +84,16 @@ class StateManager:
|
||||
return await Project.get_all_projects_old(session)
|
||||
|
||||
async def get_project_states(self, project_id: Optional[UUID], branch_id: Optional[UUID]) -> list[ProjectState]:
|
||||
async with self.session_manager as session:
|
||||
return await ProjectState.get_project_states(session, project_id, branch_id)
|
||||
return await ProjectState.get_project_states(self.current_session, project_id, branch_id)
|
||||
|
||||
async def get_branches_for_project_id(self, project_id: UUID) -> list[Branch]:
|
||||
async with self.session_manager as session:
|
||||
return await Project.get_branches_for_project_id(session, project_id)
|
||||
return await Project.get_branches_for_project_id(self.current_session, project_id)
|
||||
|
||||
async def find_user_input(self, project_state, branch_id) -> Optional[list["UserInput"]]:
|
||||
async with self.session_manager as session:
|
||||
return await UserInput.find_user_inputs(session, project_state, branch_id)
|
||||
return await UserInput.find_user_inputs(self.current_session, project_state, branch_id)
|
||||
|
||||
async def get_file_for_project(self, state_id: UUID, path: str):
|
||||
async with self.session_manager as session:
|
||||
return await Project.get_file_for_project(session, state_id, path)
|
||||
return await Project.get_file_for_project(self.current_session, state_id, path)
|
||||
|
||||
async def create_project(self, name: str, folder_name: Optional[str] = None) -> Project:
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user