Skip to main content
Open In ColabOpen on GitHub

GooseAI

GooseAI 是一个完全托管的自然语言处理即服务(NLP-as-a-Service),通过 API 提供。GooseAI 提供对 这些模型 的访问。

本笔记本介绍了如何使用 Langchain 与 GooseAI

安装 openai

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

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

导入

import os

from langchain.chains import LLMChain
from langchain_community.llms import GooseAI
from langchain_core.prompts import PromptTemplate

设置环境API密钥

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

from getpass import getpass

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

创建 GooseAI 实例

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

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)