diff --git a/pkgs/development/python-modules/llama-index-vector-stores-milvus/default.nix b/pkgs/development/python-modules/llama-index-vector-stores-milvus/default.nix new file mode 100644 index 000000000000..e1b95f532a69 --- /dev/null +++ b/pkgs/development/python-modules/llama-index-vector-stores-milvus/default.nix @@ -0,0 +1,39 @@ +{ + lib, + buildPythonPackage, + fetchPypi, + hatchling, + llama-index-core, + pymilvus, +}: + +buildPythonPackage rec { + pname = "llama-index-vector-stores-milvus"; + version = "0.9.1"; + pyproject = true; + + src = fetchPypi { + pname = "llama_index_vector_stores_milvus"; + inherit version; + hash = "sha256-8alWmT1CglPpvDqbav7AoahIOcabchsiUmrAbBeOyZI="; + }; + + build-system = [ hatchling ]; + + dependencies = [ + llama-index-core + pymilvus + ]; + + pythonImportsCheck = [ "llama_index.vector_stores.milvus" ]; + + # Module has no tests + doCheck = false; + + meta = { + description = "Llama-index vector_stores milvus integration"; + homepage = "https://pypi.org/project/llama-index-vector-stores-milvus/"; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ fab ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 3fa5006a6825..2b865168d263 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -8588,6 +8588,10 @@ self: super: with self; { callPackage ../development/python-modules/llama-index-vector-stores-google { }; + llama-index-vector-stores-milvus = + callPackage ../development/python-modules/llama-index-vector-stores-milvus + { }; + llama-index-vector-stores-postgres = callPackage ../development/python-modules/llama-index-vector-stores-postgres { };