chore(sdk): implement hack to make nexus work

.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.9.17
+current_version = 0.9.18
 commit = False
 tag = False
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\-(?P<release>[a-z]+))?

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="yea-wandb",
-    version="0.9.17",
+    version="0.9.18",
     description="Test harness wandb plugin",
     packages=["yea_wandb"],
     install_requires=[

src/yea_wandb/__init__.py

@@ -1,4 +1,4 @@
 from .setup import setup
 
 __all__ = ["setup"]
-__version__ = "0.9.17"
+__version__ = "0.9.18"

src/yea_wandb/artifact_emu.py

@@ -91,7 +91,7 @@ def create_files(self, variables):
                         "edges": [
                             {
                                 "node": {
-                                    "id": idx,
+                                    "id": str(idx),
                                     "name": af["name"],
                                     "displayName": af["name"],
                                     "uploadUrl": f"{base_url}/storage?file={af['name']}&id={af['artifactID']}",

src/yea_wandb/mock_server.py

@@ -616,9 +616,34 @@
         body = request.get_json()
         app.logger.info("graphql post body: %s", body)
 
+        # fixup body query to be more compatible with other graphql implementations
+        # lets start by changing any mutation or query that has a space after the
+        # graphql request name
+        if re.match(r"^\s*(mutation|query)\s(\w+)\s\(", body["query"]):
+            body["query"] = body["query"].replace(" (", "(")
+
         if body["variables"].get("run"):
             ctx["current_run"] = body["variables"]["run"]
 
+        if "mutation CreateRunFiles" in body["query"]:
+            requested_file = body["variables"]["files"][0]
+            upload_url = base_url + "/storage?file=%s" % requested_file
+            data = {
+                "data": {
+                    "createRunFiles": {
+                        "runID": "UnVuOnYxOmtoMXFsdmIwOnVuY2F0ZWdvcml6ZWQ6amVmZnI=",
+                        "uploadHeaders": [],
+                        "files": [
+                            {
+                                "name": requested_file,
+                                "uploadUrl": upload_url,
+                            }
+                        ],
+                    }
+                }
+            }
+            r = json.dumps(data)
+            return r
         if body["variables"].get("files"):
             requested_file = body["variables"]["files"][0]
             ctx["requested_file"] = requested_file
@@ -845,7 +870,7 @@
                 "name": "foo",
                 "uploadUrl": "",
                 "storagePath": "x/y/z",
-                "uploadheaders": [],
+                "uploadHeaders": [],
                 "artifact": {"id": "1"},
             }
             if "storagePath" not in body["query"]:
@@ -1146,7 +1171,7 @@
                             "displayName": file_spec["name"],
                             "digest": "null",
                             "uploadUrl": url,
-                            "uploadHeaders": "",
+                            "uploadHeaders": [],
                         }
                     }
                 )
@@ -1280,18 +1305,18 @@
             }
         if "mutation CreateArtifactManifest(" in body["query"]:
             manifest = {
-                "id": 1,
+                "id": "1",
                 "type": "INCREMENTAL"
                 if "incremental" in body.get("variables", {}).get("name", "")
                 else "FULL",
                 "file": {
-                    "id": 1,
+                    "id": "1",
                     "directUrl": base_url
                     + "/storage?file=wandb_manifest.json&name={}".format(
                         body.get("variables", {}).get("name", "")
                     ),
                     "uploadUrl": base_url + "/storage?file=wandb_manifest.json",
-                    "uploadHeaders": "",
+                    "uploadHeaders": [],
                 },
             }
             run_name = body.get("variables", {}).get("runName", "unknown")
@@ -1318,7 +1343,7 @@
                         body.get("variables", {}).get("name", "")
                     ),
                     "uploadUrl": base_url + "/storage?file=wandb_manifest.json",
-                    "uploadHeaders": "",
+                    "uploadHeaders": [],
                 },
             }
             return {
@@ -1357,7 +1382,7 @@
                                 "id": idx,
                                 "name": file["name"],
                                 "uploadUrl": "",
-                                "uploadheaders": [],
+                                "uploadHeaders": [],
                                 "artifact": {"id": file["artifactID"]},
                             }
                             for idx, file in enumerate(
@@ -1371,7 +1396,7 @@
             return {
                 "data": {
                     "commitArtifact": {
-                        "artifact": {"id": 1, "digest": "0000===================="}
+                        "artifact": {"id": "1", "digest": "0000===================="}
                     }
                 }
             }