Skip to main content

GooseAI

GooseAI 是一个完全托管的 NLP-as-a-Service，通过 API 提供。GooseAI 提供对这些模型的访问。

本笔记本介绍如何使用 GooseAI 使用 Langchain。

安装 openai​

使用 GooseAI API 需要安装 openai 包。使用 pip3 install openai 安装 openai。

$ pip3 install openai

导入​

import os
from langchain.llms import GooseAI
from langchain import PromptTemplate, LLMChain

设置环境 API 密钥​

确保从 GooseAI 获取您的 API 密钥。您将获得 10 美元的免费信用以测试不同的模型。

from getpass import getpass

GOOSEAI_API_KEY = getpass()
os.environ["GOOSEAI_API_KEY"] = GOOSEAI_API_KEY

创建 GooseAI 实例​

您可以指定不同的参数，如模型名称、生成的最大令牌数、温度等。

llm = GooseAI()

创建 Prompt 模板​

我们将为问题和答案创建一个 Prompt 模板。

template = """Question: {question}

Answer: 让我们一步一步地思考。"""

prompt = PromptTemplate(template=template, input_variables=["question"])

初始化 LLMChain​

llm_chain = LLMChain(prompt=prompt, llm=llm)

运行 LLMChain​

提供一个问题并运行 LLMChain。

question = "贾斯汀·比伯出生的那一年，哪个 NFL 球队赢得了超级碗？"

llm_chain.run(question)