Add sentence-transformers

Author: DouweM

pyproject.toml

@@ -59,6 +59,7 @@ outlines-llamacpp = ["pydantic-ai-slim[outlines-llamacpp]=={{ version }}"]
 outlines-mlxlm = ["pydantic-ai-slim[outlines-mlxlm]=={{ version }}; platform_system == 'Darwin' and platform_machine == 'arm64'"]
 outlines-sglang = ["pydantic-ai-slim[outlines-sglang]=={{ version }}"]
 outlines-vllm-offline = ["pydantic-ai-slim[outlines-vllm-offline]=={{ version }}"]
+sentence-transformers = ["pydantic-ai-slim[sentence-transformers]=={{ version }}"]
 
 [project.urls]
 Homepage = "https://ai.pydantic.dev"

tests/test_embeddings.py

@@ -100,8 +100,7 @@ async def test_documents(self, co_api_key: str):
 @pytest.mark.skipif(not sentence_transformers_imports_successful, reason='SentenceTransformers not installed')
 class TestSentenceTransformers:
     async def test_infer_model(self, co_api_key: str):
-        with patch.dict(os.environ, {'CO_API_KEY': co_api_key}):
-            model = infer_model('sentence-transformers:all-MiniLM-L6-v2')
+        model = infer_model('sentence-transformers:all-MiniLM-L6-v2')
         assert isinstance(model, SentenceTransformerEmbeddingModel)
         assert model.model_name == 'all-MiniLM-L6-v2'
         assert model.system == 'sentence-transformers'