Skip to content

Commit 639b861

Browse files
[Internal] Extend project creation with a customizable config that can be used in hooks
Review feedback
1 parent 0bba28a commit 639b861

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/dstack/_internal/server/services/projects.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ async def get_project_by_name(
116116
return project_model_to_project(project_model)
117117

118118

119-
class ProjectConfig(CoreModel):
119+
class ProjectHookConfig(CoreModel):
120120
"""
121121
This class can be inherited to extend the project creation configuration passed to the hooks.
122122
"""
@@ -129,7 +129,7 @@ async def create_project(
129129
user: UserModel,
130130
project_name: str,
131131
is_public: bool = False,
132-
config: Optional[ProjectConfig] = None,
132+
config: Optional[ProjectHookConfig] = None,
133133
) -> Project:
134134
user_permissions = users.get_user_permissions(user)
135135
if not user_permissions.can_create_projects:
@@ -620,7 +620,7 @@ def get_member_permissions(member_model: MemberModel) -> MemberPermissions:
620620

621621

622622
def register_create_project_hook(
623-
func: Callable[[AsyncSession, ProjectModel, Optional[ProjectConfig]], Awaitable[None]],
623+
func: Callable[[AsyncSession, ProjectModel, Optional[ProjectHookConfig]], Awaitable[None]],
624624
):
625625
_CREATE_PROJECT_HOOKS.append(func)
626626

0 commit comments

Comments
 (0)