Python function that demonstrates how to use the openai package to generate responses from the OpenAI Language Model and then store those responses in a MySQL database

OpenAImySQLStore.py

import mysql.connector
import openai

def generate_text_with_llm(prompt):
    # OpenAI API configuration
    openai.api_key = "your_openai_api_key"

    # Generate text using OpenAI's LLM
    response = openai.Completion.create(
        engine="davinci",  # Choose the appropriate engine
        prompt=prompt,
        max_tokens=100  # Adjust the desired length of generated text
    )

    return response.choices[0].text.strip()

def store_response_in_mysql(prompt, generated_text):
    # MySQL database configuration
    db_config = {
        "host": "your_host",
        "user": "your_username",
        "password": "your_password",
        "database": "your_database"
    }
    
    # Connect to the MySQL database
    connection = mysql.connector.connect(**db_config)
    cursor = connection.cursor()

    # Store response in MySQL
    query = "INSERT INTO generated_responses (prompt, response) VALUES (%s, %s)"
    values = (prompt, generated_text)
    cursor.execute(query, values)

    # Commit changes and close the database connection
    connection.commit()
    cursor.close()
    connection.close()

if __name__ == "__main__":
    prompts = [
        "Once upon a time in a galaxy far, far away...",
        "In a world where magic is real and technology is ancient...",
        "The year is 2077, and humanity has colonized Mars..."
    ]

    for prompt in prompts:
        generated_text = generate_text_with_llm(prompt)
        store_response_in_mysql(prompt, generated_text)
        print("Stored response for prompt:", prompt)