ܘh$<DdZddlmZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z mZddlmZddlmZ ddlmZmZmZmZmZmZn#e$rddlmZmZmZmZmZmZYnwxYwdd lmZe eeed fZe e eeeed fZGd d eZ Gd de Z!Gdde Z"GddeZ#GddeeZ$GddeZ%Gdde%Z&Gdde%Z'GddeeZ(GddeZ)Gdd e)Z*Gd!d"ed#$Z+Gd%d&e)Z,Gd'd(eZ-Gd)d*e-Z.Gd+d,e-Z/Gd-d.eZ0Gd/d0eZ1Gd1d2e1Z2Gd3d4e1Z3Gd5d6eZ4Gd7d8e4Z5eGd9d:e Z6Gd;deZ8Gd?d@ed#$Z9GdAdBeZ:e";d S)CzSchemas for the LangSmith API.) annotations)datetime timedelta)Enum)AnyDictListOptionalProtocolUnionruntime_checkable)UUID) TypedDict) BaseModelField PrivateAttr StrictBool StrictFloat StrictInt)LiteralNcdeZdZUdZded<ded<edZded <Gd d ZdS) ExampleBasezExample base model.r dataset_idzDict[str, Any]inputsNdefaultOptional[Dict[str, Any]]outputsceZdZdZdS)ExampleBase.ConfigTN__name__ __module__ __qualname__frozenY/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langsmith/schemas.pyConfigr 3r'r))r"r#r$__doc____annotations__rrr)r&r'r(rr,sz(-d(;(;(;G;;;;r'rcJeZdZUdZded<eejZded<dS) ExampleCreatezExample create model.Optional[UUID]iddefault_factoryr created_atN) r"r#r$r+r,rrutcnowr3r&r'r(r.r.7sD 5AAAJAAAAAAr'r.ceZdZUdZded<ded<edZded <ee Zd ed <dZ d ed<e dZ ded<e dZ d ed< ddfd Z eddZxZS)ExamplezExample model.rr0rr3NrOptional[datetime] modified_atr1z List[Run]runsr/ source_run_id Optional[str] _host_url _tenant_idkwargsrreturnNonec Vtjdi|||_||_dSzInitialize a Dataset object.Nr&)super__init__r<r=)selfr<r=r> __class__s r(rDzExample.__init__Is4 ""6""""$r'c|jrDd|jd|j}|jr |jdt |j|S|j|SdS)URL of this run within the app. /datasets/z/e//o/N)r<rr0r=str)rEpaths r(urlz Example.urlTso > -===DG==D J.IIS-A-AI4IIIn,d,, ,tr')NN)r<r;r=r/r>rr?r@r?r;)r"r#r$r+r,rr8listr9r:rr<r=rDpropertyrM __classcell__rFs@r(r6r6>s HHH&+eD&9&9&9K9999eD111D1111$(M((((*{4888I8888!,T!:!:!:J::::$(%) % % % % % % %Xr'r6cXeZdZUdZdZded<dZded<dZded<Gdd ZdS) ExampleUpdatezUpdate class for Example.Nr/rrrrceZdZdZdS)ExampleUpdate.ConfigTNr!r&r'r(r)rVfr*r'r)) r"r#r$r+rr,rrr)r&r'r(rTrT_sw##!%J%%%%'+F++++(,G,,,,r'rTceZdZdZdZdZdZdS)DataTypezEnum for dataset data types.kvllmchatN)r"r#r$r+rYrZr[r&r'r(rXrXjs#&& B C DDDr'rXcTeZdZUdZded<dZded<dZded<Gd d ZdS) DatasetBasezDataset base model.rKnameNr; descriptionzOptional[DataType] data_typeceZdZdZdS)DatasetBase.ConfigTNr!r&r'r(r)rbyr*r'r))r"r#r$r+r,r_r`r)r&r'r(r]r]rso III!%K%%%%$(I((((r'r]cNeZdZUdZdZded<eejZ ded<dS) DatasetCreatezDataset create model.Nr/r0r1rr3) r"r#r$r+r0r,rrr4r3r&r'r(rdrd}sIB 5AAAJAAAAAAr'rdc eZdZUdZded<ded<edZded <dZd ed <dZd ed <dZ ded <e dZ ded<e dZ ded<e dZ ded< ddfd ZeddZxZS)DatasetzDataset ORM model.rr0rr3Nrr7r8 Optional[int] example_count session_countlast_session_start_timer;r<r/r= _public_pathr>rr?r@c dtjdi|||_||_||_dSrB)rCrDr<r=rk)rEr<r=rkr>rFs r(rDzDataset.__init__s> ""6""""$(r'c|jrU|jr|j|jS|jr&|jdt|jd|jS|jd|jSdS)rHrJrIN)r<rkr=rKr0rEs r(rMz Dataset.urls > :  >.=$*;=== W.VVS-A-AVVTWVVVn9999 9tr')NNN) r<r;r=r/rkr;r>rr?r@rN)r"r#r$r+r,rr8rhrirjrr<r=rkrDrPrMrQrRs@r(rfrfs6 HHH&+eD&9&9&9K9999#'M''''#'M''''266666*{4888I8888!,T!:!:!:J::::"-+d";";";L;;;;$(%)&* ) ) ) ) ) ) )Xr'rfc.eZdZdZdZdZdZdZdZdZ dZ d S) RunTypeEnumz5(Deprecated) Enum for run types. Use string directly.toolchainrZ retriever embeddingpromptparserN) r"r#r$r+rqrrrZrsrtrurvr&r'r(rprps7?? D E CII F FFFr'rpceZdZUdZded< ded< ded< ded< d Zd ed < d Zd ed < d Zded< d Zd ed< d Z ded< ded< d Z d ed< d Z ded< d Z ded< d Z ded<d S)RunBasez[ Base Run schema. Contains the fundamental fields to define a run in a system. rr0rKr^r start_timerun_typeNr7end_timeOptional[dict]extrar;error serializedzOptional[List[Dict]]eventsdictrrr/reference_example_id parent_run_idOptional[List[str]]tags)r"r#r$r+r,r{r}r~rrrrrrr&r'r(rxrxs  HHH( III* MMM"$(H''''- E    =E;!%J%%%%F#'F''''LLL""G""""/+/////@$(M((((@ $D$$$$66r'rxcReZdZUdZdZded< dZded< dZded< dZd ed < dZ d ed < dZ ded < dZ d ed< dZ ded< dZ ded< dZded< dZded< dZded< ded< edZded< edZd ed<d#d$fd! Zed%d"ZxZS)&Runz$Run schema when loading from the DB.Nr/ session_idOptional[List[UUID]] child_run_idszOptional[List[Run]] child_runsrfeedback_statsr;app_path manifest_idstatusrg prompt_tokenscompletion_tokens total_tokensr7first_token_timeparent_run_idsrtrace_idrrK dotted_orderr<r>rr?r@c :|dsd|di|}tjdi|||_|js0|js+|j|j |_dSdSdS)Initialize a Run object.rr0Nr&) getrCrDr<rstriprry isoformatr0rEr<r>rFs r(rDz Run.__init__szz*%% > &**T"2"2=f=F""6"""" &&(( J1C J#'?#<#<#>#> I I ID    J J J Jr'cB|jr|jr|j|jSdS)rHN)r<rrns r(rMzRun.urls1 > 6dm 6n5dm55 5tr'Nr<r;r>rr?r@rN)r"r#r$r+rr,rrrrrrrrrrrrrrr<rDrPrMrQrRs@r(rrs..!%J%%%%-*.M....(&*J****M/3N3333&"H""""7"&K&&&&: F    .#'M''''/'+++++/"&L&&&&1+/////-+/N////!NNNCb)))L))))  +{4888I8888JJJJJJJXr'rceZdZUdZded<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#S)$ RunLikeDictz&Run-like dictionary, for type-hinting.rKr^rprzrryr|rrr7r{r}r;r~rr/rrzOptional[List[dict]]rrrinputs_s3_urlsoutputs_s3_urlsr0r session_namerinput_attachmentsoutput_attachmentsrrrNr"r#r$r+r,r&r'r(rr&s&00 III    !!!!    """"####((((%%%%&&&&NNNr'rF)totalc2eZdZUdZdZded< dZded<dS)RunWithAnnotationQueueInfoz&Run schema with annotation queue info.Nr7last_reviewed_timeadded_at)r"r#r$r+rr,rr&r'r(rrBs>00-11111.#'H''''33r'rc<eZdZUded<eeZded<dS)FeedbackSourceBaserKtyper1rmetadataN)r"r#r$r,rrrr&r'r(rrKs9 III).t)D)D)DHDDDDDDr'rc"eZdZUdZdZded<dS)APIFeedbackSourcezAPI feedback source.apizLiteral['api']rNr"r#r$r+rr,r&r'r(rrPs) D      r'rc"eZdZUdZdZded<dS)ModelFeedbackSourcezModel feedback source.modelzLiteral['model']rNrr&r'r(rrVs)  $D$$$$$$r'rceZdZdZdZ dZdS)FeedbackSourceTypezFeedback source type.rrN)r"r#r$r+APIMODELr&r'r(rr\s# C2 E""r'rceZdZUdZded< dZded< dZded< ded< d ed < dZd ed < dZd ed< dZ ded< dZ ded< dZ ded< GddZ dS) FeedbackBasezFeedback schema.rr0Nr7r3r8run_idrKkey SCORE_TYPEscore VALUE_TYPEvaluer;commentzUnion[str, dict, None] correctionOptional[FeedbackSourceBase]feedback_sourceceZdZdZdS)FeedbackBase.ConfigTNr!r&r'r(r)r}r*r'r)) r"r#r$r+r,r3r8rrrrrr)r&r'r(rres HHH(%)J)))),&*K****2LLL< HHHAE+EQ!G!!!!2)-J----!48O8888%r'rceZdZUdZded<dS)FeedbackCreatez"Schema used for creating feedback.rrNrr&r'r(rrs$,,''''%%r'rcDeZdZUdZded<ded< ded< dZded <dS) FeedbackzSchema for getting feedback.rr0rr3r8Nrr)r"r#r$r+r,rr&r'r(rrsP&& HHH,248O888822r'rceZdZUdZded< eejZded< dZ ded < dZ d ed < dZ d ed < dZ d ed< ded< e dZd ed<ddfd ZeddZeddZed dZxZS)! TracerSessionzbTracerSession schema for the API. Sessions are also referred to as "Projects" in the UI. rr0r1rryNr7r{r;r_r^rr} tenant_idrr<r>rr?r@c Htjdi|||_dS)rNr&)rCrDr<rs r(rDzTracerSession.__init__s+""6""""r'cF|jr|jd|jd|jSdS)rHrJz /projects/p/N)r<rr0rns r(rMzTracerSession.urls6 > OnNNNNTWNN Ntr'dict[str, Any]c@|j d|jvriS|jdS)zRetrieve the metadata (if any).Nrr}rns r(rzTracerSession.metadatas* : 4:!=!=Iz*%%r' List[str]c@|j d|jvrgS|jdS)zRetrieve the tags (if any).Nrrrns r(rzTracerSession.tagss* : tz!9!9Iz&!!r'rrrN)r?r)r?r)r"r#r$r+r,rrr4ryr{r_r^r}rr<rDrPrMrrrQrRs@r(rrsY  HHH  5AAAJAAAA+#'H'''')!%K%%%%)D"&*E****)OOO0*{4888I8888####### X &&&X& """X"""""r'rceZdZUdZded< ded< ded< ded< ded< ded < d ed < d ed < ded< ded<dS)TracerSessionResultzvTracerSession schema returned when reading a project by ID. Sessions are also referred to as "Projects" in the UI.rg run_countzOptional[timedelta] latency_p50 latency_p99rrrr7last_run_start_timerrrreference_dataset_idszOptional[List[Dict[str, Any]]] run_facetsNrr&r'r(rrsEE,$$$$?$$$$6=    D$$$$H++++8,,,,)////J....--r'rc@eZdZUdZded<ded<ed dZdS) BaseMessageLikezA A protocol representing objects similar to BaseMessage. rKcontentradditional_kwargsr?cdS)z,Type of the Message, used for serialization.Nr&rns r(rzBaseMessageLike.typesr'N)r?rK)r"r#r$r+r,rPrr&r'r(rrsULLL ;;;X;;;r'rc.eZdZUded<ded<ded<dS)DatasetShareSchemarr share_tokenrKrMNr"r#r$r,r&r'r(rrs1 HHHHHr'rceZdZUded<ded<dZded<eejZd ed <eejZ d ed <ded <dS) AnnotationQueuerr0rKr^Nr;r_r1rr3 updated_atr) r"r#r$r,r_rrr4r3rr&r'r(rrs| HHH III!%K%%%% 5AAAJAAAA 5AAAJAAAAOOOOOr'rc8eZdZUded<ded<ded<ded<dS)BatchIngestConfigintscale_up_qsize_triggerscale_up_nthreads_limitscale_down_nempty_trigger size_limitNrr&r'r(rrs=    """"OOOOOr'rcBeZdZUdZdZded< dZded< dZded <dS) LangSmithInfoz'Information about the LangSmith server.rrKversionNr7license_expiration_timezOptional[BatchIngestConfig]batch_ingest_config)r"r#r$r+rr,rrr&r'r(rrsS11G.266666+7;;;;;;;r'r)rs$$""""""((((((((''''''&%%%%% :y+t; < 4Y S$F G )BBBBBKBBBkBIsD)BBBBBKBBB#####k###L     #t   0707070707i070707f:::::':::z)5844444444EEEEEEEE !!!!!*!!! %%%%%,%%% ########98&&&&&\&&& 3 3 3 3 3| 3 3 3/"/"/"/"/"I/"/"/"d.....-...4 ; ; ; ; ;h ; ; ;     %    i <<<<