File: //usr/local/lib/python3.10/dist-packages/langsmith/__pycache__/schemas.cpython-310.pyc
o
���g� � @ s� d Z ddlmZ ddlmZmZmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZmZmZmZ ddlmZ ddlmZmZ zdd lmZmZm Z m Z m Z m Z W n e ym dd l!mZmZm Z m Z m Z m Z Y nw dd
l"m#Z# dd lm$Z$ ee e e d f Z%ee e&d f Z'G d
d� de�Z(e e&eee&e)f e(ee&e#f f f Z* eG dd� de��Z+G dd� de�Z,G dd� de,�Z-G dd� de�Z.G dd� de.�Z/G dd� de�Z0G dd � d e,�Z1G d d � d e,�Z2G d d � d e�Z3G d!d"� d"e�Z4G d#d$� d$e4�Z5G d%d&� d&e&e �Z6G d'd(� d(e�Z7e$d) Z8G d*d+� d+ed,d-�Z9G d.d/� d/e7�Z:G d0d1� d1e�Z;d2d3� Z<G d4d5� d5e�Z=G d6d7� d7e=�Z>G d8d9� d9e&e �Z?G d:d;� d;ed,d-�Z@G d<d=� d=e=�ZAG d>d?� d?e�ZBG d@dA� dAeB�ZCG dBdC� dCeB�ZDG dDdE� dEe �ZEG dFdG� dGe�ZFG dHdI� dIed,d-�ZGG dJdK� dKed,d-�ZHG dLdM� dMeF�ZIG dNdO� dOeF�ZJG dPdQ� dQe�ZKG dRdS� dSeK�ZLeG dTdU� dUe��ZMG dVdW� dWed,d-�ZNG dXdY� dYe�ZOG dZd[� d[ed,d-�ZPG d\d]� d]e�ZQe1�R� G d^d_� d_e�ZSG d`da� dae�ZTG dbdc� dced,d-�ZUG ddde� deed,d-�ZVG dfdg� dge�ZWG dhdi� die�ZXG djdk� dke�ZYG dldm� dme�ZZG dndo� doe�Z[G dpdq� dqe�Z\G drds� dse&e �Z]G dtdu� dued,d-�Z^G dvdw� dwed,d-�Z_G dxdy� dye�Z`G dzd{� d{e�Zad S )|z Schemas for the LangSmith API.� )� annotations)�datetime� timedelta�timezone)�Decimal)�Enum) �Any�Dict�List�
NamedTuple�Optional�Protocol�Tuple�Union�runtime_checkable)�UUID)� NotRequired� TypedDict)� BaseModel�Field� PrivateAttr�
StrictBool� StrictFloat� StrictInt)�Path)�LiteralNc @ �" e Zd ZU dZded< ded< dS )�
AttachmentaB Annotated type that will be stored as an attachment if used.
Examples:
--------
.. code-block:: python
@traceable
def my_function(bar: int, my_val: Attachment):
# my_val will be stored as an attachment
# bar will be stored as inputs
return bar
�str� mime_typezUnion[bytes, Path]�dataN��__name__�
__module__� __qualname__�__doc__�__annotations__� r' r' �</usr/local/lib/python3.10/dist-packages/langsmith/schemas.pyr 2 s
r c @ s2 e Zd ZdZdddd�Zddd d
�Zddd�ZdS )� BinaryIOLikez$Protocol for binary IO-like objects.����size�int�return�bytesc C � dS )zRead function.Nr' )�selfr+ r' r' r( �readM � zBinaryIOLike.readr �offset�whencec C r/ )zSeek function.Nr' )r0 r3 r4 r' r' r( �seekQ r2 zBinaryIOLike.seekc C r/ )zGet value function.Nr' �r0 r' r' r( �getvalueU r2 zBinaryIOLike.getvalueN)r* )r+ r, r- r. )r )r3 r, r4 r, r- r, )r- r. )r" r# r$ r% r1 r5 r7 r' r' r' r( r) I s
r) c @ s^ e Zd ZU dZded< eed�Zded< edd�Zd ed
< edd�Z d ed < G d d
� d
�Z
dS )� ExampleBasezExample base model.r �
dataset_id��default_factory�Dict[str, Any]�inputsN��default�Optional[Dict[str, Any]]�outputs�metadatac @ s e Zd ZdZdZdZdS )zExampleBase.Config�#Configuration class for the schema.TN)r" r# r$ r% �frozen�arbitrary_types_allowedr' r' r' r( �Configb s rF ) r" r# r$ r% r& r �dictr= rA rB rF r' r' r' r( r8 Z s
r8 c @ s< e Zd ZU dZded< edd� d�Zded< d Zd
ed < d S ) �
ExampleCreatezExample create model.�Optional[UUID]�idc C � t �tj�S �N�r �nowr �utcr' r' r' r( �<lambda>m � zExampleCreate.<lambda>r: r �
created_atN� Optional[Union[str, List[str]]]�split)r" r# r$ r% r& r rR rT r' r' r' r( rH i s
rH c @ s~ e Zd ZU dZded< edd� d�Zded< eed�Zd ed
< ed d �Z d
ed< ed d �Z
d
ed< d Z ded< d Z ded< d S )� ExampleUploadWithAttachmentsz Example upload with attachments.rI rJ c C rK rL rM r' r' r' r( rP u rQ z%ExampleUploadWithAttachments.<lambda>r: r rR r< r= Nr> r@ rA rB rS rT �Optional[Attachments]� attachments)
r" r# r$ r% r& r rR rG r= rA rB rT rW r' r' r' r( rU q s
rU c @ � e Zd ZU dZded< dS )� ExampleUpsertWithAttachmentsz Example create with attachments.r r9 Nr! r' r' r' r( rY } �
rY c @ s* e Zd ZU dZded< ded< ded< dS ) �AttachmentInfozInfo for an attachment.r �
presigned_urlr) �reader�
Optional[str]r Nr! r' r' r' r( r[ � s
r[ c s� e Zd ZU dZded< edd� d�Zded< eed �d
�Zded < ed d
�Z d
ed< ee
d�Z ded< d Z ded< ed d
�Z
ded< ed d
�Zded< ed d
�Zded< d"d#� fd d �
Zed$d d ��Zd d!� Z� ZS )%�ExamplezExample model.r rJ c C s t jdtjd�S )Nr )�tz)r �
fromtimestampr rO r' r' r' r( rP � s zExample.<lambda>r: r rR z$00000000-0000-0000-0000-000000000000r> r9 N�Optional[datetime]� modified_atz List[Run]�runsrI �
source_run_idz#Optional[Dict[str, AttachmentInfo]]rW r^ � _host_url�
_tenant_id�kwargsr r- �Nonec s" t � jdi |�� || _|| _dS )� Initialize a Dataset object.Nr' )�super�__init__rf rg )r0 rf rg rh �� __class__r' r( rl � s
zExample.__init__c C sL | j r$d| j� d| j� �}| jr | j � dt| j�� |� �S | j � |� �S dS )� URL of this run within the app.�
/datasets/z/e/�/o/N)rf r9 rJ rg r )r0 �pathr' r' r( �url� s z Example.urlc C �$ | j � d| j� d| j� d| j� d�S )�5Return a string representation of the RunBase object.�(id=z
, dataset_id=z, link='�'))rn rJ r9 rs r6 r' r' r( �__repr__� � $zExample.__repr__)NN)rf r^ rg rI rh r r- ri �r- r^ )r" r# r$ r% r& r rR r r9 rc �listrd re rW r rf rg rl �propertyrs rx �
__classcell__r' r' rm r( r_ � s(
� � r_ c @ rX )�
ExampleSearchz Example returned via search.r rJ Nr! r' r' r' r( r~ � rZ r~ c @ s: e Zd ZU dZeedd�Zded< eedd�Z ded< d S )
�AttachmentsOperationsz%Operations to perform on attachments.z,Mapping of old attachment names to new names)r; � descriptionzDict[str, str]�renamez List of attachment names to keep� List[str]�retainN)
r" r# r$ r% r rG r� r& r{ r� r' r' r' r( r � s
��r c @ sh e Zd ZU dZdZded< dZded< dZded< dZded < dZ ded
< dZ
d ed < G d
d� d�Z dS )�
ExampleUpdatezUpdate class for Example.NrI r9 r@ r= rA � Optional[AttachmentsOperations]�attachments_operationsrB rS rT c @ � e Zd ZdZdZdS )zExampleUpdate.ConfigrC TN�r" r# r$ r% rD r' r' r' r( rF � � rF ) r" r# r$ r% r9 r& r= rA r� rB rT rF r' r' r' r( r� � s
r� c @ st e Zd ZU dZded< eed�Zded< edd�Zd ed
< edd�Z d ed < dZ
d ed
< dZ ded< dZ ded< dS )� ExampleUpdateWithAttachmentsz Example update with attachments.r rJ r: r< r= Nr> r@ rA rB rS rT rV rW r� r� )
r" r# r$ r% r& r rG r= rA rB rT rW r� r' r' r' r( r� � s
r� c @ s e Zd ZdZdZdZdZdS )�DataTypez Enum for dataset data types.�kv�llm�chatN)r" r# r$ r% r� r� r� r' r' r' r( r� � s
r� c @ s@ e Zd ZU dZded< dZded< dZded< G d d
� d
�ZdS ) � DatasetBasezDataset base model.r �nameNr^ r� zOptional[DataType]� data_typec @ r� )zDatasetBase.ConfigrC TNr� r' r' r' r( rF � r� rF )r" r# r$ r% r& r� r� rF r' r' r' r( r� � s
r� )�remove_system_messages�convert_to_openai_message�convert_to_openai_tool�remove_extra_fields�extract_tools_from_runc @ r )�DatasetTransformationz#Schema for dataset transformations.r� rr z%Union[DatasetTransformationType, str]�transformation_typeNr! r' r' r' r( r� s
r� F)�totalc s� e Zd ZU dZded< ded< edd�Zded < dZd
ed < dZd
ed < dZ ded
< dZ
ded< dZ ded< dZ ded< e
dd�Zded< e
dd�Zded< e
dd�Zded< d d!� fd d �
Zed"d d ��Z� ZS )#�DatasetzDataset ORM model.r rJ r rR Nr> rb rc �
Optional[int]�
example_count�
session_count�last_session_start_timer@ �
inputs_schema�outputs_schemaz%Optional[List[DatasetTransformation]]�transformationsr^ rf rI rg � _public_pathrh r r- ri c sT d|v r |� d�|d< d|v r|� d�|d< t� jdi |�� || _|| _|| _dS )rj �inputs_schema_definitionr� �outputs_schema_definitionr� Nr' )�poprk rl rf rg r� )r0 rf rg r� rh rm r' r( rl s
zDataset.__init__c C sV | j r)| jr| j � | j� �S | jr | j � dt| j�� d| j� �S | j � d| j� �S dS )ro rq rp N)rf r� rg r rJ r6 r' r' r( rs + s z Dataset.url)NNN)
rf r^ rg rI r� r^ rh r r- ri rz )r" r# r$ r% r& r rc r� r� r� r� r� r� r rf rg r� rl r| rs r} r' r' rm r( r� s(
�r� c @ s& e Zd ZU dZdZded< ded< dS )�DatasetVersionz%Class representing a dataset version.N�Optional[List[str]]�tagsr �as_of)r" r# r$ r% r� r& r' r' r' r( r� 7 s
r� c C s di iS )NrB r' r' r' r' r( �_default_extra> s r� c @ s e Zd ZU dZded< ded< ded< ded< d Zd
ed < eed �Zd
ed< d Z ded< d Z
d
ed< d Z ded< ee d �Z
ded< d Zd
ed< d Zded< d Zded< d Zded< ee d �Zd ed < ed(d d!��Zed)d"d#��Zd$d%� ZG d&d'� d'�Zd S )*�RunBasea3 Base Run schema.
A Run is a span representing a single unit of work or operation within your LLM app.
This could be a single call to an LLM or chain, to a prompt formatting call,
to a runnable lambda invocation. If you are familiar with OpenTelemetry,
you can think of a run as a span.
r rJ r r� r �
start_time�run_typeNrb �end_timer: �Optional[dict]�extrar^ �error�
serializedzOptional[List[Dict]]�eventsrG r= rA rI �reference_example_id�
parent_run_idr� r� z-Union[Attachments, Dict[str, AttachmentInfo]]rW r- �dict[str, Any]c C s | j du ri | _ | j �di �S �z Retrieve the metadata (if any).NrB )r� �
setdefaultr6 r' r' r( rB } s
zRunBase.metadatac C s | j �d�S )z"Retrieve the revision ID (if any).� revision_id)rB �getr6 r' r' r( r� � s zRunBase.revision_idc C rt )ru rv z, name='z
', run_type='rw )rn rJ r� r� r6 r' r' r( rx � ry zRunBase.__repr__c @ r� )zRunBase.ConfigrC TN)r" r# r$ r% rE r' r' r' r( rF � r� rF �r- r� )r- rI )r" r# r$ r% r& r� r r� r� r� r� r� rG r= rA r� r� r� rW r| rB r� rx rF r' r' r' r( r� B sP
� r� c sD e Zd ZU dZdZded< dZded< dZded< dZd ed
< dZ d ed < dZ
ded
< dZ d ed< dZ ded< dZ
ded< dZded< dZded< dZded< dZded< dZded< dZded< ded< ed d �Zd ed < dZd ed!< edd �Zd ed"< d+d,� fd'd(�
Zed-d)d*��Z� Z S ).�Runz$Run schema when loading from the DB.NrI �
session_id�Optional[List[UUID]]�
child_run_idszOptional[List[Run]]�
child_runsr@ �feedback_statsr^ �app_path� manifest_id�statusr� �
prompt_tokens�completion_tokens� total_tokensrb �first_token_time�Optional[Decimal]�
total_cost� prompt_cost�completion_cost�parent_run_idsr �trace_id� r> r � dotted_order�Optional[bool]�
in_datasetrf rh r r- ri c s~ |� d�sd|� d�i|�}|�dd�pi }t� jdi |�d|i�� || _| j�� s;| js=| j� � � | j
� �| _dS dS dS )�Initialize a Run object.r� rJ r= Nr' ) r� r� rk rl rf r� �stripr� r� � isoformatrJ )r0 rf rh r= rm r' r( rl � s
�z Run.__init__c C s | j r| jr| j � | j� �S dS )ro N)rf r� r6 r' r' r( rs � s zRun.urlrL �rf r^ rh r r- ri rz ) r" r# r$ r% r� r& r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r rf rl r| rs r} r' r' rm r( r� � sT
r� c @ s, e Zd ZdZdZdZdZdZdZdZ dZ
d S )
� RunTypeEnumz5(Deprecated) Enum for run types. Use string directly.�tool�chainr� � retriever� embedding�prompt�parserN) r" r# r$ r% r� r� r� r� r� r� r� r' r' r' r( r� � s r� c @ s� e Zd ZU dZded< ded< ded< ded < ded
< d ed < ded
< ded< ded< ded< ded< ded< ded< ded< ded< ded< ded< ded < ded < ded < ded < d ed!< ded"< d#ed$< d%S )&� RunLikeDictz&Run-like dictionary, for type-hinting.r r� r� r� r r� r� r= rA rb r� r� r^ r� r� rI r� r� �Optional[List[dict]]r� r� r� �inputs_s3_urls�outputs_s3_urlsrJ r� � session_namer� �input_attachments�output_attachmentsr r� r� � AttachmentsrW Nr! r' r' r' r( r� � s4
r� c @ s, e Zd ZU dZdZded< dZded< dS )�RunWithAnnotationQueueInfoz&Run schema with annotation queue info.Nrb �last_reviewed_time�added_at)r" r# r$ r% r� r& r� r' r' r' r( r� s
r� c @ sJ e Zd ZU dZded< eed�Zded< dZded < dZ d
ed < dS ) �FeedbackSourceBasez�Base class for feedback sources.
This represents whether feedback is submitted from the API, model, human labeler,
etc.
r �typer: r@ rB NzOptional[Union[UUID, str]]�user_idr^ � user_name)
r" r# r$ r% r& r rG rB r� r� r' r' r' r( r� s
r� c @ � e Zd ZU dZdZded< dS )�APIFeedbackSourcezAPI feedback source.�apizLiteral['api']r� N�r" r# r$ r% r� r& r' r' r' r( r� �
r� c @ r� )�ModelFeedbackSourcezModel feedback source.�modelzLiteral['model']r� Nr� r' r' r' r( r� # r� r� c @ s e Zd ZdZdZ dZdS )�FeedbackSourceTypezFeedback source type.r� r� N)r" r# r$ r% �API�MODELr' r' r' r( r� ) s r� c @ s� e Zd ZU dZded< dZded< dZded< ded < ded
< d ed < dZd
ed< dZded< dZ ded< dZ
ded< dZ ded< dZ ded< dZ
ded< dZded< dZded< G d d � d �ZdS ) � FeedbackBasezFeedback schema.r rJ Nrb rR rc rI �run_idr� r �key�
SCORE_TYPE�score�
VALUE_TYPE�valuer^ �commentzUnion[str, dict, None]�
correction� Optional[FeedbackSourceBase]�feedback_sourcer� �comparative_experiment_id�feedback_group_idzOptional[Dict]r� c @ r� )zFeedbackBase.ConfigrC TNr� r' r' r' r( rF V r� rF )r" r# r$ r% r& rR rc r� r� r� r� r r� r r r� rF r' r' r' r( r� 2 sB
r� c @ �$ e Zd ZU dZded< ded< dS )�FeedbackCategoryz+Specific value and label pair for feedback.�floatr� r^ �labelNr! r' r' r' r( r \ �
r c @ s8 e Zd ZU dZded< ded< ded< ded< d S )
�FeedbackConfigz:Represents _how_ a feedback value ought to be interpreted.z0Literal['continuous', 'categorical', 'freeform']r� �Optional[float]�min�maxz Optional[List[FeedbackCategory]]�
categoriesNr! r' r' r' r( r e s
r c @ s( e Zd ZU dZded< dZded< dS )�FeedbackCreatez"Schema used for creating feedback.r� r NzOptional[FeedbackConfig]�feedback_config)r" r# r$ r% r& r r' r' r' r( r s s
r c @ s: e Zd ZU dZded< ded< ded< dZded < dS )
�Feedbackz Schema for getting feedback.r rJ r rR rc Nr r )r" r# r$ r% r& r r' r' r' r( r { s
r c s� e Zd ZU dZded< edd� d�Zded< d Zd
ed < d Zd ed
< d Z d ed< d Z
ded< ded< ded< e d d�Z d ed< d$d%� fdd�
Z
ed&d d ��Zed'd d ��Zed(d"d#��Z� ZS ))�
TracerSessionzbTracerSession schema for the API.
Sessions are also referred to as "Projects" in the UI.
r rJ c C rK rL rM r' r' r' r( rP � rQ zTracerSession.<lambda>r: r r� Nrb r� r^ r� r� r@ r� � tenant_idrI �reference_dataset_idr> rf rh r r- ri c s> t � jdi |�� || _| jjdu r | jjtjd�| _dS dS )r� N)�tzinfor' )rk rl rf r� r �replacer rO )r0 rf rh rm r' r( rl � s
�zTracerSession.__init__c C s$ | j r| j � d| j� d| j� �S dS )ro rq z /projects/p/N)rf r rJ r6 r' r' r( rs � s zTracerSession.urlr� c C �" | j du s
d| j vr i S | j d S r� �r� r6 r' r' r( rB � �
zTracerSession.metadatar� c C s" | j du s
d| j vr g S | j d S )zRetrieve the tags (if any).Nr� r r6 r' r' r( r� � r zTracerSession.tagsrL r� rz r� )r- r� )r" r# r$ r% r& r r� r� r� r� r� r rf rl r| rs rB r� r} r' r' rm r( r � s4
r c @ s� e Zd ZU dZded< ded< ded< ded< ded< ded < d
ed < d ed
< ded< ded< ded< ded< ded< ded< ded< dS )�TracerSessionResultzqA project, hydrated with additional information.
Sessions are also referred to as "Projects" in the UI.
r� � run_countzOptional[timedelta]� latency_p50� latency_p99r� r� r� rb �last_run_start_timer@ r� z Optional[List[Dict[str, Any]]]�
run_facetsr� r� r� r� �first_token_p50�first_token_p99r
�
error_rateNr! r' r' r' r( r � s@
r c @ s4 e Zd ZU dZded< ded< ed
dd��Zd S ) �BaseMessageLikez7A protocol representing objects similar to BaseMessage.r �contentzDict[Any, Any]�additional_kwargsr- c C r/ )z,Type of the Message, used for serialization.Nr' r6 r' r' r( r� � s zBaseMessageLike.typeN)r- r )r" r# r$ r% r& r| r� r' r' r' r( r" � s
r" c @ s. e Zd ZU dZded< ded< ded< dS )�DatasetShareSchemaz*Represents the schema for a dataset share.r r9 � share_tokenr rs Nr! r' r' r' r( r% � �
r% c @ sl e Zd ZU dZded< ded< dZded< ed d
� d �Zd ed
< edd
� d �Zd ed< ded< dS )�AnnotationQueuez Represents an annotation queue.r rJ r r� Nr^ r� c C rK rL rM r' r' r' r( rP rQ zAnnotationQueue.<lambda>r: r rR c C rK rL rM r' r' r' r( rP rQ �
updated_atr ) r" r# r$ r% r& r� r rR r) r' r' r' r( r( � s
r( c @ sL e Zd ZU dZded< ded< ded< ded< ded< d ed
< d S ) �BatchIngestConfigz"Configuration for batch ingestion.�bool�use_multipart_endpointr, �scale_up_qsize_trigger�scale_up_nthreads_limit�scale_down_nempty_trigger�
size_limitr� �size_limit_bytesNr! r' r' r' r( r*
s
r* c @ sH e Zd ZU dZdZded< dZded< dZded < dZd
ed < dS ) �
LangSmithInfoz'Information about the LangSmith server.r� r �versionNrb �license_expiration_timezOptional[BatchIngestConfig]�batch_ingest_configr@ �instance_flags) r" r# r$ r% r3 r& r4 r5 r6 r' r' r' r( r2 s
r2 c @ s< e Zd ZU dZded< ded< ded< dZded < dS )
�LangSmithSettingsz"Settings for the LangSmith tenant.r rJ � display_namer rR Nr^ �
tenant_handle)r" r# r$ r% r& r9 r' r' r' r( r7 - s
r7 c @ �. e Zd ZU dZded< ded< ded< dS ) �FeedbackIngestTokenz2Represents the schema for a feedback ingest token.r rJ r rs r �
expires_atNr! r' r' r' r( r; 9 r' r; c @ r: ) �RunEventzRun event schema.r r� zUnion[datetime, str]�timer@ rh Nr! r' r' r' r( r= D r' r= c @ �. e Zd ZU dZded< ded< ded< dS )�TimeDeltaInputzTimedelta input schema.r, �days�hours�minutesNr! r' r' r' r( r@ O r' r@ c @ r? )�DatasetDiffInfoz;Represents the difference information between two datasets.z
List[UUID]�examples_modified�examples_added�examples_removedNr! r' r' r' r( rD Z r' rD c @ s� e Zd ZU dZded< dZded< dZded< ded< d ed
< d ed < ded < dZd
ed< dZded< dZ d
ed< e
ddd��Z dS )�ComparativeExperimentz�Represents a comparative experiment.
This information summarizes evaluation results comparing
two or more models on a given dataset.
r rJ Nr^ r� r� r r rR rc r r@ r� r� �experiments_infor� r- r� c C r r� r r6 r' r' r( rB � r z ComparativeExperiment.metadatar� ) r" r# r$ r% r& r� r� r� rI r� r| rB r' r' r' r( rH e s0
rH c @ �B e Zd ZU dZded< ded< ded< ded< ded < d
S ) � PromptCommitz$Represents a Prompt with a manifest.r �owner�repo� commit_hashr< �manifestz
List[dict]�examplesNr! r' r' r' r( rK � s
rK c @ s� e Zd ZU dZded< ded< ded< dZded < dZded
< dZded < dZd ed
< dZ ded< dZ
ded< e e d�Z
ded< dZded< dZded< dZd ed< dS )�ListedPromptCommitz;Represents a listed prompt commit with associated metadata.r rJ r rL rM NrI r� �repo_id� parent_idr^ rN rb rR r) r: r� �example_run_idsr r� �
num_downloads� num_views�parent_commit_hash)r" r# r$ r% r& r� rR rS rN rR r) r r{ rT rU rV rW r' r' r' r( rQ � s8
rQ c @ s e Zd ZU dZded< dZded< dZded< ded< ded < d
ed < d
ed < d
ed< d
ed< ded< dZded< dZded< ded< ded< ded< ded< ded< dZ ded< dZ
ded < ded < dZ ded < dZ ded < dS ) �Promptz"Represents a Prompt with metadata.r � repo_handleNr^ r� �readmerJ r r rR r) r+ � is_public� is_archivedr� r� �original_repo_id�upstream_repo_idrL � full_namer, � num_likesrU rV r� �liked_by_auth_user�last_commit_hash� num_commits�original_repo_full_name�upstream_repo_full_name)
r" r# r$ r% r& r� rZ r] r^ ra rb rd re r' r' r' r( rX � s\
rX c @ r )�ListPromptsResponsez A list of prompts with metadata.z List[Prompt]�reposr, r� Nr! r' r' r' r( rf � r rf c @ s& e Zd ZdZdZ dZ dZ dZdS )�PromptSortFieldz$Enum for sorting fields for prompts.rU rV r) r` N)r" r# r$ r% rU rV r) r` r' r' r' r( rh � s rh c @ r? )�InputTokenDetailsz~Breakdown of input token counts.
Does *not* need to sum to full input token count. Does *not* need to have all keys.
r, �audio�cache_creation�
cache_readNr! r' r' r' r( ri
s
ri c @ s$ e Zd ZU dZded< ded< dS )�OutputTokenDetailsz�Breakdown of output token counts.
Does *not* need to sum to full output token count. Does *not* need to have all keys.
r, rj � reasoningNr! r' r' r' r( rm s
rm c @ rJ ) �
UsageMetadataz�Usage metadata for a message, such as token counts.
This is a standard representation of token usage that is consistent across models.
r, � input_tokens�
output_tokensr� z NotRequired[InputTokenDetails]�input_token_detailsz NotRequired[OutputTokenDetails]�output_token_detailsNr! r' r' r' r( ro / s
ro c @ r )�UpsertExamplesResponsezCResponse object returned from the upsert_examples_multipart method.r, �countr� � example_idsNr! r' r' r' r( rt G r rt )br% �
__future__r r r r �decimalr �enumr �typingr r r
r r r
r r r �uuidr �typing_extensionsr r � pydantic.v1r r r r r r � ImportError�pydantic�pathlibr r r� r r� r r. r� r) r8 rH rU rY r[ r_ r~ r r� r� r� r� �DatasetTransformationTyper� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r" r% r( r* r2 �update_forward_refsr7 r; r= r@ rD rH rK rQ rX rf rh ri rm ro rt r' r' r' r( �<module> s� , $ $�
& +
� /QG * 6&
$+1