From 9b3a025f9c806a6f8a00030c7058c689536ae5a0 Mon Sep 17 00:00:00 2001 From: xyd Date: Thu, 20 Jun 2024 21:04:50 +0800 Subject: [PATCH] fix https://github.com/langchain-ai/langchain/issues/23215 (#23216) fix bug The ZhipuAIEmbeddings class is not working. Co-authored-by: xu yandong --- libs/community/langchain_community/embeddings/zhipuai.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/community/langchain_community/embeddings/zhipuai.py b/libs/community/langchain_community/embeddings/zhipuai.py index 31f973a735..dd06216200 100644 --- a/libs/community/langchain_community/embeddings/zhipuai.py +++ b/libs/community/langchain_community/embeddings/zhipuai.py @@ -26,7 +26,7 @@ class ZhipuAIEmbeddings(BaseModel, Embeddings): # query_result = embeddings.embed_query(texts) """ - _client: Any = Field(default=None, exclude=True) #: :meta private: + client: Any = Field(default=None, exclude=True) #: :meta private: model: str = Field(default="embedding-2") """Model name""" api_key: str @@ -39,7 +39,7 @@ class ZhipuAIEmbeddings(BaseModel, Embeddings): try: from zhipuai import ZhipuAI - values["_client"] = ZhipuAI(api_key=values["api_key"]) + values["client"] = ZhipuAI(api_key=values["api_key"]) except ImportError: raise ImportError( "Could not import zhipuai python package." @@ -71,6 +71,6 @@ class ZhipuAIEmbeddings(BaseModel, Embeddings): A list of embeddings for each document in the input list. Each embedding is represented as a list of float values. """ - resp = self._client.embeddings.create(model=self.model, input=texts) + resp = self.client.embeddings.create(model=self.model, input=texts) embeddings = [r.embedding for r in resp.data] return embeddings