Skip to main content
Open In ColabOpen on GitHub

vLLM 聊天

vLLM 可以部署为一个模拟 OpenAI API 协议的服务器。这使得 vLLM 可以作为使用 OpenAI API 的应用程序的直接替代品。该服务器可以以与 OpenAI API 相同的格式进行查询。

概述​

这将帮助您开始使用vLLM 聊天模型，这些模型利用了langchain-openai包。有关所有ChatOpenAI功能和配置的详细文档，请访问API参考。

集成详情​

类包本地可序列化JS支持包下载包最新
ChatOpenAIlangchain_openai✅测试版❌PyPI - 下载量PyPI - 版本

模型特性​

特定模型功能——例如工具调用、支持多模态输入、支持令牌级流式传输等——将取决于托管模型。

设置​

请参阅vLLM文档这里。

要通过LangChain访问vLLM模型，您需要安装langchain-openai集成包。

凭证​

认证将取决于推理服务器的具体细节。

如果你想获取模型调用的自动追踪，你也可以通过取消注释以下内容来设置你的LangSmith API密钥：

# os.environ["LANGCHAIN_TRACING_V2"] = "true"
# os.environ["LANGCHAIN_API_KEY"] = getpass.getpass("Enter your LangSmith API key: ")

安装​

LangChain vLLM 集成可以通过 langchain-openai 包访问：

%pip install -qU langchain-openai

实例化​

现在我们可以实例化我们的模型对象并生成聊天完成：

from langchain_core.messages import HumanMessage, SystemMessage
from langchain_core.prompts.chat import (
    ChatPromptTemplate,
    HumanMessagePromptTemplate,
    SystemMessagePromptTemplate,
)
from langchain_openai import ChatOpenAI
inference_server_url = "http://localhost:8000/v1"

llm = ChatOpenAI(
    model="mosaicml/mpt-7b",
    openai_api_key="EMPTY",
    openai_api_base=inference_server_url,
    max_tokens=5,
    temperature=0,
)

调用​

messages = [
    SystemMessage(
        content="You are a helpful assistant that translates English to Italian."
    ),
    HumanMessage(
        content="Translate the following sentence from English to Italian: I love programming."
    ),
]
llm.invoke(messages)
AIMessage(content=' Io amo programmare', additional_kwargs={}, example=False)

链式调用​

我们可以链式我们的模型与一个提示模板，如下所示：

from langchain_core.prompts import ChatPromptTemplate

prompt = ChatPromptTemplate(
    [
        (
            "system",
            "You are a helpful assistant that translates {input_language} to {output_language}.",
        ),
        ("human", "{input}"),
    ]
)

chain = prompt | llm
chain.invoke(
    {
        "input_language": "English",
        "output_language": "German",
        "input": "I love programming.",
    }
)
API Reference:ChatPromptTemplate

API参考​

有关通过langchain-openai公开的所有功能和配置的详细文档，请访问API参考：https://python.langchain.com/api_reference/openai/chat_models/langchain_openai.chat_models.base.ChatOpenAI.html

请参考vLLM的文档。

这个页面有帮助吗？