python312Packages.docling-serve: 0.8.0 -> 0.10.1

pkgs/by-name/do/docling-serve/package.nix

@@ -2,9 +2,21 @@
   python3Packages,
   nixosTests,
   withUI ? false,
+  withTesserocr ? false,
+  withRapidocr ? false,
+  withCPU ? false,
 }:
 
-(python3Packages.toPythonApplication (python3Packages.docling-serve.override { inherit withUI; }))
+(python3Packages.toPythonApplication (
+  python3Packages.docling-serve.override {
+    inherit
+      withUI
+      withTesserocr
+      withRapidocr
+      withCPU
+      ;
+  }
+))
 // {
   passthru.tests = {
     docling-serve = nixosTests.docling-serve;

pkgs/development/python-modules/docling-serve/default.nix

@@ -12,24 +12,37 @@
   python-multipart,
   uvicorn,
   websockets,
+  tesserocr,
+  rapidocr-onnxruntime,
+  onnxruntime,
+  torch,
+  torchvision,
   gradio,
   nodejs,
   which,
   withUI ? false,
+  withTesserocr ? false,
+  withRapidocr ? false,
+  withCPU ? false,
 }:
 
 buildPythonPackage rec {
   pname = "docling-serve";
-  version = "0.8.0";
+  version = "0.10.1";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "docling-project";
     repo = "docling-serve";
     tag = "v${version}";
-    hash = "sha256-ACoqhaGiYHf2dqulxfHQDH/JIhuUlH7wyu0JY4hd0U8=";
+    hash = "sha256-ApI8I14X2BBenEZ9mLXifhgtY1DHdPljMd1LvjbNUXM=";
   };
 
+  postPatch = ''
+    substituteInPlace pyproject.toml \
+      --replace-fail '"kfp[kubernetes]>=2.10.0",' ""
+  '';
+
   build-system = [
     hatchling
     setuptools-scm
@@ -39,15 +52,20 @@ buildPythonPackage rec {
     "websockets"
   ];
 
-  dependencies = [
+  dependencies =
-    docling
+    [
-    fastapi
+      docling
-    httpx
+      fastapi
-    pydantic-settings
+      httpx
-    python-multipart
+      pydantic-settings
-    uvicorn
+      python-multipart
-    websockets
+      uvicorn
-  ] ++ lib.optionals withUI optional-dependencies.ui;
+      websockets
+    ]
+    ++ lib.optionals withUI optional-dependencies.ui
+    ++ lib.optionals withTesserocr optional-dependencies.tesserocr
+    ++ lib.optionals withRapidocr optional-dependencies.rapidocr
+    ++ lib.optionals withCPU optional-dependencies.cpu;
 
   optional-dependencies = {
     ui = [
@@ -55,6 +73,17 @@ buildPythonPackage rec {
       nodejs
       which
     ];
+    tesserocr = [
+      tesserocr
+    ];
+    rapidocr = [
+      rapidocr-onnxruntime
+      onnxruntime
+    ];
+    cpu = [
+      torch
+      torchvision
+    ];
   };
 
   pythonImportsCheck = [