tPj_(UddlmZmZddlmZmZddlmZGddeZGddeeZ dZ ee d <d Z ee d <d ed e de ddfdZd e ddfdZd e defdZy))MappingSequence)FinalProtocol)verbose_loggerc$eZdZdZdededefdZy)SupportsExecuteRawz=The one database operation create_view_tolerating_race needs.queryargsreturnc KywNselfr r s b/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/db/create_views.py execute_rawzSupportsExecuteRaw.execute_raw N)__name__ __module__ __qualname____doc__strobjectintrrrr r sGFsF6FcFrr c 4eZdZdZdededeeeeffdZy)SupportsRawQuerieszSThe database operations the view bootstrap needs: probe a relation, then create it.r r r c Kywrrrs r query_rawzSupportsRawQueries.query_rawrrN) rrrrrrrrr!rrrrr s&]_S__HWSRX[EYz.create_view_tolerating_race..,s W6SV\\^+Ws%(z*%s already created by a concurrent replica)rrdebug Exceptionanyr#)r$r%r&r,s @rcreate_view_tolerating_racer1skVnnS!!!]I6 " VW:VWW I9UUVs7A<757A<7 A9/A4/A<4A99A<c8 K |jdd{tjd |jdd{tjd |jd d{tjd  |jdd{tjd |jdd{tjd |jdd{tjd |jd d{tjd! |jd%d{tjd&y7d#t$rS}t |j t fdtDst|ddd{7Yd}~d}~wwxYw7#t$rU}t |j t fd tDsd }t|d |d{7Yd}~d}~wwxYw7#t$rU}t |j t fdtDsd}t|d|d{7Yd}~d}~wwxYw7#t$rU}t |j t fdtDsd}t|d|d{7Yd}~Cd}~wwxYw76#t$rU}t |j t fdtDsd}t|d|d{7Yd}~xd}~wwxYw7k#t$rU}t |j t fdtDsd}t|d|d{7Yd}~d}~wwxYw7#t$rU}t |j t fd"tDsd#}t|d$|d{7Yd}~d}~wwxYw7#t$rT}t |j t fd'tDsd(}t|d)|d{7Yd}~yd}~wwxYww)*a -------------------------------------------------- NOTE: Copy of `litellm/db_scripts/create_views.py`. -------------------------------------------------- Checks if the LiteLLM_VerificationTokenView and MonthlyGlobalSpend exists in the user's db. LiteLLM_VerificationTokenView: This view is used for getting the token + team data in user_api_key_auth MonthlyGlobalSpend: This view is used for the admin view to see global spend for this month If the view doesn't exist, one will be created. z5SELECT 1 FROM "LiteLLM_VerificationTokenView" LIMIT 1Nz%LiteLLM_VerificationTokenView Exists!c3&K|]}|v ywrrr*r+ error_msgs rr-z'create_missing_views..EM66Y&MLiteLLM_VerificationTokenViewa. CREATE VIEW "LiteLLM_VerificationTokenView" AS SELECT v.*, t.spend AS team_spend, t.max_budget AS team_max_budget, t.tpm_limit AS team_tpm_limit, t.rpm_limit AS team_rpm_limit, p.project_alias AS project_alias FROM "LiteLLM_VerificationToken" v LEFT JOIN "LiteLLM_TeamTable" t ON v.team_id = t.team_id LEFT JOIN "LiteLLM_ProjectTable" p ON v.project_id = p.project_id; z*SELECT 1 FROM "MonthlyGlobalSpend" LIMIT 1zMonthlyGlobalSpend Exists!c3&K|]}|v ywrrr4s rr-z'create_missing_views.._r6r7a5 CREATE OR REPLACE VIEW "MonthlyGlobalSpend" AS SELECT DATE("startTime") AS date, SUM("spend") AS spend FROM "LiteLLM_SpendLogs" WHERE "startTime" >= (CURRENT_DATE - INTERVAL '30 days') GROUP BY DATE("startTime"); MonthlyGlobalSpendz*SELECT 1 FROM "Last30dKeysBySpend" LIMIT 1zLast30dKeysBySpend Exists!c3&K|]}|v ywrrr4s rr-z'create_missing_views..tr6r7a CREATE OR REPLACE VIEW "Last30dKeysBySpend" AS SELECT L."api_key", V."key_alias", V."key_name", SUM(L."spend") AS total_spend FROM "LiteLLM_SpendLogs" L LEFT JOIN "LiteLLM_VerificationToken" V ON L."api_key" = V."token" WHERE L."startTime" >= (CURRENT_DATE - INTERVAL '30 days') GROUP BY L."api_key", V."key_alias", V."key_name" ORDER BY total_spend DESC; Last30dKeysBySpendz,SELECT 1 FROM "Last30dModelsBySpend" LIMIT 1zLast30dModelsBySpend Exists!c3&K|]}|v ywrrr4s rr-z'create_missing_views..r6r7a^ CREATE OR REPLACE VIEW "Last30dModelsBySpend" AS SELECT "model", SUM("spend") AS total_spend FROM "LiteLLM_SpendLogs" WHERE "startTime" >= (CURRENT_DATE - INTERVAL '30 days') AND "model" != '' GROUP BY "model" ORDER BY total_spend DESC; Last30dModelsBySpendz0SELECT 1 FROM "MonthlyGlobalSpendPerKey" LIMIT 1z MonthlyGlobalSpendPerKey Exists!c3&K|]}|v ywrrr4s rr-z'create_missing_views..r6r7a CREATE OR REPLACE VIEW "MonthlyGlobalSpendPerKey" AS SELECT DATE("startTime") AS date, SUM("spend") AS spend, api_key as api_key FROM "LiteLLM_SpendLogs" WHERE "startTime" >= (CURRENT_DATE - INTERVAL '30 days') GROUP BY DATE("startTime"), api_key; MonthlyGlobalSpendPerKeyz7SELECT 1 FROM "MonthlyGlobalSpendPerUserPerKey" LIMIT 1z'MonthlyGlobalSpendPerUserPerKey Exists!c3&K|]}|v ywrrr4s rr-z'create_missing_views..r6r7a CREATE OR REPLACE VIEW "MonthlyGlobalSpendPerUserPerKey" AS SELECT DATE("startTime") AS date, SUM("spend") AS spend, api_key as api_key, "user" as "user" FROM "LiteLLM_SpendLogs" WHERE "startTime" >= (CURRENT_DATE - INTERVAL '30 days') GROUP BY DATE("startTime"), "user", api_key; MonthlyGlobalSpendPerUserPerKeyz%SELECT 1 FROM "DailyTagSpend" LIMIT 1zDailyTagSpend Exists!c3&K|]}|v ywrrr4s rr-z'create_missing_views..r6r7ar CREATE OR REPLACE VIEW "DailyTagSpend" AS SELECT jsonb_array_elements_text(request_tags) AS individual_request_tag, DATE(s."startTime") AS spend_date, COUNT(*) AS log_count, SUM(spend) AS total_spend FROM "LiteLLM_SpendLogs" s GROUP BY individual_request_tag, DATE(s."startTime"); DailyTagSpendz/SELECT 1 FROM "Last30dTopEndUsersSpend" LIMIT 1zLast30dTopEndUsersSpend Exists!c3&K|]}|v ywrrr4s rr-z'create_missing_views..r6r7af CREATE VIEW "Last30dTopEndUsersSpend" AS SELECT end_user, COUNT(*) AS total_events, SUM(spend) AS total_spend FROM "LiteLLM_SpendLogs" WHERE end_user <> '' AND end_user <> user AND "startTime" >= CURRENT_DATE - INTERVAL '30 days' GROUP BY end_user ORDER BY total_spend DESC LIMIT 100; Last30dTopEndUsersSpend) r!rr.r/rr)r0r"r1)r$r, sql_queryr5s @rcreate_missing_viewsrH1s llVWWWDE0OllKLLL9:&OllKLLL9:6QllMNNN;<*UllQRRR?@(\llXYYYFG.JllFGGG45"TllPQQQ>?M X  FLLN M5LMM )  +     0 M OFLLN M5LMM   *".BINNN!O& M OFLLN M5LMM  (*".BINNN1O6 O QFLLN M5LMM  *".DiPPP'Q* S UFLLN M5LMM   *".H)TTT%U( Z \FLLN M5LMM  *".OQZ[[[)\. H JFLLN M5LMM   *"oyIIIJ" R TFLLN M5LMM   *".GSSSTsRFE=FG"G G"#I7I8IJ*&J''J*LL L0M2M/M2O3O4OP:"P7#P:<R=F G AG G GRGRG"" I+AH;/H20H;5R;IRI J$AJJJRJ$$R'J** L3AL7K:8L=RLR L M,AM'MM'!R'M,,R/M22 O;AO ?OO R ORO P4AP/#P&$P/)R/P44R7P:: RARR R RRRcPKd}|j|d{}tjd||rlt|tr\t |dkDrNt|dt r;d|dvr4|dd,|dddk(s |dddk(rtjdy y 7w) zy Run only on first time startup. If SpendLogs table already has values, then don't create views on startup. ze SELECT reltuples::BIGINT FROM pg_class WHERE oid = '"LiteLLM_SpendLogs"'::regclass; )r Nz-Estimated Row count of LiteLLM_SpendLogs = %sr reltupleszShould create viewsTF)r!rr. isinstancelistlendict)r$rGresults rshould_create_missing_viewsrQsI ,,Y,77FH&Q vt $ K!O vay$ ' 6!9 $ 1Ik " . AY{ #q (F1Ik,Bb,H23 8sB&B$B B&)collections.abcrrtypingrrlitellmrr rr"__annotations__r#rr1rHboolrQrrrrWs--""GG `+X`"XW'`e_V*<VVSVV[_V&ET#5ET$ETP*<r