From 249fd9d0d7839b5c48a7235d74a59b1ecb5467cd Mon Sep 17 00:00:00 2001
From: Sarah Clark <seclark@nextquestion.net>
Date: Sat, 31 May 2025 10:25:04 -0700
Subject: [PATCH] python3Packages.chromadb: 0.5.20 -> 0.6.3

Last of the pre-1.0.0 releases. Required by some legacy AI packages.

changelog: https://github.com/chroma-core/chroma/releases/tag/0.6.3
---
 .../python-modules/chromadb/default.nix       | 46 ++++++++++++-------
 1 file changed, 30 insertions(+), 16 deletions(-)

diff --git a/pkgs/development/python-modules/chromadb/default.nix b/pkgs/development/python-modules/chromadb/default.nix
index 0ed23ba17596..f9f226405f7e 100644
--- a/pkgs/development/python-modules/chromadb/default.nix
+++ b/pkgs/development/python-modules/chromadb/default.nix
@@ -63,25 +63,26 @@
 
 buildPythonPackage rec {
   pname = "chromadb";
-  version = "0.5.20";
+  version = "0.6.3";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "chroma-core";
     repo = "chroma";
     tag = version;
-    hash = "sha256-DQHkgCHtrn9xi7Kp7TZ5NP1EtFtTH5QOvne9PUvxsWc=";
+    hash = "sha256-yvAX8buETsdPvMQmRK5+WFz4fVaGIdNlfhSadtHwU5U=";
   };
 
   cargoDeps = rustPlatform.fetchCargoVendor {
     inherit src;
-    name = "${pname}-${version}-vendor";
-    hash = "sha256-ZtCTg8qNCiqlH7RsZxaWUNAoazdgmXP2GtpjDpRdvbk=";
+    name = "${pname}-${version}";
+    hash = "sha256-lHRBXJa/OFNf4x7afEJw9XcuDveTBIy3XpQ3+19JXn4=";
   };
 
   pythonRelaxDeps = [
     "chroma-hnswlib"
     "orjson"
+    "tokenizers"
   ];
 
   build-system = [
@@ -166,9 +167,11 @@ buildPythonPackage rec {
     # Tests are flaky / timing sensitive
     "test_fastapi_server_token_authn_allows_when_it_should_allow"
     "test_fastapi_server_token_authn_rejects_when_it_should_reject"
+
     # Issue with event loop
     "test_http_client_bw_compatibility"
-    # Issue with httpx
+
+    # httpx ReadError
     "test_not_existing_collection_delete"
   ];
 
@@ -181,6 +184,17 @@ buildPythonPackage rec {
     "chromadb/test/property/test_cross_version_persist.py"
     "chromadb/test/stress/"
     "chromadb/test/test_api.py"
+
+    # httpx failures
+    "chromadb/test/api/test_delete_database.py"
+
+    # Cannot be loaded by pytest without path hacks (fixed in 1.0.0)
+    "chromadb/test/test_logservice.py"
+    "chromadb/test/proto/test_utils.py"
+    "chromadb/test/segment/distributed/test_protobuf_translation.py"
+
+    # Hypothesis FailedHealthCheck due to nested @given tests
+    "chromadb/test/cache/test_cache.py"
   ];
 
   __darwinAllowLocalNetworking = true;
@@ -198,17 +212,17 @@ buildPythonPackage rec {
         "([0-9].+)"
       ];
     };
-  };
 
-  meta = {
-    description = "AI-native open-source embedding database";
-    homepage = "https://github.com/chroma-core/chroma";
-    changelog = "https://github.com/chroma-core/chroma/releases/tag/${version}";
-    license = lib.licenses.asl20;
-    maintainers = with lib.maintainers; [
-      fab
-      sarahec
-    ];
-    mainProgram = "chroma";
+    meta = {
+      description = "AI-native open-source embedding database";
+      homepage = "https://github.com/chroma-core/chroma";
+      changelog = "https://github.com/chroma-core/chroma/releases/tag/${version}";
+      license = lib.licenses.asl20;
+      maintainers = with lib.maintainers; [
+        fab
+        sarahec
+      ];
+      mainProgram = "chroma";
+    };
   };
 }