a
bg6 @ sZ d dl mZmZmZmZ d dlmZ d dlmZm Z m
Z
dZ dZ dZ
G dd deeZd S )
)AnyDictListOptional)
Embeddings) BaseModel
ConfigDictFieldzBAAI/bge-small-en-v1.5z9Represent this question for searching relevant passages: u9 为这个句子生成表示以用于检索相关文章：c s e Zd ZU dZdZeed< eZe ed< dZ
e e ed< e e
dZee ef ed< e e
dZee ef ed< eZe ed < d
Ze ed < ed fd
d ZedddZee eee dddZe ee dddZ ZS )IpexLLMBgeEmbeddingsa Wrapper around the BGE embedding model
with IPEX-LLM optimizations on Intel CPUs and GPUs.
To use, you should have the ``ipex-llm``
and ``sentence_transformers`` package installed. Refer to
`here `_
for installation on Intel CPU.
Example on Intel CPU:
.. code-block:: python
from langchain_community.embeddings import IpexLLMBgeEmbeddings
embedding_model = IpexLLMBgeEmbeddings(
model_name="BAAI/bge-large-en-v1.5",
model_kwargs={},
encode_kwargs={"normalize_embeddings": True},
)
Refer to
`here `_
for installation on Intel GPU.
Example on Intel GPU:
.. code-block:: python
from langchain_community.embeddings import IpexLLMBgeEmbeddings
embedding_model = IpexLLMBgeEmbeddings(
model_name="BAAI/bge-large-en-v1.5",
model_kwargs={"device": "xpu"},
encode_kwargs={"normalize_embeddings": True},
)
Nclient
model_name cache_folder)default_factory model_kwargs
encode_kwargsquery_instruction embed_instruction)kwargsc
s t jf i | z ddl}ddlm}m} W n@ tyn } z(d}td| d| d|W Y d}~n
d}~0 0 d| jvrd | jd< | jd d
vrtd | jd d |j | j
fd
| j i| j| _ || j | _ || j | _ | jd dkr| j
d| _ d| j
v rt| _dS )z$Initialize the sentence_transformer.r N)_optimize_post
_optimize_prezChttps://python.langchain.com/v0.1/docs/integrations/text_embedding/zDCould not import ipex_llm or sentence_transformers. Please refer to zD/ipex_llm/ for install required packages on Intel CPU. And refer to z;/ipex_llm_gpu/ for install required packages on Intel GPU. Zdevicecpu)r xpuzXIpexLLMBgeEmbeddings currently only supports device to be 'cpu' or 'xpu', but you have: .r
r z-zh)super__init__sentence_transformersZ ipex_llm.transformers.convertr r ImportErrorr
ValueErrorZSentenceTransformerr r
r Zhalfto DEFAULT_QUERY_BGE_INSTRUCTION_ZHr )selfr r r r excbase_url __class__ u/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/ipex_llm.pyr C sJ
z IpexLLMBgeEmbeddings.__init__Zforbidr& )extraZprotected_namespaces)textsreturnc s0 fdd|D } j j|fi j}| S )zCompute doc embeddings using a HuggingFace transformer model.
Args:
texts: The list of texts to embed.
Returns:
List of embeddings, one for each text.
c s g | ]} j |d d qS )
)r replace).0tr! r& r'
x z8IpexLLMBgeEmbeddings.embed_documents..)r encoder tolist)r! r) Z
embeddingsr& r0 r' embed_documentso s z$IpexLLMBgeEmbeddings.embed_documents)textr* c C s0 | dd}| jj| j| fi | j}| S )zCompute query embeddings using a HuggingFace transformer model.
Args:
text: The text to embed.
Returns:
Embeddings for the text.
r+ r, )r- r r3 r r r4 )r! r6 Z embeddingr& r& r' embed_query| s z IpexLLMBgeEmbeddings.embed_query) __name__
__module__ __qualname____doc__r r __annotations__DEFAULT_BGE_MODELr strr
r r dictr r r DEFAULT_QUERY_BGE_INSTRUCTION_ENr r r r Z model_configr floatr5 r7
__classcell__r& r& r$ r' r
s
# *
r
N)typingr r r r Zlangchain_core.embeddingsr Zpydanticr r r r= r@ r r
r& r& r& r' s