chore(artifacts): Add weave support to enable artifact version tags i…

…n UI (server) (#1986) * chore(weave): wip - boilerplate for opArtifactVersionTags * chore(weave): wip - starter boilerplate for artifact version tags * chore(weave): add tags field to Artifact gql schema * chore(weave): autoformatting * chore(weave): fix returnType for opArtifactVersionTags * chore(weave): handle 'artifactVersion-tags' in gql.ts * chore(weave): fix required fragment for artifact vertsion tags * chore(weave): fix python weave op for version tags * chore(weave): fix python weave op for version tags * chore(weave): fix python weave op for version tags * chore(weave): fix python weave op for version tags * chore(weave): fix python weave op for version tags * chore(weave): fix python weave op for version tags * chore(weave): fix python weave op for version tags * chore(weave): fix python weave op for version tags * chore(weave): adapt PanelArtifactVersionAliases -> PanelArtifactVersionTags * chore(weave): fix return type * chore(weave): adapt PanelArtifactVersionAliases -> PanelArtifactVersionTags * chore(weave): fixes for PanelArtifactVersionTags * chore(weave): fixes for PanelArtifactVersionTags * chore(weave): fixes for PanelArtifactVersionTags * chore(weave): fixes for PanelArtifactVersionTags * chore(weave): fixes for PanelArtifactVersionTags * chore(weave): fixes for PanelArtifactVersionTags * chore(weave): fixes for PanelArtifactVersionTags * chore(weave): drop unused Tag type for artifacts * chore(ui): cleanup and fixes * chore(ui): revert returnType for opArtifactRawTags * chore(weave): fixes for pr comments * chore(ui): require all Tag fields -- we need them anyway * chore(ui): fix/simplify/consolidate weave type definitions * chore(ui): rename artifactVersionTags op -> artifactVersionRawTags op for consistency * chore(weave): split off changes in weave-js/
wandb · Aug 5, 2024 · 83efe6f · 83efe6f
1 parent 8b0d509
commit 83efe6f
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/wb_schema.gql b/wb_schema.gql
@@ -566,6 +566,7 @@ type Artifact {
   currentManifest: ArtifactManifest
   historyStep: Int64
   ttlDurationSeconds: Int64!
+  tags: [Tag!]!
 }
 
 type ArtifactManifestConnection {

diff --git a/weave/legacy/ops_domain/artifact_version_ops.py b/weave/legacy/ops_domain/artifact_version_ops.py
@@ -258,6 +258,33 @@ def metadata(
 )
 
 
+@op(
+    name="artifactVersion-rawTags",
+    output_type=types.List(
+        types.TypedDict(
+            {
+                "id": types.String(),
+                "name": types.String(),
+                "tagCategoryName": types.String(),
+                "attributes": types.String(),
+            }
+        )
+    ),
+    plugins=wb_gql_op_plugin(
+        lambda inputs, inner: """
+        tags {
+            id
+            name
+            tagCategoryName
+            attributes
+        }
+        """
+    ),
+)
+def artifact_version_raw_tags(artifact: wdt.ArtifactVersion):
+    return artifact["tags"]
+
+
 @op(
     name="artifactVersion-createdBy",
     plugins=wb_gql_op_plugin(

diff --git a/weave/legacy/panels/panel_legacy.py b/weave/legacy/panels/panel_legacy.py
@@ -31,6 +31,7 @@ class LPanel:
     LPanel("run-overview", "PanelRunOverview"),
     LPanel("none", "PanelNone"),
     LPanel("artifactVersionAliases", "PanelArtifactVersionAliases"),
+    LPanel("artifactVersionTags", "PanelArtifactVersionTags"),
     LPanel("netron", "PanelNetron"),
     LPanel("object", "PanelObject"),
     LPanel("audio-file", "PanelAudioFile"),