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": [<prompt>]} 

{"outputs": [<output_text>]}
```
This commit is contained in:
Margaret Qian
2023-10-31 11:43:30 -07:00
committed by GitHub
parent d26ac5f999
commit acfc485808

View File

@@ -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 = {