feat: add logger

pull/21891/head
ytqh 1 year ago
parent 9845b53113
commit d451340128

@ -1,3 +1,4 @@
import logging
import time import time
from datetime import datetime from datetime import datetime
@ -10,12 +11,16 @@ 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
logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
@app.celery.task(queue="dataset") @app.celery.task(queue="dataset")
def user_profile_generate_task(): def user_profile_generate_task():
"""Generate or update user memory based on recent messages.""" """Generate or update user memory based on recent messages."""
click.echo(click.style("Starting user memory generate task.", fg="green")) logger.info("Starting user profile generate task.")
click.echo(click.style("Starting user profile generate task.", fg="green"))
start_at = time.perf_counter() start_at = time.perf_counter()
app_ids = ( app_ids = (
@ -26,7 +31,9 @@ def user_profile_generate_task():
if len(app_ids) == 0: if len(app_ids) == 0:
click.echo( click.echo(
click.style("No app_id provided, skipping memory generation.", fg="yellow") click.style(
"No app_id provided, skipping user profile generation.", fg="yellow"
)
) )
return return
@ -275,6 +282,9 @@ def process_user_health_summary(user: EndUser, new_messages: str):
result = result.strip("```json").strip("```") result = result.strip("```json").strip("```")
result = json.loads(result) result = json.loads(result)
click.echo("parsed result: " + result)
if "health_status" in result: if "health_status" in result:
user.health_status = result["health_status"] user.health_status = result["health_status"]

Loading…
Cancel
Save