wandb 0.19.8__py3-none-win_amd64.whl → 0.19.10__py3-none-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (154) hide show
  1. wandb/__init__.py +5 -1
  2. wandb/__init__.pyi +15 -8
  3. wandb/_pydantic/__init__.py +30 -0
  4. wandb/_pydantic/base.py +148 -0
  5. wandb/_pydantic/utils.py +66 -0
  6. wandb/_pydantic/v1_compat.py +284 -0
  7. wandb/apis/paginator.py +82 -38
  8. wandb/apis/public/__init__.py +2 -2
  9. wandb/apis/public/api.py +111 -53
  10. wandb/apis/public/artifacts.py +387 -639
  11. wandb/apis/public/automations.py +69 -0
  12. wandb/apis/public/files.py +2 -2
  13. wandb/apis/public/integrations.py +168 -0
  14. wandb/apis/public/projects.py +32 -2
  15. wandb/apis/public/reports.py +2 -2
  16. wandb/apis/public/runs.py +19 -11
  17. wandb/apis/public/utils.py +107 -1
  18. wandb/automations/__init__.py +81 -0
  19. wandb/automations/_filters/__init__.py +40 -0
  20. wandb/automations/_filters/expressions.py +179 -0
  21. wandb/automations/_filters/operators.py +267 -0
  22. wandb/automations/_filters/run_metrics.py +183 -0
  23. wandb/automations/_generated/__init__.py +184 -0
  24. wandb/automations/_generated/create_filter_trigger.py +21 -0
  25. wandb/automations/_generated/create_generic_webhook_integration.py +43 -0
  26. wandb/automations/_generated/delete_trigger.py +19 -0
  27. wandb/automations/_generated/enums.py +33 -0
  28. wandb/automations/_generated/fragments.py +343 -0
  29. wandb/automations/_generated/generic_webhook_integrations_by_entity.py +22 -0
  30. wandb/automations/_generated/get_triggers.py +24 -0
  31. wandb/automations/_generated/get_triggers_by_entity.py +24 -0
  32. wandb/automations/_generated/input_types.py +104 -0
  33. wandb/automations/_generated/integrations_by_entity.py +22 -0
  34. wandb/automations/_generated/operations.py +710 -0
  35. wandb/automations/_generated/slack_integrations_by_entity.py +22 -0
  36. wandb/automations/_generated/update_filter_trigger.py +21 -0
  37. wandb/automations/_utils.py +123 -0
  38. wandb/automations/_validators.py +73 -0
  39. wandb/automations/actions.py +205 -0
  40. wandb/automations/automations.py +109 -0
  41. wandb/automations/events.py +235 -0
  42. wandb/automations/integrations.py +26 -0
  43. wandb/automations/scopes.py +76 -0
  44. wandb/beta/workflows.py +9 -10
  45. wandb/bin/gpu_stats.exe +0 -0
  46. wandb/bin/wandb-core +0 -0
  47. wandb/cli/cli.py +3 -3
  48. wandb/integration/keras/keras.py +2 -1
  49. wandb/integration/langchain/wandb_tracer.py +2 -1
  50. wandb/integration/metaflow/metaflow.py +19 -17
  51. wandb/integration/sacred/__init__.py +1 -1
  52. wandb/jupyter.py +155 -133
  53. wandb/old/summary.py +0 -2
  54. wandb/proto/v3/wandb_internal_pb2.py +297 -292
  55. wandb/proto/v3/wandb_settings_pb2.py +2 -2
  56. wandb/proto/v3/wandb_telemetry_pb2.py +10 -10
  57. wandb/proto/v4/wandb_internal_pb2.py +292 -292
  58. wandb/proto/v4/wandb_settings_pb2.py +2 -2
  59. wandb/proto/v4/wandb_telemetry_pb2.py +10 -10
  60. wandb/proto/v5/wandb_internal_pb2.py +292 -292
  61. wandb/proto/v5/wandb_settings_pb2.py +2 -2
  62. wandb/proto/v5/wandb_telemetry_pb2.py +10 -10
  63. wandb/proto/v6/wandb_base_pb2.py +41 -0
  64. wandb/proto/v6/wandb_internal_pb2.py +393 -0
  65. wandb/proto/v6/wandb_server_pb2.py +78 -0
  66. wandb/proto/v6/wandb_settings_pb2.py +58 -0
  67. wandb/proto/v6/wandb_telemetry_pb2.py +52 -0
  68. wandb/proto/wandb_base_pb2.py +2 -0
  69. wandb/proto/wandb_deprecated.py +10 -0
  70. wandb/proto/wandb_internal_pb2.py +3 -1
  71. wandb/proto/wandb_server_pb2.py +2 -0
  72. wandb/proto/wandb_settings_pb2.py +2 -0
  73. wandb/proto/wandb_telemetry_pb2.py +2 -0
  74. wandb/sdk/artifacts/_generated/__init__.py +248 -0
  75. wandb/sdk/artifacts/_generated/artifact_collection_membership_files.py +43 -0
  76. wandb/sdk/artifacts/_generated/artifact_version_files.py +36 -0
  77. wandb/sdk/artifacts/_generated/create_artifact_collection_tag_assignments.py +36 -0
  78. wandb/sdk/artifacts/_generated/delete_artifact_collection_tag_assignments.py +25 -0
  79. wandb/sdk/artifacts/_generated/delete_artifact_portfolio.py +35 -0
  80. wandb/sdk/artifacts/_generated/delete_artifact_sequence.py +35 -0
  81. wandb/sdk/artifacts/_generated/enums.py +17 -0
  82. wandb/sdk/artifacts/_generated/fragments.py +186 -0
  83. wandb/sdk/artifacts/_generated/input_types.py +16 -0
  84. wandb/sdk/artifacts/_generated/move_artifact_collection.py +35 -0
  85. wandb/sdk/artifacts/_generated/operations.py +510 -0
  86. wandb/sdk/artifacts/_generated/project_artifact_collection.py +101 -0
  87. wandb/sdk/artifacts/_generated/project_artifact_collections.py +33 -0
  88. wandb/sdk/artifacts/_generated/project_artifact_type.py +24 -0
  89. wandb/sdk/artifacts/_generated/project_artifact_types.py +24 -0
  90. wandb/sdk/artifacts/_generated/project_artifacts.py +42 -0
  91. wandb/sdk/artifacts/_generated/run_input_artifacts.py +51 -0
  92. wandb/sdk/artifacts/_generated/run_output_artifacts.py +51 -0
  93. wandb/sdk/artifacts/_generated/update_artifact_portfolio.py +35 -0
  94. wandb/sdk/artifacts/_generated/update_artifact_sequence.py +35 -0
  95. wandb/sdk/artifacts/_graphql_fragments.py +56 -81
  96. wandb/sdk/artifacts/_validators.py +1 -0
  97. wandb/sdk/artifacts/artifact.py +110 -49
  98. wandb/sdk/artifacts/artifact_manifest_entry.py +2 -1
  99. wandb/sdk/artifacts/artifact_saver.py +16 -2
  100. wandb/sdk/artifacts/storage_handlers/azure_handler.py +1 -0
  101. wandb/sdk/artifacts/storage_policies/wandb_storage_policy.py +23 -2
  102. wandb/sdk/data_types/audio.py +1 -3
  103. wandb/sdk/data_types/base_types/media.py +13 -7
  104. wandb/sdk/data_types/base_types/wb_value.py +34 -11
  105. wandb/sdk/data_types/html.py +36 -9
  106. wandb/sdk/data_types/image.py +56 -37
  107. wandb/sdk/data_types/molecule.py +1 -5
  108. wandb/sdk/data_types/object_3d.py +2 -1
  109. wandb/sdk/data_types/saved_model.py +7 -9
  110. wandb/sdk/data_types/table.py +5 -0
  111. wandb/sdk/data_types/trace_tree.py +2 -0
  112. wandb/sdk/data_types/utils.py +1 -1
  113. wandb/sdk/data_types/video.py +15 -30
  114. wandb/sdk/interface/interface.py +2 -0
  115. wandb/{apis/public → sdk/internal}/_generated/__init__.py +0 -6
  116. wandb/{apis/public → sdk/internal}/_generated/server_features_query.py +3 -3
  117. wandb/sdk/internal/internal_api.py +138 -47
  118. wandb/sdk/internal/profiler.py +6 -5
  119. wandb/sdk/internal/run.py +13 -6
  120. wandb/sdk/internal/sender.py +2 -0
  121. wandb/sdk/internal/sender_config.py +8 -11
  122. wandb/sdk/internal/settings_static.py +24 -2
  123. wandb/sdk/lib/apikey.py +40 -20
  124. wandb/sdk/lib/asyncio_compat.py +1 -1
  125. wandb/sdk/lib/deprecate.py +13 -22
  126. wandb/sdk/lib/disabled.py +2 -1
  127. wandb/sdk/lib/printer.py +37 -8
  128. wandb/sdk/lib/printer_asyncio.py +46 -0
  129. wandb/sdk/lib/redirect.py +10 -5
  130. wandb/sdk/lib/run_moment.py +4 -6
  131. wandb/sdk/lib/wb_logging.py +161 -0
  132. wandb/sdk/service/server_sock.py +19 -14
  133. wandb/sdk/service/service.py +9 -7
  134. wandb/sdk/service/streams.py +5 -0
  135. wandb/sdk/verify/verify.py +6 -3
  136. wandb/sdk/wandb_config.py +44 -43
  137. wandb/sdk/wandb_init.py +323 -141
  138. wandb/sdk/wandb_login.py +13 -4
  139. wandb/sdk/wandb_metadata.py +107 -91
  140. wandb/sdk/wandb_run.py +529 -325
  141. wandb/sdk/wandb_settings.py +422 -202
  142. wandb/sdk/wandb_setup.py +52 -1
  143. wandb/util.py +29 -29
  144. {wandb-0.19.8.dist-info → wandb-0.19.10.dist-info}/METADATA +7 -7
  145. {wandb-0.19.8.dist-info → wandb-0.19.10.dist-info}/RECORD +151 -94
  146. wandb/_globals.py +0 -19
  147. wandb/apis/public/_generated/base.py +0 -128
  148. wandb/apis/public/_generated/typing_compat.py +0 -14
  149. /wandb/{apis/public → sdk/internal}/_generated/enums.py +0 -0
  150. /wandb/{apis/public → sdk/internal}/_generated/input_types.py +0 -0
  151. /wandb/{apis/public → sdk/internal}/_generated/operations.py +0 -0
  152. {wandb-0.19.8.dist-info → wandb-0.19.10.dist-info}/WHEEL +0 -0
  153. {wandb-0.19.8.dist-info → wandb-0.19.10.dist-info}/entry_points.txt +0 -0
  154. {wandb-0.19.8.dist-info → wandb-0.19.10.dist-info}/licenses/LICENSE +0 -0
@@ -0,0 +1,184 @@
1
+ # Generated by ariadne-codegen
2
+
3
+ from .create_filter_trigger import CreateFilterTrigger
4
+ from .create_generic_webhook_integration import (
5
+ CreateGenericWebhookIntegration,
6
+ CreateGenericWebhookIntegrationCreateGenericWebhookIntegration,
7
+ CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration,
8
+ CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration,
9
+ )
10
+ from .delete_trigger import DeleteTrigger
11
+ from .enums import (
12
+ AlertSeverity,
13
+ EventTriggeringConditionType,
14
+ TriggeredActionType,
15
+ TriggerScopeType,
16
+ )
17
+ from .fragments import (
18
+ ArtifactPortfolioScopeFields,
19
+ ArtifactSequenceScopeFields,
20
+ CreateFilterTriggerResult,
21
+ DeleteTriggerResult,
22
+ FilterEventFields,
23
+ GenericWebhookActionFields,
24
+ GenericWebhookActionFieldsIntegrationGenericWebhookIntegration,
25
+ GenericWebhookActionFieldsIntegrationIntegration,
26
+ GenericWebhookIntegrationConnectionFields,
27
+ GenericWebhookIntegrationConnectionFieldsEdges,
28
+ GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
29
+ GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration,
30
+ GenericWebhookIntegrationFields,
31
+ GithubIntegrationFields,
32
+ IntegrationConnectionFields,
33
+ IntegrationConnectionFieldsEdges,
34
+ IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
35
+ IntegrationConnectionFieldsEdgesNodeIntegration,
36
+ IntegrationConnectionFieldsEdgesNodeSlackIntegration,
37
+ NoOpActionFields,
38
+ NotificationActionFields,
39
+ NotificationActionFieldsIntegrationIntegration,
40
+ NotificationActionFieldsIntegrationSlackIntegration,
41
+ PageInfoFields,
42
+ ProjectConnectionFields,
43
+ ProjectConnectionFieldsEdges,
44
+ ProjectConnectionFieldsEdgesNode,
45
+ ProjectScopeFields,
46
+ QueueJobActionFields,
47
+ RunQueueFields,
48
+ SlackIntegrationConnectionFields,
49
+ SlackIntegrationConnectionFieldsEdges,
50
+ SlackIntegrationConnectionFieldsEdgesNodeIntegration,
51
+ SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration,
52
+ SlackIntegrationFields,
53
+ TriggerFields,
54
+ TriggerFieldsActionGenericWebhookTriggeredAction,
55
+ TriggerFieldsActionNoOpTriggeredAction,
56
+ TriggerFieldsActionNotificationTriggeredAction,
57
+ TriggerFieldsActionQueueJobTriggeredAction,
58
+ TriggerFieldsEventFilterEventTriggeringCondition,
59
+ TriggerFieldsScopeArtifactPortfolio,
60
+ TriggerFieldsScopeArtifactSequence,
61
+ TriggerFieldsScopeProject,
62
+ UpdateFilterTriggerResult,
63
+ UserFields,
64
+ )
65
+ from .generic_webhook_integrations_by_entity import (
66
+ GenericWebhookIntegrationsByEntity,
67
+ GenericWebhookIntegrationsByEntityEntity,
68
+ )
69
+ from .get_triggers import GetTriggers, GetTriggersSearchScope
70
+ from .get_triggers_by_entity import GetTriggersByEntity, GetTriggersByEntitySearchScope
71
+ from .input_types import (
72
+ CreateFilterTriggerInput,
73
+ CreateGenericWebhookIntegrationInput,
74
+ GenericWebhookActionInput,
75
+ NoOpTriggeredActionInput,
76
+ NotificationActionInput,
77
+ QueueJobActionInput,
78
+ TriggeredActionConfig,
79
+ UpdateFilterTriggerInput,
80
+ )
81
+ from .integrations_by_entity import IntegrationsByEntity, IntegrationsByEntityEntity
82
+ from .operations import (
83
+ CREATE_FILTER_TRIGGER_GQL,
84
+ CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL,
85
+ DELETE_TRIGGER_GQL,
86
+ GENERIC_WEBHOOK_INTEGRATIONS_BY_ENTITY_GQL,
87
+ GET_TRIGGERS_BY_ENTITY_GQL,
88
+ GET_TRIGGERS_GQL,
89
+ INTEGRATIONS_BY_ENTITY_GQL,
90
+ SLACK_INTEGRATIONS_BY_ENTITY_GQL,
91
+ UPDATE_FILTER_TRIGGER_GQL,
92
+ )
93
+ from .slack_integrations_by_entity import (
94
+ SlackIntegrationsByEntity,
95
+ SlackIntegrationsByEntityEntity,
96
+ )
97
+ from .update_filter_trigger import UpdateFilterTrigger
98
+
99
+ __all__ = [
100
+ "CREATE_FILTER_TRIGGER_GQL",
101
+ "CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL",
102
+ "DELETE_TRIGGER_GQL",
103
+ "GENERIC_WEBHOOK_INTEGRATIONS_BY_ENTITY_GQL",
104
+ "GET_TRIGGERS_BY_ENTITY_GQL",
105
+ "GET_TRIGGERS_GQL",
106
+ "INTEGRATIONS_BY_ENTITY_GQL",
107
+ "SLACK_INTEGRATIONS_BY_ENTITY_GQL",
108
+ "UPDATE_FILTER_TRIGGER_GQL",
109
+ "GetTriggers",
110
+ "GetTriggersSearchScope",
111
+ "GetTriggersByEntity",
112
+ "GetTriggersByEntitySearchScope",
113
+ "CreateFilterTrigger",
114
+ "UpdateFilterTrigger",
115
+ "DeleteTrigger",
116
+ "IntegrationsByEntity",
117
+ "IntegrationsByEntityEntity",
118
+ "SlackIntegrationsByEntity",
119
+ "SlackIntegrationsByEntityEntity",
120
+ "GenericWebhookIntegrationsByEntity",
121
+ "GenericWebhookIntegrationsByEntityEntity",
122
+ "CreateGenericWebhookIntegration",
123
+ "CreateGenericWebhookIntegrationCreateGenericWebhookIntegration",
124
+ "CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration",
125
+ "CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration",
126
+ "CreateFilterTriggerInput",
127
+ "CreateGenericWebhookIntegrationInput",
128
+ "GenericWebhookActionInput",
129
+ "NoOpTriggeredActionInput",
130
+ "NotificationActionInput",
131
+ "QueueJobActionInput",
132
+ "TriggeredActionConfig",
133
+ "UpdateFilterTriggerInput",
134
+ "ArtifactPortfolioScopeFields",
135
+ "ArtifactSequenceScopeFields",
136
+ "CreateFilterTriggerResult",
137
+ "DeleteTriggerResult",
138
+ "FilterEventFields",
139
+ "GenericWebhookActionFields",
140
+ "GenericWebhookActionFieldsIntegrationGenericWebhookIntegration",
141
+ "GenericWebhookActionFieldsIntegrationIntegration",
142
+ "GenericWebhookIntegrationConnectionFields",
143
+ "GenericWebhookIntegrationConnectionFieldsEdges",
144
+ "GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration",
145
+ "GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration",
146
+ "GenericWebhookIntegrationFields",
147
+ "GithubIntegrationFields",
148
+ "IntegrationConnectionFields",
149
+ "IntegrationConnectionFieldsEdges",
150
+ "IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration",
151
+ "IntegrationConnectionFieldsEdgesNodeIntegration",
152
+ "IntegrationConnectionFieldsEdgesNodeSlackIntegration",
153
+ "NoOpActionFields",
154
+ "NotificationActionFields",
155
+ "NotificationActionFieldsIntegrationIntegration",
156
+ "NotificationActionFieldsIntegrationSlackIntegration",
157
+ "PageInfoFields",
158
+ "ProjectConnectionFields",
159
+ "ProjectConnectionFieldsEdges",
160
+ "ProjectConnectionFieldsEdgesNode",
161
+ "ProjectScopeFields",
162
+ "QueueJobActionFields",
163
+ "RunQueueFields",
164
+ "SlackIntegrationConnectionFields",
165
+ "SlackIntegrationConnectionFieldsEdges",
166
+ "SlackIntegrationConnectionFieldsEdgesNodeIntegration",
167
+ "SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration",
168
+ "SlackIntegrationFields",
169
+ "TriggerFields",
170
+ "TriggerFieldsActionGenericWebhookTriggeredAction",
171
+ "TriggerFieldsActionNoOpTriggeredAction",
172
+ "TriggerFieldsActionNotificationTriggeredAction",
173
+ "TriggerFieldsActionQueueJobTriggeredAction",
174
+ "TriggerFieldsEventFilterEventTriggeringCondition",
175
+ "TriggerFieldsScopeArtifactPortfolio",
176
+ "TriggerFieldsScopeArtifactSequence",
177
+ "TriggerFieldsScopeProject",
178
+ "UpdateFilterTriggerResult",
179
+ "UserFields",
180
+ "AlertSeverity",
181
+ "EventTriggeringConditionType",
182
+ "TriggerScopeType",
183
+ "TriggeredActionType",
184
+ ]
@@ -0,0 +1,21 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/automations/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Optional
7
+
8
+ from pydantic import Field
9
+
10
+ from wandb._pydantic import GQLBase
11
+
12
+ from .fragments import CreateFilterTriggerResult
13
+
14
+
15
+ class CreateFilterTrigger(GQLBase):
16
+ create_filter_trigger: Optional[CreateFilterTriggerResult] = Field(
17
+ alias="createFilterTrigger"
18
+ )
19
+
20
+
21
+ CreateFilterTrigger.model_rebuild()
@@ -0,0 +1,43 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/automations/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Literal, Optional, Union
7
+
8
+ from pydantic import Field
9
+
10
+ from wandb._pydantic import GQLBase, Typename
11
+
12
+ from .fragments import GenericWebhookIntegrationFields
13
+
14
+
15
+ class CreateGenericWebhookIntegration(GQLBase):
16
+ create_generic_webhook_integration: Optional[
17
+ CreateGenericWebhookIntegrationCreateGenericWebhookIntegration
18
+ ] = Field(alias="createGenericWebhookIntegration")
19
+
20
+
21
+ class CreateGenericWebhookIntegrationCreateGenericWebhookIntegration(GQLBase):
22
+ integration: Union[
23
+ CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration,
24
+ CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration,
25
+ ] = Field(discriminator="typename__")
26
+
27
+
28
+ class CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration(
29
+ GQLBase
30
+ ):
31
+ typename__: Typename[
32
+ Literal["GitHubOAuthIntegration", "Integration", "SlackIntegration"]
33
+ ]
34
+
35
+
36
+ class CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration(
37
+ GenericWebhookIntegrationFields
38
+ ):
39
+ typename__: Typename[Literal["GenericWebhookIntegration"]]
40
+
41
+
42
+ CreateGenericWebhookIntegration.model_rebuild()
43
+ CreateGenericWebhookIntegrationCreateGenericWebhookIntegration.model_rebuild()
@@ -0,0 +1,19 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/automations/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Optional
7
+
8
+ from pydantic import Field
9
+
10
+ from wandb._pydantic import GQLBase
11
+
12
+ from .fragments import DeleteTriggerResult
13
+
14
+
15
+ class DeleteTrigger(GQLBase):
16
+ delete_trigger: Optional[DeleteTriggerResult] = Field(alias="deleteTrigger")
17
+
18
+
19
+ DeleteTrigger.model_rebuild()
@@ -0,0 +1,33 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: core/api/graphql/schemas/schema-latest.graphql
3
+
4
+ from __future__ import annotations
5
+
6
+ from enum import Enum
7
+
8
+
9
+ class EventTriggeringConditionType(str, Enum):
10
+ CREATE_ARTIFACT = "CREATE_ARTIFACT"
11
+ UPDATE_ARTIFACT_ALIAS = "UPDATE_ARTIFACT_ALIAS"
12
+ ADD_ARTIFACT_ALIAS = "ADD_ARTIFACT_ALIAS"
13
+ LINK_MODEL = "LINK_MODEL"
14
+ RUN_METRIC = "RUN_METRIC"
15
+ RUN_METRIC_CHANGE = "RUN_METRIC_CHANGE"
16
+
17
+
18
+ class TriggerScopeType(str, Enum):
19
+ PROJECT = "PROJECT"
20
+ ARTIFACT_COLLECTION = "ARTIFACT_COLLECTION"
21
+
22
+
23
+ class AlertSeverity(str, Enum):
24
+ INFO = "INFO"
25
+ WARN = "WARN"
26
+ ERROR = "ERROR"
27
+
28
+
29
+ class TriggeredActionType(str, Enum):
30
+ QUEUE_JOB = "QUEUE_JOB"
31
+ NOTIFICATION = "NOTIFICATION"
32
+ GENERIC_WEBHOOK = "GENERIC_WEBHOOK"
33
+ NO_OP = "NO_OP"
@@ -0,0 +1,343 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/automations/
3
+
4
+ from __future__ import annotations
5
+
6
+ from datetime import datetime
7
+ from typing import List, Literal, Optional, Union
8
+
9
+ from pydantic import Field
10
+ from typing_extensions import Annotated
11
+
12
+ from wandb._pydantic import GQLBase, GQLId, Typename
13
+
14
+ from .enums import AlertSeverity, EventTriggeringConditionType
15
+
16
+
17
+ class ArtifactPortfolioScopeFields(GQLBase):
18
+ typename__: Typename[Literal["ArtifactPortfolio"]] = "ArtifactPortfolio"
19
+ id: GQLId
20
+ name: str
21
+
22
+
23
+ class ArtifactSequenceScopeFields(GQLBase):
24
+ typename__: Typename[Literal["ArtifactSequence"]] = "ArtifactSequence"
25
+ id: GQLId
26
+ name: str
27
+
28
+
29
+ class CreateFilterTriggerResult(GQLBase):
30
+ typename__: Typename[Literal["CreateFilterTriggerPayload"]] = (
31
+ "CreateFilterTriggerPayload"
32
+ )
33
+ trigger: Optional[TriggerFields]
34
+
35
+
36
+ class DeleteTriggerResult(GQLBase):
37
+ typename__: Typename[Literal["DeleteTriggerPayload"]] = "DeleteTriggerPayload"
38
+ success: bool
39
+
40
+
41
+ class FilterEventFields(GQLBase):
42
+ typename__: Typename[Literal["FilterEventTriggeringCondition"]] = (
43
+ "FilterEventTriggeringCondition"
44
+ )
45
+ event_type: EventTriggeringConditionType = Field(alias="eventType")
46
+ filter: str
47
+
48
+
49
+ class GenericWebhookActionFields(GQLBase):
50
+ typename__: Typename[Literal["GenericWebhookTriggeredAction"]] = (
51
+ "GenericWebhookTriggeredAction"
52
+ )
53
+ integration: Union[
54
+ GenericWebhookActionFieldsIntegrationIntegration,
55
+ GenericWebhookActionFieldsIntegrationGenericWebhookIntegration,
56
+ ] = Field(discriminator="typename__")
57
+ request_payload: Optional[str] = Field(alias="requestPayload")
58
+
59
+
60
+ class GenericWebhookActionFieldsIntegrationIntegration(GQLBase):
61
+ typename__: Typename[
62
+ Literal["GitHubOAuthIntegration", "Integration", "SlackIntegration"]
63
+ ]
64
+
65
+
66
+ class GenericWebhookIntegrationConnectionFields(GQLBase):
67
+ typename__: Typename[Literal["IntegrationConnection"]] = "IntegrationConnection"
68
+ page_info: PageInfoFields = Field(alias="pageInfo")
69
+ edges: List[GenericWebhookIntegrationConnectionFieldsEdges]
70
+
71
+
72
+ class GenericWebhookIntegrationConnectionFieldsEdges(GQLBase):
73
+ cursor: str
74
+ node: Optional[
75
+ Annotated[
76
+ Union[
77
+ GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration,
78
+ GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
79
+ ],
80
+ Field(discriminator="typename__"),
81
+ ]
82
+ ]
83
+
84
+
85
+ class GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration(GQLBase):
86
+ typename__: Typename[
87
+ Literal["GitHubOAuthIntegration", "Integration", "SlackIntegration"]
88
+ ]
89
+
90
+
91
+ class GenericWebhookIntegrationFields(GQLBase):
92
+ typename__: Typename[Literal["GenericWebhookIntegration"]] = (
93
+ "GenericWebhookIntegration"
94
+ )
95
+ id: GQLId
96
+ name: str
97
+ url_endpoint: str = Field(alias="urlEndpoint")
98
+ created_at: datetime = Field(alias="createdAt")
99
+
100
+
101
+ class GithubIntegrationFields(GQLBase):
102
+ typename__: Typename[Literal["GitHubOAuthIntegration"]] = "GitHubOAuthIntegration"
103
+ id: GQLId
104
+
105
+
106
+ class IntegrationConnectionFields(GQLBase):
107
+ typename__: Typename[Literal["IntegrationConnection"]] = "IntegrationConnection"
108
+ page_info: PageInfoFields = Field(alias="pageInfo")
109
+ edges: List[IntegrationConnectionFieldsEdges]
110
+
111
+
112
+ class IntegrationConnectionFieldsEdges(GQLBase):
113
+ cursor: str
114
+ node: Optional[
115
+ Annotated[
116
+ Union[
117
+ IntegrationConnectionFieldsEdgesNodeIntegration,
118
+ IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
119
+ IntegrationConnectionFieldsEdgesNodeSlackIntegration,
120
+ ],
121
+ Field(discriminator="typename__"),
122
+ ]
123
+ ]
124
+
125
+
126
+ class IntegrationConnectionFieldsEdgesNodeIntegration(GQLBase):
127
+ typename__: Typename[Literal["GitHubOAuthIntegration", "Integration"]]
128
+
129
+
130
+ class NoOpActionFields(GQLBase):
131
+ typename__: Typename[Literal["NoOpTriggeredAction"]] = "NoOpTriggeredAction"
132
+ no_op: Optional[bool] = Field(alias="noOp")
133
+
134
+
135
+ class NotificationActionFields(GQLBase):
136
+ typename__: Typename[Literal["NotificationTriggeredAction"]] = (
137
+ "NotificationTriggeredAction"
138
+ )
139
+ integration: Union[
140
+ NotificationActionFieldsIntegrationIntegration,
141
+ NotificationActionFieldsIntegrationSlackIntegration,
142
+ ] = Field(discriminator="typename__")
143
+ title: Optional[str]
144
+ message: Optional[str]
145
+ severity: Optional[AlertSeverity]
146
+
147
+
148
+ class NotificationActionFieldsIntegrationIntegration(GQLBase):
149
+ typename__: Typename[
150
+ Literal["GenericWebhookIntegration", "GitHubOAuthIntegration", "Integration"]
151
+ ]
152
+
153
+
154
+ class PageInfoFields(GQLBase):
155
+ end_cursor: Optional[str] = Field(alias="endCursor")
156
+ has_next_page: bool = Field(alias="hasNextPage")
157
+
158
+
159
+ class ProjectConnectionFields(GQLBase):
160
+ typename__: Typename[Literal["ProjectConnection"]] = "ProjectConnection"
161
+ page_info: PageInfoFields = Field(alias="pageInfo")
162
+ edges: List[ProjectConnectionFieldsEdges]
163
+
164
+
165
+ class ProjectConnectionFieldsEdges(GQLBase):
166
+ cursor: str
167
+ node: Optional[ProjectConnectionFieldsEdgesNode]
168
+
169
+
170
+ class ProjectConnectionFieldsEdgesNode(GQLBase):
171
+ triggers: List[TriggerFields]
172
+
173
+
174
+ class ProjectScopeFields(GQLBase):
175
+ typename__: Typename[Literal["Project"]] = "Project"
176
+ id: GQLId
177
+ name: str
178
+
179
+
180
+ class QueueJobActionFields(GQLBase):
181
+ typename__: Typename[Literal["QueueJobTriggeredAction"]] = "QueueJobTriggeredAction"
182
+ queue: Optional[RunQueueFields]
183
+ template: str
184
+
185
+
186
+ class RunQueueFields(GQLBase):
187
+ typename__: Typename[Literal["RunQueue"]] = "RunQueue"
188
+ id: GQLId
189
+ name: str
190
+
191
+
192
+ class SlackIntegrationConnectionFields(GQLBase):
193
+ typename__: Typename[Literal["IntegrationConnection"]] = "IntegrationConnection"
194
+ page_info: PageInfoFields = Field(alias="pageInfo")
195
+ edges: List[SlackIntegrationConnectionFieldsEdges]
196
+
197
+
198
+ class SlackIntegrationConnectionFieldsEdges(GQLBase):
199
+ cursor: str
200
+ node: Optional[
201
+ Annotated[
202
+ Union[
203
+ SlackIntegrationConnectionFieldsEdgesNodeIntegration,
204
+ SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration,
205
+ ],
206
+ Field(discriminator="typename__"),
207
+ ]
208
+ ]
209
+
210
+
211
+ class SlackIntegrationConnectionFieldsEdgesNodeIntegration(GQLBase):
212
+ typename__: Typename[
213
+ Literal["GenericWebhookIntegration", "GitHubOAuthIntegration", "Integration"]
214
+ ]
215
+
216
+
217
+ class SlackIntegrationFields(GQLBase):
218
+ typename__: Typename[Literal["SlackIntegration"]] = "SlackIntegration"
219
+ id: GQLId
220
+ team_name: str = Field(alias="teamName")
221
+ channel_name: str = Field(alias="channelName")
222
+
223
+
224
+ class TriggerFields(GQLBase):
225
+ typename__: Typename[Literal["Trigger"]] = "Trigger"
226
+ id: GQLId
227
+ created_at: datetime = Field(alias="createdAt")
228
+ created_by: UserFields = Field(alias="createdBy")
229
+ updated_at: Optional[datetime] = Field(alias="updatedAt")
230
+ name: str
231
+ description: Optional[str]
232
+ enabled: bool
233
+ scope: Union[
234
+ TriggerFieldsScopeProject,
235
+ TriggerFieldsScopeArtifactSequence,
236
+ TriggerFieldsScopeArtifactPortfolio,
237
+ ] = Field(discriminator="typename__")
238
+ event: TriggerFieldsEventFilterEventTriggeringCondition
239
+ action: Union[
240
+ TriggerFieldsActionQueueJobTriggeredAction,
241
+ TriggerFieldsActionNotificationTriggeredAction,
242
+ TriggerFieldsActionGenericWebhookTriggeredAction,
243
+ TriggerFieldsActionNoOpTriggeredAction,
244
+ ] = Field(discriminator="typename__")
245
+
246
+
247
+ class UpdateFilterTriggerResult(GQLBase):
248
+ typename__: Typename[Literal["UpdateFilterTriggerPayload"]] = (
249
+ "UpdateFilterTriggerPayload"
250
+ )
251
+ trigger: Optional[TriggerFields]
252
+
253
+
254
+ class UserFields(GQLBase):
255
+ typename__: Typename[Literal["User"]] = "User"
256
+ id: GQLId
257
+ username: Optional[str]
258
+
259
+
260
+ class TriggerFieldsScopeArtifactPortfolio(ArtifactPortfolioScopeFields):
261
+ typename__: Typename[Literal["ArtifactPortfolio"]]
262
+
263
+
264
+ class TriggerFieldsScopeArtifactSequence(ArtifactSequenceScopeFields):
265
+ typename__: Typename[Literal["ArtifactSequence"]]
266
+
267
+
268
+ class TriggerFieldsEventFilterEventTriggeringCondition(FilterEventFields):
269
+ typename__: Typename[Literal["FilterEventTriggeringCondition"]]
270
+
271
+
272
+ class TriggerFieldsActionGenericWebhookTriggeredAction(GenericWebhookActionFields):
273
+ typename__: Typename[Literal["GenericWebhookTriggeredAction"]]
274
+
275
+
276
+ class GenericWebhookActionFieldsIntegrationGenericWebhookIntegration(
277
+ GenericWebhookIntegrationFields
278
+ ):
279
+ typename__: Typename[Literal["GenericWebhookIntegration"]]
280
+
281
+
282
+ class GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
283
+ GenericWebhookIntegrationFields
284
+ ):
285
+ typename__: Typename[Literal["GenericWebhookIntegration"]]
286
+
287
+
288
+ class IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
289
+ GenericWebhookIntegrationFields
290
+ ):
291
+ typename__: Typename[Literal["GenericWebhookIntegration"]]
292
+
293
+
294
+ class TriggerFieldsActionNoOpTriggeredAction(NoOpActionFields):
295
+ typename__: Typename[Literal["NoOpTriggeredAction"]]
296
+
297
+
298
+ class TriggerFieldsActionNotificationTriggeredAction(NotificationActionFields):
299
+ typename__: Typename[Literal["NotificationTriggeredAction"]]
300
+
301
+
302
+ class TriggerFieldsScopeProject(ProjectScopeFields):
303
+ typename__: Typename[Literal["Project"]]
304
+
305
+
306
+ class TriggerFieldsActionQueueJobTriggeredAction(QueueJobActionFields):
307
+ typename__: Typename[Literal["QueueJobTriggeredAction"]]
308
+
309
+
310
+ class IntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
311
+ typename__: Typename[Literal["SlackIntegration"]]
312
+
313
+
314
+ class NotificationActionFieldsIntegrationSlackIntegration(SlackIntegrationFields):
315
+ typename__: Typename[Literal["SlackIntegration"]]
316
+
317
+
318
+ class SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
319
+ typename__: Typename[Literal["SlackIntegration"]]
320
+
321
+
322
+ ArtifactPortfolioScopeFields.model_rebuild()
323
+ ArtifactSequenceScopeFields.model_rebuild()
324
+ CreateFilterTriggerResult.model_rebuild()
325
+ DeleteTriggerResult.model_rebuild()
326
+ FilterEventFields.model_rebuild()
327
+ GenericWebhookActionFields.model_rebuild()
328
+ GenericWebhookIntegrationConnectionFields.model_rebuild()
329
+ GenericWebhookIntegrationFields.model_rebuild()
330
+ GithubIntegrationFields.model_rebuild()
331
+ IntegrationConnectionFields.model_rebuild()
332
+ NoOpActionFields.model_rebuild()
333
+ NotificationActionFields.model_rebuild()
334
+ PageInfoFields.model_rebuild()
335
+ ProjectConnectionFields.model_rebuild()
336
+ ProjectScopeFields.model_rebuild()
337
+ QueueJobActionFields.model_rebuild()
338
+ RunQueueFields.model_rebuild()
339
+ SlackIntegrationConnectionFields.model_rebuild()
340
+ SlackIntegrationFields.model_rebuild()
341
+ TriggerFields.model_rebuild()
342
+ UpdateFilterTriggerResult.model_rebuild()
343
+ UserFields.model_rebuild()
@@ -0,0 +1,22 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/automations/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Optional
7
+
8
+ from wandb._pydantic import GQLBase
9
+
10
+ from .fragments import GenericWebhookIntegrationConnectionFields
11
+
12
+
13
+ class GenericWebhookIntegrationsByEntity(GQLBase):
14
+ entity: Optional[GenericWebhookIntegrationsByEntityEntity]
15
+
16
+
17
+ class GenericWebhookIntegrationsByEntityEntity(GQLBase):
18
+ integrations: Optional[GenericWebhookIntegrationConnectionFields]
19
+
20
+
21
+ GenericWebhookIntegrationsByEntity.model_rebuild()
22
+ GenericWebhookIntegrationsByEntityEntity.model_rebuild()
@@ -0,0 +1,24 @@
1
+ # Generated by ariadne-codegen
2
+ # Source: tools/graphql_codegen/automations/
3
+
4
+ from __future__ import annotations
5
+
6
+ from typing import Optional
7
+
8
+ from pydantic import Field
9
+
10
+ from wandb._pydantic import GQLBase
11
+
12
+ from .fragments import ProjectConnectionFields
13
+
14
+
15
+ class GetTriggers(GQLBase):
16
+ search_scope: Optional[GetTriggersSearchScope] = Field(alias="searchScope")
17
+
18
+
19
+ class GetTriggersSearchScope(GQLBase):
20
+ projects: Optional[ProjectConnectionFields]
21
+
22
+
23
+ GetTriggers.model_rebuild()
24
+ GetTriggersSearchScope.model_rebuild()