From 2f6cd392ad63d5f2d21a2a5b45038dc127a2c3a9 Mon Sep 17 00:00:00 2001
From: Ben Sima <ben@bsima.me>
Date: Wed, 5 Jun 2024 22:46:44 -0400
Subject: Add llm-sentence-transformers

Needed this for some LLM projects, basically I need it whenever I want to make
embeddings because it has all of the embedding models.
---
 Biz/Bild/Deps/llm-sentence-transformers.nix | 42 +++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 Biz/Bild/Deps/llm-sentence-transformers.nix

(limited to 'Biz/Bild/Deps/llm-sentence-transformers.nix')

diff --git a/Biz/Bild/Deps/llm-sentence-transformers.nix b/Biz/Bild/Deps/llm-sentence-transformers.nix
new file mode 100644
index 0000000..4d63c83
--- /dev/null
+++ b/Biz/Bild/Deps/llm-sentence-transformers.nix
@@ -0,0 +1,42 @@
+{ buildPythonPackage, fetchFromGitHub, lib, llm, sentence-transformers
+, pytestCheckHook, setuptools, pythonOlder, }:
+buildPythonPackage rec {
+  pname = "llm-sentence-transformers";
+  version = "0.2";
+  pyproject = true;
+
+  disabled = pythonOlder "3.8";
+
+  src = fetchFromGitHub {
+    owner = "simonw";
+    repo = pname;
+    rev = "refs/tags/${version}";
+    hash = "sha256-1NlKPWekdVLrNkIMWXLCRWn54RlAEuEDWMCDnQHNkBc=";
+  };
+
+  nativeBuildInputs = [ setuptools ];
+
+  buildInputs = [ llm sentence-transformers ];
+
+  propagatedBuildInputs = [ sentence-transformers ];
+
+  # fails because of some pydantic warnings
+  doCheck = false;
+  disabledTests = [
+    # wants to mkdir in the /homeless-shelter
+    "test_sentence_transformers"
+  ];
+
+  nativeCheckInputs = [ pytestCheckHook ];
+
+  pythonImportsCheck = [ "llm_sentence_transformers" ];
+
+  meta = with lib; {
+    homepage = "https://github.com/taketwo/llm-sentence-transformers";
+    description = "LLM plugin for embeddings using sentence-transformers";
+    changelog =
+      "https://github.com/taketwo/llm-sentence-transformers/releases/tag/${version}";
+    license = licenses.asl20;
+    maintainers = with maintainers; [ bsima ];
+  };
+}
-- 
cgit v1.2.3