LangChain自定义Embedding封装 之 ERNIE Bot

LangChain自定义Embedding封装 之  ERNIE Bot

百度飞浆平台的  ERNIE Bot

导入下面方法 和 环境 ，即可验证

embedding = ERNIE_Bot_embedding()

class ERNIE_Bot_embedding(BaseModel, Embeddings):client: Any@root_validator()def validate_environment(cls, values: Dict) -> Dict:import osvalues["client"] = os.environ.get("EB_ACCESS_TOKEN")print(values)return valuesdef embed_query(self, text: str) -> List[float]:# print(text)import erniebotresponse = erniebot.Embedding.create(model="ernie-text-embedding",input=[text])# print(response.data[0]['embedding'])return response.data[0]['embedding']def embed_documents(self, texts: List[str]) -> List[List[float]]:# print(texts)return [self.embed_query(text) for text in texts]

自己封装了好久，结果是两个单词看错了，啊啊啊。