ninehills / langchain-wenxin

langchain baidu wenxinworkshop wrapper

langchain-wenxin - Langchain Baidu WENXINWORKSHOP wrapper

请迁移到： https://python.langchain.com/docs/integrations/chat/baidu_qianfan_endpoint 本项目后续不再维护

---

Table of Contents

• Installation
• License

Installation

pip install langchain-wenxin

Document

WENXINWORKSHOP API: https://cloud.baidu.com/doc/WENXINWORKSHOP/s/flfmc9do2

How to use

export BAIDU_API_KEY="xxxxx"
export BAIDU_SECRET_KEY="xxxxx"

from langchain_wenxin.llms import Wenxin

# Wenxin model
llm = Wenxin(model="ernie-bot-turbo")
print(llm("你好"))

# stream call
for i in llm.stream("你好"):
    print(i)

# async call
import asyncio
print(asyncio.run(llm._acall("你好")))

# Wenxin chat model
from langchain_wenxin.chat_models import ChatWenxin
from langchain.schema import HumanMessage
llm = ChatWenxin()
print(llm([HumanMessage(content="你好")]))

# Wenxin embeddings model
from langchain_wenxin.embeddings import WenxinEmbeddings
wenxin_embed = WenxinEmbeddings(truncate="END")
print(wenxin_embed.embed_query("hello"))
print(wenxin_embed.embed_documents(["hello"]))

Support models:

• ernie-bot: Standard model, https://cloud.baidu.com/doc/WENXINWORKSHOP/s/jlil56u11
  • Also named wenxin for compatibility.
• ernie-bot-turbo: Fast model, https://cloud.baidu.com/doc/WENXINWORKSHOP/s/4lilb2lpf
  • Also named eb-instant for compatibility.
• other endpoints: eg: bloomz_7b1 or other custom endpoint.

Qianfan Private

You can set BAIDU_API_URL and BAIDU_ACCESS_CODE in environment variables.

Development

# Create virtual environment
hatch env create
# Activate virtual environment
hatch shell
# Run test
export BAIDU_API_KEY="xxxxxxxx"
export BAIDU_SECRET_KEY="xxxxxxxx"
hatch run test

License

langchain-wenxin is distributed under the terms of the MIT license.