Skip to main content

Predibase(预测数据库)

了解如何在Predibase上使用LangChain模型。

设置(Setup)

  • 创建一个Predibase账户和API密钥
  • 使用pip install predibase安装Predibase Python客户端。
  • 使用您的API密钥进行身份验证。

LLM

Predibase通过实现LLM模块与LangChain集成。您可以在下面看到一个简短的示例,或者在LLM > 集成 > Predibase下找到完整的笔记本。

import os
os.environ["PREDIBASE_API_TOKEN"] = "{PREDIBASE_API_TOKEN}"

from langchain.llms import Predibase

model = Predibase(model = 'vicuna-13b', predibase_api_key=os.environ.get('PREDIBASE_API_TOKEN'))

response = model("Can you recommend me a nice dry wine?")
print(response)