Use truenas_api_client

truenas · Jun 3, 2024 · 5b8d726 · 5b8d726
1 parent ba0dccb
commit 5b8d726
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ixdiagnose/utils/formatter.py b/ixdiagnose/utils/formatter.py
@@ -1,7 +1,8 @@
-from middlewared.client.ejson import dumps as middleware_dumps, loads # noqa
-from middlewared.utils import get
 from typing import Callable, Dict, List, Union
 
+from truenas_api_client.ejson import dumps as middleware_dumps, loads # noqa
+from middlewared.utils import get
+
 
 def dumps(*args, **kwargs) -> str:
     kwargs.setdefault('indent', 4)

diff --git a/ixdiagnose/utils/middleware.py b/ixdiagnose/utils/middleware.py
@@ -2,7 +2,7 @@
 
 from dataclasses import dataclass
 from ixdiagnose.config import conf
-from middlewared.client import Client
+from truenas_api_client import Client
 from typing import Any, Callable, List, Optional