from datetime import datetime from decimal import Decimal import pytz import sqlalchemy as sa from flask import jsonify from flask_login import current_user from flask_restx import Resource, fields, reqparse from controllers.console import api, console_ns from controllers.console.app.wraps import get_app_model from controllers.console.wraps import account_initialization_required, setup_required from core.app.entities.app_invoke_entities import InvokeFrom from extensions.ext_database import db from libs.helper import DatetimeString from libs.login import login_required from models import AppMode, Message @console_ns.route("/apps//statistics/daily-messages") class DailyMessageStatistic(Resource): @api.doc("get_daily_message_statistics") @api.doc(description="Get daily message statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "Daily message statistics retrieved successfully", fields.List(fields.Raw(description="Daily message count data")), ) @get_app_model @setup_required @login_required @account_initialization_required def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() sql_query = """SELECT DATE(DATE_TRUNC('day', created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz )) AS date, COUNT(*) AS message_count FROM messages WHERE app_id = :app_id AND invoke_from != :invoke_from""" arg_dict = {"tz": account.timezone, "app_id": app_model.id, "invoke_from": InvokeFrom.DEBUGGER.value} timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at >= :start" arg_dict["start"] = start_datetime_utc if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at < :end" arg_dict["end"] = end_datetime_utc sql_query += " GROUP BY date ORDER BY date" response_data = [] with db.engine.begin() as conn: rs = conn.execute(sa.text(sql_query), arg_dict) for i in rs: response_data.append({"date": str(i.date), "message_count": i.message_count}) return jsonify({"data": response_data}) @console_ns.route("/apps//statistics/daily-conversations") class DailyConversationStatistic(Resource): @api.doc("get_daily_conversation_statistics") @api.doc(description="Get daily conversation statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "Daily conversation statistics retrieved successfully", fields.List(fields.Raw(description="Daily conversation count data")), ) @get_app_model @setup_required @login_required @account_initialization_required def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc stmt = ( sa.select( sa.func.date( sa.func.date_trunc("day", sa.text("created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz")) ).label("date"), sa.func.count(sa.distinct(Message.conversation_id)).label("conversation_count"), ) .select_from(Message) .where(Message.app_id == app_model.id, Message.invoke_from != InvokeFrom.DEBUGGER.value) ) if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) stmt = stmt.where(Message.created_at >= start_datetime_utc) if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) stmt = stmt.where(Message.created_at < end_datetime_utc) stmt = stmt.group_by("date").order_by("date") response_data = [] with db.engine.begin() as conn: rs = conn.execute(stmt, {"tz": account.timezone}) for row in rs: response_data.append({"date": str(row.date), "conversation_count": row.conversation_count}) return jsonify({"data": response_data}) @console_ns.route("/apps//statistics/daily-end-users") class DailyTerminalsStatistic(Resource): @api.doc("get_daily_terminals_statistics") @api.doc(description="Get daily terminal/end-user statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "Daily terminal statistics retrieved successfully", fields.List(fields.Raw(description="Daily terminal count data")), ) @get_app_model @setup_required @login_required @account_initialization_required def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() sql_query = """SELECT DATE(DATE_TRUNC('day', created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz )) AS date, COUNT(DISTINCT messages.from_end_user_id) AS terminal_count FROM messages WHERE app_id = :app_id AND invoke_from != :invoke_from""" arg_dict = {"tz": account.timezone, "app_id": app_model.id, "invoke_from": InvokeFrom.DEBUGGER.value} timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at >= :start" arg_dict["start"] = start_datetime_utc if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at < :end" arg_dict["end"] = end_datetime_utc sql_query += " GROUP BY date ORDER BY date" response_data = [] with db.engine.begin() as conn: rs = conn.execute(sa.text(sql_query), arg_dict) for i in rs: response_data.append({"date": str(i.date), "terminal_count": i.terminal_count}) return jsonify({"data": response_data}) @console_ns.route("/apps//statistics/token-costs") class DailyTokenCostStatistic(Resource): @api.doc("get_daily_token_cost_statistics") @api.doc(description="Get daily token cost statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "Daily token cost statistics retrieved successfully", fields.List(fields.Raw(description="Daily token cost data")), ) @get_app_model @setup_required @login_required @account_initialization_required def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() sql_query = """SELECT DATE(DATE_TRUNC('day', created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz )) AS date, (SUM(messages.message_tokens) + SUM(messages.answer_tokens)) AS token_count, SUM(total_price) AS total_price FROM messages WHERE app_id = :app_id AND invoke_from != :invoke_from""" arg_dict = {"tz": account.timezone, "app_id": app_model.id, "invoke_from": InvokeFrom.DEBUGGER.value} timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at >= :start" arg_dict["start"] = start_datetime_utc if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at < :end" arg_dict["end"] = end_datetime_utc sql_query += " GROUP BY date ORDER BY date" response_data = [] with db.engine.begin() as conn: rs = conn.execute(sa.text(sql_query), arg_dict) for i in rs: response_data.append( {"date": str(i.date), "token_count": i.token_count, "total_price": i.total_price, "currency": "USD"} ) return jsonify({"data": response_data}) @console_ns.route("/apps//statistics/average-session-interactions") class AverageSessionInteractionStatistic(Resource): @api.doc("get_average_session_interaction_statistics") @api.doc(description="Get average session interaction statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "Average session interaction statistics retrieved successfully", fields.List(fields.Raw(description="Average session interaction data")), ) @setup_required @login_required @account_initialization_required @get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT, AppMode.ADVANCED_CHAT]) def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() sql_query = """SELECT DATE(DATE_TRUNC('day', c.created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz )) AS date, AVG(subquery.message_count) AS interactions FROM ( SELECT m.conversation_id, COUNT(m.id) AS message_count FROM conversations c JOIN messages m ON c.id = m.conversation_id WHERE c.app_id = :app_id AND m.invoke_from != :invoke_from""" arg_dict = {"tz": account.timezone, "app_id": app_model.id, "invoke_from": InvokeFrom.DEBUGGER.value} timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) sql_query += " AND c.created_at >= :start" arg_dict["start"] = start_datetime_utc if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) sql_query += " AND c.created_at < :end" arg_dict["end"] = end_datetime_utc sql_query += """ GROUP BY m.conversation_id ) subquery LEFT JOIN conversations c ON c.id = subquery.conversation_id GROUP BY date ORDER BY date""" response_data = [] with db.engine.begin() as conn: rs = conn.execute(sa.text(sql_query), arg_dict) for i in rs: response_data.append( {"date": str(i.date), "interactions": float(i.interactions.quantize(Decimal("0.01")))} ) return jsonify({"data": response_data}) @console_ns.route("/apps//statistics/user-satisfaction-rate") class UserSatisfactionRateStatistic(Resource): @api.doc("get_user_satisfaction_rate_statistics") @api.doc(description="Get user satisfaction rate statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "User satisfaction rate statistics retrieved successfully", fields.List(fields.Raw(description="User satisfaction rate data")), ) @get_app_model @setup_required @login_required @account_initialization_required def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() sql_query = """SELECT DATE(DATE_TRUNC('day', m.created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz )) AS date, COUNT(m.id) AS message_count, COUNT(mf.id) AS feedback_count FROM messages m LEFT JOIN message_feedbacks mf ON mf.message_id=m.id AND mf.rating='like' WHERE m.app_id = :app_id AND m.invoke_from != :invoke_from""" arg_dict = {"tz": account.timezone, "app_id": app_model.id, "invoke_from": InvokeFrom.DEBUGGER.value} timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) sql_query += " AND m.created_at >= :start" arg_dict["start"] = start_datetime_utc if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) sql_query += " AND m.created_at < :end" arg_dict["end"] = end_datetime_utc sql_query += " GROUP BY date ORDER BY date" response_data = [] with db.engine.begin() as conn: rs = conn.execute(sa.text(sql_query), arg_dict) for i in rs: response_data.append( { "date": str(i.date), "rate": round((i.feedback_count * 1000 / i.message_count) if i.message_count > 0 else 0, 2), } ) return jsonify({"data": response_data}) @console_ns.route("/apps//statistics/average-response-time") class AverageResponseTimeStatistic(Resource): @api.doc("get_average_response_time_statistics") @api.doc(description="Get average response time statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "Average response time statistics retrieved successfully", fields.List(fields.Raw(description="Average response time data")), ) @setup_required @login_required @account_initialization_required @get_app_model(mode=AppMode.COMPLETION) def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() sql_query = """SELECT DATE(DATE_TRUNC('day', created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz )) AS date, AVG(provider_response_latency) AS latency FROM messages WHERE app_id = :app_id AND invoke_from != :invoke_from""" arg_dict = {"tz": account.timezone, "app_id": app_model.id, "invoke_from": InvokeFrom.DEBUGGER.value} timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at >= :start" arg_dict["start"] = start_datetime_utc if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at < :end" arg_dict["end"] = end_datetime_utc sql_query += " GROUP BY date ORDER BY date" response_data = [] with db.engine.begin() as conn: rs = conn.execute(sa.text(sql_query), arg_dict) for i in rs: response_data.append({"date": str(i.date), "latency": round(i.latency * 1000, 4)}) return jsonify({"data": response_data}) @console_ns.route("/apps//statistics/tokens-per-second") class TokensPerSecondStatistic(Resource): @api.doc("get_tokens_per_second_statistics") @api.doc(description="Get tokens per second statistics for an application") @api.doc(params={"app_id": "Application ID"}) @api.expect( api.parser() .add_argument("start", type=str, location="args", help="Start date (YYYY-MM-DD HH:MM)") .add_argument("end", type=str, location="args", help="End date (YYYY-MM-DD HH:MM)") ) @api.response( 200, "Tokens per second statistics retrieved successfully", fields.List(fields.Raw(description="Tokens per second data")), ) @get_app_model @setup_required @login_required @account_initialization_required def get(self, app_model): account = current_user parser = reqparse.RequestParser() parser.add_argument("start", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") parser.add_argument("end", type=DatetimeString("%Y-%m-%d %H:%M"), location="args") args = parser.parse_args() sql_query = """SELECT DATE(DATE_TRUNC('day', created_at AT TIME ZONE 'UTC' AT TIME ZONE :tz )) AS date, CASE WHEN SUM(provider_response_latency) = 0 THEN 0 ELSE (SUM(answer_tokens) / SUM(provider_response_latency)) END as tokens_per_second FROM messages WHERE app_id = :app_id AND invoke_from != :invoke_from""" arg_dict = {"tz": account.timezone, "app_id": app_model.id, "invoke_from": InvokeFrom.DEBUGGER.value} timezone = pytz.timezone(account.timezone) utc_timezone = pytz.utc if args["start"]: start_datetime = datetime.strptime(args["start"], "%Y-%m-%d %H:%M") start_datetime = start_datetime.replace(second=0) start_datetime_timezone = timezone.localize(start_datetime) start_datetime_utc = start_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at >= :start" arg_dict["start"] = start_datetime_utc if args["end"]: end_datetime = datetime.strptime(args["end"], "%Y-%m-%d %H:%M") end_datetime = end_datetime.replace(second=0) end_datetime_timezone = timezone.localize(end_datetime) end_datetime_utc = end_datetime_timezone.astimezone(utc_timezone) sql_query += " AND created_at < :end" arg_dict["end"] = end_datetime_utc sql_query += " GROUP BY date ORDER BY date" response_data = [] with db.engine.begin() as conn: rs = conn.execute(sa.text(sql_query), arg_dict) for i in rs: response_data.append({"date": str(i.date), "tps": round(i.tokens_per_second, 4)}) return jsonify({"data": response_data})