From ee5ed7aeeed2ec8c1777c14d9e49e20b9f0682e0 Mon Sep 17 00:00:00 2001 From: ytqh Date: Sun, 9 Mar 2025 11:31:10 +0800 Subject: [PATCH] change log --- api/schedule/user_memory_generate_task.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/api/schedule/user_memory_generate_task.py b/api/schedule/user_memory_generate_task.py index de09a7e0d9..d49efabb88 100644 --- a/api/schedule/user_memory_generate_task.py +++ b/api/schedule/user_memory_generate_task.py @@ -134,7 +134,7 @@ def process_user(user: EndUser, memory_app_id: str, app_id: str): if user.memory_updated_at: message_query = message_query.filter(Message.created_at > user.memory_updated_at) - latest_messages = message_query.order_by(asc(Message.created_at)).limit(10).all() + latest_messages = message_query.order_by(asc(Message.created_at)).all() # Skip if no messages found (unlikely due to our query, but just to be safe) if not latest_messages: @@ -163,15 +163,12 @@ def process_user(user: EndUser, memory_app_id: str, app_id: str): } } - click.echo(click.style(f"Args: {args}", fg="green")) - # Call the memory generation service + click.echo(click.style(f"Start to generate memory for user {user.id} in app {app_id}", fg="green")) response = AppGenerateService.generate( app_model=memory_app_model, user=user, args=args, invoke_from=InvokeFrom.SCHEDULER, streaming=False ) - click.echo(click.style(f"Response: {response}, type: {type(response)}", fg="green")) - # Save the updated memory to the user if ( response @@ -182,6 +179,6 @@ def process_user(user: EndUser, memory_app_id: str, app_id: str): ): user.memory = response["data"]["outputs"]["result"] user.memory_updated_at = latest_messages[-1].created_at - click.echo(click.style(f"Updated memory for user {user.id}", fg="green")) + click.echo(click.style(f"Updated memory for user {user.id} in app {app_id}", fg="green")) else: - click.echo(click.style(f"Failed to update memory for user {user.id}, invalid response", fg="yellow")) + click.echo(click.style(f"Failed to update memory for user {user.id}, invalid response format", fg="yellow"))