Updated `LocalAIEmbeddings` docstring to better explain why `openai` (#10946)

Fixes my misgivings in
https://github.com/langchain-ai/langchain/issues/10912
pull/11206/head
James Braza 11 months ago committed by GitHub
parent 2c114fcb5e
commit b11f21c25f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -120,16 +120,19 @@ async def async_embed_with_retry(embeddings: LocalAIEmbeddings, **kwargs: Any) -
class LocalAIEmbeddings(BaseModel, Embeddings):
"""LocalAI embedding models.
To use, you should have the ``openai`` python package installed, and the
environment variable ``OPENAI_API_KEY`` set to a random string. You need to
specify ``OPENAI_API_BASE`` to point to your LocalAI service endpoint.
Since LocalAI and OpenAI have 1:1 compatibility between APIs, this class
uses the ``openai`` Python package's ``openai.Embedding`` as its client.
Thus, you should have the ``openai`` python package installed, and defeat
the environment variable ``OPENAI_API_KEY`` by setting to a random string.
You also need to specify ``OPENAI_API_BASE`` to point to your LocalAI
service endpoint.
Example:
.. code-block:: python
from langchain.embeddings import LocalAIEmbeddings
openai = LocalAIEmbeddings(
openai_api_key="random-key",
openai_api_key="random-string",
openai_api_base="http://localhost:8080"
)

Loading…
Cancel
Save