Skip to main content

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"