GooseAI
GooseAI 是一個完全託管的 NLP 即服務 (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 執行個體
您可以指定不同的參數，例如模型名稱、產生的最大 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)