tPj<"UdZddlmZddlZddlmZmZmZddlmZm Z m Z m Z ddl Z ddl mZddlmZdd lmZer dd lmZdd lmZne Zd Zd ed<dZd ed<GddeZdgZy)z;Focus export logger orchestrating DB pull/transform/upload.) annotationsN)datetime timedeltatimezone) TYPE_CHECKINGAnyFinalcast)verbose_logger) CustomLogger)FocusTimeWindow)AsyncIOSchedulerFocusExportEnginefocus_export_usage_datar FOCUS_USAGE_DATA_JOB_NAMEiDEFAULT_DRY_RUN_LIMITceZdZdZdddddddd dfdZddZdddd ddZefddZdd Z e dd Z dd Z dd Z dd Z ddZddZxZS) FocusLoggerzOCoordinates Focus export jobs across transformer/serializer/destination layers.N)provider export_format frequencycron_offset_minuteinterval_secondsprefixdestination_configc t |d i||xstjdxsdj |_|xstjdxsdj |_|xstjdxsdj |_||nttjdd|_ ||ntjd } d|_ |  t| |_ tjd } ||n| r| nd |_||_d|_y#ttf$rtjd | YZwxYw)NFOCUS_PROVIDERs3 FOCUS_FORMATparquetFOCUS_FREQUENCYhourlyFOCUS_CRON_OFFSET5FOCUS_INTERVAL_SECONDSz2Invalid FOCUS_INTERVAL_SECONDS value: %s, ignoring FOCUS_PREFIX focus_exports)super__init__osgetenvlowerrrrintrr ValueError TypeErrorr warningr_destination_config_engine) selfrrrrrrrkwargs raw_interval env_prefix __class__s l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/focus/focus_logger.pyr,zFocusLogger.__init__s? "6"!HRYY/?%@HDOOQ +Uryy/HUI\\^#Oryy1B'COxVVX"4"@ c"))TgilJmFn 3C2N.TVT]T]^vTw ,0  # (+L(9% IIn5 %+%76Jj\k #5 15  * &&H  s,D//%EEc|j?ddlm}||j|j|j |j |_|jS)z+Instantiate the heavy export engine lazily.r r)rrrr)r5 export_enginerrrrr4)r6rs r;_ensure_enginezFocusLogger._ensure_engine@sG <<  8,"00{{#'#;#; DL ||)limitstart_time_utc end_time_utccKt|t|z r td|r6|r4t|||j}|j ||d{y|j |d{y7 7w)a(Public hook to trigger export immediately. When called without time bounds (manual /vantage/export with no start/end), exports **all** available data instead of the last scheduled window. The hourly/daily window only applies to automatic scheduler runs. z9start_time_utc and end_time_utc must be provided together start_timeend_timerwindowr@Nr@)boolr1rr_export_window _export_all)r6r@rArBrHs r;export_usage_datazFocusLogger.export_usage_dataMs  $|"4 4XY Y l+)%..F %%V5%A A A"""/ / / B 0s$AA<A8A<2A:3A<:A<cbK|j}|j|d{S7w)z*Return transformed data without uploading.rIN)r>dry_run_export_usage_datar6r@engines r;rOz%FocusLogger.dry_run_export_usage_datais.++-55E5BBBBs &/-/cKddlm}d}|t|dd}| t|dd}|rz|jrn|j t d{}|st jdy |jd{|jt d{y|jd{y7m7>7"#|jt d{7wxYw7/w)z@Entry point for scheduler jobs to run export cycle with locking.r)proxy_logging_objNdb_spend_update_writerpod_lock_manager) cronjob_idz(Focus export: unable to acquire pod lock) litellm.proxy.proxy_serverrSgetattr redis_cache acquire_lockrr debug_run_scheduled_export release_lock)r6rSrUwriteracquireds r;initialize_focus_export_jobz'FocusLogger.initialize_focus_export_jobns@  (#$57OQUVF!#*63Et#L  0 < <$4$A$AMf$A$ggH$$%OP Z00222&33?X3YYY,,. . .h 3Y&33?X3YYY .sfAC*B>C*.CCCC*C C*8C(9C*CC*C%C!C%%C*cJKtjjtDcgc]}t |tur|}}|st j dytt|d}|j}|j|jfi|ycc}ww)zBRegister the export cron/interval job with the provided scheduler.) callback_typez5No Focus export logger registered; skipping schedulerNr) litellmlogging_callback_managerget_custom_loggers_for_typertyper r[r _build_scheduler_triggeradd_jobr`) schedulercb focus_loggers focus_loggertrigger_kwargss r; init_focus_export_background_jobz,FocusLogger.init_focus_export_background_jobs66RRalRm4 Bx;& 4 4   !X Y "; a0@A , E E G   4 4  4 s'B#BA B#cx|jdk(r|jxsd}d|dS|jdk(r&tdtd|j}d|ddS|jd k(r;td|j}td |dz}td|dz}d||dd St d |j) z:Return scheduler configuration for the selected frequency.interval<)triggersecondsr$r;cron)rrminuteseconddaily)rrhourrvrwUnsupported frequency: )rrmaxminrr1)r6rsrv total_minutesrzs r;rgz$FocusLogger._build_scheduler_triggers >>Z '!228bG)g> > >>X %CD$;$;<=F%1E E >>W $#&q$*A*A#BMb-2"56D]R/0F%tvQRS S24>>2BCDDr?cK|jtjtj}|j |dd{y7w)z7Execute the scheduled export for the configured window.NrG)_compute_time_windowrnowrutcrK)r6rHs r;r\z!FocusLogger._run_scheduled_exports<11(,,x||2LM!!t!<< export_allrPs r;rLzFocusLogger._export_alls- ++-e,,,s &0.0cfK|j}|j||d{y7w)NrG)r> export_window)r6rHr@rQs r;rKzFocusLogger._export_windows/ ++-""&">>>s '1/1c|jtj}|jdk(r$|j ddd}|t dz }n}|jdk(r%|j dddd}|t dz }nI|jd k(r"t |j xsd }|}||z }ntd |jt|||j S)z?Derive the time window to export based on configured frequency.r$r)rvrw microsecondr )hoursrx)rzrvrwr)daysrprq)rsr{rD) astimezonerrrreplacerrr1r)r6rnow_utcrFrErps r;rz FocusLogger._compute_time_windows 5 >>X %aqIH!IA$66J ^^w &AaqQH!I1$55J ^^z )'0E0E0KLHH!H,J6t~~6FGH H!nn  r?)r str | Nonerrrrr int | Nonerrrrrzdict[str, Any] | Noner7rreturnNone)rr)r@rrAdatetime | NonerBrrr)r@rrdict[str, Any])rr)rirrr)rr)r@rrr)rHrr@rrr)rrrr)__name__ __module__ __qualname____doc__r,r>rMrrOr` staticmethodrnrgr\rLrKr __classcell__)r:s@r;rrs<Y $$( $)-'+!48!6!6" !6  !6 ' !6%!6!62!6!6 !6F  !*.(, 00( 0 & 0  08CXC /, #   .E$= --  -? ? ?  ? r?r)r __future__rr-rrrtypingrrr r rclitellm._loggingr "litellm.integrations.custom_loggerr destinationsrapscheduler.schedulers.asynciorr=rr__annotations__rr__all__r*r?r;rscA" 2222+;)?0#<5<"u"~ ,~ B /r?