o -wi(@sbUdZddlmZddlmZmZmZmZmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZmZddlmZmZmZdd lmZdd lmZmZmZdd l m!Z!dd l"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,er{ddl-m.Z.Gddde#Z/GdddeZ0GdddeZ1GdddeZ2Gddde!Z3GdddeZ4Gddde4Z5Gddde5Z6Gd d!d!e5Z7Gd"d#d#e5Z8Gd$d%d%e4Z9Gd&d'd'e9Z:e ee6e7e8e:fe d(d)fZ;ee;j<Z=d*e>d+<Gd,d-d-Z?Gd.d/d/Z@eAe2Ae0Ae6Ae7Ae8Ae:Adgd0d1e=Dd-d/d2d3ZBd4S)5z$Events that trigger W&B Automations.) annotations) TYPE_CHECKINGAnyLiteralOptionalUnion)Field) AnnotatedSelfget_args)GQLBaseSerializedToJson ensure_jsonfield_validatormodel_validatorpydantic_isinstance)AndMongoLikeFilterOr)FilterableField)MetricChangeFilterMetricThresholdFilter MetricVal)FilterEventFields)LenientStrEnum simplify_op) InputActionInputActionTypesSavedActionTypes)ArtifactCollectionScopeAutomationScope ProjectScope NewAutomationc@s(eZdZdZdZdZdZdZdZdZ dS) EventTypez.The type of event that triggers an automation.UPDATE_ARTIFACT_ALIASCREATE_ARTIFACTADD_ARTIFACT_ALIAS LINK_MODEL RUN_METRICRUN_METRIC_CHANGEN) __name__ __module__ __qualname____doc__r&r'r( LINK_ARTIFACTRUN_METRIC_THRESHOLDr+r2r2U/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/automations/events.pyr% sr%c@seZdZUeZded<dS)_WrappedSavedEventFilter!SerializedToJson[MongoLikeFilter]filterN)r,r-r.rr6__annotations__r2r2r2r3r48s r4c@s`eZdZUdZded<dZded<eddedd d Zed ddddZ e dddZ dS)_WrappedMetricFilterNzOptional[MetricThresholdFilter]threshold_filterzOptional[MetricChangeFilter] change_filterbeforemodevrreturncCs,t|tr ||dSt|tr||dS|S)N)r9)r:rrrclsr>r2r2r3_wrap_metric_filter@s    z(_WrappedMetricFilter._wrap_metric_filterafterr cCsddd|D}|sdttt|j}td|t|dkr0dtt|}td||S)NcSsg|] \}}|dur|qSNr2).0namevalr2r2r3 Ksz@_WrappedMetricFilter._ensure_exactly_one_set..z, zExpected one of: rz)Expected exactly one metric filter, got: )joinmapreprtype model_fields ValueErrorlen)self set_fields all_names set_namesr2r2r3_ensure_exactly_one_setIs z,_WrappedMetricFilter._ensure_exactly_one_setr%cCs(|jdurtjS|jdurtjStd)Nz6Expected one of: `threshold_filter` or `change_filter`)r9r%r1r:r+ RuntimeError)rQr2r2r3 event_typeWs  z_WrappedMetricFilter.event_typer>rr?r)r?r )r?r%) r,r-r.r9r7r:r classmethodrCrUpropertyrWr2r2r2r3r8<s    r8c@s`eZdZUeZded<ded<dZded< edd edd dZ e ddd dddZ dS)RunMetricFilterzGAnnotated[SerializedToJson[MongoLikeFilter], Field(alias='run_filter')]runzAAnnotated[_WrappedMetricFilter, Field(alias='run_metric_filter')]metricNzkAnnotated[Optional[SerializedToJson[MetricThresholdFilter]], Field(alias='metric_filter', deprecated=True)]legacy_metric_filterr;r<r>rr?cCst|ttfr ||dS|S)N)r]r@rAr2r2r3rCns z#RunMetricFilter._wrap_metric_filterrDrcCs"t|}t|tr |St|gdS)Nand_)rrrrBr>v_newr2r2r3_wrap_run_filterysz RunMetricFilter._wrap_run_filterrX)r>rr?r) r,r-r.rr\r7r^rrYrCrrcr2r2r2r3r[`s   r[c@s"eZdZUdZded<ded<dS) SavedEventz+A triggering event from a saved automation.z(Annotated[EventType, Field(frozen=True)]rWzBSerializedToJson[Union[_WrappedSavedEventFilter, RunMetricFilter]]r6N)r,r-r.r/r7r2r2r2r3rds rdc@s<eZdZUded<ded< ded<dd d ZdddZdS)_BaseEventInputr%rWr!scopezSerializedToJson[Any]r6actionrr?r$cCs:ddlm}t|ttfr|||dStdt|j)zFDefine a new Automation in which this event triggers the given action.rr#)eventrgzExpected a valid action, got: ) automationsr$ isinstancerr TypeErrorrMr.)rQrgr$r2r2r3thens  z_BaseEventInput.thenothercCs ||S)zPImplements `event >> action` to define an Automation with this event and action.)rl)rQrmr2r2r3 __rshift__s z_BaseEventInput.__rshift__N)rgrr?r$)rmrr?r$)r,r-r.r7rlrnr2r2r2r3res   rec@s4eZdZUeZded< edddd dd Zd S) _BaseMutationEventInputr5r6rDr<r>rr?cCs.t|}t|tr |nt|gd}t|gdS)zEnsure the given filter is wrapped like: `{"$or": [{"$and": []}]}`. This is awkward but necessary, because the frontend expects this format. r_)or_)rrrrrar2r2r3 _wrap_filters z$_BaseMutationEventInput._wrap_filterNrX)r,r-r.rr6r7rrqr2r2r2r3ros  roc@ eZdZUdZejZded<dS)OnLinkArtifactz)A new artifact is linked to a collection.z Literal[EventType.LINK_ARTIFACT]rWN)r,r-r.r/r%r0rWr7r2r2r2r3rs rsc@rr)OnAddArtifactAliasz'A new alias is assigned to an artifact.z%Literal[EventType.ADD_ARTIFACT_ALIAS]rWN)r,r-r.r/r%r(rWr7r2r2r2r3rurtruc@s(eZdZUdZejZded<ded<dS)OnCreateArtifactzA new artifact is created.z"Literal[EventType.CREATE_ARTIFACT]rWr rfN)r,r-r.r/r%r'rWr7r2r2r2r3rvs rvc@seZdZUded<dS)_BaseRunEventInputr"rfN)r,r-r.r7r2r2r2r3rws rwc@s<eZdZUdZded<ded< eddedd d Zd S) OnRunMetricz0A run metric satisfies a user-defined condition.zDLiteral[EventType.RUN_METRIC_THRESHOLD, EventType.RUN_METRIC_CHANGE]rWz!SerializedToJson[RunMetricFilter]r6r;r<datarr?cCs>t|tr|d}rtt|}i|d|jjiS|S)a Infer the event type at validation time from the inner filter. This allows this class to accommodate both "threshold" and "change" metric filter types, which are can only be determined after parsing and validating the inner JSON data. r6rW)rjdictgetr[model_validate_jsonrr]rW)rBry raw_filter parsed_filterr2r2r3_infer_event_types zOnRunMetric._infer_event_typeN)ryrr?r)r,r-r.r/r7rrYrr2r2r2r3rxs rxrW) discriminatorztuple[type, ...]InputEventTypesc@s$eZdZeddZed ddZd S) RunEvent display_name) server_namerGstrr?rcCs t|dS)z!Define a metric filter condition.rG)rrr2r2r3r]s zRunEvent.metricN)rGrr?r)r,r-r.rrG staticmethodr]r2r2r2r3rs rc@seZdZeZdS) ArtifactEventN)r,r-r.raliasr2r2r2r3r s rccs|]}|jVqdSrE)r,)rFrBr2r2r3 srrrN)Cr/ __future__rtypingrrrrrpydanticrtyping_extensionsr r r wandb._pydanticr r rrrr_filtersrrr_filters.expressionsr_filters.run_metricsrrr _generatedr _validatorsrractionsrrrscopesr r!r"rir$r%r4r8r[rdrerorsrurvrwrx InputEvent __origin__rr7rr model_rebuild__all__r2r2r2r3sx      $