AB Testing using the endpoint in Python

import requests
import nebuly
nebuly.init("your_nebuly_api_key")
import openai

# Set up the OpenAI client
openai.api_key = "your_openai_api_key"


def cast_type(value: str, type: str) -> str:
    if type == "string":
        return value
    elif type == "int":
        return int(value)
    elif type == "float":
        return float(value)
    elif type == "bool":
        return value.lower() == "true"
    else:
        raise ValueError(f"Unknown type: {type}")


def generate_response(user_id: str, messages: List[Dict[str, str]]) -> str:
    
    # Get the variant for the user
    variant = requests.post(
        "https://backend.nebuly.com/api/external/variants",
        json={
            "user": user_id,
            "feature_flags": ["{experiment_id}"]
        },
        headers={"Authorization": f"Bearer {nebuly_api_key}"}
    )

    # Prepare the messages
    system_message = {"role": "system", "content": "You are a helpful assistant."}
    messages = [system_message] + messages

    # Manage the variant and edit the messages if needed
    variant_list = variant.json()["variants"]
    extra_params = {}
    for variant in variant_list:
        if variant["kind"] == "prompt":
            messages.append({"role": "system", "content": variant["prompt"]})
        config_params = variant["config_params"]
        for config_param in config_params:
            extra_params[config_param["name"]] = cast_type(config_param["value"], config_param["type"])

    # Generate the response using OpenAI
    response = openai.chat.completions.create(
        model="gpt-3.5-turbo",
        messages=messages,
        # nebuly params
        feature_flags=["{experiment_id}"]
        # params needed by the variant
        **extra_params
    )

    return response.choices[0].message['content']