
    ^h6                     @    S SK Jr  S SKJr  \rS\4S jrS\S\4S jrg)    )Any)verbose_loggerdbc                   #     U R                  S5      I Sh  vN   [        S5         U R                  S5      I Sh  vN   [        S5         U R                  S5      I Sh  vN   [        S5         U R                  S5      I Sh  vN   [        S5         U R                  S5      I Sh  vN   [        S5         U R                  S5      I Sh  vN   [        S5         U R                  S5      I Sh  vN   [        S5         U R                  S5      I Sh  vN   [        S 5        g GN! [         a)    U R                  S5      I Sh  vN    [        S5         GN9f = f GN(! [         a)    SnU R                  US	9I Sh  vN    [        S
5         GNMf = f GN<! [         a)    SnU R                  US	9I Sh  vN    [        S5         GNaf = f GNP! [         a)    SnU R                  US	9I Sh  vN    [        S5         GNuf = f GNd! [         a)    SnU R                  US	9I Sh  vN    [        S5         GNf = f GNx! [         a)    SnU R                  US	9I Sh  vN    [        S5         GNf = f GN! [         a)    SnU R                  US	9I Sh  vN    [        S5         GNf = f GN! [         a(    S!nU R                  US	9I Sh  vN    [        S"5         gf = f7f)#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!a  
                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
                FROM "LiteLLM_VerificationToken" v
                LEFT JOIN "LiteLLM_TeamTable" t ON v.team_id = t.team_id;
            z&LiteLLM_VerificationTokenView Created!z*SELECT 1 FROM "MonthlyGlobalSpend" LIMIT 1zMonthlyGlobalSpend Exists!a5  
        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");
        queryzMonthlyGlobalSpend Created!z*SELECT 1 FROM "Last30dKeysBySpend" LIMIT 1zLast30dKeysBySpend Exists!a   
        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;
        zLast30dKeysBySpend Created!z,SELECT 1 FROM "Last30dModelsBySpend" LIMIT 1zLast30dModelsBySpend Exists!a^  
        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;
        zLast30dModelsBySpend Created!z0SELECT 1 FROM "MonthlyGlobalSpendPerKey" LIMIT 1z MonthlyGlobalSpendPerKey Exists!a  
            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;
        z!MonthlyGlobalSpendPerKey Created!z7SELECT 1 FROM "MonthlyGlobalSpendPerUserPerKey" LIMIT 1z'MonthlyGlobalSpendPerUserPerKey Exists!a  
            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;
        z(MonthlyGlobalSpendPerUserPerKey Created!z#SELECT 1 FROM DailyTagSpend LIMIT 1zDailyTagSpend Exists!ap  
        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");
        zDailyTagSpend Created!z/SELECT 1 FROM "Last30dTopEndUsersSpend" LIMIT 1zLast30dTopEndUsersSpend Exists!af  
        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;
        z Last30dTopEndUsersSpend Created!)	query_rawprint	Exceptionexecute_raw)r   	sql_querys     W/home/james-whalen/.local/lib/python3.13/site-packages/litellm/proxy/db/create_views.pycreate_missing_viewsr      s    8llVWWW56&-llKLLL*+$-llKLLL*+4/llMNNN,-(3llQRRR01&:llI
 	
 	
 	78,(llDEEE%& 2llPQQQ/0  [ 	X 8nn

 	
 	
 	67!8& 	M -	 nn9n---+,-$ 	M -	( nn9n---+,/-4 	O /	 nn9n----.%/( 	S 3	 nn9n---12#3&	
  :	  nn9n---89':, 	F (		 nn9n---&'(  	R 2		 nn9n---01
2s  K3D/ D,D/ E( E%E( F! "F#F! 3G GG H ,H-H =I I	I "J 6J7J J> J;J> +K3,D/ /E"EE"K3!E""K3%E( (FF	FK3FK3F! !G?G GK3GK3G H8G;9H	K3HK3H I1H42IK3IK3	I I?*I-+I?;K3>I??K3J J8#J&$J84K37J88K3;J> >K0KK0-K3/K00K3returnc                   #    SnU R                  US9I Sh  vN n[        R                  " SR                  U5      5        U(       a  [	        U[
        5      (       al  [        U5      S:  a]  [	        US   [        5      (       aE  SUS   ;   a<  US   S   (       a/  US   S   S:X  d  US   S   S:X  a  [        R                  " S5        g	g
 N7f)zm
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 = {}r   	reltupleszShould create viewsTF)r	   r   debugformat
isinstancelistlendict)r   r   results      r   should_create_missing_viewsr      s     I <<i<00FHOOPVWXvt$$K!Ovay$''6!9$1Ik"AY{#q(F1Ik,Bb,H23 1s   CC
B3CN)typingr   litellmr   _dbr   boolr        r   <module>r"      s2     "	|3 |~# $ r!   