
    jLD              
         U d dl mZ d dlmZmZmZmZmZ d dlm	Z	 d dl
mZmZmZmZmZmZmZmZ erZd dlmZmZmZmZ d dlmZ d dl
mZ d dlmZmZ d d	lmZmZm Z  d d
l!m"Z" d dl#m$Z$m%Z%m&Z&m'Z' d dl(m)Z* d dl+m,Z,m-Z-m.Z.m/Z/ d dl0m1Z1 d dl2m3Z3  G d de          Z4 G d de          Z5 G d de          Z6 G d de          Z7 G d de          Z8 G d de          Z9 G d de          Z: G d de          Z; G d  d!e          Z< G d" d#e          Z= G d$ d%e          Z>ed&         Z?d'e@d(<   eeAd)         d)f         ZBd'e@d*<   eeAd+         d+f         ZCd'e@d,<   e?ee?         z  ZDd'e@d-<   eAeE         eAeF         z  eAeG         z  eAeH         z  eAd.         z  eAd/         z  eAd0         z  eAd1         z  eAeIe                  z  eAeJed2f                  z  eAeK         z  eAeL         z  eAd3         z  eAd4         z  ZMd'e@d5<   eeHe?eMz  d4z  f         eeHeJeHe?eMz  d4z  f         z           z  ZNd'e@d6<   eeHe?f         ZOd'e@d7<   eeEeFd3f         ZPd'e@d8<   ed9         ZQd'e@d:<   ePeQz  eHz  eGz  eKz  ZRd'e@d;<   eeRd<eIe         f         ZSd'e@d=<   ed>d?eHf         ZTd'e@d@<   eSeTz  d4z  ZUd'e@dA<   edB         ZVd'e@dC<   edD         ZWd'e@dE<   dFZXd'e@dG<   edHeXf         ZYd'e@dI<   edJ         ZZd'e@dK<   edL         Z[d'e@dM<   edN         Z\d'e@dO<   edP         Z]d'e@dQ<   edR         Z^d'e@dS<   edT         Z_d'e@dU<   eJedV         ef         Z`d'e@dW<   eJedX         eaeEed?eHf         f         f         Zbd'e@dY<   eJedZ         eaeEeIeH         f         f         eJed[         ed\gd\f         f         z  Zcd'e@d]<   ed^         Zdd'e@d_<   ed`         Zed'e@da<   edb         Zfd'e@dc<   edd         Zgd'e@de<   edf         Zhd'e@dg<   edh         Zid'e@di<   edj         Zjd'e@dk<   edR         Zkd'e@dl<   edm         Zld'e@dn<   edo         Zmd'e@dp<   edq         Znd'e@dr<   eds         Zod'e@dt<   edu         Zpd'e@dv<   edw         Zqd'e@dx<   edy         Zrd'e@dz<   ed{         Zsd'e@d|<   ed}         Ztd'e@d~<   ed         Zud'e@d<   ed         Zvd'e@d<   ed         Zwd'e@d<   ed         Zxd'e@d<   ed         Zyd'e@d<   ed         Zzd'e@d<   ed         Z{d'e@d<   ed         Z|d'e@d<   ed         Z}d'e@d<   ed         Z~d'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   ed         Zd'e@d<   eee         d?dddd!d#ddf	         Zd'e@d<   eeeHeeRz  d4z  f         ee         e7e:e;dde>d\f	         Zd'e@d<   eeYeJeYd2f         z  eHeeHeHf         z  f         Zd'e@d<   eeYeeH         z  eHeeHef         z  eeHeeHef         z           z  f         Zd'e@d<   eeYeJeY         z  eHf         eeH         z  eGz  Zd'e@d<   eeYeJeHd2f         eEz  f         eEz  Zd'e@d<   eeHeHeeH         z  f         eeH         z  eGz  Zd'e@d<   ed         Zd'e@d<    edd\dd?d>          Z edd\d          ZeJeEeEeEf         Zd'e@d<   ed         Zd'e@d<    G dĄ de          Z G dƄ de          Z G dȄ de          Z G dʄ de          Zed         Zd'e@d<   ed	         Zd'e@d<   eez  ez  ez  ez  Zd'e@d<   eEZd'e@d<   eeeeeE         d?d<f         Zd'e@d<   eHZd'e@d<   eeeeH         d?d<f         Zd'e@d<   eeeG         d?d<f         Zd'e@d<   eez  Zd'e@d<   eez  ez  Zd'e@d<   eed         df         Zd'e@d<   ed         Zd'e@d<   eHe	z  eeK         z  eKz  eIeH         z  eIe	         z  eIeeK                  z  eIeK         z  Zd'e@d<   eegeKf         eegeHf         z  Zed         Zd'e@d<   g dߢZ G d d          ZeegeaeHeHf         f         Zd'e@d<   eaeHeHf         ez  Zd'e@d<   eaeHef         Zd'e@d<   d4S )    )annotations)Callable
CollectionIterableMappingSequence)Path)IOTYPE_CHECKINGAnyLiteralProtocol	TypedDictTypeVarUnion)datedatetimetime	timedelta)Decimal)	TypeAlias)
ConnectionEngine)AsyncConnectionAsyncEngineAsyncSession)Session)	DataFrameExpr	LazyFrameSeries)numpy)DataTypeDataTypeClassIntegerTypeTemporalType)	GPUEngine)Selectorc                      e Zd ZdZ	 dd	dZdS )
ArrowArrayExportablezGType protocol for Arrow C Data Interface via Arrow PyCapsule Interface.Nrequested_schemaobject | Nonereturntuple[object, object]c                    d S N selfr+   s     S/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/_typing.py__arrow_c_array__z&ArrowArrayExportable.__arrow_c_array__#   s     #    r0   )r+   r,   r-   r.   )__name__
__module____qualname____doc__r5   r1   r6   r4   r*   r*       s8        QQ 15$ $ $ $ $ $ $r6   r*   c                      e Zd ZdZdd	dZdS )
ArrowStreamExportablezIType protocol for Arrow C Stream Interface via Arrow PyCapsule Interface.Nr+   r,   r-   objectc                    d S r0   r1   r2   s     r4   __arrow_c_stream__z(ArrowStreamExportable.__arrow_c_stream__+         r6   r0   )r+   r,   r-   r=   )r7   r8   r9   r:   r?   r1   r6   r4   r<   r<   (   s%        SSWWWWWWWr6   r<   c                      e Zd ZdZddZdS )ArrowSchemaExportablezIType protocol for Arrow C Schema Interface via Arrow PyCapsule Interface.r-   r=   c                    d S r0   r1   )r3   s    r4   __arrow_c_schema__z(ArrowSchemaExportable.__arrow_c_schema__1   r@   r6   N)r-   r=   )r7   r8   r9   r:   rD   r1   r6   r4   rB   rB   .   s"        SS//////r6   rB   c                  :    e Zd ZdZddZddZddZdd	Zdd
ZdS )
NumpyArrayz?Protocol to match NumPy Arrays without needing NumPy installed.argsr   kwargsr-   c                    d S r0   r1   r3   rG   rH   s      r4   byteswapzNumpyArray.byteswap7   r@   r6   c                    d S r0   r1   rJ   s      r4   	conjugatezNumpyArray.conjugate8   r@   r6   c                    d S r0   r1   rJ   s      r4   ravelzNumpyArray.ravel9   r@   r6   c                    d S r0   r1   rJ   s      r4   searchsortedzNumpyArray.searchsorted:   r@   r6   c                    d S r0   r1   rJ   s      r4   swapaxeszNumpyArray.swapaxes;   r@   r6   NrG   r   rH   r   r-   r   )	r7   r8   r9   r:   rK   rM   rO   rQ   rS   r1   r6   r4   rF   rF   4   sR        II====>>>>::::AAAA======r6   rF   c                  "    e Zd ZdZd	dZd	dZdS )
PyArrowArrayz
    Protocol to match PyArrow arrays without needing PyArrow installed.

    Only use for function arguments, not return types.
    rG   r   rH   r-   c                    d S r0   r1   rJ   s      r4   bufferszPyArrowArray.buffersE   r@   r6   c                    d S r0   r1   rJ   s      r4   tolistzPyArrowArray.tolistF   r@   r6   NrT   )r7   r8   r9   r:   rX   rZ   r1   r6   r4   rV   rV   >   s4          =<<<;;;;;;r6   rV   c                      e Zd ZdZddZdS )	PyArrowChunkedArrayz
    Protocol to match PyArrow chunked arrays without needing PyArrow installed.

    Only use for function arguments, not return types.
    rG   r   rH   r-   c                    d S r0   r1   rJ   s      r4   
iterchunkszPyArrowChunkedArray.iterchunksP   r@   r6   NrT   )r7   r8   r9   r:   r^   r1   r6   r4   r\   r\   I   s)          @?????r6   r\   c                  B    e Zd ZdZddZddZddZdd	Zdd
ZddZ	dS )PyArrowTablez
    Protocol to match PyArrow tables without needing PyArrow installed.

    Only use for function arguments, not return types.
    rG   r   rH   r-   c                    d S r0   r1   rJ   s      r4   filterzPyArrowTable.filterZ   r@   r6   c                    d S r0   r1   rJ   s      r4   group_byzPyArrowTable.group_by[   r@   r6   c                    d S r0   r1   rJ   s      r4   
add_columnzPyArrowTable.add_column\   r@   r6   c                    d S r0   r1   rJ   s      r4   remove_columnzPyArrowTable.remove_column]   r@   r6   c                    d S r0   r1   rJ   s      r4   takezPyArrowTable.take^   r@   r6   c                    d S r0   r1   rJ   s      r4   	to_pandaszPyArrowTable.to_pandas_   r@   r6   NrT   )
r7   r8   r9   r:   rb   rd   rf   rh   rj   rl   r1   r6   r4   r`   r`   S   sd          <;;;====????BBBB9999>>>>>>r6   r`   c                  2    e Zd ZdZddZddZddZdd	Zd
S )PandasDataFramez
    Protocol to match pandas dataframes without needing pandas-stubs installed.

    Only use for function arguments, not return types.
    rG   r   rH   r-   c                    d S r0   r1   rJ   s      r4   wherezPandasDataFrame.wherei   r@   r6   c                    d S r0   r1   rJ   s      r4   groupbyzPandasDataFrame.groupbyj   r@   r6   c                    d S r0   r1   rJ   s      r4   unstackzPandasDataFrame.unstackk   r@   r6   c                    d S r0   r1   rJ   s      r4   pivot_tablezPandasDataFrame.pivot_tablel   r@   r6   NrT   )r7   r8   r9   r:   rp   rr   rt   rv   r1   r6   r4   rn   rn   b   sL          ;:::<<<<<<<<@@@@@@r6   rn   c                  "    e Zd ZdZd	dZd	dZdS )
PandasSeriesz
    Protocol to match pandas series without needing pandas-stubs installed.

    Only use for function arguments, not return types.
    rG   r   rH   r-   c                    d S r0   r1   rJ   s      r4   to_framezPandasSeries.to_framev   r@   r6   c                    d S r0   r1   rJ   s      r4   isnazPandasSeries.isnaw   r@   r6   NrT   )r7   r8   r9   r:   rz   r|   r1   r6   r4   rx   rx   o   s4          >===999999r6   rx   c                  "    e Zd ZdZd	dZd	dZdS )
PandasIndexz
    Protocol to match pandas indexes without needing pandas-stubs installed.

    Only use for function arguments, not return types.
    rG   r   rH   r-   c                    d S r0   r1   rJ   s      r4   	to_serieszPandasIndex.to_series   r@   r6   c                    d S r0   r1   rJ   s      r4   r|   zPandasIndex.isna   r@   r6   NrT   )r7   r8   r9   r:   r   r|   r1   r6   r4   r~   r~   z   s4          ?>>>999999r6   r~   c                  "    e Zd ZdZd	dZd	dZdS )
TorchTensorz
    Protocol to match PyTorch tensors without needing PyTorch installed.

    Only use for function arguments, not return types.
    rG   r   rH   r-   c                    d S r0   r1   rJ   s      r4   cudazTorchTensor.cuda   r@   r6   c                    d S r0   r1   rJ   s      r4   backwardzTorchTensor.backward   r@   r6   NrT   )r7   r8   r9   r:   r   r   r1   r6   r4   r   r      s4          :999======r6   r   )r$   r#   r   PolarsDataTyper&   PolarsTemporalTyper%   PolarsIntegerTypeOneOrMoreDataTypesr   r   r   r   .r   NPythonDataTypeSchemaDefinition
SchemaDictNumericLiteral)r   r   r   r   TemporalLiteralNonNestedLiteralznp.ndarray[Any, Any]PythonLiteralr   r!   IntoExprColumnIntoExpr)eqneqgtltgt_eqlt_eqComparisonOperator)leftcenterrightLEFTCENTERRIGHT	Alignmentr(   SelectorTypestrColumnNameOrSelector)earliestlatestraisenull	Ambiguous)uncompressedsnappydeflateAvroCompression)	necessaryalwaysnon_numericneverCsvQuoteStyle)physicallexicalCategoricalOrdering)r   gzipzstdCsvCompression)utf8z
utf8-lossyCsvEncodingziceberg-column-mappingColumnMappingicebergDefaultFieldValuesziceberg-position-deletezdelta-deletion-vectorr   DeletionFiles)forwardr   minmaxmeanzerooneFillNullStrategy)fullmixedFloatFmt)cfortran
IndexOrder)r   lz4r   IpcCompression)zm:mzm:1z1:mz1:1JoinValidation)r   r   	datapointLabel)noner   r   
left_right
right_leftMaintainOrderJoinNdjsonCompression)r   r   NonExistent)ignoredropNullBehavior)autocolumns
row_groupsprefilteredr   ParallelStrategy)r   r   r   r   brotlir   ParquetCompression)	r   r   firstlastsumr   medianlenitemPivotAgg)nearesthigherlowermidpointlinearequiprobableQuantileMethod)averager   r   denseordinalrandom
RankMethod)r   r   r   Roll)half_to_evenhalf_away_from_zeroto_zero	RoundMode)binaryjsonSerializationFormat)littlebig
Endianness)
bkbmbgbtbbytes	kilobytes	megabytes	gigabytes	terabytesSizeUnit)	windowr   mondaytuesday	wednesdaythursdayfridaysaturdaysundayStartBy)dataallSyncOnCloseMethod)nsusmsTimeUnit)NFCNFKCNFDNFKDUnicodeForm)r   r   anyr   UniqueKeepStrategy)vertical
horizontalUnstackDirection)thread_local	threadingMapElementsStrategy)r   r   r   AsofJoinStrategy)r   r   bothr   ClosedInterval)r   r   InterpolationMethod)innerr   r   r   semianticrossouterJoinStrategy)first_non_null	max_widthListToStructWidthStrategy)
r"  vertical_relaxeddiagonaldiagonal_relaxedr#  align
align_fullalign_inner
align_leftalign_right)pearsonspearmanCorrelationMethod)adbc
connectorxDbReadEngine)
sqlalchemyr@  DbWriteEngine)replaceappendfailDbWriteMode)r  r  r  sd)arraydictJaxExportType)colrowOrientation)r   r   r   SearchSortedSide)tensordatasetrL  TorchExportType)hexbase64TransferEncoding)group_to_rowsjoinexplodeWindowMappingStrategy)plaintreeExplainFormat	ArrayLikeFrameInitTypesColumnFormatDictConditionalFormatDictColumnTotalsDefinitionColumnWidthsDefinitionRowTotalsDefinition)fastbalanced	expensiveParametricProfileNames
PolarsTyper    	FrameType
BufferInfo)calamineopenpyxlxlsx2csvExcelSpreadsheetEnginec                  2    e Zd ZU dZded<   ded<   ded<   dS )SeriesBufferszUnderlying buffers of a Series.r!   valueszSeries | NonevalidityoffsetsN)r7   r8   r9   r:   __annotations__r1   r6   r4   rr  rr  p  s9         ))NNNr6   rr  c                      e Zd ZddZdS )BasicConnectionrG   r   rH   r-   c                    dS )zReturn a cursor object.Nr1   rJ   s      r4   cursorzBasicConnection.cursor{  r@   r6   NrT   )r7   r8   r9   rz  r1   r6   r4   rx  rx  z  s(        & & & & & &r6   rx  c                      e Zd ZddZdS )BasicCursorrG   r   rH   r-   c                    dS )zExecute a query.Nr1   rJ   s      r4   executezBasicCursor.execute  r@   r6   NrT   )r7   r8   r9   r~  r1   r6   r4   r|  r|    s(             r6   r|  c                      e Zd ZddZddZdS )	CursorrG   r   rH   r-   c                    dS )zFetch all results.Nr1   rJ   s      r4   fetchallzCursor.fetchall  r@   r6   c                    dS )zFetch results in batches.Nr1   rJ   s      r4   	fetchmanyzCursor.fetchmany  r@   r6   NrT   )r7   r8   r9   r  r  r1   r6   r4   r  r    s<        ! ! ! !( ( ( ( ( (r6   r  )r   r   r   AlchemyConnectionAlchemyAsyncConnectionConnectionOrCursorSingleIndexSelectorMultiIndexSelectorSingleNameSelectorMultiNameSelectorBooleanMaskSingleColSelectorMultiColSelector)r   z	in-memory	streaminggpur'   
EngineType)irr   	PlanStage
FileSource)functionrenamed_parameterstreaming_parameternonkeyword_argumentsparameter_as_multi_positionalDeprecationType)]r   r   r*   r<   r(  r   r  rl  r   r*  ra  r   rc  rd  r   ConcatMethodrb  r  r?  r   r   r  rB  rD  rH  r  r   r  EpochTimeUnitrp  r^  r  r   r   r`  rk  r   r+  r   r   r   JSONEncoderrM  r1  r   r   r4  r   r'  r  r  r  r   r   r   r   r   r   rP  r   ri  r   r   r   r   r   rj  r   r   r   r   r   re  r   r   rQ  r   r   rr  r  r  r  r
  r  r  r   r  rT  rW  r  r!  r$  r[  c                  &    e Zd ZU dZddZded<   dS )	ParquetMetadataContextz
    The context given when writing file-level parquet metadata.

    .. warning::
        This functionality is considered **experimental**. It may be removed or
        changed at any point without it being considered a breaking change.
    arrow_schemar   r-   Nonec                   || _         d S r0   )r  )r3   r  s     r4   __init__zParquetMetadataContext.__init__3  s    (r6   N)r  r   r-   r  )r7   r8   r9   r:   r  rv  r1   r6   r4   r  r  *  s>          ) ) ) ) r6   r  ParquetMetadataFnParquetMetadataStorageOptionsDict)
__future__r   collections.abcr   r   r   r   r   pathlibr	   typingr
   r   r   r   r   r   r   r   r   r   r   r   decimalr   r   sqlalchemy.enginer   r   sqlalchemy.ext.asyncior   r   r   sqlalchemy.ormr   polarsr   r   r    r!   polars._dependenciesr"   nppolars.datatypesr#   r$   r%   r&   polars.lazyframe.engine_configr'   polars.selectorsr(   r*   r<   rB   rF   rV   r\   r`   rn   rx   r~   r   r   rv  typer   r   r   intfloatboolr   listtupler  r=   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rL  r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r
  r  r  r  r  r!  r$  r'  r(  r*  r+  r1  r4  r  r?  rB  rD  rH  r  rM  rP  rQ  rT  rW  r[  r^  r_  r`  ra  rb  rc  rd  re  ri  rj  rk  rl  rp  rr  rx  r|  r  r  r  r  r  sliceranger  r  r  r  r  r  r  r  r  r  r  __all__r  r  r  r  r1   r6   r4   <module>r     s   " " " " " " " M M M M M M M M M M M M M M      	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	  *888888888888      44444444QQQQQQQQQQ&&&&&&999999999999000000SSSSSSSSSSSS888888))))))$ $ $ $ $8 $ $ $X X X X XH X X X0 0 0 0 0H 0 0 0> > > > > > > >< < < < <8 < < <@ @ @ @ @( @ @ @? ? ? ? ?8 ? ? ?
A 
A 
A 
A 
Ah 
A 
A 
A: : : : :8 : : :: : : : :( : : :> > > > >( > > > ""=> > > > > %d>&:N&J K  K K K K$T-%8-%GH  H H H H ..1I I  I I I II
5k
4j 3i 6l	
 6l : ; 49o 5c?	 5k
 6l 9o 4j     $ C.04778sU3 ?$ FFGGHI       ^ 34
 4 4 4 4!#ui"78 8 8 8 8"#JK K K K K,>DtKeS  S S S S !13I4PS9!TU U U U U!&(C"78 8 8 8 8#n4t; ; ; ; ; '(Q R  R R R RST	 T T T T % $ $ $ $"'|(;"<  < < < < DE	 E E E E$%HI I I I I"#PQ Q Q Q Q!()>!?  ? ? ? ?#$BC C C C C !56 6 6 6 6 $% 	     !&IS%#"6677!      
'+
,d3S	>.B
BCG+,h}k7Q.RRST      &>      o. . . . ./
 / / / /#$AB B B B B#$>? ? ? ? ?78 8 8 8 8&7        ''EF  F F F F 1 1 1 1 1!"23 3 3 3 3%:      !(=!      I     $F       UV
 V V V V89 9 9 9 9OP	 P P P P!()9!:  : : : :0
 0 0 0 0	     
 
 
 
 
  '}5  5 5 5 5./ / / / / !=> > > > > '(F G  G G G G%&>?  ? ? ? ?!()D!E  E E E E &&FG  G G G G#$CD D D D D!()<!=  = = = =!F     (//L'M  M M M M 	  ''<=  = = = =!"67 7 7 7 7"#78 8 8 8 8 !<= = = = =23"?3 3 3 3 3 . . . . .%&<=  = = = =$%@A A A A A%o6  6 6 6 6#*+M#N  N N N N"?3 3 3 3 3 SM
	 
 
 
 
 "C--445SM
 
 
 
 
 &5!5s!:;;'#s(
     
 $+:c?*'#s(
hsWS#X->'>??A$       5)=#>>CDsm
       %S/C"7783>      Cz#&&'*S/9D@     
 %,,K$L  L L L L W\;XvNN
GKk::	c3m,
 , , , , %,,N$O  O O O O    I   & & & & &h & & &
    (   
( ( ( ( ([ ( ( (  %%FG  G G G G$)4%      k!F*->>AWW     
 "%  $ $ $ $ %		SM	!      !$  # # # #$	SM       TN    
  35GG  G G G G03DD{R  R R R R 34kA
     /0	 0 0 0 0 
i  3i	
 4j 2e9o 5k  	 	 	 	 ue|$xs
';;$%    ^ ^ ^B         ()?(@$sCx.(PQ  Q Q Q Q!#s(^.?? ? ? ? ? $S#X  . . . . . .r6   