diff --git a/pkgs/development/python-modules/aistudio-sdk/default.nix b/pkgs/development/python-modules/aistudio-sdk/default.nix new file mode 100644 index 000000000000..5d55607efa26 --- /dev/null +++ b/pkgs/development/python-modules/aistudio-sdk/default.nix @@ -0,0 +1,50 @@ +{ + lib, + buildPythonPackage, + fetchPypi, + bce-python-sdk, + click, + prettytable, + psutil, + requests, + tqdm, +}: + +let + version = "0.3.8"; + + format = "wheel"; +in +buildPythonPackage { + pname = "aistudio-sdk"; + inherit version format; + + # No source code dist available + src = fetchPypi { + pname = "aistudio_sdk"; + inherit version format; + dist = "py3"; + python = "py3"; + hash = "sha256-v8lq9yQ6wu4zAwFISapAKHF8zlr6Yir4z+Oh1E0ZQdY="; + }; + + dependencies = [ + bce-python-sdk + click + prettytable + psutil + requests + tqdm + ]; + + pythonImportsCheck = [ "aistudio_sdk" ]; + + meta = { + description = "Python client library for the AIStudio API"; + homepage = "https://pypi.org/project/aistudio-sdk"; + license = lib.licenses.unfree; + mainProgram = "aistudio"; + maintainers = with lib.maintainers; [ kyehn ]; + sourceProvenance = with lib.sourceTypes; [ fromSource ]; + }; +}