AHj6DdZddlmZddlmZmZmZddlZGddZy)z4Database connection and data extraction for LiteLLM.)datetime)AnyOptionalListNc \eZdZdZdZ d deedeedeedejfdZ y) LiteLLMDatabasez;Handle LiteLLM PostgreSQL database connections and queries.c.ddlm} | td|S)Nr) prisma_clientzDatabase not connected. Connect a database to your proxy - https://docs.litellm.ai/docs/simple_proxy#managing-auth---virtual-keys)litellm.proxy.proxy_serverr Exception)selfr s _/root/tools/cai/cai_env/lib/python3.12/site-packages/litellm/integrations/cloudzero/database.py_ensure_prisma_clientz%LiteLLMDatabase._ensure_prisma_clients)<0  T Nlimitstart_time_utc end_time_utcreturncK|j}d}||g}| |jt||dz } |j j |g|d{}tj|dS#ttf$r t dwxYw79#t$r}tdt|d}~wwxYww)z8Retrieve usage data from LiteLLM daily user spend table.aj SELECT dus.id, dus.date, dus.user_id, dus.api_key, dus.model, dus.model_group, dus.custom_llm_provider, dus.prompt_tokens, dus.completion_tokens, dus.spend, dus.api_requests, dus.successful_requests, dus.failed_requests, dus.cache_creation_input_tokens, dus.cache_read_input_tokens, dus.created_at, dus.updated_at, vt.team_id, vt.key_alias as api_key_alias, tt.team_alias, ut.user_email as user_email FROM "LiteLLM_DailyUserSpend" dus LEFT JOIN "LiteLLM_VerificationToken" vt ON dus.api_key = vt.token LEFT JOIN "LiteLLM_TeamTable" tt ON vt.team_id = tt.team_id LEFT JOIN "LiteLLM_UserTable" ut ON dus.user_id = ut.user_id WHERE ($1::timestamptz IS NULL OR dus.updated_at >= $1::timestamptz) AND ($2::timestamptz IS NULL OR dus.updated_at <= $2::timestamptz) ORDER BY dus.date DESC, dus.created_at DESC Nzlimit must be an integerz LIMIT $3)infer_schema_lengthzError retrieving usage data: ) rappendint TypeError ValueErrordb query_rawpl DataFramer str) r rrrclientqueryparams db_responsees rget_usage_datazLiteLLMDatabase.get_usage_data(s++- B      = c%j) [ E F 3 3 3E CF CCK<< F Fz* = !;<< = D F;CF8DE E FsPCA8C BBB7C8BCB B=!B88B==C)NNN) __name__ __module__ __qualname____doc__rrrrrrr%rrrrsWE $-1+/ =F}=F!*=Fx( =F  =Frr) r)rtypingrrrpolarsrrr*rrr-s$&;&&JFJFr