跳到主要内容

针对自定义 LangServe 模型服务器运行 Playground

LangSmith Playground 允许您使用自己的自定义模型。您可以部署一个模型服务器,通过 LangServe(一个用于服务 LangChain 应用程序的开源库)公开您模型的 API。在后台,Playground 将与您的模型服务器交互以生成响应。

部署自定义模型服务器

为了您的方便,我们提供了一个示例模型服务器,您可以将其用作参考。您可以在这里找到示例模型服务器 我们强烈建议使用示例模型服务器作为起点。

根据您的模型是 instruct 风格还是 chat 风格的模型,您需要分别实现 custom_model.pycustom_chat_model.py

添加可配置字段

通常,使用不同的参数配置您的模型非常有用。这些参数可能包括 temperature、model_name、max_tokens 等。

为了使您的模型在 LangSmith Playground 中可配置,您需要在模型服务器中添加可配置字段。这些字段可用于从 Playground 更改模型参数。

您可以通过在 config.py 文件中实现 with_configurable_fields 函数来添加可配置字段。您可以

def with_configurable_fields(self) -> Runnable:
"""Expose fields you want to be configurable in the playground. We will automatically expose these to the
playground. If you don't want to expose any fields, you can remove this method."""
return self.configurable_fields(n=ConfigurableField(
id="n",
name="Num Characters",
description="Number of characters to return from the input prompt.",
))

在 LangSmith Playground 中使用模型

部署模型服务器后,您可以在 LangSmith Playground 中使用它。进入 Playground,然后为 chat 风格模型或 instruct 风格模型选择 ChatCustomModelCustomModel 提供商。

输入 URL。Playground 将自动检测可用的端点和可配置字段。然后,您可以使用所需的参数调用模型。

ChatCustomModel in Playground

如果一切设置正确,您应该在 Playground 中看到模型的响应以及在 with_configurable_fields 中指定的可配置字段。

此处查看如何存储您的模型配置以供以后使用。


此页对您有帮助吗?


您可以留下详细的反馈 在 GitHub 上.