diff --git a/pkgs/development/python-modules/llama-index-vector-stores-faiss/default.nix b/pkgs/development/python-modules/llama-index-vector-stores-faiss/default.nix new file mode 100644 index 000000000000..fdc0a13cfab1 --- /dev/null +++ b/pkgs/development/python-modules/llama-index-vector-stores-faiss/default.nix @@ -0,0 +1,39 @@ +{ + lib, + buildPythonPackage, + faiss, + fetchPypi, + hatchling, + llama-index-core, +}: + +buildPythonPackage (finalAttrs: { + pname = "llama-index-vector-stores-faiss"; + version = "0.6.0"; + pyproject = true; + + src = fetchPypi { + pname = "llama_index_vector_stores_faiss"; + inherit (finalAttrs) version; + hash = "sha256-AL/rbLdXHg6FZWbLTxDIm0FbYQjxUdmtSO6cMdpWP14="; + }; + + build-system = [ hatchling ]; + + dependencies = [ + faiss + llama-index-core + ]; + + pythonImportsCheck = [ "llama_index.vector_stores.faiss" ]; + + meta = { + description = "LlamaIndex Vector Store Integration for Faiss"; + homepage = "https://github.com/run-llama/llama_index/tree/main/llama-index-integrations/vector_stores/llama-index-vector-stores-faiss"; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ + fab + kilyanni + ]; + }; +}) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index e8558e3a9521..00c28c4b354e 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -9163,6 +9163,10 @@ self: super: with self; { callPackage ../development/python-modules/llama-index-vector-stores-chroma { }; + llama-index-vector-stores-faiss = + callPackage ../development/python-modules/llama-index-vector-stores-faiss + { }; + llama-index-vector-stores-google = callPackage ../development/python-modules/llama-index-vector-stores-google { };