Extraindo dados estruturados de texto

Desenvolvimento de sistemas de IA com a API OpenAI

Francesca Donadoni

Curriculum Manager, DataCamp

Implementando function calling

"Estamos buscando um(a) Data Scientist altamente qualificado(a) para se juntar ao nosso time inovador na sede da empresa em San Francisco, CA. Neste cargo, você vai trabalhar em análises e modelos complexos que orientam decisões estratégicas. Requisitos: Mínimo de 3 anos de experiência em data science com Python e AWS, Azure ou GCP."

from openai import OpenAI

client = OpenAI(api_key="ENTER YOUR KEY 
                HERE")

response= client.chat.completions.create(
    model="gpt-4o-mini",
    messages=messages,

    tools=function_definition,

)

Configurando function calling

function_definition = [{

        'type': 'function',

        'function': {

          'name': 'extract_job_info',
          'description': 'Get the job information from the body of the input text',
          'parameters': {
            ...

            }

    }]

Configurando function calling

function_definition = [{
        'type': 'function',
        'function': {
          'name': 'extract_job_info',
          'description': 'Get the job information from the body of the input text',

          'parameters': {'type': 'object',
                         'properties': 
                            'job': {'type': 'string',
                                    'description': 'Job title'},
                            'location': {'type': 'string',
                                         'description': 'Office location'},
              ...
           }

    }]

Resposta

response= client.chat.completions.create(
    model="gpt-4o-mini",
    messages=messages,
    tools=function_definition
)

print(response.choices[0].message.tool_calls[0].function.arguments)

{"job":"Data Scientist","location":"San Francisco, CA"}

Vamos praticar!

Desenvolvimento de sistemas de IA com a API OpenAI