OpenAI e a integração com LangChain
Aprenda a como interagir com os modelos da OpenAI usando uma biblioteca chamada LangChain. A OpenAI oferece um espectro de modelos com diferentes níveis de potência adequados para diferentes tarefas.
2 min readMay 20, 2023
Este exemplo aborda como usar LangChain para interagir com modelos OpenAI
# get a token: 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
from langchain.llms import OpenAI
from langchain import PromptTemplate, LLMChain
template = """Question: {question}
Answer: Let's think step by step."""
prompt = PromptTemplate(template=template, input_variables=["question"])
llm = OpenAI()
llm_chain = LLMChain(prompt=prompt, llm=llm)
question = "What NFL team won the Super Bowl in the year Justin Beiber was born?"
llm_chain.run(question)
Vamos analisar o exemplo acima passo a passo:
- Primeiramente, você precisa obter uma chave de API da OpenAI, que será usada para autenticar suas solicitações à API. Essa chave pode ser obtida na página de chaves da API da OpenAI (https://platform.openai.com/account/api-keys).
- Depois de obter a chave da API, o script Python usa a função
getpass
para solicitar a entrada da chave. Essa chave é então armazenada na variávelOPENAI_API_KEY
. - A chave da API é então armazenada como uma variável de ambiente para que possa ser usada pela biblioteca LangChain.
- A biblioteca LangChain é então importada, e uma nova instância da classe
OpenAI
é criada. - Em seguida, um modelo de mensagem é definido usando a classe
PromptTemplate
. Neste caso, o modelo é uma pergunta seguida de uma resposta inicial. - O modelo de mensagem e a instância de
OpenAI
são passados para a classeLLMChain
(Linguistic Logic Model Chain), que encapsula a lógica para fazer perguntas ao modelo da OpenAI. - Uma pergunta é então definida — “Que equipe da NFL venceu o Super Bowl no ano em que Justin Bieber nasceu?” — e passada para o método
run
da instânciallm_chain
. - O modelo da OpenAI então processa a pergunta e retorna uma resposta: ‘Justin Bieber nasceu em 1994, então estamos procurando o vencedor do Super Bowl daquele ano. O Super Bowl em 1994 foi o Super Bowl XXVIII, e o vencedor foi o Dallas Cowboys.’
Reference
https://python.langchain.com/en/latest/reference/modules/llms.html
Source
LangChain — Índice
Clique no link abaixo para acessar a página que serve como um índice abrangente, reunindo todos os links para os nossos tutoriais sobre LangChain.