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]
自己封装了好久,结果是两个单词看错了,啊啊啊。