openai.alternatives.perplexity.py

import requests

perplexity_api_key = os.getenv('PERPLEXITY_API_KEY_1')

purl = "https://api.perplexity.ai/chat/completions"
pheaders = {
    "accept": "application/json",
    "authorization": f'Bearer {perplexity_api_key}',
    "Content-Type": "application/json"
}

def get_perplexity_completion(prompt, model="llama-2-70b-chat"):
    messages = [{"role": "user", "content": prompt}]
    data = {
        "messages": messages,
        "model": model,
        "temperature": 0.2
    }
    perplexity_response = requests.post(purl, headers=pheaders, json=data)
    return perplexity_response.json()["choices"][0]["message"]["content"]