diff --git a/.github/workflows/test-integrations-ai.yml b/.github/workflows/test-integrations-ai.yml index 10171ce196..e497ba4280 100644 --- a/.github/workflows/test-integrations-ai.yml +++ b/.github/workflows/test-integrations-ai.yml @@ -104,7 +104,7 @@ jobs: strategy: fail-fast: false matrix: - python-version: ["3.8","3.9","3.11","3.12"] + python-version: ["3.8","3.9","3.10","3.11","3.12","3.13"] # python3.6 reached EOL and is no longer being supported on # new versions of hosted runners on Github Actions # ubuntu-20.04 is the last version that supported python3.6 diff --git a/scripts/populate_tox/populate_tox.py b/scripts/populate_tox/populate_tox.py index 8f588a1b26..912cc15bd5 100644 --- a/scripts/populate_tox/populate_tox.py +++ b/scripts/populate_tox/populate_tox.py @@ -78,7 +78,6 @@ "fastapi", "gcp", "httpx", - "huggingface_hub", "langchain", "langchain_notiktoken", "openai", diff --git a/tests/integrations/huggingface_hub/test_huggingface_hub.py b/tests/integrations/huggingface_hub/test_huggingface_hub.py index e017ce2449..090b0e4f3e 100644 --- a/tests/integrations/huggingface_hub/test_huggingface_hub.py +++ b/tests/integrations/huggingface_hub/test_huggingface_hub.py @@ -1,4 +1,5 @@ import itertools +from unittest import mock import pytest from huggingface_hub import ( @@ -9,8 +10,6 @@ from sentry_sdk import start_transaction from sentry_sdk.integrations.huggingface_hub import HuggingfaceHubIntegration -from unittest import mock # python 3.3 and above - def mock_client_post(client, post_mock): # huggingface-hub==0.28.0 deprecates the `post` method @@ -33,7 +32,7 @@ def test_nonstreaming_chat_completion( ) events = capture_events() - client = InferenceClient("some-model") + client = InferenceClient() if details_arg: post_mock = mock.Mock( return_value=b"""[{ @@ -92,7 +91,7 @@ def test_streaming_chat_completion( ) events = capture_events() - client = InferenceClient("some-model") + client = InferenceClient() post_mock = mock.Mock( return_value=[ @@ -116,7 +115,6 @@ def test_streaming_chat_completion( ) ) assert len(response) == 2 - print(response) if details_arg: assert response[0].token.text + response[1].token.text == "the model response" else: @@ -142,7 +140,7 @@ def test_bad_chat_completion(sentry_init, capture_events): sentry_init(integrations=[HuggingfaceHubIntegration()], traces_sample_rate=1.0) events = capture_events() - client = InferenceClient("some-model") + client = InferenceClient() post_mock = mock.Mock(side_effect=OverloadedError("The server is overloaded")) mock_client_post(client, post_mock) @@ -160,7 +158,7 @@ def test_span_origin(sentry_init, capture_events): ) events = capture_events() - client = InferenceClient("some-model") + client = InferenceClient() post_mock = mock.Mock( return_value=[ b"""data:{ diff --git a/tox.ini b/tox.ini index 0cc8a0cce2..50c4dcf4ac 100644 --- a/tox.ini +++ b/tox.ini @@ -151,6 +151,13 @@ envlist = # These come from the populate_tox.py script. Eventually we should move all # integration tests there. + # ~~~ AI ~~~ + {py3.8,py3.10,py3.11}-huggingface_hub-v0.22.2 + {py3.8,py3.10,py3.11}-huggingface_hub-v0.25.2 + {py3.8,py3.12,py3.13}-huggingface_hub-v0.28.1 + {py3.8,py3.12,py3.13}-huggingface_hub-v0.30.2 + + # ~~~ DBs ~~~ {py3.7,py3.11,py3.12}-clickhouse_driver-v0.2.9 @@ -519,6 +526,13 @@ deps = # These come from the populate_tox.py script. Eventually we should move all # integration tests there. + # ~~~ AI ~~~ + huggingface_hub-v0.22.2: huggingface_hub==0.22.2 + huggingface_hub-v0.25.2: huggingface_hub==0.25.2 + huggingface_hub-v0.28.1: huggingface_hub==0.28.1 + huggingface_hub-v0.30.2: huggingface_hub==0.30.2 + + # ~~~ DBs ~~~ clickhouse_driver-v0.2.9: clickhouse-driver==0.2.9