OpenAI (开放人工智能)
OpenAI 提供了一系列不同功率级别的模型,适用于不同的任务。
本示例介绍如何使用 LangChain 与 OpenAI
模型 进行交互。
# 获取令牌:https://platform.openai.com/account/api-keys
from getpass import getpass
OPENAI_API_KEY = getpass()
import os
os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
如果您需要指定组织 ID,可以使用以下单元格。但是,如果您只是属于一个组织或打算使用默认组织,则不需要指定。您可以在此处检查您的默认组织。
要指定您的组织,可以使用以下代码:
OPENAI_ORGANIZATION = getpass()
os.environ["OPENAI_ORGANIZATION"] = OPENAI_ORGANIZATION
from langchain.llms import OpenAI
from langchain import PromptTemplate, LLMChain
template = """问题:{question}
回答:让我们逐步思考。"""
prompt = PromptTemplate(template=template, input_variables=["question"])
llm = OpenAI()
如果您想手动指定 OpenAI API 密钥和/或组织 ID,可以使用以下代码:
llm = OpenAI(openai_api_key="YOUR_API_KEY", openai_organization="YOUR_ORGANIZATION_ID")
如果不适用于您,请删除 openai_organization 参数。
llm_chain = LLMChain(prompt=prompt, llm=llm)
question = "贾斯汀·比伯出生的那一年,哪个 NFL 球队赢得了超级碗?"
llm_chain.run(question)
'贾斯汀·比伯出生于1994年,所以在1994年赢得超级碗的 NFL 球队是达拉斯牛仔队。'
如果您在使用显式代理,请使用 OPENAI_PROXY 环境变量进行传递
os.environ["OPENAI_PROXY"] = "http://proxy.yourcompany.com:8080"