blob: b04dfd063d4e8ba94d8c562b5a351d5e0fcb0ad1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# Maintainer: Carl Smedstad <carsme@archlinux.org>
pkgname=python-cohere
_pkgname=cohere-python
pkgver=5.5.5
pkgrel=1
pkgdesc="Python Library for Accessing the Cohere API"
arch=(any)
url="https://github.com/cohere-ai/cohere-python"
license=(MIT)
depends=(
python
python-boto3
python-botocore
python-fastavro
python-httpx
python-pydantic
python-requests
python-tokenizers
python-typing_extensions
)
makedepends=(
python-build
python-installer
python-poetry
python-wheel
)
checkdepends=(
python-pytest
python-pytest-asyncio
)
source=("$pkgname-$pkgver.tar.gz::$url/archive/$pkgver.tar.gz")
sha256sums=('87f8ae468b03e720d9d9f9d32462c91a3cfdc41e0299f9488a87b2dc295b1b90')
_archive="$_pkgname-$pkgver"
build() {
cd "$_archive"
python -m build --wheel --no-isolation
}
check() {
cd "$_archive"
rm -rf tmp_install
python -m installer --destdir=tmp_install dist/*.whl
local site_packages=$(python -c "import site; print(site.getsitepackages()[0])")
export PYTHONPATH="$PWD/tmp_install/$site_packages:$PYTHONPATH"
# Only run tests that do not require a valid API key. The one below is a mock
# one and is required for test collection.
export CO_API_KEY=sk-dBAe8c5a9bc4294cca9bed292cd61e0ff9030bB94647adfb
pytest \
tests/test_async_client.py::TestClient::test_context_manager \
tests/test_async_client.py::TestClient::test_deprecated_fn \
tests/test_async_client.py::TestClient::test_moved_fn \
tests/test_async_client.py::TestClient::test_stream_equals_true \
tests/test_async_client.py::TestClient::test_token_falls_back_on_env_variable \
tests/test_client.py::TestClient::test_context_manager \
tests/test_client.py::TestClient::test_deprecated_fn \
tests/test_client.py::TestClient::test_moved_fn \
tests/test_client.py::TestClient::test_stream_equals_true \
tests/test_client.py::TestClient::test_token_falls_back_on_env_variable \
tests/test_embed_utils.py::TestClient::test_merge_embeddings_by_type \
tests/test_embed_utils.py::TestClient::test_merge_embeddings_floats \
tests/test_embed_utils.py::TestClient::test_merge_partial_embeddings_floats
}
package() {
cd "$_archive"
python -m installer --destdir="$pkgdir" dist/*.whl
install -Dm644 -t "$pkgdir/usr/share/licenses/$pkgname" LICENSE
}
|