|
|
|
@ -1,36 +1,35 @@ |
|
|
|
|
{ lib |
|
|
|
|
, buildPythonPackage |
|
|
|
|
, fetchFromGitHub |
|
|
|
|
, pythonOlder |
|
|
|
|
|
|
|
|
|
# Python dependencies |
|
|
|
|
, matplotlib |
|
|
|
|
, numpy |
|
|
|
|
, openpyxl |
|
|
|
|
, pandas |
|
|
|
|
, pandas-stubs |
|
|
|
|
, plotly |
|
|
|
|
, pytest-mock |
|
|
|
|
, pytestCheckHook |
|
|
|
|
, pythonOlder |
|
|
|
|
, requests |
|
|
|
|
, scikit-learn |
|
|
|
|
, tenacity |
|
|
|
|
, tqdm |
|
|
|
|
, typing-extensions |
|
|
|
|
, wandb |
|
|
|
|
|
|
|
|
|
# Check dependencies |
|
|
|
|
, pytest-mock |
|
|
|
|
, pytestCheckHook |
|
|
|
|
}: |
|
|
|
|
|
|
|
|
|
buildPythonPackage rec { |
|
|
|
|
pname = "openai"; |
|
|
|
|
version = "0.23.0"; |
|
|
|
|
version = "0.23.1"; |
|
|
|
|
format = "setuptools"; |
|
|
|
|
|
|
|
|
|
disabled = pythonOlder "3.7.1"; |
|
|
|
|
|
|
|
|
|
# Use GitHub source since PyPi source does not include tests |
|
|
|
|
src = fetchFromGitHub { |
|
|
|
|
owner = "openai"; |
|
|
|
|
repo = "openai-python"; |
|
|
|
|
rev = "v${version}"; |
|
|
|
|
sha256 = "sha256-VH1XR2FocRX5AYpCruAKwQUXjXqvdJsVwKdtot5Bo+Y="; |
|
|
|
|
hash = "sha256-4RdER6ecvHGXTLZ1GnBNI1hIETI8O/t+kuOXiQhMigs="; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
propagatedBuildInputs = [ |
|
|
|
@ -39,18 +38,40 @@ buildPythonPackage rec { |
|
|
|
|
pandas |
|
|
|
|
pandas-stubs |
|
|
|
|
requests |
|
|
|
|
scikit-learn |
|
|
|
|
tenacity |
|
|
|
|
tqdm |
|
|
|
|
wandb |
|
|
|
|
typing-extensions |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
passthru.optional-dependencies = { |
|
|
|
|
wandb = [ |
|
|
|
|
wandb |
|
|
|
|
]; |
|
|
|
|
embeddings = [ |
|
|
|
|
matplotlib |
|
|
|
|
plotly |
|
|
|
|
scikit-learn |
|
|
|
|
tenacity |
|
|
|
|
]; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
pythonImportsCheck = [ |
|
|
|
|
"openai" |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
checkInputs = [ |
|
|
|
|
pytestCheckHook |
|
|
|
|
pytest-mock |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
pytestFlagsArray = [ |
|
|
|
|
"openai/tests" |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
pythonImportsCheck = [ "openai" ]; |
|
|
|
|
checkInputs = [ pytestCheckHook pytest-mock ]; |
|
|
|
|
pytestFlagsArray = [ "openai/tests" ]; |
|
|
|
|
OPENAI_API_KEY = "sk-foo"; |
|
|
|
|
|
|
|
|
|
disabledTestPaths = [ |
|
|
|
|
"openai/tests/test_endpoints.py" # requires a real API key |
|
|
|
|
# Requires a real API key |
|
|
|
|
"openai/tests/test_endpoints.py" |
|
|
|
|
"openai/tests/test_file_cli.py" |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
@ -58,6 +79,6 @@ buildPythonPackage rec { |
|
|
|
|
description = "Python client library for the OpenAI API"; |
|
|
|
|
homepage = "https://github.com/openai/openai-python"; |
|
|
|
|
license = licenses.mit; |
|
|
|
|
maintainers = [ maintainers.malo ]; |
|
|
|
|
maintainers = with maintainers; [ malo ]; |
|
|
|
|
}; |
|
|
|
|
} |
|
|
|
|