openai_fake_data_generator.py

import openai
frmo typing import List
from presidio_analyzer import RecognizerResult
from presidio_anonymizer import AnonymizerEngine


def set_openai_key(openai_key:string):
    """Set the OpenAI API key.
    :param openai_key: the open AI key (https://help.openai.com/en/articles/4936850-where-do-i-find-my-secret-api-key)
    """
    openai.api_key = openai_key


def call_completion_model(prompt:str, model:str="text-davinci-003", max_tokens:int=512) ->str:
    """Creates a request for the OpenAI Completion service and returns the response.
    
    :param prompt: The prompt for the completion model
    :param model: OpenAI model name
    :param temperature: Model's temperature parameter
    """

    response = openai.Completion.create(
        model=model,
        prompt= prompt,
        max_tokens=max_tokens
    )

    return response['choices'][0].text


def create_prompt(anonymized_text: str) -> str:
    """
    Create the prompt with instructions to GPT-3.
    
    :param anonymized_text: Text with placeholders instead of PII values, e.g. My name is <PERSON>.
    """

    prompt = f"""
    Your role is to create synthetic text based on de-identified text with placeholders instead of personally identifiable information.
    Replace the placeholders (e.g. , , {{DATE}}, {{ip_address}}) with fake values.

    Instructions:

    Use completely random numbers, so every digit is drawn between 0 and 9.
    Use realistic names that come from diverse genders, ethnicities and countries.
    If there are no placeholders, return the text as is and provide an answer.
    input: How do I change the limit on my credit card {{credit_card_number}}?
    output: How do I change the limit on my credit card 2539 3519 2345 1555?
    input: {anonymized_text}
    output:
    """
    return prompt