Skip to content

Commit bb63d7a

Browse files
authored
Feat:add time step (#327)
* fix time * add timed step for llm * fix:ci code * fix:test * remove * update
1 parent ee89e68 commit bb63d7a

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/memos/api/routers/product_router.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,11 +80,15 @@ def register_user(user_req: UserRegisterRequest):
8080
"""Register a new user with configuration and default cube."""
8181
try:
8282
# Get configuration for the user
83+
time_start_register = time.time()
8384
user_config, default_mem_cube = APIConfig.create_user_config(
8485
user_name=user_req.user_id, user_id=user_req.user_id
8586
)
8687
logger.info(f"user_config: {user_config.model_dump(mode='json')}")
8788
logger.info(f"default_mem_cube: {default_mem_cube.config.model_dump(mode='json')}")
89+
logger.info(
90+
f"time register api : create user config time user_id: {user_req.user_id} time is: {time.time() - time_start_register}"
91+
)
8892
mos_product = get_mos_product_instance()
8993

9094
# Register user with default config and mem cube
@@ -96,7 +100,9 @@ def register_user(user_req: UserRegisterRequest):
96100
default_mem_cube=default_mem_cube,
97101
mem_cube_id=user_req.mem_cube_id,
98102
)
99-
103+
logger.info(
104+
f"time register api : register time user_id: {user_req.user_id} time is: {time.time() - time_start_register}"
105+
)
100106
if result["status"] == "success":
101107
return UserRegisterResponse(
102108
message="User registered successfully",

0 commit comments

Comments
 (0)