Dennis Zheleznyak 19d290a988 fix(sagemaker): Support TEI raw array response format for embeddings (#20487) ...

HuggingFace Text Embeddings Inference (TEI) returns embeddings as raw
arrays [[0.1, 0.2, ...]] instead of wrapped format {"embedding": [...]}.

This change handles both formats:
- Raw array: [[...]] (TEI, some HF models)
- Wrapped: {"embedding": [[...]]} (standard HF format)

Fixes SagemakerError: "HF response missing 'embedding' field" when using
TEI containers on SageMaker.

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>

2026-02-12 19:39:05 +05:30

..

test_sagemaker_common_utils.py

…

test_sagemaker_embedding_voyage.py

fix(sagemaker): Support TEI raw array response format for embeddings (#20487)

2026-02-12 19:39:05 +05:30