63 lines
1.1 KiB
Nix

{
lib,
buildPythonPackage,
fetchFromGitHub,
hatchling,
hatch-vcs,
httpx,
pydantic,
pillow,
pytest-asyncio,
pytest-httpserver,
pytestCheckHook,
}:
buildPythonPackage rec {
pname = "ollama";
version = "0.5.3";
pyproject = true;
src = fetchFromGitHub {
owner = "ollama";
repo = "ollama-python";
tag = "v${version}";
hash = "sha256-QpG8bo6tarhW4NpdD4MQ9DWR/w8KjT2zEQyirYtTjJ0=";
};
pythonRelaxDeps = [ "httpx" ];
build-system = [
hatchling
hatch-vcs
];
dependencies = [
httpx
pydantic
];
nativeCheckInputs = [
pillow
pytest-asyncio
pytest-httpserver
pytestCheckHook
];
__darwinAllowLocalNetworking = true;
pythonImportsCheck = [ "ollama" ];
disabledTestPaths = [
# Don't test the examples
"examples/"
];
meta = {
description = "Ollama Python library";
homepage = "https://github.com/ollama/ollama-python";
changelog = "https://github.com/ollama/ollama-python/releases/tag/${src.tag}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ fab ];
};
}