Add embeddings for LocalAI #8134
Conversation
|
The latest updates on your projects. Learn more about Vercel for Git ↗︎
|
dosubot
bot
added
Ɑ: embeddings
mudler
force-pushed
the
localai_embeddings
branch
from
99131ec to
0addb58
Compare
Signed-off-by: mudler <mudler@localai.io>
mudler
force-pushed
the
localai_embeddings
branch
from
0addb58 to
76ebaf2
Compare
hwchase17
added
the
needs documentation
Signed-off-by: mudler <mudler@localai.io>
Signed-off-by: mudler <mudler@localai.io>
|
@hwchase17 done! will follow up along with #5256 and once we have complete integration with LocalAI I'll update also the documentation page accordingly. Taking this opportunity to ask - is there any interest into adding e.g. voice capabilities? LocalAI supports tts and audio-to-text as well |
baskaryan
removed
the
needs documentation
|
looks awesome, thanks @mudler! there definitely is interest in voice but adding other modalities is a big change that we want to be super thoughtful about, and we haven't had the time to think it through just yet. very open to suggestions on the interface if you're eager to see it in langchain |
|
|
||
| # https://stackoverflow.com/questions/76469415/getting-embeddings-of-length-1-from-langchain-openaiembeddings | ||
| def _check_response(response: dict) -> dict: | ||
| if any(len(d["embedding"]) == 1 for d in response["data"]): |
There was a problem hiding this comment.
Pardon for commenting old code. But the SO thread discusses OpenAI service, so it's hardly the case in LocalAI. Isn't it?
This is why I want to remove this retry condition in new integration package https://github.com/mkhludnev/langchain-localai WDYT?
This package spin off from the discussion #22399 (comment)
| """Call out to LocalAI's embedding endpoint.""" | ||
| # handle large input text | ||
| if self.model.endswith("001"): | ||
| # See: https://github.com/openai/openai-python/issues/418#issuecomment-1525939500 |
There was a problem hiding this comment.
Should we bother about OpenAI specifics when integrate with LocalAI? I don't think so. I'm going to wipe it there https://github.com/mkhludnev/langchain-localai
Description:
This PR adds embeddings for LocalAI ( https://github.com/go-skynet/LocalAI ), a self-hosted OpenAI drop-in replacement. As LocalAI can re-use OpenAI clients it is mostly following the lines of the OpenAI embeddings, however when embedding documents, it just uses string instead of sending tokens as sending tokens is best-effort depending on the model being used in LocalAI. Sending tokens is also tricky as token id's can mismatch with the model - so it's safer to just send strings in this case.
Partly related to: #5256
Dependencies: No new dependencies
Twitter: @mudler_it
Maintainers: @rlancemartin, @eyurtsev, @hwchase17