mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
6b2a57161a
- **Description:** add support for kwargs in`MlflowEmbeddings` `embed_document()` and `embed_query()` so that all the arguments required by Cohere API (and others?) can be passed down to the server. - **Issue:** #15234 - **Dependencies:** MLflow with MLflow Deployments (`pip install mlflow[genai]`) **Tests** Now this code [adapted from the docs](https://python.langchain.com/docs/integrations/providers/mlflow#embeddings-example) for the Cohere API works locally. ```python """ Setup ----- export COHERE_API_KEY=... mlflow deployments start-server --config-path examples/deployments/cohere/config.yaml Run --- python /path/to/this/file.py """ embeddings = MlflowCohereEmbeddings(target_uri="http://127.0.0.1:5000", endpoint="embeddings") print(embeddings.embed_query("hello")[:3]) print(embeddings.embed_documents(["hello", "world"])[0][:3]) ``` Output ``` [0.060455322, 0.028793335, -0.025848389] [0.031707764, 0.021057129, -0.009361267] ``` |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| test_deterministic_embedding.py | ||
| test_edenai.py | ||
| test_embaas.py | ||
| test_gradient_ai.py | ||
| test_huggingface.py | ||
| test_imports.py | ||
| test_infinity.py | ||
| test_llm_rails.py | ||
| test_openai.py | ||
| test_vertexai.py | ||