langchain/langchain/embeddings
Daniel King a9b97aa6f4
Update output format of MosaicML endpoint to be more flexible (#6060)
There will likely be another change or two coming over the next couple
weeks as we stabilize the API, but putting this one in now which just
makes the integration a bit more flexible with the response output
format.

```
(langchain) danielking@MML-1B940F4333E2 langchain % pytest tests/integration_tests/llms/test_mosaicml.py tests/integration_tests/embeddings/test_mosaicml.py 
=================================================================================== test session starts ===================================================================================
platform darwin -- Python 3.10.11, pytest-7.3.1, pluggy-1.0.0
rootdir: /Users/danielking/github/langchain
configfile: pyproject.toml
plugins: asyncio-0.20.3, mock-3.10.0, dotenv-0.5.2, cov-4.0.0, anyio-3.6.2
asyncio: mode=strict
collected 12 items                                                                                                                                                                        

tests/integration_tests/llms/test_mosaicml.py ......                                                                                                                                [ 50%]
tests/integration_tests/embeddings/test_mosaicml.py ......                                                                                                                          [100%]

=================================================================================== slowest 5 durations ===================================================================================
4.76s call     tests/integration_tests/llms/test_mosaicml.py::test_retry_logic
4.74s call     tests/integration_tests/llms/test_mosaicml.py::test_mosaicml_llm_call
4.13s call     tests/integration_tests/llms/test_mosaicml.py::test_instruct_prompt
0.91s call     tests/integration_tests/llms/test_mosaicml.py::test_short_retry_does_not_loop
0.66s call     tests/integration_tests/llms/test_mosaicml.py::test_mosaicml_extra_kwargs
=================================================================================== 12 passed in 19.70s ===================================================================================
```

#### Who can review?

  @hwchase17
  @dev2049
2023-06-15 22:15:39 -07:00
..
__init__.py add dashscope text embedding (#5929) 2023-06-11 21:14:20 -07:00
aleph_alpha.py Documentation fixes (linting and broken links) (#5563) 2023-06-01 13:06:17 -07:00
base.py FAISS and embedding support (#48) 2022-11-01 21:29:39 -07:00
bedrock.py Fix bedrock auth validation (#5574) 2023-06-01 12:35:06 -07:00
cohere.py Created fix for 5475 (#5659) 2023-06-03 16:47:52 -07:00
dashscope.py add dashscope text embedding (#5929) 2023-06-11 21:14:20 -07:00
deepinfra.py Add DeepInfra embeddings integration with tests and examples, better exception handling for Deep Infra LLM (#5854) 2023-06-07 19:14:30 -07:00
elasticsearch.py Documentation fixes (linting and broken links) (#5563) 2023-06-01 13:06:17 -07:00
embaas.py Add embaas document extraction api endpoints (#6048) 2023-06-12 19:13:52 -07:00
fake.py add fake embeddings class (#1503) 2023-03-07 15:23:46 -08:00
google_palm.py Exponential back-off support for Google PaLM api (#4001) 2023-05-15 17:21:11 -07:00
huggingface_hub.py Minor: Remove duplicated word in error message (#2706) 2023-04-11 13:10:33 -07:00
huggingface.py encoding_kwargs for InstructEmbeddings (#5450) 2023-05-30 11:57:04 -07:00
jina.py changed ValueError to ImportError (#5103) 2023-05-22 15:24:45 -07:00
llamacpp.py llama-cpp: add gpu layers parameter (#4739) 2023-05-15 16:01:48 -07:00
minimax.py Add MiniMax embeddings (#5174) 2023-05-25 06:57:49 -07:00
modelscope_hub.py Harrison/modelscope (#5156) 2023-05-24 08:06:45 -07:00
mosaicml.py Update output format of MosaicML endpoint to be more flexible (#6060) 2023-06-15 22:15:39 -07:00
openai.py Fix openai proxy error (#5914) 2023-06-08 23:15:06 -07:00
sagemaker_endpoint.py Fix Sagemaker Batch Endpoints (#3249) 2023-04-22 08:49:51 -07:00
self_hosted_hugging_face.py Removed duplicate BaseModel dependencies (#2471) 2023-04-06 12:45:16 -07:00
self_hosted.py Removed duplicate BaseModel dependencies (#2471) 2023-04-06 12:45:16 -07:00
sentence_transformer.py Sentence Transformers Aliasing (#3541) 2023-04-25 23:29:20 -07:00
tensorflow_hub.py changed ValueError to ImportError (#5103) 2023-05-22 15:24:45 -07:00
vertexai.py Add pagination for Vertex AI embeddings (#5325) 2023-05-29 06:57:41 -07:00