Skip to content

Commit 6b8cf85

Browse files
authored
Feat:add self defined memcube id for reg user (#295)
* Chore: Change version to v1.0.1 * feat: add mem_cube id for user reg
1 parent 99d8e19 commit 6b8cf85

File tree

3 files changed

+7
-1
lines changed

3 files changed

+7
-1
lines changed

src/memos/api/product_models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ class UserRegisterRequest(BaseRequest):
4242
user_id: str = Field(
4343
default_factory=lambda: str(uuid.uuid4()), description="User ID for registration"
4444
)
45+
mem_cube_id: str | None = Field(None, description="Cube ID for registration")
4546
user_name: str | None = Field(None, description="User name for registration")
4647
interests: str | None = Field(None, description="User interests")
4748

src/memos/api/routers/product_router.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ def register_user(user_req: UserRegisterRequest, g: Annotated[G, Depends(get_g_o
106106
interests=user_req.interests,
107107
config=user_config,
108108
default_mem_cube=default_mem_cube,
109+
mem_cube_id=user_req.mem_cube_id,
109110
)
110111

111112
if result["status"] == "success":

src/memos/mem_os/product.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -801,6 +801,7 @@ def user_register(
801801
interests: str | None = None,
802802
default_mem_cube: GeneralMemCube | None = None,
803803
default_cube_config: GeneralMemCubeConfig | None = None,
804+
mem_cube_id: str | None = None,
804805
) -> dict[str, str]:
805806
"""Register a new user with configuration and default cube.
806807
@@ -836,7 +837,10 @@ def user_register(
836837
default_cube_name = f"{user_name}_{user_id}_default_cube"
837838
mem_cube_name_or_path = os.path.join(CUBE_PATH, default_cube_name)
838839
default_cube_id = self.create_cube_for_user(
839-
cube_name=default_cube_name, owner_id=user_id, cube_path=mem_cube_name_or_path
840+
cube_name=default_cube_name,
841+
owner_id=user_id,
842+
cube_path=mem_cube_name_or_path,
843+
cube_id=mem_cube_id,
840844
)
841845

842846
if default_mem_cube:

0 commit comments

Comments
 (0)