o
    ưib                     @  s@   d Z ddlmZ ddlZddlZddlmZ G dd deZdS )z$Parquet serializer for Focus export.    )annotationsN   )FocusSerializerc                   @  s   e Zd ZdZdZd
ddZd	S )FocusParquetSerializerz3Serialize normalized Focus frames to Parquet bytes.Zparquetframepl.DataFramereturnbytesc                 C  s8   |  s|ntj|jd}t }|j|dd | S )z/Encode the provided frame as a parquet payload.)schemaZsnappy)compression)Zis_emptyplZ	DataFramer
   ioBytesIOZwrite_parquetgetvalue)selfr   targetbuffer r   e/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/focus/serializers/parquet.py	serialize   s   z FocusParquetSerializer.serializeN)r   r   r   r	   )__name__
__module____qualname____doc__	extensionr   r   r   r   r   r      s    r   )	r   
__future__r   r   Zpolarsr   baser   r   r   r   r   r   <module>   s    