Blog Formação DEV

Como usar a API da OpenAi com Python

Daremos os primeiros passos para introduzir inteligência artificial em seus projetos Python. Iremos desde a configuração do ambiente até os primeiros exemplos de código, permitindo que você tenha uma base para criar seus próprios projetos com a openai.
Como usar a API da OpenAi com Python
Texto de: Marlliton Souza

Introdução

Antes de começarmos a usar a API da OpenAI é necessário criar uma conta na openai (caso ainda não possua uma). Em seguida, faça login, nague até a tela API Keys e clique em "Create new secret key":

A chave que você vai criar precisará de um nome, e você pode nomear a sua chave como achar melhor. Não se esqueça de copiar essa chave, pois ela não será acessível em outros momentos além de sua criação, e ela será necessária nos passos seguintes.

Iniciando projeto

Supondo que você já tenha o python instalado em sua máquina, partiremos para a configuração do projeto que é bem simples.

Para isso, você precisará criar uma pasta com o nome que você desejar para o projeto, eu chamei a minha de chat-gpt-e-python.

Depois crie o arquivo main.py e um .env no seu editor de código como na imagem abaixo:

Dentro do arquivo .env iremos guardar a nossa API KEY, para fazer isso basta definir uma variável OPENAI_API_KEY e o seu valor será a chave gerada na introdução.

Seu arquivo .env deverá se parecer com isso:

Instalando a biblioteca OpenAI para Python

Aqui você pode abrir o terminal do VS Code ou outro terminal de sua preferência e rodar o seguinte comando: pip install openai. Esse comando fará a instalação globalmente na sua máquina, mas se você tiver experiência com a configuração de ambientes virtuais, também pode criar um ambiente virtual para fazer essa instalação.

Ao final da execução do comando a cima, a biblioteca já estará instalada e já podemos começar nossos exemplos.

Enviando sua primeira requisição

Nesse passo, para que de fato possamos interagir com a API do ChatGPT você precisa ter créditos em sua conta. O valor mínimo que pode ser adicionado na conta é de 5 dólares. Com esse valor você já pode fazer muita coisa, dependendo do modelo que você estiver usando do GPT.

Para adicionar créditos basta seguir o caminho indicado na imagem abaixo:

Finalmente, podemos fazer nossa requisição. Abra o arquivo main.py e digite o seguinte:

from openai import OpenAI

client = OpenAI()

completition = client.chat.completions.create(
  model="gpt-3.5-turbo",
  messages=[
    {
      "role": "system", 
      "content": "Você é um assistente de programação para explicar conceitos complexos de forma criativa."
    },
    {
      "role": "user", 
      "content": "O que é uma recursão em programação?"
    }
  ]
)

print(completition.choices[0].message.content) 

As mensagens são organizadas em um array, onde cada mensagem tem um papel role e um conteúdo content. Todas as mensagens presentes no array serão enviadas para a requisição e servem como contexto para a resposta final que será gerada.

Então, a “role” : “system” é usada para fornecer contexto com instruções iniciais ao modelo. Essas instruções são descritas no atributo content que serve como uma espécie configuração inicial para o modelo entender o propósito da conversa.

No exemplo, nós estamos dizendo para o ChatGPT se comportar como um assistente de programação para explicar conceitos complexos da área.

Já a “role” : “user” com o seu content é a mensagem do usuário, ou seja, é o conceito que o usuário quer como resposta do ChatGPT, no caso, recursão. Esse tipo de informação é o equivalente às mensagens que enviados dentro da própria interface do ChatGPT.

Além das mensagens, na configuração da nossa requisição também consta a chave model que indica o modelo de inteligência artificial que será utilizado para respostas da nossa consulta.

Por fim, temos a impressão da mensagem obtida como resultado da nossa consulta. Essa mensagem está envolta em alguns objetos e por isso a forma de acesso a essa informação acaba sendo um pouco longa, como completition.choices[0].message.content

Executando o exemplo

Existem diversas formas de executar esse exemplo, você pode usar o terminal, ou alguma extensão como o code-runner. Vou optar por executar no terminal, mas você pode executar como achar melhor.

Então, no terminal basta digitar python3 src/main.py, alguns instantes depois você terá uma resposta parecida como essa:

Sua resposta provavelmente não será igual à resposta que obtive, mas em resumo é isso.

Conclusão

Ao longo do artigo, vimos como fazer a configuração e como consumir a API do ChatGPT de forma simples.

Esse é o pontapé inicial para que você consiga criar seus projetos e tirar a sua ideia do papel. Não deixe de conferir outros tópicos da documentação oficial.

Leia também

Docker: comandos básicos e criação de um container postgres

Jogos para estudar CSS

Sobre o autor
Cod3r

Cod3r

Com mais de 400 mil alunos, a Cod3r é uma das principais escolas de tecnologia do País. Um de seus produtos mais importantes é a Formação DEV, com objetivo de preparar os profissionais para o mercado.

Ótimo! Inscreveu-se com sucesso.

Bem-vindo de volta! Registou-se com sucesso.

Assinou com sucesso o Blog Formação DEV .

Sucesso! Verifique o seu e-mail para obter o link mágico para se inscrever.

As suas informações de pagamento foram atualizadas.

Seu pagamento não foi atualizado.