From 52f7d62215d1007650a8610c1d2cd0efd9edb028 Mon Sep 17 00:00:00 2001
From: porink0424 <daichi2120@gmail.com>
Date: Wed, 10 Jul 2024 18:15:35 +0900
Subject: [PATCH] Add APIClientProvider to widget

---
 jupyterlab/src/widget.tsx | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/jupyterlab/src/widget.tsx b/jupyterlab/src/widget.tsx
index d30ecf9d..3a7d207e 100644
--- a/jupyterlab/src/widget.tsx
+++ b/jupyterlab/src/widget.tsx
@@ -1,6 +1,10 @@
 import { ReactWidget } from '@jupyterlab/ui-components';
 import React from 'react';
 import { JupyterLabEntrypoint } from './components/JupyterLabEntrypoint';
+import { JupyterlabAPIClient } from './apiClient';
+import { APIClientProvider } from '@optuna/optuna-dashboard';
+
+const jupyterlabAPIClient = new JupyterlabAPIClient();
 
 export class OptunaDashboardWidget extends ReactWidget {
   constructor() {
@@ -9,6 +13,10 @@ export class OptunaDashboardWidget extends ReactWidget {
   }
 
   render(): JSX.Element {
-    return <JupyterLabEntrypoint />;
+    return (
+      <APIClientProvider apiClient={jupyterlabAPIClient}>
+        <JupyterLabEntrypoint />
+      </APIClientProvider>
+    );
   }
 }