feat: fix memory

pull/21891/head
ytqh 1 year ago
parent c7125a6152
commit 21b8d76077

@ -1,14 +1,13 @@
import time import time
from datetime import datetime from datetime import datetime
import click
from sqlalchemy import asc, func, or_
import app import app
import click
from configs import dify_config from configs import dify_config
from core.app.entities.app_invoke_entities import InvokeFrom from core.app.entities.app_invoke_entities import InvokeFrom
from models.model import App, EndUser, Message, db from models.model import App, EndUser, Message, db
from services.app_generate_service import AppGenerateService from services.app_generate_service import AppGenerateService
from sqlalchemy import asc, func, or_
@app.celery.task(queue="dataset") @app.celery.task(queue="dataset")
@ -194,7 +193,6 @@ def process_user_memory(user: EndUser, new_messages: str):
# Call the memory generation service # Call the memory generation service
click.echo(click.style(f"Start to generate memory for user {user.id}", fg="green")) click.echo(click.style(f"Start to generate memory for user {user.id}", fg="green"))
try:
response = AppGenerateService.generate( response = AppGenerateService.generate(
app_model=memory_app_model, app_model=memory_app_model,
user=user, user=user,
@ -212,12 +210,6 @@ def process_user_memory(user: EndUser, new_messages: str):
user.memory = result user.memory = result
click.echo(click.style(f"Updated memory for user {user.id}", fg="green")) click.echo(click.style(f"Updated memory for user {user.id}", fg="green"))
except Exception as e:
click.echo(
click.style(
f"Failed to update memory for user {user.id}, {str(e)}", fg="yellow"
)
)
def process_user_health_summary(user: EndUser, new_messages: str): def process_user_health_summary(user: EndUser, new_messages: str):

Loading…
Cancel
Save