From bc370b861ca74f92ce8117a9b5aef75324e6622f Mon Sep 17 00:00:00 2001 From: jlonge4 Date: Thu, 23 Nov 2023 23:10:56 -0500 Subject: [PATCH] reno & black --- haystack/nodes/retriever/_embedding_encoder.py | 8 +++----- .../aws-bedrock-embedding-encoder-a978884c1a2c8237.yaml | 4 ++++ 2 files changed, 7 insertions(+), 5 deletions(-) create mode 100644 releasenotes/notes/aws-bedrock-embedding-encoder-a978884c1a2c8237.yaml diff --git a/haystack/nodes/retriever/_embedding_encoder.py b/haystack/nodes/retriever/_embedding_encoder.py index ceb55e9a9f..7f8bc9d159 100644 --- a/haystack/nodes/retriever/_embedding_encoder.py +++ b/haystack/nodes/retriever/_embedding_encoder.py @@ -451,11 +451,8 @@ def embed(self, text: str) -> np.ndarray: input_body["inputText"] = text body = json.dumps(input_body) response = self.client.invoke_model( - body=body, - modelId=self.model, - accept="application/json", - contentType="application/json", - ) + body=body, modelId=self.model, accept="application/json", contentType="application/json" + ) response_body = json.loads(response.get("body").read()) return np.array(response_body.get("embedding")) @@ -487,6 +484,7 @@ def train( def save(self, save_dir: Union[Path, str]): raise NotImplementedError(f"Saving is not implemented for {self.__class__}") + _EMBEDDING_ENCODERS: Dict[str, Callable] = { "farm": _DefaultEmbeddingEncoder, "transformers": _DefaultEmbeddingEncoder, diff --git a/releasenotes/notes/aws-bedrock-embedding-encoder-a978884c1a2c8237.yaml b/releasenotes/notes/aws-bedrock-embedding-encoder-a978884c1a2c8237.yaml new file mode 100644 index 0000000000..caa1f479b8 --- /dev/null +++ b/releasenotes/notes/aws-bedrock-embedding-encoder-a978884c1a2c8237.yaml @@ -0,0 +1,4 @@ +--- +features: + - | + Adding Bedrock Embeddings Encoder to use as a retriever.