j< `ddlmZddlZddlmZejeZGddZdS)) annotationsN)Anyc,eZdZ dddZ dddZdS)ErpAiBudgetMixinNchat agent_codestr prompt_tokensintcompletion_tokenscost_vnd Any | None task_type user_email str | None query_previewreturndict[str, Any]c ||z}|d} t|dz | zd}d} ||||||||r t|dkr |dddzn|d} |5} | 5} | | | | | cdddcdddS#1swxYwYddddS#1swxYwYdS) uOGhi nhận vết sử dụng token và chi phí ngân sách cho nhân viên AI.NgI@g@@a INSERT INTO erp_ai_agent_usage_logs ( agent_code, task_type, prompt_tokens, completion_tokens, total_tokens, cost_vnd, user_email, query_preview ) VALUES ( %(agent_code)s, %(task_type)s, %(prompt_tokens)s, %(completion_tokens)s, %(total_tokens)s, %(cost_vnd)s, %(user_email)s, %(query_preview)s ) RETURNING *; z...)rrr r total_tokensr rr)roundlenget_connectioncursorexecutecommitfetchone)selfrr r r rrrr cost_per_1ksqlparamsconncurs @C:\Projects\DSCons\app\core\postgres\clients\ai_budget_client.pylog_ai_agent_usagez#ErpAiBudgetMixin.log_ai_agent_usage s%'88   KlV3{BAFFH %"*!2( $!$]!3!3c!9!9,DSD1E99    " " "dDKKMM "S KKV $ $ $ KKMMM<<>> " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " " "s6#C'8>C6 C'C C'C C''C+.C+cld}g}|r|dz }||d|d}|5}|5}||||}dddn #1swxYwYdddn #1swxYwYg}d} d} d} d} |D]?} t | d}t | d}|d krt ||z d zd nd}t | d }t | d }|d krt ||z d zd nd}t | d}t | d}|d krt ||z d zd nd}t | d}t | d}|d krt ||z d zd nd}| |z } | |z } | |z } | |z } || d| d| d| dt | d| d| d|||t| dt| dd|||t| dt| dd|||t| dt| dd|||t| d t| d!dd" A|t | d#t | d#t | d#t | d#d$S)%u^Tổng hợp ngân sách đã sử dụng theo ngày, tuần, tháng, năm cho nhân viên AI.zWHERE b.is_active = TRUEz AND b.agent_code = %sa SELECT b.agent_code, b.display_name, b.role_title, b.avatar_icon, b.daily_budget_vnd, b.weekly_budget_vnd, b.monthly_budget_vnd, b.yearly_budget_vnd, b.cost_per_1k_tokens_vnd, b.model_name, b.provider, COALESCE(SUM(l.cost_vnd) FILTER (WHERE l.created_at >= date_trunc('day', NOW())), 0) AS day_spent_vnd, COALESCE(SUM(l.total_tokens) FILTER (WHERE l.created_at >= date_trunc('day', NOW())), 0) AS day_tokens, COALESCE(COUNT(l.id) FILTER (WHERE l.created_at >= date_trunc('day', NOW())), 0) AS day_requests, COALESCE(SUM(l.cost_vnd) FILTER (WHERE l.created_at >= date_trunc('week', NOW())), 0) AS week_spent_vnd, COALESCE(SUM(l.total_tokens) FILTER (WHERE l.created_at >= date_trunc('week', NOW())), 0) AS week_tokens, COALESCE(COUNT(l.id) FILTER (WHERE l.created_at >= date_trunc('week', NOW())), 0) AS week_requests, COALESCE(SUM(l.cost_vnd) FILTER (WHERE l.created_at >= date_trunc('month', NOW())), 0) AS month_spent_vnd, COALESCE(SUM(l.total_tokens) FILTER (WHERE l.created_at >= date_trunc('month', NOW())), 0) AS month_tokens, COALESCE(COUNT(l.id) FILTER (WHERE l.created_at >= date_trunc('month', NOW())), 0) AS month_requests, COALESCE(SUM(l.cost_vnd) FILTER (WHERE l.created_at >= date_trunc('year', NOW())), 0) AS year_spent_vnd, COALESCE(SUM(l.total_tokens) FILTER (WHERE l.created_at >= date_trunc('year', NOW())), 0) AS year_tokens, COALESCE(COUNT(l.id) FILTER (WHERE l.created_at >= date_trunc('year', NOW())), 0) AS year_requests FROM erp_ai_agent_budgets b LEFT JOIN erp_ai_agent_usage_logs l ON b.agent_code = l.agent_code a GROUP BY b.agent_code, b.display_name, b.role_title, b.avatar_icon, b.daily_budget_vnd, b.weekly_budget_vnd, b.monthly_budget_vnd, b.yearly_budget_vnd, b.cost_per_1k_tokens_vnd, b.model_name, b.provider ORDER BY b.agent_code ASC; Ng day_spent_vnddaily_budget_vndrgY@week_spent_vndweekly_budget_vndmonth_spent_vndmonthly_budget_vndyear_spent_vndyearly_budget_vndr display_name role_title avatar_iconcost_per_1k_tokens_vnd model_nameprovider day_tokens day_requests) spent_vnd budget_vnd percentagetokensrequests_count week_tokens week_requests month_tokensmonth_requests year_tokens year_requests) rr2r3r4r5r6r7dayweekmonthyear)agentstotal_spent_today_vndtotal_spent_week_vndtotal_spent_month_vndtotal_spent_year_vnd)appendrrrfetchallfloatrr )r r where_clauser#r"r$r%rows agents_data total_today total_week total_month total_yearr day_spent day_budgetday_pct week_spent week_budgetweek_pct month_spent month_budget month_pct year_spent year_budgetyear_pcts r&get_ai_agent_budget_summaryz,ErpAiBudgetMixin.get_ai_agent_budget_summary7sV2   & 4 4L MM* % % % 67   @ " " "dDKKMM "S KKV $ $ $<<>>D " " " " " " " " " " " " " " " " " " " " " " " " " " " " " "     F F Aa011Iq!3455J>H1nny:-5:::RU q!1233J"5 677K@KazK/%7!<<>>  q!1233J"5 677K@KazK/%7!<<rqs""""""  8 $ $p p p p p p p p p p rk