wandb 0.19.9__py3-none-any.whl → 0.19.10__py3-none-any.whl

wandb/automations/_generated/delete_trigger.py

@@ -0,0 +1,19 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/automations/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import DeleteTriggerResult
+
+
+class DeleteTrigger(GQLBase):
+    delete_trigger: Optional[DeleteTriggerResult] = Field(alias="deleteTrigger")
+
+
+DeleteTrigger.model_rebuild()

wandb/automations/_generated/enums.py

@@ -0,0 +1,33 @@
+# Generated by ariadne-codegen
+# Source: core/api/graphql/schemas/schema-latest.graphql
+
+from __future__ import annotations
+
+from enum import Enum
+
+
+class EventTriggeringConditionType(str, Enum):
+    CREATE_ARTIFACT = "CREATE_ARTIFACT"
+    UPDATE_ARTIFACT_ALIAS = "UPDATE_ARTIFACT_ALIAS"
+    ADD_ARTIFACT_ALIAS = "ADD_ARTIFACT_ALIAS"
+    LINK_MODEL = "LINK_MODEL"
+    RUN_METRIC = "RUN_METRIC"
+    RUN_METRIC_CHANGE = "RUN_METRIC_CHANGE"
+
+
+class TriggerScopeType(str, Enum):
+    PROJECT = "PROJECT"
+    ARTIFACT_COLLECTION = "ARTIFACT_COLLECTION"
+
+
+class AlertSeverity(str, Enum):
+    INFO = "INFO"
+    WARN = "WARN"
+    ERROR = "ERROR"
+
+
+class TriggeredActionType(str, Enum):
+    QUEUE_JOB = "QUEUE_JOB"
+    NOTIFICATION = "NOTIFICATION"
+    GENERIC_WEBHOOK = "GENERIC_WEBHOOK"
+    NO_OP = "NO_OP"

wandb/automations/_generated/fragments.py

@@ -0,0 +1,343 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/automations/
+
+from __future__ import annotations
+
+from datetime import datetime
+from typing import List, Literal, Optional, Union
+
+from pydantic import Field
+from typing_extensions import Annotated
+
+from wandb._pydantic import GQLBase, GQLId, Typename
+
+from .enums import AlertSeverity, EventTriggeringConditionType
+
+
+class ArtifactPortfolioScopeFields(GQLBase):
+    typename__: Typename[Literal["ArtifactPortfolio"]] = "ArtifactPortfolio"
+    id: GQLId
+    name: str
+
+
+class ArtifactSequenceScopeFields(GQLBase):
+    typename__: Typename[Literal["ArtifactSequence"]] = "ArtifactSequence"
+    id: GQLId
+    name: str
+
+
+class CreateFilterTriggerResult(GQLBase):
+    typename__: Typename[Literal["CreateFilterTriggerPayload"]] = (
+        "CreateFilterTriggerPayload"
+    )
+    trigger: Optional[TriggerFields]
+
+
+class DeleteTriggerResult(GQLBase):
+    typename__: Typename[Literal["DeleteTriggerPayload"]] = "DeleteTriggerPayload"
+    success: bool
+
+
+class FilterEventFields(GQLBase):
+    typename__: Typename[Literal["FilterEventTriggeringCondition"]] = (
+        "FilterEventTriggeringCondition"
+    )
+    event_type: EventTriggeringConditionType = Field(alias="eventType")
+    filter: str
+
+
+class GenericWebhookActionFields(GQLBase):
+    typename__: Typename[Literal["GenericWebhookTriggeredAction"]] = (
+        "GenericWebhookTriggeredAction"
+    )
+    integration: Union[
+        GenericWebhookActionFieldsIntegrationIntegration,
+        GenericWebhookActionFieldsIntegrationGenericWebhookIntegration,
+    ] = Field(discriminator="typename__")
+    request_payload: Optional[str] = Field(alias="requestPayload")
+
+
+class GenericWebhookActionFieldsIntegrationIntegration(GQLBase):
+    typename__: Typename[
+        Literal["GitHubOAuthIntegration", "Integration", "SlackIntegration"]
+    ]
+
+
+class GenericWebhookIntegrationConnectionFields(GQLBase):
+    typename__: Typename[Literal["IntegrationConnection"]] = "IntegrationConnection"
+    page_info: PageInfoFields = Field(alias="pageInfo")
+    edges: List[GenericWebhookIntegrationConnectionFieldsEdges]
+
+
+class GenericWebhookIntegrationConnectionFieldsEdges(GQLBase):
+    cursor: str
+    node: Optional[
+        Annotated[
+            Union[
+                GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration,
+                GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
+            ],
+            Field(discriminator="typename__"),
+        ]
+    ]
+
+
+class GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration(GQLBase):
+    typename__: Typename[
+        Literal["GitHubOAuthIntegration", "Integration", "SlackIntegration"]
+    ]
+
+
+class GenericWebhookIntegrationFields(GQLBase):
+    typename__: Typename[Literal["GenericWebhookIntegration"]] = (
+        "GenericWebhookIntegration"
+    )
+    id: GQLId
+    name: str
+    url_endpoint: str = Field(alias="urlEndpoint")
+    created_at: datetime = Field(alias="createdAt")
+
+
+class GithubIntegrationFields(GQLBase):
+    typename__: Typename[Literal["GitHubOAuthIntegration"]] = "GitHubOAuthIntegration"
+    id: GQLId
+
+
+class IntegrationConnectionFields(GQLBase):
+    typename__: Typename[Literal["IntegrationConnection"]] = "IntegrationConnection"
+    page_info: PageInfoFields = Field(alias="pageInfo")
+    edges: List[IntegrationConnectionFieldsEdges]
+
+
+class IntegrationConnectionFieldsEdges(GQLBase):
+    cursor: str
+    node: Optional[
+        Annotated[
+            Union[
+                IntegrationConnectionFieldsEdgesNodeIntegration,
+                IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
+                IntegrationConnectionFieldsEdgesNodeSlackIntegration,
+            ],
+            Field(discriminator="typename__"),
+        ]
+    ]
+
+
+class IntegrationConnectionFieldsEdgesNodeIntegration(GQLBase):
+    typename__: Typename[Literal["GitHubOAuthIntegration", "Integration"]]
+
+
+class NoOpActionFields(GQLBase):
+    typename__: Typename[Literal["NoOpTriggeredAction"]] = "NoOpTriggeredAction"
+    no_op: Optional[bool] = Field(alias="noOp")
+
+
+class NotificationActionFields(GQLBase):
+    typename__: Typename[Literal["NotificationTriggeredAction"]] = (
+        "NotificationTriggeredAction"
+    )
+    integration: Union[
+        NotificationActionFieldsIntegrationIntegration,
+        NotificationActionFieldsIntegrationSlackIntegration,
+    ] = Field(discriminator="typename__")
+    title: Optional[str]
+    message: Optional[str]
+    severity: Optional[AlertSeverity]
+
+
+class NotificationActionFieldsIntegrationIntegration(GQLBase):
+    typename__: Typename[
+        Literal["GenericWebhookIntegration", "GitHubOAuthIntegration", "Integration"]
+    ]
+
+
+class PageInfoFields(GQLBase):
+    end_cursor: Optional[str] = Field(alias="endCursor")
+    has_next_page: bool = Field(alias="hasNextPage")
+
+
+class ProjectConnectionFields(GQLBase):
+    typename__: Typename[Literal["ProjectConnection"]] = "ProjectConnection"
+    page_info: PageInfoFields = Field(alias="pageInfo")
+    edges: List[ProjectConnectionFieldsEdges]
+
+
+class ProjectConnectionFieldsEdges(GQLBase):
+    cursor: str
+    node: Optional[ProjectConnectionFieldsEdgesNode]
+
+
+class ProjectConnectionFieldsEdgesNode(GQLBase):
+    triggers: List[TriggerFields]
+
+
+class ProjectScopeFields(GQLBase):
+    typename__: Typename[Literal["Project"]] = "Project"
+    id: GQLId
+    name: str
+
+
+class QueueJobActionFields(GQLBase):
+    typename__: Typename[Literal["QueueJobTriggeredAction"]] = "QueueJobTriggeredAction"
+    queue: Optional[RunQueueFields]
+    template: str
+
+
+class RunQueueFields(GQLBase):
+    typename__: Typename[Literal["RunQueue"]] = "RunQueue"
+    id: GQLId
+    name: str
+
+
+class SlackIntegrationConnectionFields(GQLBase):
+    typename__: Typename[Literal["IntegrationConnection"]] = "IntegrationConnection"
+    page_info: PageInfoFields = Field(alias="pageInfo")
+    edges: List[SlackIntegrationConnectionFieldsEdges]
+
+
+class SlackIntegrationConnectionFieldsEdges(GQLBase):
+    cursor: str
+    node: Optional[
+        Annotated[
+            Union[
+                SlackIntegrationConnectionFieldsEdgesNodeIntegration,
+                SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration,
+            ],
+            Field(discriminator="typename__"),
+        ]
+    ]
+
+
+class SlackIntegrationConnectionFieldsEdgesNodeIntegration(GQLBase):
+    typename__: Typename[
+        Literal["GenericWebhookIntegration", "GitHubOAuthIntegration", "Integration"]
+    ]
+
+
+class SlackIntegrationFields(GQLBase):
+    typename__: Typename[Literal["SlackIntegration"]] = "SlackIntegration"
+    id: GQLId
+    team_name: str = Field(alias="teamName")
+    channel_name: str = Field(alias="channelName")
+
+
+class TriggerFields(GQLBase):
+    typename__: Typename[Literal["Trigger"]] = "Trigger"
+    id: GQLId
+    created_at: datetime = Field(alias="createdAt")
+    created_by: UserFields = Field(alias="createdBy")
+    updated_at: Optional[datetime] = Field(alias="updatedAt")
+    name: str
+    description: Optional[str]
+    enabled: bool
+    scope: Union[
+        TriggerFieldsScopeProject,
+        TriggerFieldsScopeArtifactSequence,
+        TriggerFieldsScopeArtifactPortfolio,
+    ] = Field(discriminator="typename__")
+    event: TriggerFieldsEventFilterEventTriggeringCondition
+    action: Union[
+        TriggerFieldsActionQueueJobTriggeredAction,
+        TriggerFieldsActionNotificationTriggeredAction,
+        TriggerFieldsActionGenericWebhookTriggeredAction,
+        TriggerFieldsActionNoOpTriggeredAction,
+    ] = Field(discriminator="typename__")
+
+
+class UpdateFilterTriggerResult(GQLBase):
+    typename__: Typename[Literal["UpdateFilterTriggerPayload"]] = (
+        "UpdateFilterTriggerPayload"
+    )
+    trigger: Optional[TriggerFields]
+
+
+class UserFields(GQLBase):
+    typename__: Typename[Literal["User"]] = "User"
+    id: GQLId
+    username: Optional[str]
+
+
+class TriggerFieldsScopeArtifactPortfolio(ArtifactPortfolioScopeFields):
+    typename__: Typename[Literal["ArtifactPortfolio"]]
+
+
+class TriggerFieldsScopeArtifactSequence(ArtifactSequenceScopeFields):
+    typename__: Typename[Literal["ArtifactSequence"]]
+
+
+class TriggerFieldsEventFilterEventTriggeringCondition(FilterEventFields):
+    typename__: Typename[Literal["FilterEventTriggeringCondition"]]
+
+
+class TriggerFieldsActionGenericWebhookTriggeredAction(GenericWebhookActionFields):
+    typename__: Typename[Literal["GenericWebhookTriggeredAction"]]
+
+
+class GenericWebhookActionFieldsIntegrationGenericWebhookIntegration(
+    GenericWebhookIntegrationFields
+):
+    typename__: Typename[Literal["GenericWebhookIntegration"]]
+
+
+class GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
+    GenericWebhookIntegrationFields
+):
+    typename__: Typename[Literal["GenericWebhookIntegration"]]
+
+
+class IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
+    GenericWebhookIntegrationFields
+):
+    typename__: Typename[Literal["GenericWebhookIntegration"]]
+
+
+class TriggerFieldsActionNoOpTriggeredAction(NoOpActionFields):
+    typename__: Typename[Literal["NoOpTriggeredAction"]]
+
+
+class TriggerFieldsActionNotificationTriggeredAction(NotificationActionFields):
+    typename__: Typename[Literal["NotificationTriggeredAction"]]
+
+
+class TriggerFieldsScopeProject(ProjectScopeFields):
+    typename__: Typename[Literal["Project"]]
+
+
+class TriggerFieldsActionQueueJobTriggeredAction(QueueJobActionFields):
+    typename__: Typename[Literal["QueueJobTriggeredAction"]]
+
+
+class IntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
+    typename__: Typename[Literal["SlackIntegration"]]
+
+
+class NotificationActionFieldsIntegrationSlackIntegration(SlackIntegrationFields):
+    typename__: Typename[Literal["SlackIntegration"]]
+
+
+class SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
+    typename__: Typename[Literal["SlackIntegration"]]
+
+
+ArtifactPortfolioScopeFields.model_rebuild()
+ArtifactSequenceScopeFields.model_rebuild()
+CreateFilterTriggerResult.model_rebuild()
+DeleteTriggerResult.model_rebuild()
+FilterEventFields.model_rebuild()
+GenericWebhookActionFields.model_rebuild()
+GenericWebhookIntegrationConnectionFields.model_rebuild()
+GenericWebhookIntegrationFields.model_rebuild()
+GithubIntegrationFields.model_rebuild()
+IntegrationConnectionFields.model_rebuild()
+NoOpActionFields.model_rebuild()
+NotificationActionFields.model_rebuild()
+PageInfoFields.model_rebuild()
+ProjectConnectionFields.model_rebuild()
+ProjectScopeFields.model_rebuild()
+QueueJobActionFields.model_rebuild()
+RunQueueFields.model_rebuild()
+SlackIntegrationConnectionFields.model_rebuild()
+SlackIntegrationFields.model_rebuild()
+TriggerFields.model_rebuild()
+UpdateFilterTriggerResult.model_rebuild()
+UserFields.model_rebuild()

wandb/automations/_generated/generic_webhook_integrations_by_entity.py

@@ -0,0 +1,22 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/automations/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from wandb._pydantic import GQLBase
+
+from .fragments import GenericWebhookIntegrationConnectionFields
+
+
+class GenericWebhookIntegrationsByEntity(GQLBase):
+    entity: Optional[GenericWebhookIntegrationsByEntityEntity]
+
+
+class GenericWebhookIntegrationsByEntityEntity(GQLBase):
+    integrations: Optional[GenericWebhookIntegrationConnectionFields]
+
+
+GenericWebhookIntegrationsByEntity.model_rebuild()
+GenericWebhookIntegrationsByEntityEntity.model_rebuild()

wandb/automations/_generated/get_triggers.py

@@ -0,0 +1,24 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/automations/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ProjectConnectionFields
+
+
+class GetTriggers(GQLBase):
+    search_scope: Optional[GetTriggersSearchScope] = Field(alias="searchScope")
+
+
+class GetTriggersSearchScope(GQLBase):
+    projects: Optional[ProjectConnectionFields]
+
+
+GetTriggers.model_rebuild()
+GetTriggersSearchScope.model_rebuild()

wandb/automations/_generated/get_triggers_by_entity.py

@@ -0,0 +1,24 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/automations/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase
+
+from .fragments import ProjectConnectionFields
+
+
+class GetTriggersByEntity(GQLBase):
+    search_scope: Optional[GetTriggersByEntitySearchScope] = Field(alias="searchScope")
+
+
+class GetTriggersByEntitySearchScope(GQLBase):
+    projects: Optional[ProjectConnectionFields]
+
+
+GetTriggersByEntity.model_rebuild()
+GetTriggersByEntitySearchScope.model_rebuild()

wandb/automations/_generated/input_types.py

@@ -0,0 +1,104 @@
+# Generated by ariadne-codegen
+# Source: core/api/graphql/schemas/schema-latest.graphql
+
+from __future__ import annotations
+
+from typing import Optional
+
+from pydantic import Field
+
+from wandb._pydantic import GQLBase, GQLId
+
+from .enums import (
+    AlertSeverity,
+    EventTriggeringConditionType,
+    TriggeredActionType,
+    TriggerScopeType,
+)
+
+
+class CreateGenericWebhookIntegrationInput(GQLBase):
+    entity_name: str = Field(alias="entityName")
+    url_endpoint: str = Field(alias="urlEndpoint")
+    name: str
+    secret_ref: Optional[str] = Field(alias="secretRef", default=None)
+    access_token_ref: Optional[str] = Field(alias="accessTokenRef", default=None)
+    client_mutation_id: Optional[str] = Field(alias="clientMutationId", default=None)
+
+
+class QueueJobActionInput(GQLBase):
+    queue_id: GQLId = Field(alias="queueID")
+    template: str
+
+
+class NotificationActionInput(GQLBase):
+    integration_id: GQLId = Field(alias="integrationID")
+    title: Optional[str] = None
+    message: Optional[str] = None
+    severity: Optional[AlertSeverity] = None
+
+
+class GenericWebhookActionInput(GQLBase):
+    integration_id: GQLId = Field(alias="integrationID")
+    request_payload: Optional[str] = Field(alias="requestPayload", default=None)
+
+
+class NoOpTriggeredActionInput(GQLBase):
+    no_op: Optional[bool] = Field(alias="noOp", default=None)
+
+
+class TriggeredActionConfig(GQLBase):
+    queue_job_action_input: Optional[QueueJobActionInput] = Field(
+        alias="queueJobActionInput", default=None
+    )
+    notification_action_input: Optional[NotificationActionInput] = Field(
+        alias="notificationActionInput", default=None
+    )
+    generic_webhook_action_input: Optional[GenericWebhookActionInput] = Field(
+        alias="genericWebhookActionInput", default=None
+    )
+    no_op_action_input: Optional[NoOpTriggeredActionInput] = Field(
+        alias="noOpActionInput", default=None
+    )
+
+
+class CreateFilterTriggerInput(GQLBase):
+    name: str
+    description: Optional[str] = None
+    triggering_event_type: EventTriggeringConditionType = Field(
+        alias="triggeringEventType"
+    )
+    scope_type: TriggerScopeType = Field(alias="scopeType")
+    scope_id: GQLId = Field(alias="scopeID")
+    event_filter: str = Field(alias="eventFilter")
+    triggered_action_type: TriggeredActionType = Field(alias="triggeredActionType")
+    triggered_action_config: TriggeredActionConfig = Field(
+        alias="triggeredActionConfig"
+    )
+    enabled: bool
+    client_mutation_id: Optional[str] = Field(alias="clientMutationId", default=None)
+
+
+class UpdateFilterTriggerInput(GQLBase):
+    id: GQLId
+    name: Optional[str] = None
+    description: Optional[str] = None
+    triggering_event_type: Optional[EventTriggeringConditionType] = Field(
+        alias="triggeringEventType", default=None
+    )
+    scope_type: Optional[TriggerScopeType] = Field(alias="scopeType", default=None)
+    scope_id: Optional[GQLId] = Field(alias="scopeID", default=None)
+    event_filter: Optional[str] = Field(alias="eventFilter", default=None)
+    triggered_action_type: Optional[TriggeredActionType] = Field(
+        alias="triggeredActionType", default=None
+    )
+    triggered_action_config: Optional[TriggeredActionConfig] = Field(
+        alias="triggeredActionConfig", default=None
+    )
+    enabled: Optional[bool] = None
+    client_mutation_id: Optional[str] = Field(alias="clientMutationId", default=None)
+
+
+TriggeredActionConfig.model_rebuild()
+CreateFilterTriggerInput.model_rebuild()
+UpdateFilterTriggerInput.model_rebuild()

wandb/automations/_generated/integrations_by_entity.py

@@ -0,0 +1,22 @@
+# Generated by ariadne-codegen
+# Source: tools/graphql_codegen/automations/
+
+from __future__ import annotations
+
+from typing import Optional
+
+from wandb._pydantic import GQLBase
+
+from .fragments import IntegrationConnectionFields
+
+
+class IntegrationsByEntity(GQLBase):
+    entity: Optional[IntegrationsByEntityEntity]
+
+
+class IntegrationsByEntityEntity(GQLBase):
+    integrations: Optional[IntegrationConnectionFields]
+
+
+IntegrationsByEntity.model_rebuild()
+IntegrationsByEntityEntity.model_rebuild()