async def test_embed_with_entity_context(self, model):
        """Test embedding with entity context (should just be logged)."""
        mock_embedding = [0.1, 0.2, 0.3] * 512
        mock_response = CreateEmbeddingResponse(
            object="list",
            data=[
                Embedding(
                    object="embedding",
                    embedding=mock_embedding,
                    index=0
                )
            ],
            model="text-embedding-3-small",
            usage={"prompt_tokens": 5, "total_tokens": 5}
        )

        with patch.object(model._client.embeddings, 'create', new_callable=AsyncMock) as mock_create:
            mock_create.return_value = mock_response

            result = await model.embed("Test text", entity_context="test_entity")

            # Verify the result is the same (entity_context only affects logging)
            assert result == mock_embedding

            # Verify the API call is the same
            mock_create.assert_called_once_with(
                input="Test text",
                model="text-embedding-3-small",
                encoding_format="float",
            )
Test embedding with entity context (should just be logged).
backend/tests/platform/embedding_models/test_openai_text2vec.py