llm-lmstudio: init at 0.2.1

pkgs/development/python-modules/llm-lmstudio/default.nix

@@ -0,0 +1,59 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+  setuptools,
+  wheel,
+  httpx,
+  llm,
+  requests,
+  pytest,
+  pytest-asyncio,
+  pytest-mock,
+  pytest-vcr,
+}:
+
+buildPythonPackage rec {
+  pname = "llm-lmstudio";
+  version = "0.2.1";
+  pyproject = true;
+
+  src = fetchFromGitHub {
+    owner = "agustif";
+    repo = "llm-lmstudio";
+    rev = "v${version}";
+    hash = "sha256-dPrkvoVLDIk/JxUhvUUyjMzovH+Q/O13eTdA6qvdKxY=";
+  };
+
+  build-system = [
+    setuptools
+    wheel
+  ];
+
+  dependencies = [
+    httpx
+    llm
+    requests
+  ];
+
+  optional-dependencies = {
+    test = [
+      pytest
+      pytest-asyncio
+      pytest-mock
+      pytest-vcr
+    ];
+  };
+
+  pythonImportsCheck = [
+    "llm_lmstudio"
+  ];
+
+  meta = {
+    description = "Plugin to use local models via LM Studio API with https://llm.datasette.io";
+    homepage = "https://github.com/agustif/llm-lmstudio";
+    changelog = "https://github.com/agustif/llm-lmstudio/blob/${src.rev}/CHANGELOG.md";
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [ dwt ];
+  };
+}

pkgs/development/python-modules/llm/default.nix

@@ -81,6 +81,7 @@ let
       llm-hacker-news ? false,
       llm-jq ? false,
       llm-llama-server ? false,
+      llm-lmstudio ? false,
       llm-mistral ? false,
       llm-ollama ? false,
       llm-openai-plugin ? false,

pkgs/top-level/python-packages.nix

@@ -8904,6 +8904,8 @@ self: super: with self; {
 
   llm-llama-server = callPackage ../development/python-modules/llm-llama-server { };
 
+  llm-lmstudio = callPackage ../development/python-modules/llm-lmstudio { };
+
   llm-mistral = callPackage ../development/python-modules/llm-mistral { };
 
   llm-ollama = callPackage ../development/python-modules/llm-ollama { };