A Golang replacement for the Kubeflow Jupyter Web APIs / Un remplacement Golang pour les API de Web de Jupyter, partie de Kubeflow
Jupyter Application Programming Interfaces
A Golang replacement for the [Kubeflow][kubeflow] Jupyter Web APIs.
How to Contribute
See CONTRIBUTING.md
Development Environment
Note that the frontend will report errors when calling/api/namespaces when run locally. This
issue does not arise in production, as the /api/namespaces endpoint is unused.
To initialize the .env file for the development environment, use task env. You will need to fill out your kubeflow cloud account and kubeflow namespace information manually. The thunder-tests folder contains configuration for testing requests against the backend. Use the vscode THUNDER CLIENT extension to load the tests.
Run API Server
The API server will connect to the Kubeflow cluster from your current kubectl context. See Connecting a Kubeflow cluster below for options.
- Install [Go][go]
- Change directory to project root:
cd jupyter-apis - Run
go run . -spawner-config samples/spawneruiconfig.yaml
task go:dev -w -- -spawner-config samples/spawneruiconfig.yamlwill live-reload the Go server upon changes.
You can use the vscode debugger to run the backend, just copy the below contents to a file at path .vscode/launch.json.
{ "version": "0.2.0", "configurations": [ { "name": "Debug jupyter-api backend", "type": "go", "request": "launch", "mode": "debug", "program": ".", "args": [ "-spawner-config", "samples/spawneruiconfig.yaml", ], "envFile": "${workspaceFolder}/.env" } ] }
Run Front-End
The front-end is configured to proxy requests to the local API server. It requires an environment variable (KFUSERID) to specify the current user – this is passed to the API server as an HTTP header.
The following can be pasted in a script and executed. This uses the latest node lts/gallium version(v16.20.0) with npm v8(8.19.4). NOTE: user is when using vagrant. Use the email adress if it is the dev cluser (please never connect to prod directly)
cd frontend/common/kubeflow-common-lib
npm i
npm run build
npm link ./dist/kubeflow
cd ../../jupyter npm i npm link kubeflow KFUSERID=user npm start
For the kubecost data to be retrievable when running locally, the following will need to be executed kubectl port-forward -n kubecost-system deployment/kubecost-cost-analyzer 9090
Testing backend Rest API
To test the backend, install the Thunder Client extension. After being installed, it should appear in the vs code sidebar (it might require a restart first). First thing is to make sure that the backend is running. Then, open Thunder Client from the vs code sidebar. You should be able to see multiple requests that can be made against the backend under the "Collections" tab, in the "Golang kubeflow" dropdown. From there, just select a request and hit "Send". Some requests require certain parameters to have values. Those can be filled up under the "Env" tab.
Running intergration tests
We use Cypress to make our end-to-end tests. To run integration tests locally, first make sure that the jupyter-apis app is up and running. Then, from the jupyter-apis/frontend/jupyter directory, run either npm run ui-test to open the Cypress UI, or npm run ui-test-ci to run the cypress tests just in the terminal.
Connecting a Kubeflow Cluster
The API server will connect to the Kubeflow cluster from your current kubectl context. Here are a couple options for setting that up.
Local MiniKF Cluster
This deploys a Kubeflow cluster on your local machine and requires at least 50GB of disk space and the recommanded RAM is 12 Gb. First, create the miniKF cluster:
- Install Vagrant and Virtual Box.
- Create a new directory and run
vagrant init arrikto/minikfand then
vagrant up (takes about 20 minutes to boot).
- Navigate to
http://10.10.10.10. - Follow on-screen steps to start Kubeflow and Rok (takes about another 20 minutes).
- From here, you can use Kubeflow and Rok.
kubectl to connect to your new cluster:
- Download the miniKF
kubectlconfig file fromhttp://10.10.10.10. - Use the downloaded configuration for
kubectl, either replacing, or merging
~/.kube/config.
- Ensure
kubectl config current-contextis pointing to your local cluster.
KFUSERID can be the default user that was created for your miniKF
cluster (typically user).
Each time you need to start the cluster, navigate to the directory you created and run vagrant up.
Note: after some experimentation, it was found that the vagrant file could be modified on line 57 to use 8gb instead of the default 12.
config.vm.provider "virtualbox" do |vb|
# Display the VirtualBox GUI when booting the machine
#vb.gui = true
# Customize the amount of memory on the VM: vb.memory = "8192" end
Remote AKS Cluster
- Install Azure CLI
azandkubectl - Login with
azand set your subscription - Run
az aks get-credentialsfor the desired remote cluster - Ensure
kubectl config current-contextis pointing to correct cluster
On platform testing
Any push to an open PR that has the auto-deploy label on it allows developers to opt-in to on-platform testing. For example, when you need to build in github and test on platform (or want someone else to be able to pull your image):
- open a PR and add the auto-deploy label
- push to your PR and watch the GitHub Action CI
- access your image in Kubeflow DEV via a custom image from any of:
Whats Different?
Routes are defined in this repository here.
Upstream, the endpoints are structures via request type (e.g. GET, PUT, DELETE).
Note
- that not all endpoints are included in the golang implementation
- to find the upstream endpoint, load the Upstream
| Request Type | Golang Endpoint | Upstream Python Endpoint | Purpose | | --- | --- | --- | --- | | GET | /api/config | /api/config | Gets the spawnerui_config.yaml | | GET | /api/gpus | /api/gpus | Reads the GPU vendors from the spawner config | | GET | /api/storageclasses | /api/storageclasses | list all storageclasses | | GET | /api/storageclasses/default | /api/storageclasses/default | gets the storage class with the is-default-class annotation | | GET | /api/namespaces/{namespace}/cost/allocation | Not found | Get the kubecost Allocation API | | GET | /api/namespaces | /api/namespaces | Get the list of namespaces | | GET | /api/namespaces/{namespace} | Not found | Get namespace metadata | | GET | /api/namespaces/{namespace}/notebooks | /api/namespaces/\PVCs | | GET | /api/namespaces/{namespace}/pvcs/{pvc} | /api/namespaces/\PVC | | DELETE | /api/namespaces/{namespace}/pvcs/{pvc} | /api/namespaces/\PVC | | GET | /api/namespaces/{namespace}/pvcs/{pvc}/pods | /api/namespaces/\PVC | | GET | /api/namespaces/{namespace}/pvcs/{pvc}/events | /api/namespaces/\PVC | | GET | /api/namespaces/{namespace}/poddefaults | /api/namespaces/\PodDefaults for a given namespace |