github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-23 19:39:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Repair embeddings/llamacpp's embed_query method (#29935)

Browse Source 
**Description:** As commented on the commit
[41b6a86](41b6a86bbe)
it introduced a bug for when we do an embedding request and the model
returns a non-nested list. Typically it's the case for model
**_nomic-embed-text_**.

- I added the unit test, and ran `make format`, `make lint` and `make
test` from the `community` package.
- No new dependency.

---------

Co-authored-by: Chester Curme <chester.curme@gmail.com>
This commit is contained in:
Julien Elkaim
2025-02-24 03:32:17 +08:00
committed by GitHub
parent 5ca4933b9d
commit e586bffe51
2 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/embeddings/llamacpp.py
View File

@@ -139,7 +139,7 @@ class LlamaCppEmbeddings(BaseModel, Embeddings):
Embeddings for the text.
"""
embedding = self.client.embed(text)
if not isinstance(embedding, list):
return list(map(float, embedding))
else:
if embedding and isinstance(embedding, list) and isinstance(embedding[0], list):
return list(map(float, embedding[0]))
else:
return list(map(float, embedding))