o Zh&@sdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddl mZmZmZmZmZddlZddlmZddlmZddlmZdd lmZeejefZ Gd d d Z!dddZ"dS)zThe Async LangSmith Client.) annotationsN)AsyncGenerator AsyncIteratorMappingSequence)AnyLiteralOptionalUnioncast)client)schemas)utils)_beta_decoratorc@steZdZdZdZ     d#d$d d Zd%ddZddZddZe ddZ e d&ddZ d'dd Z d(d)d%d&Z dd'd(d)d*d/d0Zddd1d+d:d;Zd,d>d?Zd-dAdBZdddddddddddddddddCd.dZd[Zdd\d/d^d_Zd0dadbZd1dcddZd2dfdgZ  d3d4dhdiZdddjd5dkdlZd6dodpZ  d3d7drdsZd8dtduZd9dwdxZ   d:d;d{d|ZdddZ d(dddddd?ddZ!ddddd@ddZ"dAddZ#ddddddBddZ$dCddZ%ddddddDddZ&ddddEddZ'dFddZ(dddGddZ)dHddZ*dIddĄZ+dJddƄZ,dKdd˄Z-e.j/dd͜dLddЄZ0e.j/dMdd҄Z1e.j/ddӜdNddׄZ2dOddڄZ3dPdd݄Z4dQddZ5 dRdSddZ6dTddZ7dUddZ8dVddZ9dWddZ:dWddZ;ddddeddddXddZ?dYddZ@ddddddZd d ZAdd d[ddZBddddddd\ddZCd]ddZDddd^ddZEddddd_ddZFddd`ddZGddddddd dad!d"ZHdS(b AsyncClientz4Async Client for interacting with the LangSmith API.) _retry_config_client_web_url _settingsNapi_url Optional[str]api_key timeout_msWOptional[Union[int, tuple[Optional[int], Optional[int], Optional[int], Optional[int]]]] retry_configOptional[Mapping[str, Any]]web_urlcCs|pddi|_ddi}t|}t|}|r||tj<t||t|tr0|ddddf}nt|t r?t dd|D}nd }t j |||d |_ ||_ d|_dS) zInitialize the async client. max_retriesz Content-Typeapplication/jsonNcSs g|] }|dur |dndqS)Nr ).0tr!r!M/var/www/html/lang_env/lib/python3.10/site-packages/langsmith/async_client.py <s z(AsyncClient.__init__.. )base_urlheaderstimeout)rls_utilsZ get_api_keyZ get_api_url ls_clientZ X_API_KEYZ_validate_api_key_if_hosted isinstanceinttuplehttpxrrrr)selfrrrrr_headersZtimeout_r!r!r$__init__"s$        zAsyncClient.__init__returncs|S)zEnter the async client.r!r0r!r!r$ __aenter__EszAsyncClient.__aenter__cs|IdHdS)zExit the async client.N)aclose)r0exc_typeexc_valexc_tbr!r!r$ __aexit__IszAsyncClient.__aexit__cs|jIdHdS)zClose the async client.N)rr6r4r!r!r$r6MszAsyncClient.aclosecCs t|jjSN)strrr'r4r!r!r$_api_urlQs zAsyncClient._api_urlr<cCst|j|jS)zThe web host url.)r*Z get_host_urlrr=r4r!r!r$ _host_urlUszAsyncClient._host_urlmethodendpointkwargsrhttpx.Responsec stt|jdd}d|vr |d}dd|D}||d<t|D]}z|jj||fi|IdH}t ||WSt j y} z||j dkr\t d|d |d t| |j d krktd |||j d kr}td|dt| |j dkrtd|dt| |j dkrtd|dt| |j dkrtd|dt| td|d |d t| d} ~ wt jy} z ||dkrtdt| td|IdHWYd} ~ q$d} ~ wwt d)z(Make an async HTTP request with retries.rrparamscSi|] \}}|dur||qSr;r!r"kvr!r!r$ gz6AsyncClient._arequest_with_retries..NizServer error caused failure to  z in LangSmith API. iz(Client took too long to send request to izRate limit exceeded for z. izAuthentication failed for izResource not found for iz Conflict for z Failed to zRequest error: z0Unexpected error connecting to the LangSmith API)r r-rgetitemsrangerrequestr*raise_for_status_with_textr/ZHTTPStatusError status_codeZLangSmithAPIErrorreprZLangSmithRequestTimeoutZLangSmithRateLimitErrorZLangSmithAuthErrorLangSmithNotFoundErrorZLangSmithConflictErrorLangSmithError RequestErrorZLangSmithConnectionErrorasynciosleep) r0r?r@rArrCZfiltered_paramsattemptresponseer!r!r$_arequest_with_retriesZsj            z"AsyncClient._arequest_with_retriespathrCOptional[dict[str, Any]]AsyncIterator[dict[str, Any]]cCs|pi}|dd}|dd|d< ||d<|jd||dIdH}|}|s+dS|D]}|Vq-t||dkr=dS|t|7}q) zGet a paginated list of items.offsetrlimitdTGETrCN)rMr\jsonlen)r0r]rCr`rZrNitemr!r!r$_aget_paginated_lists   z AsyncClient._aget_paginated_listPOSTruns)bodyrequest_methoddata_keyrkOptional[dict]rlrmAsyncIterator[dict]c Cs|r|ni} |j||t|dIdH}|}|s dS||s'dS||D]}|Vq+|d} | s:dS| dsAdS| d|d<q )z%Get a cursor paginated list of items.TcontentNcursorsnextcursor)copyr\r+ _dumps_jsonrerM) r0r]rkrlrmZparams_rZZ response_bodyrunrrr!r!r$_aget_cursor_paginated_lists,       z'AsyncClient._aget_cursor_paginated_list) project_name revision_idnameinputsdict[str, Any]run_typeryrzOptional[ls_client.ID_TYPE]NonecsN||dp t|||pt|d|}|jddt|dIdHdS)z Create a run.id)r{rr|r~Z session_namerzri/runsrpN)rMuuiduuid4r*Zget_tracer_projectr\r+rv)r0r{r|r~ryrzrAZ run_creater!r!r$ create_runs   zAsyncClient.create_runrun_idls_client.ID_TYPEcsDi|dt|i}|jddt|t|dIdHdS)z Update a run.rPATCH/runs/rpN)r+_as_uuidr\rv)r0rrAdatar!r!r$ update_runszAsyncClient.update_runls_schemas.Runc4|ddt|IdH}tjdi|S)z Read a run.rcrNr!)r\r+r ls_schemasRunre)r0rrZr!r!r$read_runs  zAsyncClient.read_run) project_idryr~trace_idreference_example_idqueryfilter trace_filter tree_filteris_root parent_run_id start_timeerrorrun_idsselectrar?Optional[Union[ls_client.ID_TYPE, Sequence[ls_client.ID_TYPE]]]#Optional[Union[str, Sequence[str]]]rrrrrrrOptional[bool]rrOptional[datetime.datetime]rr%Optional[Sequence[ls_client.ID_TYPE]]rOptional[Sequence[str]]ra Optional[int]AsyncIterator[ls_schemas.Run]c sBg}t|tjtfr||n t|tr|||dur@t|tr'|g}tjfdd|DIdH}|dd|D|rD|nd||rK|gnd|||| | | | rX| nd| |||d|}|rmdd|D|d<dd | D}d }j d |d 2z3dHW}t j di|V|d 7}|dur||krdSq6dS)a!List runs from the LangSmith API. Parameters ---------- project_id : UUID or None, default=None The ID(s) of the project to filter by. project_name : str or None, default=None The name(s) of the project to filter by. run_type : str or None, default=None The type of the runs to filter by. trace_id : UUID or None, default=None The ID of the trace to filter by. reference_example_id : UUID or None, default=None The ID of the reference example to filter by. query : str or None, default=None The query string to filter by. filter : str or None, default=None The filter string to filter by. trace_filter : str or None, default=None Filter to apply to the ROOT run in the trace tree. This is meant to be used in conjunction with the regular `filter` parameter to let you filter runs by attributes of the root run within a trace. tree_filter : str or None, default=None Filter to apply to OTHER runs in the trace tree, including sibling and child runs. This is meant to be used in conjunction with the regular `filter` parameter to let you filter runs by attributes of any run within a trace. is_root : bool or None, default=None Whether to filter by root runs. parent_run_id : UUID or None, default=None The ID of the parent run to filter by. start_time : datetime or None, default=None The start time to filter by. error : bool or None, default=None Whether to filter by error status. run_ids : List[str or UUID] or None, default=None The IDs of the runs to filter by. limit : int or None, default=None The maximum number of runs to return. **kwargs : Any Additional keyword arguments. Yields: ------ Run The runs. Examples: -------- List all runs in a project: .. code-block:: python project_runs = client.list_runs(project_name="") List LLM and Chat runs in the last 24 hours: .. code-block:: python todays_llm_runs = client.list_runs( project_name="", start_time=datetime.now() - timedelta(days=1), run_type="llm", ) List root traces in a project: .. code-block:: python root_runs = client.list_runs(project_name="", is_root=1) List runs without errors: .. code-block:: python correct_runs = client.list_runs(project_name="", error=False) List runs and only return their inputs/outputs (to speed up the query): .. code-block:: python input_output_runs = client.list_runs( project_name="", select=["inputs", "outputs"] ) List runs by run ID: .. code-block:: python run_ids = [ "a36092d2-4ad5-4fb4-9c0d-0dba9a2ed836", "9398e6be-964f-4aa4-8ae9-ad78cd4b7074", ] selected_runs = client.list_runs(id=run_ids) List all "chain" type runs that took more than 10 seconds and had `total_tokens` greater than 5000: .. code-block:: python chain_runs = client.list_runs( project_name="", filter='and(eq(run_type, "chain"), gt(latency, 10), gt(total_tokens, 5000))', ) List all runs called "extractor" whose root of the trace was assigned feedback "user_score" score of 1: .. code-block:: python good_extractor_runs = client.list_runs( project_name="", filter='eq(name, "extractor")', trace_filter='and(eq(feedback_key, "user_score"), eq(feedback_score, 1))', ) List all runs that started after a specific timestamp and either have "error" not equal to null or a "Correctness" feedback score equal to 0: .. code-block:: python complex_runs = client.list_runs( project_name="", filter='and(gt(start_time, "2023-07-15T12:34:56Z"), or(neq(error, null), and(eq(feedback_key, "Correctness"), eq(feedback_score, 0.0))))', ) List all runs where `tags` include "experimental" or "beta" and `latency` is greater than 2 seconds: .. code-block:: python tagged_runs = client.list_runs( project_name="", filter='and(or(has(tags, "experimental"), has(tags, "beta")), gt(latency, 2))', ) Ncsg|]}j|dqS)ry) read_project)r"r{r4r!r$r%sz)AsyncClient.list_runs..cSsg|]}|jqSr!)r)r"projectr!r!r$r%s)sessionr~Zreference_examplerrrrrZ parent_runrrrtracercSg|] }tt|qSr!r<r+rr"id_r!r!r$r%srcSrDr;r!rEr!r!r$rHrIz)AsyncClient.list_runs..rz /runs/query)rkrKr!)r,rUUIDr<appendlistextendrWZgather isoformatrNrxrr)r0rryr~rrrrrrrrrrrrrarAZ project_idsZprojectsZ body_queryrkixrwr!r4r$ list_runssX        zAsyncClient.list_runs)share_idrcstt|d}t|t|ptd}|jdd|dt|dIdH}t|| d}|j d |d S) aGet a share link for a run asynchronously. Args: run_id (ID_TYPE): The ID of the run to share. share_id (Optional[ID_TYPE], optional): Custom share ID. If not provided, a random UUID will be generated. Returns: str: The URL of the shared run. Raises: httpx.HTTPStatusError: If the API request fails. r)r share_tokenPUTr/sharerpNr/public//r) r+rr<rrr\rvr*rQrer>)r0rrZrun_id_rrZrr!r!r$ share_runs    zAsyncClient.share_runboolcs"|t|dIdH}|duS)z)Get share state for a run asynchronously.rN)read_run_shared_linkr+r)r0rlinkr!r!r$ run_is_sharedszAsyncClient.run_is_sharedcs`|ddt|ddIdH}t||}|dus#d|vr%dS|jd|ddS) aNRetrieve the shared link for a specific run asynchronously. Args: run_id (ID_TYPE): The ID of the run. Returns: Optional[str]: The shared link for the run, or None if the link is not available. Raises: httpx.HTTPStatusError: If the API request fails. rcrrrNrrr)r\r+rr*rQrer>)r0rrZresultr!r!r$rs  z AsyncClient.read_run_shared_linkls_schemas.TracerSessionc>d|i|}|jddt|dIdH}tjdi|S)zCreate a project.r{ri /sessionsrpNr!)r\r+rvr TracerSessionre)r0ryrArrZr!r!r$create_project    zAsyncClient.create_projectc|r|ddt|IdH}n|r#|jddd|idIdH}ntd|}t|trE|s;td|d t j d i|d St j d i|S) zRead a project.rc /sessions/Nrr{rd2Either project_name or project_id must be providedzProject not foundrr!) r\r+r ValueErrorrer,rr*rTrr)r0ryrrZrr!r!r$r$    zAsyncClient.read_project)ryrcsl|dur |dur td|dur|j|dIdH}t|j}|s%td|ddt|IdHdS)a Delete a project from LangSmith. Parameters ---------- project_name : str or None, default=None The name of the project to delete. project_id : str or None, default=None The ID of the project to delete. NrrzProject not foundDELETEr)rrr<rr\r+r)r0ryrrr!r!r$delete_projects  zAsyncClient.delete_project dataset_namels_schemas.Datasetcr)zCreate a dataset.r{ri /datasetsrpNr!)r\r+rvrDatasetre)r0rrArrZr!r!r$create_dataset6rzAsyncClient.create_dataset dataset_idcr) zRead a dataset.rc /datasets/Nrr{rdz2Either dataset_name or dataset_id must be providedzDataset rrr!) r\r+rrrer,rr*rTrr)r0rrrZrr!r!r$ read_datasetBrzAsyncClient.read_datasetcs$|ddt|IdHdS)zDelete a dataset.rrN)r\r+r)r0rr!r!r$delete_dataset\s zAsyncClient.delete_dataset!AsyncIterator[ls_schemas.Dataset]cKs6|jd|d2z3dHW}tjdi|Vq6dS)zList datasets.rrdNr!)rhrr)r0rAdatasetr!r!r$ list_datasetscszAsyncClient.list_datasetsoutputsls_schemas.Examplec s~|dur |dur td|dur|j|dIdH}|j}||t|d|}|jddt|dIdH}tjdi| S) zCreate an example.Nz2Either dataset_id or dataset_name must be providedr)r|rrri /examplesrpr!) rrrr<r\r+rvrExamplere) r0r|rrrrArrrZr!r!r$create_exampleks    zAsyncClient.create_example example_idcr)zRead an example.rcz /examples/Nr!)r\r+rrrre)r0rrZr!r!r$ read_examples  zAsyncClient.read_example)rr!AsyncIterator[ls_schemas.Example]cKsr|}|rt||d<n|r|j|dIdH}|j|d<|jd|d2z3dHW}tjdi|Vq&6dS)zList examples.rrNrrdr!)rur+rrrrhrr)r0rrrArCrexampler!r!r$ list_exampless zAsyncClient.list_exampleskeyscoreOptional[float]value Optional[Any]commentls_schemas.Feedbackc sPtj|dd||||d|}|jddt|dIdH}tjdi|S) aCreate feedback for a run. Args: run_id (Optional[ls_client.ID_TYPE]): The ID of the run to provide feedback for. Can be None for project-level feedback. key (str): The name of the metric or aspect this feedback is about. score (Optional[float]): The score to rate this run on the metric or aspect. value (Optional[Any]): The display value or non-numeric value for this feedback. comment (Optional[str]): A comment about this feedback. **kwargs: Additional keyword arguments to include in the feedback data. Returns: ls_schemas.Feedback: The created feedback object. Raises: httpx.HTTPStatusError: If the API request fails. T)Z accept_null)rrrrrri /feedbackrpNr!)r+Z _ensure_uuidr\rvrFeedbackre) r0rrrrrrArrZr!r!r$create_feedbacks   zAsyncClient.create_feedback)r correctionrmetadata token_or_urlUnion[str, uuid.UUID]Union[float, int, bool, None](Union[float, int, bool, str, dict, None]rUnion[dict, None]Union[str, None]rc srtj||jdd\}}||jkrtd||jddt|t|||||ddIdH} t| dS) aCreate feedback from a presigned token or URL. Args: token_or_url (Union[str, uuid.UUID]): The token or URL from which to create feedback. score (Union[float, int, bool, None], optional): The score of the feedback. Defaults to None. value (Union[float, int, bool, str, dict, None], optional): The value of the feedback. Defaults to None. correction (Union[dict, None], optional): The correction of the feedback. Defaults to None. comment (Union[str, None], optional): The comment of the feedback. Defaults to None. metadata (Optional[dict], optional): Additional metadata for the feedback. Defaults to None. Raises: ValueError: If the source API URL is invalid. Returns: None: This method does not return anything. rK)Z num_partszInvalid source API URL. riz/feedback/tokens/)rrrrrrpN) r+Z_parse_token_or_urlr=rr\rrvr*rQ) r0rrrrrrZsource_api_urlZ token_uuidrZr!r!r$create_feedback_from_tokens&   z&AsyncClient.create_feedback_from_token) expirationfeedback_config feedback_id feedback_keyr0Optional[datetime.datetime | datetime.timedelta]r#Optional[ls_schemas.FeedbackConfig]rls_schemas.FeedbackIngestTokencs||||p ttd}|durtjdddd|d<n1t|tjr*||d<n$t|tjrDtj|j |j d|j dd d|d<n t d t |d |j d d t|dIdH}tjdi|S)aCreate a pre-signed URL to send feedback data to. This is useful for giving browser-based clients a way to upload feedback data directly to LangSmith without accessing the API key. Args: run_id: feedback_key: expiration: The expiration time of the pre-signed URL. Either a datetime or a timedelta offset from now. Default to 3 hours. feedback_config: FeedbackConfig or None. If creating a feedback_key for the first time, this defines how the metric should be interpreted, such as a continuous score (w/ optional bounds), or distribution over categorical values. feedback_id: The ID of the feedback to create. If not provided, a new feedback will be created. Returns: The pre-signed URL for uploading feedback data. )rrrrNrr)dayshoursminutesZ expires_inZ expires_ati<zInvalid expiration type: z3. Expected datetime.datetime or datetime.timedelta.riz/feedback/tokensrpr!)r<rrrZTimeDeltaInputr,datetimer timedeltarsecondsrtyper\r+rvZFeedbackIngestTokenre)r0rrrrrrkrZr!r!r$create_presigned_feedback_tokens:!     z+AsyncClient.create_presigned_feedback_tokencr)zRead feedback.rc /feedback/Nr!)r\r+rrrrer0rrZr!r!r$ read_feedback;s  zAsyncClient.read_feedback)rrfeedback_source_typerar1Optional[Sequence[ls_schemas.FeedbackSourceType]]"AsyncIterator[ls_schemas.Feedback]c Ks|r dd|Dnd|durt|dndd|}|dur"||d<|dur*||d<d}|jd |d 2z3dHW}tjd i|V|d 7}|durQ||krQdSq36dS) zList feedback.cSrr!rrr!r!r$r%Psz-AsyncClient.list_feedback..Nrb)rwrarsourcerrrdrKr!)minrhrr) r0rrrrarArCrfeedbackr!r!r$ list_feedbackDs$ zAsyncClient.list_feedbackID_TYPEcs0|ddt|dIdH}t|dS)zDelete a feedback by ID. Args: feedback_id (Union[UUID, str]): The ID of the feedback to delete. Returns: None rr rNr\r+rr*rQr r!r!r$delete_feedback`s   zAsyncClient.delete_feedback) queue_idsr{ name_containsrarOptional[list[ID_TYPE]]r)AsyncIterator[ls_schemas.AnnotationQueue]cCs|durddt|Dnd|||durt|dndd}d}|jd|d2z3dHW}tjd i|V|d 7}|durE||krEdSq'6dS) aList the annotation queues on the LangSmith API. Args: queue_ids (Optional[List[Union[UUID, str]]]): The IDs of the queues to filter by. name (Optional[str]): The name of the queue to filter by. name_contains (Optional[str]): The substring that the queue name should contain. limit (Optional[int]): The maximum number of queues to return. Yields: The annotation queues. NcSs$g|]\}}t|d|dqS)z queue_ids[])r+rr"irr!r!r$r%sz6AsyncClient.list_annotation_queues..rb)Zidsr{rrar/annotation-queuesrdrKr!) enumeraterrhrAnnotationQueue)r0rr{rrarCrrr!r!r$list_annotation_queuesqs( z"AsyncClient.list_annotation_queues) descriptionqueue_idr#r$Optional[ID_TYPE]ls_schemas.AnnotationQueuecsh|||dur t|nttd}|jdddd|DdIdH}t|tjdi| S) aCreate an annotation queue on the LangSmith API. Args: name (str): The name of the annotation queue. description (Optional[str]): The description of the annotation queue. queue_id (Optional[Union[UUID, str]]): The ID of the annotation queue. Returns: AnnotationQueue: The created annotation queue object. N)r{r#rrircSrDr;r!rEr!r!r$rHrIz7AsyncClient.create_annotation_queue..rer!) r<rrr\rNr*rQrr!re)r0r{r#r$rkrZr!r!r$create_annotation_queues z#AsyncClient.create_annotation_queuecs|j|gdIdHS)zRead an annotation queue with the specified queue ID. Args: queue_id (Union[UUID, str]): The ID of the annotation queue to read. Returns: AnnotationQueue: The annotation queue object. )rN)r" __anext__)r0r$r!r!r$read_annotation_queues z!AsyncClient.read_annotation_queue)r#cs:|jddt|d||ddIdH}t|dS)a}Update an annotation queue with the specified queue_id. Args: queue_id (Union[UUID, str]): The ID of the annotation queue to update. name (str): The new name for the annotation queue. description (Optional[str]): The new description for the annotation queue. Defaults to None. Returns: None r/annotation-queues/r$)r{r#r'Nr)r0r$r{r#rZr!r!r$update_annotation_queues z#AsyncClient.update_annotation_queuecs@|jddt|dddi|jjdIdH}t|dS)zDelete an annotation queue with the specified queue ID. Args: queue_id (Union[UUID, str]): The ID of the annotation queue to delete. Returns: None rr+r$Acceptr)r(N)r\r+rrr(r*rQ)r0r$rZr!r!r$delete_annotation_queues  z#AsyncClient.delete_annotation_queue list[ID_TYPE]csD|jddt|ddddt|DdIdH}t|dS) a7Add runs to an annotation queue with the specified queue ID. Args: queue_id (Union[UUID, str]): The ID of the annotation queue. run_ids (List[Union[UUID, str]]): The IDs of the runs to be added to the annotation queue. Returns: None rir+r$rc Ss(g|]\}}tt|d|dqS)zrun_ids[rrrr!r!r$r%sz.r'N)r\r+rr r*rQ)r0r$rrZr!r!r$add_runs_to_annotation_queues  z(AsyncClient.add_runs_to_annotation_queuec s>|ddt|ddt|dIdH}t|dS)a?Delete a run from an annotation queue with the specified queue ID and run ID. Args: queue_id (Union[UUID, str]): The ID of the annotation queue. run_id (Union[UUID, str]): The ID of the run to be added to the annotation queue. Returns: None rr+r$rrNr)r0r$rrZr!r!r$ delete_run_from_annotation_queues   z,AsyncClient.delete_run_from_annotation_queueindexr-%ls_schemas.RunWithAnnotationQueueInfocsPdt|dd}|d|d|IdH}t|tjdi|S)aGet a run from an annotation queue at the specified index. Args: queue_id (Union[UUID, str]): The ID of the annotation queue. index (int): The index of the run to retrieve. Returns: RunWithAnnotationQueueInfo: The run at the specified index. Raises: LangSmithNotFoundError: If the run is not found at the given index. LangSmithError: For other API-related errors. r+r$z/runrc/Nr!)r+rr\r*rQrZRunWithAnnotationQueueInfore)r0r$r2r'rZr!r!r$get_run_from_annotation_queue!s  z)AsyncClient.get_run_from_annotation_queuelatest)tagr7csHt|d}|jdd|dtd|i|dIdH}t|dS)a1Enable dataset indexing. Examples are indexed by their inputs. This enables searching for similar examples by inputs with ``client.similar_examples()``. Args: dataset_id (UUID): The ID of the dataset to index. tag (str, optional): The version of the dataset to index. If 'latest' then any updates to the dataset (additions, updates, deletions of examples) will be reflected in the index. Returns: None Raises: requests.HTTPError rrirz/indexr7rpNr+rr\rvr*rQ)r0rr7rArespr!r!r$ index_dataset6s   zAsyncClient.index_datasetcsDt|d}|jdd|dti|dIdH}t|dS)aSync dataset index. This already happens automatically every 5 minutes, but you can call this to force a sync. Args: dataset_id (UUID): The ID of the dataset to sync. Returns: None Raises: requests.HTTPError rrirz /index/syncrpNr8)r0rrAr9r!r!r$sync_indexed_datasetWs    z AsyncClient.sync_indexed_dataset)rdictlist[ls_schemas.ExampleSearch]c st|d}||d|}|r||d<|jdd|dt|dIdH}t|g}|d D]} |tj d i| d|iq3|S) aRetrieve the dataset examples whose inputs best match the current inputs. **Note**: Must have few-shot indexing enabled for the dataset. See ``client.index_dataset()``. Args: inputs (dict): The inputs to use as a search query. Must match the dataset input schema. Must be JSON serializable. limit (int): The maximum number of examples to return. dataset_id (str or UUID): The ID of the dataset to search over. filter (str, optional): A filter string to apply to the search results. Uses the same syntax as the `filter` parameter in `list_runs()`. Only a subset of operations are supported. Defaults to None. kwargs (Any): Additional keyword args to pass as part of request body. Returns: List of ExampleSearch objects. Example: .. code-block:: python from langsmith import Client client = Client() await client.similar_examples( {"question": "When would i use the runnable generator"}, limit=3, dataset_id="...", ) .. code-block:: pycon [ ExampleSearch( inputs={'question': 'How do I cache a Chat model? What caches can I use?'}, outputs={'answer': 'You can use LangChain\'s caching layer for Chat Models. This can save you money by reducing the number of API calls you make to the LLM provider, if you\'re often requesting the same completion multiple times, and speed up your application.\n\n```python\n\nfrom langchain.cache import InMemoryCache\nlangchain.llm_cache = InMemoryCache()\n\n# The first time, it is not yet in cache, so it should take longer\nllm.predict(\'Tell me a joke\')\n\n```\n\nYou can also use SQLite Cache which uses a SQLite database:\n\n```python\n rm .langchain.db\n\nfrom langchain.cache import SQLiteCache\nlangchain.llm_cache = SQLiteCache(database_path=".langchain.db")\n\n# The first time, it is not yet in cache, so it should take longer\nllm.predict(\'Tell me a joke\') \n```\n'}, metadata=None, id=UUID('b2ddd1c4-dff6-49ae-8544-f48e39053398'), dataset_id=UUID('01b6ce0f-bfb6-4f48-bbb8-f19272135d40') ), ExampleSearch( inputs={'question': "What's a runnable lambda?"}, outputs={'answer': "A runnable lambda is an object that implements LangChain's `Runnable` interface and runs a callbale (i.e., a function). Note the function must accept a single argument."}, metadata=None, id=UUID('f94104a7-2434-4ba7-8293-6a283f4860b4'), dataset_id=UUID('01b6ce0f-bfb6-4f48-bbb8-f19272135d40') ), ExampleSearch( inputs={'question': 'Show me how to use RecursiveURLLoader'}, outputs={'answer': 'The RecursiveURLLoader comes from the langchain.document_loaders.recursive_url_loader module. Here\'s an example of how to use it:\n\n```python\nfrom langchain.document_loaders.recursive_url_loader import RecursiveUrlLoader\n\n# Create an instance of RecursiveUrlLoader with the URL you want to load\nloader = RecursiveUrlLoader(url="https://example.com")\n\n# Load all child links from the URL page\nchild_links = loader.load()\n\n# Print the child links\nfor link in child_links:\n print(link)\n```\n\nMake sure to replace "https://example.com" with the actual URL you want to load. The load() method returns a list of child links found on the URL page. You can iterate over this list to access each child link.'}, metadata=None, id=UUID('0308ea70-a803-4181-a37d-39e95f138f8c'), dataset_id=UUID('01b6ce0f-bfb6-4f48-bbb8-f19272135d40') ), ] r)r|rarrirz/searchrpNexamplesr!) r+rr\rvr*rQrerrZ ExampleSearch) r0r|rarrrAreqr9r>exr!r!r$similar_examplesqs& D   zAsyncClient.similar_examplesls_schemas.LangSmithSettingscsD|jdur|ddIdH}t|tjdi||_|jS)zzGet the settings for the current tenant. Returns: dict: The settings for the current tenant. Nrcz /settingsr!)rr\r*rQrZLangSmithSettingsre)r0rZr!r!r$ _get_settingss   zAsyncClient._get_settingsownercs"|IdH}|dkp|j|kS)zCheck if the current workspace has the same handle as owner. Args: owner (str): The owner to check against. Returns: bool: True if the current tenant is the owner, False otherwise. N-)rC tenant_handle)r0rDsettingsr!r!r$_current_tenant_is_owners z$AsyncClient._current_tenant_is_owneractionls_utils.LangSmithUserErrorcs.|IdH}td|d|jd|S)NzCannot z% for another tenant. Current tenant: z, Requested tenant: )rCr*LangSmithUserErrorrF)r0rIrDrGr!r!r$_owner_conflict_errorsz!AsyncClient._owner_conflict_errorrKrprompt_owner_and_namer`csF|jdd|d||ddIdH}|d}|r!|dd SdS) aGet the latest commit hash for a prompt. Args: prompt_owner_and_name (str): The owner and name of the prompt. limit (int, default=1): The maximum number of commits to fetch. Defaults to 1. offset (int, default=0): The number of commits to skip. Defaults to 0. Returns: Optional[str]: The latest commit hash, or None if no commits are found. rc /commits/r4)rar`rdNcommitsr commit_hash)r\re)r0rMrar`rZrOr!r!r$_get_latest_commit_hashs   z#AsyncClient._get_latest_commit_hashprompt_identifierlikedict[str, int]csHt|\}}}|jdd|d|d|idIdH}||S)aLike or unlike a prompt. Args: prompt_identifier (str): The identifier of the prompt. like (bool): True to like the prompt, False to unlike it. Returns: A dictionary with the key 'likes' and the count of likes as the value. Raises: requests.exceptions.HTTPError: If the prompt is not found or another error occurs. riz/likes/r4rSr'N)r*parse_prompt_identifierr\raise_for_statusre)r0rRrSrD prompt_name_rZr!r!r$_like_or_unlike_prompts z"AsyncClient._like_or_unlike_promptc stt|\}}}||s |jd|d|d|ddS|IdH}|jd|d|ddd|jS)zGet a URL for a prompt. Args: prompt_identifier (str): The identifier of the prompt. Returns: str: The URL for the prompt. z/hub/r4:Nz /prompts/z?organizationId=)r*rUrHr>rCr)r0rRrDrWrPrGr!r!r$_get_prompt_urls  $zAsyncClient._get_prompt_urlcs||IdH}|r dSdS)zCheck if a prompt exists. Args: prompt_identifier (str): The identifier of the prompt. Returns: bool: True if the prompt exists, False otherwise. NTF) get_prompt)r0rRpromptr!r!r$_prompt_exists/s zAsyncClient._prompt_existsc|j|ddIdHS)zLike a prompt. Args: prompt_identifier (str): The identifier of the prompt. Returns: Dict[str, int]: A dictionary with the key 'likes' and the count of likes as the value. TrSNrYr0rRr!r!r$ like_prompt; zAsyncClient.like_promptcr`)zUnlike a prompt. Args: prompt_identifier (str): The identifier of the prompt. Returns: Dict[str, int]: A dictionary with the key 'likes' and the count of likes as the value. FraNrbrcr!r!r$ unlike_promptGrezAsyncClient.unlike_promptrbFZdesc)rar` is_public is_archived sort_fieldsort_directionrrgrhrils_schemas.PromptSortFieldrjLiteral['desc', 'asc']ls_schemas.ListPromptsResponsec s||||rdn|dur dnd|rdndt|tjr|jn||||r#dndd}|jddt|dIdH} tjdi| S) a#List prompts with pagination. Args: limit (int, default=100): The maximum number of prompts to return. Defaults to 100. offset (int, default=0): The number of prompts to skip. Defaults to 0. is_public (Optional[bool]): Filter prompts by if they are public. is_archived (Optional[bool]): Filter prompts by if they are archived. sort_field (PromptSortField): The field to sort by. Defaults to "updated_at". sort_direction (Literal["desc", "asc"], default="desc"): The order to sort by. Defaults to "desc". query (Optional[str]): Filter prompts by a search query. Returns: ListPromptsResponse: A response object containing the list of prompts. trueNfalse)rar`rgrhrirjrZ match_prefixrc/repos/rdr!)r,rPromptSortFieldrr\ _exclude_noneZListPromptsResponsere) r0rar`rgrhrirjrrCrZr!r!r$ list_promptsSs      zAsyncClient.list_promptsOptional[ls_schemas.Prompt]csbt|\}}}z|dd|d|IdH}tjdi|dWStjy0YdSw)aGet a specific prompt by its identifier. Args: prompt_identifier (str): The identifier of the prompt. The identifier should be in the format "prompt_name" or "owner/prompt_name". Returns: Optional[Prompt]: The prompt object. Raises: requests.exceptions.HTTPError: If the prompt is not found or another error occurs. rcrpr4Nrepor!)r*rUr\rPromptrerTr0rRrDrWrXrZr!r!r$r]s zAsyncClient.get_promptr#readmetagsrgryrzls_schemas.Promptc s|IdH}|r|jstdt|\}}} |j|dIdHs,|d|IdH||p0d|p3d|p6g|d} |jdd| d IdH} | t j d i| d S) aCreate a new prompt. Does not attach prompt object, just creates an empty prompt. Args: prompt_identifier (str): The identifier of the prompt. The identifier should be in the formatof owner/name:hash, name:hash, owner/name, or name description (Optional[str]): A description of the prompt. readme (Optional[str]): A readme for the prompt. tags (Optional[Sequence[str]]): A list of tags for the prompt. is_public (bool): Whether the prompt should be public. Defaults to False. Returns: Prompt: The created prompt object. Raises: ValueError: If the current tenant is not the owner. HTTPError: If the server request fails. NCannot create a public prompt without first creating a LangChain Hub handle. You can add a handle by creating a public prompt at: https://smith.langchain.com/prompts)rDzcreate a prompt)Z repo_handler#ryrzrgrirpr'rur!) rCrFr*rKrUrHrLr\rVrrvre) r0rRr#ryrzrgrGrDrWrXrerZr!r!r$ create_prompts$ zAsyncClient.create_promptparent_commit_hashobjectrcs||IdHstdzddlm}Wn ty!tdw||}t|}t|\}}} |d|} |dksB|durJ| | IdH}||d} |j d d | | d IdH} | d d } | | d| IdHS)aCreate a commit for an existing prompt. Args: prompt_identifier (str): The identifier of the prompt. object (Any): The LangChain object to commit. parent_commit_hash (Optional[str]): The hash of the parent commit. Defaults to latest commit. Returns: str: The url of the prompt commit. Raises: HTTPError: If the server request fails. ValueError: If the prompt does not exist. Nz0Prompt does not exist, you must create it first.r)dumpszvThe client.create_commit function requires the langchain_corepackage to run. Install with `pip install langchain_core`r4r6)Z parent_commitmanifestrirNr'commitrPrZ) r_r*rTZlangchain_core.load.dumpr ImportErrorreloadsrUrQr\r\)r0rRrrrZ json_objectZ manifest_dictrDrWrXrMZ request_dictrZrPr!r!r$ create_commits4     zAsyncClient.create_commit)r#ryrzrgrhc s|IdH}|r|jstdi}|dur||d<|dur#||d<|dur+||d<|dur3||d<|dur;||d<t|\} } } |jdd | d | |d IdH} | | S) a$Update a prompt's metadata. To update the content of a prompt, use push_prompt or create_commit instead. Args: prompt_identifier (str): The identifier of the prompt to update. description (Optional[str]): New description for the prompt. readme (Optional[str]): New readme for the prompt. tags (Optional[Sequence[str]]): New list of tags for the prompt. is_public (Optional[bool]): New public status for the prompt. is_archived (Optional[bool]): New archived status for the prompt. Returns: Dict[str, Any]: The updated prompt data as returned by the server. Raises: ValueError: If the prompt_identifier is empty. HTTPError: If the server request fails. Nr|r#ryrgrhrzrrpr4r')rCrFrr*rUr\rVre) r0rRr#ryrzrgrhrGrerDrWrXrZr!r!r$ update_prompts.  zAsyncClient.update_promptcs^t|\}}}||IdHs|d|IdH|dd|d|IdH}|dS)a6Delete a prompt. Args: prompt_identifier (str): The identifier of the prompt to delete. Returns: bool: True if the prompt was successfully deleted, False otherwise. Raises: ValueError: If the current tenant is not the owner of the prompt. Nzdelete a promptrrpr4)r*rUrHrLr\rVrwr!r!r$ delete_promptBs  zAsyncClient.delete_prompt include_modelrls_schemas.PromptCommitc s`t|\}}}|dd|d|d||rdndIdH}tjdi||d|S) aPull a prompt object from the LangSmith API. Args: prompt_identifier (str): The identifier of the prompt. Returns: PromptCommit: The prompt object. Raises: ValueError: If no commits are found for the prompt. rcrNr4z?include_model=truer}NrDrur!)r*rUr\rZ PromptCommitre)r0rRrrDrWrPrZr!r!r$pull_prompt_commitWs   zAsyncClient.pull_prompt_commitrar`r3AsyncGenerator[ls_schemas.ListedPromptCommit, None]cCst|\}}}|durtd|n|||d}d} ||d<|jdd|d |d |d IdH} | } | d } | d } | sAdS| D]}|durP| |krPdStjdi||d |V| d7} qC|t| 7}|| krodSq)aList commits for a given prompt. Args: prompt_identifier (str): The identifier of the prompt in the format 'owner/repo_name'. limit (Optional[int]): The maximum number of commits to return. If None, returns all commits. Defaults to None. offset (int, default=0): The number of commits to skip before starting to return results. Defaults to 0. include_model (bool, default=False): Whether to include the model information in the commit data. Defaults to False. Yields: A ListedPromptCommit object for each commit. Note: This method uses pagination to retrieve commits. It will make multiple API calls if necessary to retrieve all commits or up to the specified limit. NrbrrTr`rcrNr4rdrOtotalrrKr!)r*rUrr\rerZListedPromptCommitrf)r0rRrar`rrDrWrXrCrrZvalrNritr!r!r$list_prompt_commitsvs<    zAsyncClient.list_prompt_commitscsNz(ddlm}ddlm}ddlm}ddlm}ddlm }ddl m }m } Wn t y4t dwzdd lm} Wnt yMtjd d } Ynw|j||d Id H} | |t| j} Wd n1snwYt| |st| | rt| j|rt| |r| nt| | rt| j|r| jnd } | d urtd| jd uri| _| j| j| j| jd|r%t| | r%t| j|r%t| j dkr%t| j!|s%t| j!|rt| j!j"|rt#| | j| j!j"B}t|j dkr|j d}| | j|j$dii| j!j%|j!} | S|} | St| j!|r$| j| j!B} | S | S)aPull a prompt and return it as a LangChain PromptTemplate. This method requires `langchain_core`. Args: prompt_identifier (str): The identifier of the prompt. include_model (Optional[bool], default=False): Whether to include the model information in the prompt data. Returns: Any: The prompt object in the specified format. r)BaseLanguageModel)r)BaseOutputParser)BasePromptTemplate)StructuredPrompt)RunnableBindingRunnableSequenceztThe client.pull_prompt function requires the langchain_corepackage to run. Install with `pip install langchain_core`)suppress_langchain_beta_warningcss dVdSr;r!r!r!r!r$rs z@AsyncClient.pull_prompt..suppress_langchain_beta_warningrNz-Prompt object is not a valid prompt template.)Z lc_hub_ownerZ lc_hub_repoZlc_hub_commit_hashrLrrKr!)&Z#langchain_core.language_models.baserZlangchain_core.load.loadrZlangchain_core.output_parsersrZlangchain_core.promptsrZ!langchain_core.prompts.structuredrZlangchain_core.runnables.baserrrZlangchain_core._apir contextlibcontextmanagerrrerrr,firstr*rUrupdaterDrurPrfZstepslastboundr bindrA)r0rRrrrrrrrrrZ prompt_objectr^Zprompt_templateseqZ rebound_llmr!r!r$ pull_prompts                zAsyncClient.pull_prompt)rrrgr#ryrzc s||IdHr$tdd||||fDr#|j|||||dIdHn|j||dur-|nd|||dIdH|durD|j|dIdHS|j|||dIdH}|S) aPush a prompt to the LangSmith API. Can be used to update prompt metadata or prompt content. If the prompt does not exist, it will be created. If the prompt exists, it will be updated. Args: prompt_identifier (str): The identifier of the prompt. object (Optional[Any]): The LangChain object to push. parent_commit_hash (str): The parent commit hash. Defaults to "latest". is_public (Optional[bool]): Whether the prompt should be public. If None (default), the current visibility status is maintained for existing prompts. For new prompts, None defaults to private. Set to True to make public, or False to make private. description (Optional[str]): A description of the prompt. Defaults to an empty string. readme (Optional[str]): A readme for the prompt. Defaults to an empty string. tags (Optional[Sequence[str]]): A list of tags for the prompt. Defaults to an empty list. Returns: str: The URL of the prompt. Ncss|]}|duVqdSr;r!)r"paramr!r!r$ 8s z*AsyncClient.push_prompt..rxF)rgr#ryrz)rRr)r_anyrr~r\r) r0rRrrrgr#ryrzurlr!r!r$ push_prompts8&    zAsyncClient.push_prompt)NNNNN) rrrrrrrrrr)r3r)r3r<)r?r<r@r<rArr3rBr;)r]r<rCr^r3r_) r]r<rkrnrlr<rmr<r3ro)r{r<r|r}r~r<ryrrzrrArr3r)rrrArr3r)rrr3r)$rrryrr~rrrrrrrrrrrrrrrrrrrrrrrrrrarrArr3r)rrrrr3r<)rrr3r)rrr3r)ryr<rArr3r)NN)ryrrrr3r)ryrrrr3r)rr<rArr3r)rrrrr3r)rrr3r)rArr3r)NNN) r|r}rr^rrrrrArr3r)rrr3r)rrrrrArr3r)rrrr<rrrrrrrArr3r)rrrrrrrrrrrrnr3r) rrrr<rrrrrrr3r)rrr3r) rrrrrrrarrArr3r)rrr3r) rrr{rrrrarr3r)r{r<r#rr$r%r3r&)r$rr3r&)r$rr{r<r#rr3r)r$rr3r)r$rrr/r3r)r$rrrr3r)r$rr2r-r3r3)rrr7r<rArr3r)rrrArr3r) r|r<rar-rrrrrArr3r=)r3rB)rDr<r3r)rIr<rDr<r3rJ)rKr)rMr<rar-r`r-r3r)rRr<rSrr3rT)rRr<r3r<)rRr<r3r)rRr<r3rT)rar-r`r-rgrrhrrirkrjrlrrr3rm)rRr<r3rt) rRr<r#rryrrzrrgrr3r{)rRr<rrrrr3r<)rRr<r#rryrrzrrgrrhrr3r})rRr<r3r)rRr<rrr3r) rRr<rarr`r-rrr3r)rRr<rrr3r)rRr<rrrr<rgrr#rryrrzrr3r<)I__name__ __module__ __qualname____doc__ __slots__r2r5r:r6propertyr=r>r\rhrxrrrrrrrrrrrrrrrrrrrr r rrr"r(r*r,r.r0r1r5ls_betaZ warn_betar:r;rArCrHrLrQrYr\r_rdrfrrqZ updated_atrsr]r~rrrrrrrr!r!r!r$rsF #  ?$   J         *: A  1 #      W      2:;:#9grdr<r3cCsdd|DS)z&Exclude None values from a dictionary.cSrDr;r!rEr!r!r$rHYrIz!_exclude_none..)rN)rr!r!r$rrWsrr)rr<r3r<)#r __future__rrWrrrercollections.abcrrrrtypingrrr r r r/Z langsmithr r+r rrr*Zlangsmith._internalrrrr<rrrrr!r!r!r$s>     H