Skip to main content
Open In ColabOpen on GitHub

GooseAI

GooseAI 是一项全托管的 NLP 即服务，通过 API 提供。GooseAI 提供对这些模型的访问。

本 Notebook 将介绍如何将 Langchain 与 GooseAI 结合使用。

安装 openai​

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

%pip install --upgrade --quiet  langchain-openai

Imports​

import os

from langchain.chains import LLMChain
from langchain_community.llms import GooseAI
from langchain_core.prompts import PromptTemplate
API Reference:LLMChain | GooseAI | PromptTemplate

设置 Environment API 密钥​

请确保从 GooseAI 获取您的 API 密钥。您将获得 10 美元的免费积分来测试不同的模型。

from getpass import getpass

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

创建 GooseAI 实例​

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

llm = GooseAI()

创建提示模板​

我们将为问答创建提示模板。

template = """Question: {question}

Answer: Let's think step by step."""

prompt = PromptTemplate.from_template(template)

初始化 LLMChain​

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

运行 LLMChain​

提供一个问题并运行 LLMChain。

question = "What NFL team won the Super Bowl in the year Justin Beiber was born?"

llm_chain.run(question)