File: //home/arjun/.local/lib/python3.10/site-packages/langsmith/__pycache__/run_trees.cpython-310.pyc
o
���gaX � @ s� U d Z ddlmZ ddlZddlZddlZddlmZmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZ ddlmZmZ z
ddlmZmZ W n eyV ddlmZmZ Y nw ddlZddlZ ddl m Z dd l m Z dd
l!m"Z"m#Z#m$Z$m%Z%m&Z& e�'e(�Z)d Z*e�+e*� d ��Z,e,�-d
�Z.e�+e*� d��Z/e�+e*� d��Z0e�+e*� d��Z1da2de3d< e�4� Z5d(dd�Z6G dd� de j7�Z8G dd � d �Z9d)d d!�Z:d*d&d'�Z;ddgZ<dS )+z Schemas for the LangSmith API.� )� annotationsN)�datetime�timezone) �Any�Dict�List�Mapping�Optional�Sequence�Tuple�Union�cast)�UUID�uuid4)�Field�root_validator)�schemas)�utils)�ID_TYPE�
RUN_TYPE_T�Client� _dumps_json� _ensure_uuidz
langsmith-�trace�utf-8�metadata�tags�project�Optional[Client]�_CLIENT� init_kwargsr �returnr c K s" t d u rt d u rtdi | ��a t S )N� )r r )r r" r" �F/home/arjun/.local/lib/python3.10/site-packages/langsmith/run_trees.py�get_cached_client* s r$ c
sx e Zd ZU dZded< eed�Zded< edd�Zded < ed
d � d�Z d ed
< eddd�Z
ded< ee ddhid�Z ded< edd � dd�Z
ded< edd d �Zd ed < eed�Zd ed!< ee d�Zd"ed#< ee d�Zd$ed%< eddd�Zd&ed'< ed(d)d*�Zded+< ed(d,d*�Zded-< ed.d/d*�Zd0ed1< G d2d3� d3�Zedd4�d�d8d9��Zed.d4�d�d:d;��Zed�d=d>��Z ed�d@dA��Z � fdBdC�Z d�dFdG�Z d�dJdK�Z d�dMdN�Z!d�dPdQ�Z"d�dSdT�Z#ddddddU�d�d]d^�Z$ d�dddddddddddd_� d�dgdh�Z%didj� Z&d�d�dmdn�Z'd�dodp�Z(d�dqdr�Z)d�dsdt�Z*e+d�dwdx��Z,e+d�d{d|��Z-e+d�dd���Z.d�d�d��Z/d�d�� Z0� Z1S )��RunTreez1Run Schema with back-references for posting runs.�str�name)�default_factoryr �id�chain)�default�run_typec C s t �tj�S �N)r �nowr �utcr" r" r" r# �<lambda>8 � zRunTree.<lambda>r �
start_timeNT)r+ �exclude�Optional[RunTree]�
parent_run�__all__�
parent_run_id)r( r3 z
List[RunTree]�
child_runsc C s t �� pdS )Nr+ )r �get_tracer_projectr" r" r" r# r0 ? r1 � project_name)r( �alias� session_name�
project_id)r+ r; �Optional[UUID]�
session_idr �extra�Optional[List[str]]r z
List[Dict]�eventsz
Optional[Any]� ls_client� z!The order of the run in the tree.)r+ � description� dotted_orderzThe trace id of the run.�trace_idFz3Whether to allow filesystem access for attachments.zOptional[bool]� dangerously_allow_filesystemc @ s e Zd ZdZdZdZdZdS )zRunTree.Configz Pydantic model configuration.T�ignoreN)�__name__�
__module__� __qualname__�__doc__�arbitrary_types_allowed� allow_population_by_field_namer@ r" r" r" r# �ConfigQ s
rP )�pre�values�dictr! c C sx |� d�du r-|� d�dur-d|d v r |d d |d< nd|d v r-|d d d |d< |� d�du r8d|d< d|v rD|�d�|d< n d |v rO|�d �|d< |� d�sXd|d< |� d
�durf|d
j|d < d|vrot� |d< d |vr�d
|v r|d
j|d < n|d |d < tt|�d
i �� |� d�du r�g |d< |� d�du r�g |d< |� d�du r�i |d< |� d�du r�i |d< |S )zAssign name to the run.r' N�
serializedr) ����Unnamed�clientrC �_clientr5 r7 rG r@ rB r �outputs� attachments)�get�popr) r rG r
rS �
setdefault)�clsrR r" r" r# �infer_defaultsX s>
zRunTree.infer_defaultsc C sZ |� d�}|r
|�� r
|S t|d |d �}|d r'|d jd | |d< |S ||d< |S )z#Ensure the dotted order of the run.rF r2 r) r5 �.)r[ �strip� _create_current_dotted_orderrF )r^ rR �current_dotted_orderr" r" r# �ensure_dotted_order| s
���zRunTree.ensure_dotted_orderr c C s | j du r t� | _ | j S )zReturn the client.N)rC r$ ��selfr" r" r# rW � s
zRunTree.clientr c C s | j S r- )rC re r" r" r# rX � s zRunTree._clientc s |dkr || _ dS t� �||�S )zSet the _client specially.rX N)rC �super� __setattr__)rf r' �value�� __class__r" r# rh � s
zRunTree.__setattr__�Union[Sequence[str], str]�Nonec C s0 t |t�r|g}| jdu rg | _| j�|� dS )zAdd tags to the run.N)�
isinstancer& r �extend)rf r r" r" r# �add_tags� s
zRunTree.add_tagsr �Dict[str, Any]c C s2 | j du ri | _ tt| j ��di �}|�|� dS )zAdd metadata to the run.Nr )r@ r
rS r] �update)rf r � metadata_r" r" r# � add_metadata� s
zRunTree.add_metadatarY c C � | j du ri | _ | j �|� dS �z�Upsert the given outputs into the run.
Args:
outputs (Dict[str, Any]): A dictionary containing the outputs to be added.
Returns:
None
N)rY rr )rf rY r" r" r# � add_outputs� �
zRunTree.add_outputs�inputsc C ru rv )ry rr )rf ry r" r" r# �
add_inputs� rx zRunTree.add_inputs�TUnion[ls_schemas.RunEvent, Sequence[ls_schemas.RunEvent], Sequence[dict], dict, str]c C sh | j du rg | _ t|t�r| j �|� dS t|t�r,| j �dt�tj�� � |d�� dS | j �
|� dS )a~ Add an event to the list of events.
Args:
events (Union[ls_schemas.RunEvent, Sequence[ls_schemas.RunEvent],
Sequence[dict], dict, str]):
The event(s) to be added. It can be a single event, a sequence
of events, a sequence of dictionaries, a dictionary, or a string.
Returns:
None
N�event�r' �time�message) rB rn rS �appendr& r r. r r/ � isoformatro )rf rB r" r" r# � add_event� s
��zRunTree.add_event)rY �error�end_timerB r �Optional[Dict]r� �
Optional[str]r� �Optional[datetime]�'Optional[Sequence[ls_schemas.RunEvent]]�Optional[Dict[str, Any]]c C sn |pt �tj�| _|dur| js|| _n| j�|� |dur!|| _|dur*| �|� |dur5| � |� dS dS )z/Set the end time of the run and all child runs.N)
r r. r r/ r� rY rr r� r� rt )rf rY r� r� rB r r" r" r# �end� s
�z RunTree.end) �run_idrT ry rY r� �reference_example_idr2 r� r r@ rZ r r� �Optional[ID_TYPE]rT r� rZ � Optional[ls_schemas.Attachments]c C s� |pd|i}t di d|�dt|��d|�d|pi �d|p i �d|�d|�d|�d | p/t�tj��d
|
�d | p7i �d | �d
| j�d| j�d| �d|
pYi �d| j��}| j �
|� |S �d| j��}| j �
|� |S )z Add a child run to the run tree.r' r) rT ry rY r� r, r� r2 r� r@ r5 r: rC r rZ rH Nr" ) r% r r r. r r/ r< rC rH r8 r� )rf r' r, r� rT ry rY r� r� r2 r� r r@ rZ � serialized_�runr" r" r# � create_child s` ���������� ��
� �� �
����� ��� zRunTree.create_childc C sF | j h d�dd�}| jd ur| j�� |d< | jd ur!| j�� |d< |S )N> ry rY r8 T)r3 � exclude_nonery rY )rS ry �copyrY )rf � self_dictr" r" r# �_get_dicts_safe0 s �
zRunTree._get_dicts_safe�exclude_child_runs�boolc C s~ | � � }| jjd i |�� |�d� }r-dd� |D �}| j�dt�tj ��
� t |�d�� |s;| j D ]
}|j
dd� q2dS dS )
z,Post the run tree to the API asynchronously.rZ c S s g | ]}t |��qS r" )r& )�.0r' r" r" r# �
<listcomp>B s z RunTree.post.<locals>.<listcomp>�uploaded_attachmentr} F)r� Nr" )r� rW �
create_runr[ rB r� r r. r r/ r� �setr8 �post)rf r� �kwargsrZ �keys� child_runr" r" r# r� = s ��
�z RunTree.postc s� | j s| �� dd� | j�� D �}z |r,tdd� | jD �d�� � r,� fdd�|�� D �}W n tyG } zt�d|� �� W Y d}~nd}~ww | j j
| j | j | j
rW| j
�� nd| jr`| j�� nd| j| j| j| j | j| j| j| j| j|d� dS ) z5Patch the run tree to the API in a background thread.c S s i | ] \}}t |t�r||�qS r" )rn �tuple�r� �a�vr" r" r# �
<dictcomp>R s
��z!RunTree.patch.<locals>.<dictcomp>c s s" � | ] }|� d �dkr|V qdS )r' r� N)r[ )r� �evr" r" r# � <genexpr>Y s � ��z RunTree.patch.<locals>.<genexpr>Nc s" i | ]
\}}|� d vr||�qS )r r" r� ��uploadedr" r# r� a s
�z'Error filtering attachments to upload: )r' r� ry rY r� r7 r� r� rF rG rB r r@ rZ )r� r� rZ �items�nextrB � Exception�logger�warningrW �
update_runr' r) ry r� rY r� r7 r� rF rG r r@ )rf rZ �er" r� r# �patchN sL ���
�� ��
�z
RunTree.patchc C s dS )z"Wait for all _futures to complete.Nr" re r" r" r# �waity s z RunTree.waitc C s | j j| d�S )zReturn the URL of the run.)r� )rW � get_run_urlre r" r" r# �get_url} s zRunTree.get_urlr� r c K s t |i}tt| j|fi |���S )zwCreate a new 'child' span from the provided dotted order.
Returns:
RunTree: The new span.
)�LANGSMITH_DOTTED_ORDERr
r% � from_headers)r^ rF r� �headersr" r" r# �from_dotted_order� s �zRunTree.from_dotted_order�config�Optional[dict]c
s� zddl m}m} ddlm}m} ddlm� W n ty) } ztd�|�d}~ww |du r=|t |t
�r9t ||�nd�}nt ||�}|� d� } r�t | ||f�r�| j
r�t� fdd �| jD �d� }
r�|
j� t| j
�� } r�| jr�| j} | j|d
< | j|d < | j|d < | j|d
< | j|d< tt| jp�g |� dg � ��|d< | j|d< |� di �}
|
� di �}|� | j � nt |
d�r�| j
|
j v r�|
j | j
d } ndS |
j!|d< |
j"|d< t#j$| fi |��S dS )z�Create a new 'child' span from the provided runnable config.
Requires langchain to be installed.
Returns:
Optional[RunTree]: The new span or None if
no parent span information is found.
r )�AsyncCallbackManager�CallbackManager)�RunnableConfig�
ensure_config��LangChainTracerz{RunTree.from_runnable_config requires langchain-core to be installed. You can install it with `pip install langchain-core`.N� callbacksc 3 s � | ]
}t |� �r|V qd S r- )rn )r� �tr� r" r# r� � s � z/RunTree.from_runnable_config.<locals>.<genexpr>r, ry rY r2 r� r r' r@ r � order_map� rW r: )%� langchain_core.callbacks.managerr� r� �langchain_core.runnablesr� r� � langchain_core.tracers.langchainr� � ImportErrorrn rS r
r[ r7 r� �handlers�run_mapr& rF r, ry rY r2 r� �sortedr� r r' r] rr r �hasattrr� rW r: r% r� )r^ r� r� r� r� r� r� r� �config_�cb�tracerr� rF �extra_rs r" r� r# �from_runnable_config� s^ �����
� ����
"
z RunTree.from_runnable_configr� �-Mapping[Union[str, bytes], Union[str, bytes]]c K s| |� � }ttt |�t��}|s#ttt |�t��}|s dS |�d�}|� � }t
|�}|d d }||d< |d d |d< ||d< t |�d krO|d
d |d < |�d �pYt �
tj�|d < |�d
�pbd|d
< |�d�pkd|d< t�|�} | jsy| jr�|�di �|d< |d �di �|d d< i | j�|d d �}
|
|d d< tt| j|�dg � ��} | |d< | jr�| j|d< tdi |��S )a� Create a new 'parent' span from the provided headers.
Extracts parent span information from the headers and creates a new span.
Metadata and tags are extracted from the baggage header.
The dotted order and trace id are extracted from the trace header.
Returns:
Optional[RunTree]: The new span or None if
no parent span information is found.
Nr r r� rG rU r) rF � ���r7 r2 r, r* r' �parentr@ r r r: r" )r� r
r r& r[ r� �bytes� LANGSMITH_DOTTED_ORDER_BYTES�decodera �_parse_dotted_order�lenr r. r r/ �_Baggager� r r r] r� r� r: r% ) r^ r� r� � init_args�langsmith_trace�langsmith_trace_bytes�parent_dotted_order�parsed_dotted_orderrG �baggager r r" r" r# r� � sF �
�
�
zRunTree.from_headers�Dict[str, str]c C sB i }| j r | j|t� < t| j�di �| j| jd�}|�� |d< |S )z.Return the RunTree as a dictionary of headers.r �r r r: r� ) rG rF r� r� r@ r[ r r< � to_header)rf r� r� r" r" r# �
to_headers s � zRunTree.to_headersc C s& d| j � d| j� d| j� d| j� d� S )z5Return a string representation of the RunTree object.z RunTree(id=z, name='z
', run_type='z', dotted_order='z'))r) r' r, rF re r" r" r# �__repr__ s ���zRunTree.__repr__)rR rS r! rS )r! r )r! r )r rl r! rm )r rq r! rm )rY rq r! rm )ry rq r! rm )rB r{ r! rm ) rY r� r� r� r� r� rB r� r r� r! rm )r* ) r' r& r, r r� r� rT r� ry r� rY r� r� r� r� r> r2 r� r� r� r rA r@ r� rZ r� r! r% )T)r� r� r! rm )r! rm �r! r& )rF r& r� r r! r% )r� r� r� r r! r4 )r� r� r� r r! r4 )r! r� )2rJ rK rL rM �__annotations__r r r) r, r2 r5 �listr8 r< r? rS r@ r rB rC rF rG rH rP r r_ rd �propertyrW rX rh rp rt rw rz r� r� r� r� r� r� r� r� � classmethodr� r� r� r� r� �
__classcell__r" r" rj r# r% 2 s�
���� #
'���)
+
@
8
r% c @ sH e Zd ZdZ ddd d
�Zedd
d��Zeddd��Zddd�ZdS )r� zBaggage header information.Nr �Optional[Dict[str, str]]r rA r: r� c C s |pi | _ |pg | _|| _dS )z Initialize the Baggage object.Nr� )rf r r r: r" r" r# �__init__# s
z_Baggage.__init__� header_valuer! c
C s� |s| � S i }g }d}z8|� d�D ]0}|� dd�\}}|tkr)t�tj�|��}q|tkr7tj�|�� d�}q|tkrAtj�|�}qW n t y] } zt
� d|� �� W Y d}~nd}~ww | |||d�S )z4Create a Baggage object from the given header value.N�,�=r� z Error parsing baggage header: r� ) �split�LANGSMITH_METADATA�json�loads�urllib�parse�unquote�LANGSMITH_TAGS�LANGSMITH_PROJECTr� r� r� ) r^ r� r r r: �item�keyri r� r" r" r# � from_header. s* �� ��z_Baggage.from_headerr� � Mapping[Union[str, bytes], Any]c C sB d|v r | � |d �S d|v r | � tt|d ��d��S | � d �S )Nr� s baggager )r� r
r� r� )r^ r� r" r" r# r� D s
z_Baggage.from_headersr& c C s� g }| j rt| j �}|�t� dtj�|�� �� | jr/d�| j�}|�t� dtj�|�� �� | j rA|�t� dtj�| j �� �� d�|�S )z,Return the Baggage object as a header value.z metadata=r� ztags=zproject=)
r r r� �LANGSMITH_PREFIXr� r� �quoter �joinr: )rf r� �serialized_metadata�serialized_tagsr" r" r# r� M s
� ��
z_Baggage.to_header)NNN)r r� r rA r: r� )r� r� r! r� )r� r� r! r� r� ) rJ rK rL rM r� r� r� r� r� r" r" r" r# r� s � r� rF r& �List[Tuple[datetime, UUID]]c C s | � d�}dd� |D �S )z Parse the dotted order string.r` c S s0 g | ]}t �|d d� d�t|dd � �f�qS )Ni����%Y%m%dT%H%M%S%fZ)r �strptimer )r� �partr" r" r# r� d s "��z'_parse_dotted_order.<locals>.<listcomp>)r� )rF �partsr" r" r# r� a s
�r� r2 r� r� r> c C s, | pt �tj�}|p t� }|�d�t|� S )z Create the current dotted order.r )r r. r r/ r �strftimer& )r2 r� �st�id_r" r" r# rb j s
rb )r r r! r )rF r& r! r )r2 r� r� r> r! r& )=rM �
__future__r r� �logging�sysr r �typingr r r r r r
r r r
�uuidr r � pydantic.v1r r r� �pydantic� threading� urllib.parser� � langsmithr �
ls_schemasr �langsmith.clientr r r r r � getLoggerrJ r� r� �internr� �encoder� r� r� r� r r� �Lock�_LOCKr$ �RunBaser% r� r� rb r6 r" r" r" r# �<module> sH , �
q
A