SageTeX installation and usage

What is the nix way to use SageTeX?

I have the following packages already in my home.nix:

texliveFull
sage
sagetex

Problems:

  1. LaTeX cannot not find the sagetex.sty file, which is to be expected. See here. How can I automatically add this file to the path?
  2. When I copy the .sty file to my project root, my .tex file compiles. However, running Sage on the generated .sage file throws the following error:
ModuleNotFoundError: No module named 'sagetex'

I guess this is some python module, which I couldn’t locate on my system, but probably should be included with SageMath itself (?). How should this be handled?