Modal
Modal 云平台提供了从本地计算机上的 Python 脚本按需访问无服务器云计算的便捷方式。使用 modal
运行您自己的自定义 LLM 模型,而不是依赖于 LLM API。
此示例介绍如何使用 LangChain 与 modal
HTTPS Web 端点进行交互。
使用 LangChain 进行问答是另一个如何将 LangChain 与 Modal
一起使用的示例。在该示例中,Modal 端到端运行 LangChain 应用程序,并使用 OpenAI 作为其 LLM API。
%pip install --upgrade --quiet modal
# Register an account with Modal and get a new token.
!modal token new
Launching login page in your browser window...
If this is not showing up, please copy this URL into your web browser manually:
https://modal.com/token-flow/tf-Dzm3Y01234mqmm1234Vcu3
langchain.llms.modal.Modal
集成类要求您部署一个 Modal 应用程序,该应用程序具有符合以下 JSON 界面的 Web 端点
- LLM 提示作为
"prompt"
键下的str
值接受 - LLM 响应作为
"prompt"
键下的str
值返回
示例请求 JSON
{
"prompt": "Identify yourself, bot!",
"extra": "args are allowed",
}
示例响应 JSON
{
"prompt": "This is the LLM speaking",
}
满足此界面的示例“虚拟”Modal Web 端点函数将是
...
...
class Request(BaseModel):
prompt: str
@stub.function()
@modal.web_endpoint(method="POST")
def web(request: Request):
_ = request # ignore input
return {"prompt": "hello world"}
- 有关设置满足此界面的端点的基础知识,请参阅 Modal 的Web 端点指南。
- 有关自定义 LLM 的入门指南,请参阅 Modal 的“使用 AutoGPTQ 运行 Falcon-40B”开源 LLM 示例!
部署 Modal Web 端点后,您可以将其 URL 传递到 langchain.llms.modal.Modal
LLM 类。然后,此类可以用作链中的构建块。
from langchain.chains import LLMChain
from langchain_community.llms import Modal
from langchain_core.prompts import PromptTemplate
template = """Question: {question}
Answer: Let's think step by step."""
prompt = PromptTemplate.from_template(template)
endpoint_url = "https://ecorp--custom-llm-endpoint.modal.run" # REPLACE ME with your deployed Modal web endpoint's URL
llm = Modal(endpoint_url=endpoint_url)
llm_chain = LLMChain(prompt=prompt, llm=llm)
question = "What NFL team won the Super Bowl in the year Justin Beiber was born?"
llm_chain.run(question)