diff --git a/src/pipelines/ingest.py b/src/pipelines/ingest.py index 3272635..c1cb3a4 100644 --- a/src/pipelines/ingest.py +++ b/src/pipelines/ingest.py @@ -185,7 +185,7 @@ def _embed_text_bedrock(text: str) -> tuple[float, ...]: ) response_body = _json.loads(response["body"].read()) - return tuple(response_body["embedding"]) + return tuple(response_body["embeddings"][0]["embedding"]) # ---------------------------------------------------------------------------