Add LLM-Based Recommendations for Broken Pipeline Runs

package.json

@@ -63,83 +63,104 @@
     "watch-tests": "tsc -p . -w --outDir out"
   },
   "contributes": {
-    "configuration": {
-      "title": "ZenML",
-      "properties": {
-        "zenml-python.args": {
-          "default": [],
-          "description": "Arguments passed in. Each argument is a separate item in the array.",
-          "items": {
-            "type": "string"
+    "configuration": [
+      {
+        "title": "ZenML",
+        "properties": {
+          "zenml-python.args": {
+            "default": [],
+            "description": "Arguments passed in. Each argument is a separate item in the array.",
+            "items": {
+              "type": "string"
+            },
+            "scope": "resource",
+            "type": "array"
           },
-          "scope": "resource",
-          "type": "array"
-        },
-        "zenml-python.path": {
-          "default": [],
-          "scope": "resource",
-          "items": {
+          "zenml-python.path": {
+            "default": [],
+            "scope": "resource",
+            "items": {
+              "type": "string"
+            },
+            "type": "array"
+          },
+          "zenml-python.importStrategy": {
+            "default": "useBundled",
+            "enum": [
+              "useBundled",
+              "fromEnvironment"
+            ],
+            "enumDescriptions": [
+              "Always use the bundled version of `<pytool-module>`.",
+              "Use `<pytool-module>` from environment, fallback to bundled version only if `<pytool-module>` not available in the environment."
+            ],
+            "scope": "window",
             "type": "string"
           },
-          "type": "array"
-        },
-        "zenml-python.importStrategy": {
-          "default": "useBundled",
-          "enum": [
-            "useBundled",
-            "fromEnvironment"
-          ],
-          "enumDescriptions": [
-            "Always use the bundled version of `<pytool-module>`.",
-            "Use `<pytool-module>` from environment, fallback to bundled version only if `<pytool-module>` not available in the environment."
-          ],
-          "scope": "window",
-          "type": "string"
-        },
-        "zenml-python.interpreter": {
-          "default": [],
-          "description": "When set to a path to python executable, extension will use that to launch the server and any subprocess.",
-          "scope": "resource",
-          "items": {
+          "zenml-python.interpreter": {
+            "default": [],
+            "description": "When set to a path to python executable, extension will use that to launch the server and any subprocess.",
+            "scope": "resource",
+            "items": {
+              "type": "string"
+            },
+            "type": "array"
+          },
+          "zenml-python.showNotifications": {
+            "default": "off",
+            "description": "Controls when notifications are shown by this extension.",
+            "enum": [
+              "off",
+              "onError",
+              "onWarning",
+              "always"
+            ],
+            "enumDescriptions": [
+              "All notifications are turned off, any errors or warning are still available in the logs.",
+              "Notifications are shown only in the case of an error.",
+              "Notifications are shown for errors and warnings.",
+              "Notifications are show for anything that the server chooses to show."
+            ],
+            "scope": "machine",
             "type": "string"
           },
-          "type": "array"
-        },
-        "zenml-python.showNotifications": {
-          "default": "off",
-          "description": "Controls when notifications are shown by this extension.",
-          "enum": [
-            "off",
-            "onError",
-            "onWarning",
-            "always"
-          ],
-          "enumDescriptions": [
-            "All notifications are turned off, any errors or warning are still available in the logs.",
-            "Notifications are shown only in the case of an error.",
-            "Notifications are shown for errors and warnings.",
-            "Notifications are show for anything that the server chooses to show."
-          ],
-          "scope": "machine",
-          "type": "string"
-        },
-        "zenml.serverUrl": {
-          "type": "string",
-          "default": "",
-          "description": "ZenML Server URL"
-        },
-        "zenml.accessToken": {
-          "type": "string",
-          "default": "",
-          "description": "Access token for the ZenML server"
-        },
-        "zenml.activeStackId": {
-          "type": "string",
-          "default": "",
-          "description": "Active stack id for the ZenML server"
+          "zenml.serverUrl": {
+            "type": "string",
+            "default": "",
+            "description": "ZenML Server URL"
+          },
+          "zenml.accessToken": {
+            "type": "string",
+            "default": "",
+            "description": "Access token for the ZenML server"
+          },
+          "zenml.activeStackId": {
+            "type": "string",
+            "default": "",
+            "description": "Active stack id for the ZenML server"
+          }
+        }
+      },
+      {
+        "title": "ZenML AI",
+        "properties": {
+          "zenml.llm-model": {
+            "enum": [
+              "anthropic.claude-3-5-sonnet-20240620",
+              "anthropic.claude-3-opus-20240229",
+              "anthropic.claude-3-haiku-20240307",
+              "gemini.gemini-1.5-pro",
+              "gemini.gemini-1.5-flash",
+              "gemini.gemini-1.0-pro",
+              "openai.gpt-4o",
+              "openai.gpt-4o-mini",
+              "openai.gpt-4-turbo",
+              "openai.gpt-3.5-turbo"
+            ]
+          }
         }
       }
-    },
+    ],
     "commands": [
       {
         "command": "zenml.promptForInterpreter",
@@ -297,6 +318,29 @@
         "title": "Restart LSP Server",
         "icon": "$(debug-restart)",
         "category": "ZenML Environment"
+      },
+      {
+        "command": "zenml.registerLLMAPIKey",
+        "title": "Register LLM API Key",
+        "icon": "$(add)",
+        "category": "ZenML Secrets"
+      },
+      {
+        "command": "zenml.sendOpenAIMessage",
+        "title": "Send OpenAI API Message",
+        "category": "ZenML GenAI"
+      },
+      {
+        "command": "zenml.displayNextCodeRecommendation",
+        "title": "Next Recommendation",
+        "category": "ZenML GenAI",
+        "when": "resourceFilename in zenml.aiCodeRecommendations && resourceScheme == zenml-stepfixer"
+      },
+      {
+        "command": "zenml.acceptCodeRecommendation",
+        "title": "Accept Change",
+        "category": "ZenML GenAI",
+        "when": "resourceFilename in zenml.aiCodeRecommendations && resourceScheme == zenml-stepfixer"
       }
     ],
     "viewsContainers": {
@@ -469,6 +513,18 @@
           "command": "zenml.setPythonInterpreter",
           "group": "inline"
         }
+      ],
+      "editor/title": [
+        {
+          "command": "zenml.displayNextCodeRecommendation",
+          "group": "navigation",
+          "when": "resourceFilename in zenml.aiCodeRecommendations && resourceScheme == zenml-stepfixer"
+        },
+        {
+          "command": "zenml.acceptCodeRecommendation",
+          "group": "navigation",
+          "when": "resourceFilename in zenml.aiCodeRecommendations && resourceScheme == zenml-stepfixer"
+        }
       ]
     }
   },
@@ -493,7 +549,7 @@
     "sinon": "^17.0.1",
     "ts-loader": "^9.5.1",
     "ts-node": "^10.9.2",
-    "typescript": "^5.3.3",
+    "typescript": "^5.6.2",
     "webpack": "^5.94.0",
     "webpack-cli": "^5.1.4"
   },
@@ -504,8 +560,10 @@
     "dagre": "^0.8.5",
     "fs-extra": "^11.2.0",
     "hbs": "^4.2.0",
+    "simple-git": "^3.26.0",
     "svg-pan-zoom": "github:bumbu/svg-pan-zoom",
     "svgdom": "^0.1.19",
+    "token.js": "^0.4.4",
     "vscode-languageclient": "^9.0.1"
   }
 }

resources/dag-view/dag.css

@@ -155,3 +155,36 @@ g.svg-pan-zoom-control {
   width: 100%;
   height: 100%;
 }
+
+#context-menu {
+  position: absolute;
+  background: white;
+  box-shadow: 0 4px 5px 3px rgba(0, 0, 0, 0.2);
+  border-radius: 5px;
+}
+
+#context-menu ul {
+  list-style: none;
+  padding: 0px;
+}
+
+#context-menu li {
+  font-weight: 500;
+  font-size: 14px;
+  padding: 10px 40px 10px 20px;
+  cursor: pointer;
+}
+
+#context-menu li:hover {
+  color: #333;
+  background: #e0e0e0;
+}
+
+#loading-ai {
+  display: none;
+}
+
+.context-menu-subitem-indicator {
+  font-size: 0.7rem;
+  vertical-align: text-top;
+}