o
    +i?                     @  s  U d Z ddlmZ ddlZddlZddlZddlZddlmZm	Z	 ddl
m
Z
mZ ddlmZmZmZmZ ddlmZmZmZ ddlmZ dd	lmZ dd
lmZ z
ddlmZmZ W n eyl   ddlmZmZ Y nw ddl Z ddl!Z!ddl"Z#ddl$m%  m&Z& ddl'm(Z) ddl'm*Z* ddl+m,Z,m-Z-m.Z.m/Z/m0Z0 e1e2Z3G dd deddZ4dZ5e6e5 dZ7e78dZ9e6e5 dZ:e6e5 dZ;e6e5 dZ<e6e5 dZ=e6dZ>ede? Z@e!A ZBe jCee	e4   dddZDe jCeeE  dddZFede? ZGdZHdSd#d$ZIeGeGeGeGeGfdTd-d.ZJdUd1d2ZKG d3d4 d4e)jLZMG d5d6 d6ZNejOd7d8dVd;d<ZPdWd=d>ZQdXdBdCZRdYdEdFZSdZdJdKZTe&jUaVd&eWdL< d4d4gZXd[dQdRZYdS )\zSchemas for the LangSmith API.    )annotationsN)MappingSequence)datetimetimezone)AnyOptionalUnioncast)NAMESPACE_DNSUUIDuuid5)	TypedDict)uuid7)uuid7_from_datetime)Fieldroot_validator)schemas)utils)ID_TYPE
RUN_TYPE_TClient_dumps_json_ensure_uuidc                   @  s.   e Zd ZU ded< ded< ded< ded< dS )WriteReplicaOptional[str]api_urlapi_keyproject_nameOptional[dict]updatesN)__name__
__module____qualname____annotations__ r%   r%   P/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/langsmith/run_trees.pyr   '   s
   
 r   F)totalz
langsmith-traceutf-8metadatatagsprojectreplicas__omit_auto_outputs	_REPLICASdefault_DISTRIBUTED_PARENT_ID$   init_kwargsr   returnr   c                  K  sX   t d u r*t t d u rtdi | a W d    t S W d    t S 1 s%w   Y  t S )Nr%   )_CLIENT_LOCKr   )r4   r%   r%   r&   get_cached_clientH   s   

r8   clientOptional[Client]enabledOptional[bool]r   r   Optional[list[str]]Optional[dict[str, Any]]c                 C  s   t K | tur	| a|turtj| |t_|tur#tj| |t_|tur0tj	| |t_
|turEtj| |t_W d   dS W d   dS 1 sPw   Y  dS )a	  Configure global LangSmith tracing context.

    This function allows you to set global configuration options for LangSmith
    tracing that will be applied to all subsequent traced operations. It modifies
    context variables that control tracing behavior across your application.

    Do this once at startup to configure the global settings in code.

    If, instead, you wish to only configure tracing for a single invocation,
    use the `tracing_context` context manager instead.

    Args:
        client: A LangSmith Client instance to use for all tracing operations.

            If provided, this client will be used instead of creating new clients.

            Pass `None` to explicitly clear the global client.
        enabled: Whether tracing is enabled.

            Can be:

            - `True`: Enable tracing and send data to LangSmith
            - `False`: Disable tracing completely
            - `'local'`: Enable tracing but only store data locally
            - `None`: Clear the setting (falls back to environment variables)
        project_name: The LangSmith project name where traces will be sent.

            This determines which project dashboard will display your traces.

            Pass `None` to explicitly clear the project name.
        tags: A list of tags to be applied to all traced runs.

            Tags are useful for filtering and organizing runs in the LangSmith UI.

            Pass `None` to explicitly clear all global tags.
        metadata: A dictionary of metadata to attach to all traced runs.

            Metadata can store any additional context about your runs.

            Pass `None` to explicitly clear all global metadata.

    Examples:
        Basic configuration:
        >>> import langsmith as ls
        >>> # Enable tracing with a specific project
        >>> ls.configure(enabled=True, project_name="my-project")

        Set global trace masking:
        >>> def hide_keys(data):
        ...     if not data:
        ...         return {}
        ...     return {k: v for k, v in data.items() if k not in ["key1", "key2"]}
        >>> ls.configure(
        ...     client=ls.Client(
        ...         hide_inputs=hide_keys,
        ...         hide_outputs=hide_keys,
        ...     )
        ... )

        Adding global tags and metadata:
        >>> ls.configure(
        ...     tags=["production", "v1.0"],
        ...     metadata={"environment": "prod", "version": "1.0.0"},
        ... )

        Disabling tracing:
        >>> ls.configure(enabled=False)
    N)r7   	_SENTINELr6   _context_TRACING_ENABLEDset_GLOBAL_TRACING_ENABLED_PROJECT_NAME_GLOBAL_PROJECT_NAME_TAGS_GLOBAL_TAGS	_METADATA_GLOBAL_METADATA)r9   r;   r   r+   r*   r%   r%   r&   	configureQ   s$   L"rJ   data!ls_schemas.ExtractedUsageMetadatac                 C  s.   h d}t |  | }|rtd| | S )z2Validate that the dict only contains allowed keys.>
   
input_cost
total_costoutput_costinput_tokenstotal_tokensoutput_tokensinput_cost_detailsinput_token_detailsoutput_cost_detailsoutput_token_detailsz#Unexpected keys in usage metadata: )rB   keys
ValueError)rK   allowed_keys
extra_keysr%   r%   r&   !validate_extracted_usage_metadata   s
   r[   c                      s  e Zd ZU dZded< eedZded< eddZded	< ed
d dZ	ded< edddZ
ded< edddZded< eeddhidZded< edd ddZded< edddZd ed!< eedZd"ed#< eedZd$ed%< eedZd&ed'< 	 edddZd(ed)< ed*d+d,Zded-< ed*d.d,Zded/< ed0d1d,Zd2ed3< edd4d,Zd5ed6< G d7d8 d8Zedd9dd<d=Zed0d9dd>d?ZeddAdBZeddDdEZ fdFdGZ e!e!e!e!e!dHddQdRZ"ddTdUZ#ddWdXZ$ddYdZZ%dd[d\Z&dd^d_Z'dddddd`ddgdhZ(	dddddddddddddiddqdrZ)dsdt Z*ddwdxZ+	dddzd{Z,ddd~dZ-d0ddddZ.dddZ/dddZ0e1dddZ2e1dddZ3e1dddZ4dddZ5dd Z6  Z7S )RunTreez1Run Schema with back-references for posting runs.strname)default_factoryr   idchainr0   run_typec                   C  s   t tjS N)r   nowr   utcr%   r%   r%   r&   <lambda>       zRunTree.<lambda>r   
start_timeNT)r1   excludeOptional[RunTree]
parent_runr   parent_dotted_order__all__parent_run_id)r_   ri   zlist[RunTree]
child_runsc                   C  s   t  pdS )Nr1   )r   get_tracer_projectr%   r%   r%   r&   rf      rg   r   )r_   aliassession_name
project_id)r1   rq   Optional[UUID]
session_iddictextrar=   r+   z
list[dict]eventszOptional[Any]	ls_client z!The order of the run in the tree.)r1   descriptiondotted_orderzThe trace id of the run.trace_idFz3Whether to allow filesystem access for attachments.r<   dangerously_allow_filesystemz8Projects to replicate this run to with optional updates. Optional[Sequence[WriteReplica]]r-   c                   @  s   e Zd ZdZdZdZdZdS )zRunTree.ConfigzPydantic model configuration.TignoreN)r!   r"   r#   __doc__arbitrary_types_allowedallow_population_by_field_namerw   r%   r%   r%   r&   Config   s
    r   )prevaluesr5   c                 C  s  | ddu r-| ddur-d|d v r|d d |d< nd|d v r-|d d d |d< | ddu r8d|d< d|v rD|d|d< nd	|v rO|d	|d< | dsXd|d< |d
d}|durl|j|d< |j|d< d|vrd|v r|d durt|d |d< nt |d< d|vr|dur|j|d< n|d |d< tt|	di  | ddu rg |d< | ddu rg |d< | ddu ri |d< | ddu ri |d< | ddu rt
  |d< t|d |d< |S )zAssign name to the run.r^   N
serializedr`   Unnamedr9   ry   _clientrk   rn   rl   rh   r}   rw   rx   r+   outputsattachmentsr-   )getpopr`   r|   r   r   r}   r
   rv   
setdefaultr/   _ensure_write_replicas)clsr   rk   r%   r%   r&   infer_defaults   sL   



zRunTree.infer_defaultsc                 C  s^   | d}|r| r|S t|d |d }| d}|dur)|d | |d< |S ||d< |S )z#Ensure the dotted order of the run.r|   rh   r`   rl   N.)r   strip_create_current_dotted_order)r   r   current_dotted_orderrl   r%   r%   r&   ensure_dotted_order  s   

zRunTree.ensure_dotted_orderr   c                 C  s   | j du r	t | _ | j S )zReturn the client.N)ry   r8   selfr%   r%   r&   r9   .  s   
zRunTree.clientr:   c                 C  s   | j S rc   )ry   r   r%   r%   r&   r   7  s   zRunTree._clientc                   s    |dkr	|| _ dS t ||S )zSet the `_client` specially.r   N)ry   super__setattr__)r   r^   value	__class__r%   r&   r   <  s   
zRunTree.__setattr__)inputsr   r+   r*   usage_metadatar   Optional[Mapping[str, Any]]r   Optional[Sequence[str]]r*   r   +Optional[ls_schemas.ExtractedUsageMetadata]Nonec                C  s   |t ur	t|| _|t ur| jdi |pi  |t ur/d| jd< |du r*i | _nt|| _|t urEd| jt< |du r@i | _	nt|| _	|t urVt
|| jdi d< dS dS )a  Set the inputs, outputs, tags, and metadata of the run.

        If performed, this will override the default behavior of the
        end() method to ignore new outputs (that would otherwise be added)
        by the @traceable decorator.

        If your LangChain or LangGraph versions are sufficiently up-to-date,
        this will also override the default behavior of `LangChainTracer`.

        Args:
            inputs: The inputs to set.
            outputs: The outputs to set.
            tags: The tags to set.
            metadata: The metadata to set.
            usage_metadata: Usage information to set.

        Returns:
            None
        r*   Finputs_is_truthyNTr   )NOT_PROVIDEDlistr+   rw   r   updater   rv   OVERRIDE_OUTPUTSr   r[   )r   r   r   r+   r*   r   r%   r%   r&   rB   D  s$   




zRunTree.setUnion[Sequence[str], str]c                 C  s0   t |tr|g}| jdu rg | _| j| dS )zAdd tags to the run.N)
isinstancer]   r+   extend)r   r+   r%   r%   r&   add_tagsw  s
   

zRunTree.add_tagsdict[str, Any]c                 C  s2   | j du ri | _ tt| j di }|| dS )zAdd metadata to the run.Nr*   )rw   r
   rv   r   r   )r   r*   	metadata_r%   r%   r&   add_metadata  s   
zRunTree.add_metadatac                 C  s    | j du ri | _ | j | dS )zUpsert the given outputs into the run.

        Args:
            outputs: A dictionary containing the outputs to be added.
        N)r   r   )r   r   r%   r%   r&   add_outputs  s   
zRunTree.add_outputsc                 C  s*   | j du ri | _ | j | d| jd< dS )zUpsert the given inputs into the run.

        Args:
            inputs: A dictionary containing the inputs to be added.
        NFr   )r   r   rw   )r   r   r%   r%   r&   
add_inputs  s   
zRunTree.add_inputsTUnion[ls_schemas.RunEvent, Sequence[ls_schemas.RunEvent], Sequence[dict], dict, str]c                 C  sh   | j du rg | _ t|tr| j | dS t|tr,| j dttj	 |d dS | j 
| dS )a  Add an event to the list of events.

        Args:
            events: 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)rx   r   rv   appendr]   r   rd   r   re   	isoformatr   )r   rx   r%   r%   r&   	add_event  s   


zRunTree.add_event)r   errorend_timerx   r*   r   r   r   Optional[datetime]'Optional[Sequence[ls_schemas.RunEvent]]r>   c                C  sz   |pt tj| _| jts |dur | js|| _n| j	| |dur'|| _
|dur0| | |dur;| | dS dS )z/Set the end time of the run and all child runs.N)r   rd   r   re   r   rw   r   r   r   r   r   r   r   )r   r   r   r   rx   r*   r%   r%   r&   end  s   

zRunTree.end)run_idr   r   r   r   reference_example_idrh   r   r+   rw   r   r   r   Optional[ID_TYPE]r   r   r    Optional[ls_schemas.Attachments]c                C  s   |pd|i}t di d|dt|d|d|pi d|pi d|d|d|d	|	p/ttjd
|
d|p7i d| d| jd| jd| jd|d|pWi d| j	}|S d| j	}|S )z Add a child run to the run tree.r^   r`   r   r   r   r   rb   r   rh   r   rw   rk   r   r-   ry   r+   r   r~   Nr%   )
r\   r   r   rd   r   re   rr   r-   ry   r~   )r   r^   rb   r   r   r   r   r   r   rh   r   r+   rw   r   serialized_runr%   r%   r&   create_child  s`   	
zRunTree.create_childc                 C  sF   | j h ddd}| jd ur| j |d< | jd ur!| j |d< |S )N>   r   r   ro   T)ri   exclude_noner   r   )rv   r   copyr   )r   	self_dictr%   r%   r&   _get_dicts_safe  s   

zRunTree._get_dicts_safe	parent_idrun_dictc           
      C  s   | d }rU|d}d}t|}t|D ]\}}|t d }t||kr+|} nq|durU||d d }	d|	|d< |	rOt|	d t d |d< n|d |d< t| d|krf|dd dS dS )	zSlice the parent id from dotted order.

        Additionally check if the current run is a child of the parent. If so, update
        the parent_run_id to None, and set the trace id to the new root id after
        parent_id.
        r|   r   N   r   r}   r`   rn   )r   splitr]   	enumerateTIMESTAMP_LENGTHjoinr   r   )
r   r   r   r|   segs	start_idxidxpartseg_idtrimmed_segsr%   r%   r&   _slice_parent_id  s&   
zRunTree._slice_parent_idr    c                 C  st  |   }|| jkr|S |r|ddrt }|r| || |d }tt| d| }|d}|r?tt| d| }nd}|d}	|	rStt|	 d| }
nd}
|dr|d d	}g }|dd
 D ]}tt|t d  d| }|	|dt  t
|  qi|	|d
 dt  t
|  d	|}nd}t|}||||
||d |r|| |S )zDRewrites ids/dotted_order for a given project with optional updates.rerootFr`   :r}   Nrn   r|   r   r   )r`   r}   rn   r|   rr   )r   rr   r   r2   r   r   r   r   r   r   r]   r   r   deepish_copyr   )r   r   r    r   distributed_parent_idold_idnew_id	old_trace	new_traceparent
new_parentr   rebuiltr   repldotteddupr%   r%   r&   _remap_for_project.  sP   



 
	
zRunTree._remap_for_projectexclude_child_runsboolc           	      C  s   | j r1| j D ])}|dp| j}|d}| ||}| jjdi ||d|dd qn|  }| jjdi | | jr[dd | jD }| j	dt
tj t|d	 |si| jD ]
}|jd
d q`dS dS )z,Post the run tree to the API asynchronously.r   r    r   r   )r   r   c                 S  s   g | ]}t |qS r%   )r]   ).0r^   r%   r%   r&   
<listcomp>t  s    z RunTree.post.<locals>.<listcomp>uploaded_attachmentr   F)r   Nr%   )r-   r   rr   r   r9   
create_runr   r   rx   r   r   rd   r   re   r   rB   ro   post)	r   r   replicar   r    r   kwargsrW   	child_runr%   r%   r&   r   d  s6   




zRunTree.post)exclude_inputsr   c             
     s~  | j s|   dd | j D }z|r,tdd | jD d  r, fdd| D }W n tyG } ztd|  W Y d}~nd}~ww | j	r| j	D ]}|
dpW| j}|
d	}| ||}| jjdi d
|d
 d|d d|
dd|
dd|rdn|d d|d d|
dd|
dd|
dd|
dd|
dd|
dd|
dd|
dd|
dd|
dd|d|
dd|
d qNdS | jjdi d
| jd| jdtt| jd| jd|rdn
| jr| j ndd| jr| j ndd| jd| jd| jd| jd| j d| jd| jd| jd| jd| jd| dS )zPatch the run tree to the API in a background thread.

        Args:
            exclude_inputs: Whether to exclude inputs from the patch request.
        c                 S  s    i | ]\}}t |tr||qS r%   )r   tupler   avr%   r%   r&   
<dictcomp>  s
    z!RunTree.patch.<locals>.<dictcomp>c                 s  s"    | ]}| d dkr|V  qdS )r^   r   N)r   )r   evr%   r%   r&   	<genexpr>  s    z RunTree.patch.<locals>.<genexpr>Nc                   s"   i | ]\}}| d  vr||qS )r   r%   r   uploadedr%   r&   r     s
    z'Error filtering attachments to upload: r   r    r^   r   r`   rb   rh   r   r   r   rn   rr   r   r   r|   r}   rx   r+   rw   r   r   r   r%   )r   r   r   itemsnextrx   	Exceptionloggerwarningr-   r   rr   r   r9   
update_runr^   r`   r
   r   rb   rh   r   r   r   r   rn   r   r|   r}   r+   rw   )r   r   r   er   r   r    r   r%   r   r&   patch  s   







	










zRunTree.patchc                 C  s   dS )z$Wait for all `_futures` to complete.Nr%   r   r%   r%   r&   wait  s   zRunTree.waitc                 C  s   | j j| dS )zReturn the URL of the run.)r   )r9   get_run_urlr   r%   r%   r&   get_url  s   zRunTree.get_urlr   r   c                 K  s    t |i}tt| j|fi |S )zwCreate a new 'child' span from the provided dotted order.

        Returns:
            RunTree: The new span.
        )LANGSMITH_DOTTED_ORDERr
   r\   from_headers)r   r|   r   headersr%   r%   r&   from_dotted_order  s   zRunTree.from_dotted_orderconfigc              
     s  zddl m}m} ddlm}m} ddlm  W n ty) } ztd|d}~ww |du r=|t	|t
r9t||nd}nt||}|d }	rt	|	||fr|	jrt fdd	|	jD d }
r|
jt|	j }r|jr|j}|j|d
< |j|d< |j|d< |j|d< |j|d< tt|jpg |dg  |d< |j|d< |di }|di }||j nt|
dr|	j|
j v r|
j |	j d }ndS |
j!|d< |
j"|d< t#j$|fi |S dS )zCreate a new 'child' span from the provided runnable config.

        Requires `langchain` to be installed.

        Returns:
            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  qd S rc   )r   )r   tr  r%   r&   r     s    z/RunTree.from_runnable_config.<locals>.<genexpr>rb   r   r   rh   r   r+   r^   rw   r*   	order_mapr   r9   r   )% langchain_core.callbacks.managerr  r  langchain_core.runnablesr  r   langchain_core.tracers.langchainr  ImportErrorr   rv   r
   r   rn   r   handlersrun_mapr]   r|   rb   r   r   rh   r   sortedrB   r+   r^   r   r   r*   hasattrr  r9   r   r\   r
  )r   r  r   r  r  r  r  r  config_cbtracerr   r|   extra_r   r%   r  r&   from_runnable_config  s^   





"


zRunTree.from_runnable_configr	  -Mapping[Union[str, bytes], Union[str, bytes]]c                 K  s  |  }ttt |t}|s#ttt |t}|sdS |d}|	 }t
|}|d d }||d< |d d |d< ||d< t|d	krO|d
 d |d< |dpYttj|d< |dpbd|d< |dpkd|d< t|}	|	jsy|	jr|di |d< |d di |d d< i |	j|d d }
|
|d d< tt|	j|dg  }||d< |	jr|	j|d< |	jr|	j|d< tdi |}tt|j |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:
            The new span or `None` if no parent span information is found.
        Nr)   r   r   r}   r   r`   r|      rn   rh   rb   ra   r^   r   rw   r*   r+   r   r-   r%   )r   r
   r   r]   r   r  bytesLANGSMITH_DOTTED_ORDER_BYTESdecoder   _parse_dotted_orderlenr   rd   r   re   _Baggager  r*   r+   r   r  rB   r   r-   r\   r2   r`   )r   r	  r   	init_argslangsmith_tracelangsmith_trace_bytesrl   parsed_dotted_orderr}   baggager*   r+   run_treer%   r%   r&   r  )  sN   



zRunTree.from_headersdict[str, str]c                 C  sF   i }| j r| j|t < t| jdi | j| j| jd}|	 |d< |S )z0Return the `RunTree` as a dictionary of headers.r*   r*   r+   r   r-   r/  )
r}   r|   r  r*  rw   r   r+   rr   r-   	to_header)r   r	  r/  r%   r%   r&   
to_headersj  s   zRunTree.to_headersc              	   C  s&   d| j  d| j d| j d| j d	S )z7Return a string representation of the `RunTree` object.zRunTree(id=z, name='z', run_type='z', dotted_order='z'))r`   r^   rb   r|   r   r%   r%   r&   __repr__x  s   zRunTree.__repr__)r   rv   r5   rv   )r5   r   )r5   r:   )r   r   r   r   r+   r   r*   r   r   r   r5   r   )r+   r   r5   r   )r*   r   r5   r   )r   r   r5   r   )r   r   r5   r   )rx   r   r5   r   )r   r   r   r   r   r   rx   r   r*   r>   r5   r   )ra   )r^   r]   rb   r   r   r   r   r   r   r   r   r   r   r   r   rt   rh   r   r   r   r+   r=   rw   r   r   r   r5   r\   )r   r]   r   rv   r5   r   rc   )r   r]   r    r   r5   rv   )T)r   r   r5   r   )r   r   r5   r   )r5   r   r5   r]   )r|   r]   r   r   r5   r\   )r  r   r   r   r5   rj   )r	  r"  r   r   r5   rj   )r5   r1  )8r!   r"   r#   r   r$   r   r   r`   rb   rh   rk   rl   r   ro   rr   ru   rv   rw   r+   rx   ry   r|   r}   r~   r-   r   r   r   r   propertyr9   r   r   r   rB   r   r   r   r   r   r   r   r   r   r   r   r  r  r  classmethodr
  r!  r  r4  r5  __classcell__r%   r%   r   r&   r\      s   
 ,
3




%*
6
Q
?
@r\   c                   @  sJ   e Zd ZdZ				ddddZedddZedddZdddZdS )r*  zBaggage header information.Nr*   Optional[dict[str, str]]r+   r=   r   r   r-   r   c                 C  s(   |pi | _ |pg | _|| _|pg | _dS )zInitialize the Baggage object.Nr2  )r   r*   r+   r   r-   r%   r%   r&   __init__  s   

z_Baggage.__init__header_valuer5   c              
   C  sh  |s|  S i }g }d}d}z| dD ]|}| dd\}}|tkr+ttj|}q|tkr9tj| d}q|tkrDtj|}q|t	krttj|}	g }
|	D ]7}t
|ttfrvt|dkrv|
tddt|d |d d qUt
|tr|
tt| qUtd|  qU|
}qW n ty } ztd	|  W Y d}~nd}~ww | ||||d
S )z4Create a Baggage object from the given header value.N,=r   r#  r   r   r   r   r    z#Unknown replica format in baggage: zError parsing baggage header: r2  )r   LANGSMITH_METADATAjsonloadsurllibparseunquoteLANGSMITH_TAGSLANGSMITH_PROJECTLANGSMITH_REPLICASr   r   r   r)  r   r   r]   rv   r
   r   r   r   )r   r<  r*   r+   r   r-   itemkeyr   replicas_dataparsed_replicasreplica_itemr  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|  tt|d dS |  d S )Nr/  s   baggager)   )rN  r
   r%  r'  )r   r	  r%   r%   r&   r    s
   
z_Baggage.from_headersr]   c                 C  s   g }| j rt| j }|t dtj|  | jr/d| j}|t dtj|  | j	rA|t dtj| j	  | j
rWt| j
}|t dtj|  d|S )z,Return the Baggage object as a header value.z	metadata=r=  ztags=zproject=z	replicas=)r*   r   r   LANGSMITH_PREFIXrC  rD  quoter+   r   r   r-   )r   r   serialized_metadataserialized_tagsserialized_replicasr%   r%   r&   r3    s*   


z_Baggage.to_header)NNNN)r*   r:  r+   r=   r   r   r-   r   )r<  r   r5   r*  )r	  rO  r5   r*  r6  )	r!   r"   r#   r   r;  r8  rN  r  r3  r%   r%   r%   r&   r*    s    2r*  r   )maxsizeenv_varlist[WriteReplica]c           	   
   C  s  | sg S zt | }t|trdg }|D ]M}t|ts&tdt|j  q|	d}|	d}t|t
sAtdt|j  qt|t
sRtdt|j  q|t|d|ddd q|W S t|trt| g }| D ](\}}|d}t|t
r|t||ddd qstd	| d
t|j  qs|W S tdt|j  g W S  tjy     ty } ztd|  g W  Y d}~S d}~ww )zParse write replicas from LANGSMITH_RUNS_ENDPOINTS environment variable value.

    Supports array format [{"api_url": "x", "api_key": "y"}] and object format
    {"url": "key"}.
    zBInvalid item type in LANGSMITH_RUNS_ENDPOINTS: expected dict, got r   r   zGInvalid api_url type in LANGSMITH_RUNS_ENDPOINTS: expected string, got zGInvalid api_key type in LANGSMITH_RUNS_ENDPOINTS: expected string, got /Nr?  z7Invalid value type in LANGSMITH_RUNS_ENDPOINTS for URL z: expected string, got u   Invalid LANGSMITH_RUNS_ENDPOINTS – must be valid JSON list of objects with api_url and api_key properties, or object mapping url->apiKey, got u   Invalid LANGSMITH_RUNS_ENDPOINTS – must be valid JSON list of objects with api_url and api_key properties, or object mapping url->apiKey: )rA  rB  r   r   rv   r   r   typer!   r   r]   r   r   rstrip_check_endpoint_env_unsetr   r   LangSmithUserErrorr   )	rV  parsedr-   rI  r   r   urlrJ  r  r%   r%   r&   "_parse_write_replicas_from_env_var  s   









	r_  c                  C  s   t d} t| S )zFGet write replicas from LANGSMITH_RUNS_ENDPOINTS environment variable.RUNS_ENDPOINTS)r   get_env_varr_  )rV  r%   r%   r&   _get_write_replicas_from_env@  s   
rb  r]  r1  r   c                 C  s2   ddl }| r|ds|drtddS dS )zECheck if endpoint environment variables conflict with runs endpoints.r   NLANGSMITH_ENDPOINTLANGCHAIN_ENDPOINTz]You cannot provide both LANGSMITH_ENDPOINT / LANGCHAIN_ENDPOINT and LANGSMITH_RUNS_ENDPOINTS.)osgetenvr   r\  )r]  re  r%   r%   r&   r[  G  s   r[  r   c                 C  s   | du rt  S t| S )z(Convert replicas to WriteReplica format.N)rb  r   )r-   r%   r%   r&   r   R  s   r   r|   r]   list[tuple[datetime, UUID]]c                 C  s   |  d}dd |D S )zParse the dotted order string.r   c                 S  s4   g | ]}t |d t  dt|t d  fqS )N%Y%m%dT%H%M%S%fZ)r   strptimer   r   )r   r   r%   r%   r&   r   `  s    z'_parse_dotted_order.<locals>.<listcomp>)r   )r|   partsr%   r%   r&   r(  ]  s   
r(  r6   rh   r   r   rt   c                 C  s.   | pt tj}|pt|}|dt| S )z Create the current dotted order.rh  )r   rd   r   re   r   strftimer]   )rh   r   stid_r%   r%   r&   r   m  s   r   )r4   r   r5   r   )
r9   r:   r;   r<   r   r   r+   r=   r*   r>   )rK   rL   r5   rL   )rV  r   r5   rW  )r5   rW  )r]  r1  r5   r   )r-   r   r5   rW  )r|   r]   r5   rg  )rh   r   r   rt   r5   r]   )Zr   
__future__r   	functoolsrA  loggingsyscollections.abcr   r   r   r   typingr   r   r	   r
   uuidr   r   r   typing_extensionsr   langsmith._internal._uuidr   langsmith.uuidr   pydantic.v1r   r   r  pydanticcontextvars	threadingurllib.parserC  langsmith._internal._context	_internalr@   	langsmithr   
ls_schemasr   langsmith.clientr   r   r   r   r   	getLoggerr!   r   r   rP  internr  encoder&  r@  rF  rG  rH  r   objectr   Lockr7   
ContextVarr/   r]   r2   r?   r   r8   rJ   r[   RunBaser\   r*  	lru_cacher_  rb  r[  r   r(  _GLOBAL_CLIENTr6   r$   rm   r   r%   r%   r%   r&   <module>   s    





]     @
e
Z


