From f368888e7e37a0e6b8bec689cd7d0098599581d4 Mon Sep 17 00:00:00 2001 From: ccurme Date: Tue, 18 Aug 2026 18:00:36 -0400 Subject: [PATCH] fix(partners): isolate unit tests from network [closes #39727] (#39729) Co-authored-by: pufuki Co-authored-by: open-swe[bot] --- libs/partners/exa/Makefile | 7 ++----- libs/partners/mistralai/Makefile | 2 +- .../tests/unit_tests/test_embeddings.py | 18 +++++++++++------- libs/partners/nomic/Makefile | 7 ++----- 4 files changed, 16 insertions(+), 18 deletions(-) diff --git a/libs/partners/exa/Makefile b/libs/partners/exa/Makefile index bff0451743..556b46fd04 100644 --- a/libs/partners/exa/Makefile +++ b/libs/partners/exa/Makefile @@ -12,15 +12,12 @@ PYTEST_EXTRA ?= integration_test integration_tests: TEST_FILE=tests/integration_tests/ -test: - uv run --group test --group test_integration pytest $(PYTEST_EXTRA) $(TEST_FILE) +test tests: + uv run --group test pytest $(PYTEST_EXTRA) --disable-socket --allow-unix-socket $(TEST_FILE) integration_test integration_tests: uv run --group test --group test_integration pytest -v --tb=short -n auto --benchmark-disable $(PYTEST_EXTRA) $(TEST_FILE) -tests: - uv run --group test pytest $(PYTEST_EXTRA) $(TEST_FILE) - test_watch: uv run --group test ptw --snapshot-update --now . -- -vv $(TEST_FILE) diff --git a/libs/partners/mistralai/Makefile b/libs/partners/mistralai/Makefile index d1dd17af90..609b5cdd34 100644 --- a/libs/partners/mistralai/Makefile +++ b/libs/partners/mistralai/Makefile @@ -14,7 +14,7 @@ PYTEST_EXTRA ?= integration_test integration_tests: TEST_FILE=$(INTEGRATION_TEST_FILE) test tests: - uv run --group test pytest $(PYTEST_EXTRA) $(TEST_FILE) + uv run --group test pytest $(PYTEST_EXTRA) --disable-socket --allow-unix-socket $(TEST_FILE) test_watch: uv run --group test ptw --snapshot-update --now . -- -vv $(TEST_FILE) diff --git a/libs/partners/mistralai/tests/unit_tests/test_embeddings.py b/libs/partners/mistralai/tests/unit_tests/test_embeddings.py index ddd6cad6af..7caaa9c357 100644 --- a/libs/partners/mistralai/tests/unit_tests/test_embeddings.py +++ b/libs/partners/mistralai/tests/unit_tests/test_embeddings.py @@ -1,6 +1,6 @@ import os from typing import cast -from unittest.mock import MagicMock +from unittest.mock import MagicMock, patch import httpx from pydantic import SecretStr @@ -15,12 +15,16 @@ os.environ["MISTRAL_API_KEY"] = "foo" def test_mistral_init() -> None: - for model in [ - MistralAIEmbeddings(model="mistral-embed", mistral_api_key="test"), # type: ignore[call-arg] - MistralAIEmbeddings(model="mistral-embed", api_key="test"), # type: ignore[arg-type] - ]: - assert model.model == "mistral-embed" - assert cast("SecretStr", model.mistral_api_key).get_secret_value() == "test" + with patch( + "langchain_mistralai.embeddings.Tokenizer.from_pretrained", + return_value=MagicMock(), + ): + for model in [ + MistralAIEmbeddings(model="mistral-embed", mistral_api_key="test"), # type: ignore[call-arg] + MistralAIEmbeddings(model="mistral-embed", api_key="test"), # type: ignore[arg-type] + ]: + assert model.model == "mistral-embed" + assert cast("SecretStr", model.mistral_api_key).get_secret_value() == "test" def test_is_retryable_error_timeout() -> None: diff --git a/libs/partners/nomic/Makefile b/libs/partners/nomic/Makefile index 5a084aa96a..638763b811 100644 --- a/libs/partners/nomic/Makefile +++ b/libs/partners/nomic/Makefile @@ -12,15 +12,12 @@ PYTEST_EXTRA ?= integration_test integration_tests: TEST_FILE = tests/integration_tests/ -test: - uv run --group test --group test_integration pytest $(PYTEST_EXTRA) $(TEST_FILE) +test tests: + uv run --group test pytest $(PYTEST_EXTRA) --disable-socket --allow-unix-socket $(TEST_FILE) integration_test integration_tests: uv run --group test --group test_integration pytest -v --tb=short -n auto --benchmark-disable $(PYTEST_EXTRA) $(TEST_FILE) -tests: - uv run --group test pytest $(PYTEST_EXTRA) $(TEST_FILE) - test_watch: uv run --group test ptw --snapshot-update --now . -- -vv $(TEST_FILE)