You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
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
12 months ago
..
__init__.py add dashscope text embedding (#5929) 12 months ago
aleph_alpha.py Documentation fixes (linting and broken links) (#5563) 1 year ago
base.py FAISS and embedding support (#48) 2 years ago
bedrock.py Fix bedrock auth validation (#5574) 1 year ago
cohere.py Created fix for 5475 (#5659) 1 year ago
dashscope.py add dashscope text embedding (#5929) 12 months ago
deepinfra.py Add DeepInfra embeddings integration with tests and examples, better exception handling for Deep Infra LLM (#5854) 12 months ago
elasticsearch.py Documentation fixes (linting and broken links) (#5563) 1 year ago
embaas.py Add embaas document extraction api endpoints (#6048) 12 months ago
fake.py add fake embeddings class (#1503) 1 year ago
google_palm.py Exponential back-off support for Google PaLM api (#4001) 1 year ago
huggingface.py `encoding_kwargs` for InstructEmbeddings (#5450) 1 year ago
huggingface_hub.py Minor: Remove duplicated word in error message (#2706) 1 year ago
jina.py changed ValueError to ImportError (#5103) 1 year ago
llamacpp.py llama-cpp: add gpu layers parameter (#4739) 1 year ago
minimax.py Add MiniMax embeddings (#5174) 1 year ago
modelscope_hub.py Harrison/modelscope (#5156) 1 year ago
mosaicml.py Update output format of MosaicML endpoint to be more flexible (#6060) 12 months ago
openai.py Fix openai proxy error (#5914) 12 months ago
sagemaker_endpoint.py Fix Sagemaker Batch Endpoints (#3249) 1 year ago
self_hosted.py Removed duplicate BaseModel dependencies (#2471) 1 year ago
self_hosted_hugging_face.py Removed duplicate BaseModel dependencies (#2471) 1 year ago
sentence_transformer.py Sentence Transformers Aliasing (#3541) 1 year ago
tensorflow_hub.py changed ValueError to ImportError (#5103) 1 year ago
vertexai.py Add pagination for Vertex AI embeddings (#5325) 1 year ago