From acfc4858089555ede179ca59921d1dea66d64282 Mon Sep 17 00:00:00 2001 From: Margaret Qian Date: Tue, 31 Oct 2023 11:43:30 -0700 Subject: [PATCH] Update MosaicML Embedding Input Key (#12657) This input key was missed in the last update PR: https://github.com/langchain-ai/langchain/pull/7391 The input/output formats are intended to be like this: ``` {"inputs": []} {"outputs": []} ``` --- libs/langchain/langchain/embeddings/mosaicml.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langchain/langchain/embeddings/mosaicml.py b/libs/langchain/langchain/embeddings/mosaicml.py index d7eb9b1526b..934ffb83901 100644 --- a/libs/langchain/langchain/embeddings/mosaicml.py +++ b/libs/langchain/langchain/embeddings/mosaicml.py @@ -64,7 +64,7 @@ class MosaicMLInstructorEmbeddings(BaseModel, Embeddings): def _embed( self, input: List[Tuple[str, str]], is_retry: bool = False ) -> List[List[float]]: - payload = {"input_strings": input} + payload = {"inputs": input} # HTTP headers for authorization headers = {