yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-08 04:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

removing async test for non-existant model (#9208)

Browse Source
This commit is contained in:
David S. Batista 2025-04-10 12:38:35 +02:00 committed by GitHub
parent bac29d9337
commit 45aa9608b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
test/components/embedders/test_hugging_face_api_text_embedder.py
View File

@ -317,28 +317,3 @@ class TestHuggingFaceAPITextEmbedderAsync:
assert isinstance(result["embedding"], list)
assert all(isinstance(x, float) for x in result["embedding"])
assert len(result["embedding"]) == 384 # MiniLM-L6-v2 has 384 dimensions
@pytest.mark.integration
@pytest.mark.asyncio
@pytest.mark.skipif(os.environ.get("HF_API_TOKEN", "") == "", reason="HF_API_TOKEN is not set")
async def test_run_async_error_handling_with_real_api(self):
"""
Integration test that verifies error handling with a real API.
This test requires a valid Hugging Face API token.
"""
# Use an invalid model name to trigger an error
invalid_model_name = "invalid-model-name-that-does-not-exist"
embedder = HuggingFaceAPITextEmbedder(
api_type=HFEmbeddingAPIType.SERVERLESS_INFERENCE_API, api_params={"model": invalid_model_name}
)
# Test with a simple text
text = "This is a test sentence for embedding."
# The request should fail with an appropriate error
with pytest.raises(Exception) as excinfo:
await embedder.run_async(text=text)
# Verify that the error message contains information about the invalid model
assert invalid_model_name in str(excinfo.value) or "model" in str(excinfo.value).lower()