From 76d2933460ba5562f379ce9ed652584285bc66af Mon Sep 17 00:00:00 2001 From: zsimjee Date: Tue, 21 Nov 2023 13:33:07 -0800 Subject: [PATCH] correct embedding access pattern fix embeddings access pattern in the other validator --- guardrails/validators.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/guardrails/validators.py b/guardrails/validators.py index 58708271d..2feef7c43 100644 --- a/guardrails/validators.py +++ b/guardrails/validators.py @@ -710,7 +710,7 @@ def __init__( self._document = document embedding_response = self.client.create_embedding(input=[document], model=model) - embedding = embedding_response["data"][0]["embedding"] # type: ignore + embedding = embedding_response[0] # type: ignore self._document_embedding = np.array(embedding) self._model = model self._threshold = float(threshold) @@ -736,7 +736,7 @@ def validate(self, value: Any, metadata: Dict) -> ValidationResult: ) value_embedding = np.array( - embedding_response["data"][0]["embedding"] # type: ignore + embedding_response[0] # type: ignore ) similarity = self.cosine_similarity(