github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-16 08:06:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply patch [skip ci]

Browse Source
This commit is contained in:
open-swe[bot] 2025-07-31 00:13:19 +00:00
parent 2f8756d5bd
commit 72df882ae9
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/partners/huggingface/langchain_huggingface/embeddings/huggingface.py
View File

@ -2,6 +2,8 @@ from __future__ import annotations
from typing import Any, Optional from typing import Any, Optional
import numpy as np
import torch
from langchain_core.embeddings import Embeddings from langchain_core.embeddings import Embeddings
from pydantic import BaseModel, ConfigDict, Field from pydantic import BaseModel, ConfigDict, Field
@ -18,9 +20,9 @@ _MIN_OPTIMUM_VERSION = "1.22"
class HuggingFaceEmbeddings(BaseModel, Embeddings): class HuggingFaceEmbeddings(BaseModel, Embeddings):
"""HuggingFace sentence_transformers embedding models. """HuggingFace transformer embedding models.
To use, you should have the ``sentence_transformers`` python package installed. To use, you should have the ``transformers`` python package installed.
Example: Example:
.. code-block:: python .. code-block:: python
@ -171,3 +173,4 @@ class HuggingFaceEmbeddings(BaseModel, Embeddings):
else self.encode_kwargs else self.encode_kwargs
) )
return self._embed([text], embed_kwargs)[0] return self._embed([text], embed_kwargs)[0]