from openai import OpenAI
from portkey_ai import PORTKEY_GATEWAY_URL, createHeaders
from google.colab import userdata
client = OpenAI(
    api_key= userdata.get('DEEPINFRA_API_KEY'), ## replace it your Mistral API key
    base_url=PORTKEY_GATEWAY_URL,
    default_headers=createHeaders(
        provider="deepinfra",
        api_key= userdata.get('PORTKEY_API_KEY'), ## replace it your Portkey API key
    )
)
chat_complete = client.chat.completions.create(
    model="meta-llama/Meta-Llama-3-70B-Instruct",
    messages=[{"role": "user",
               "content": "Who are you?"}],
)
print(chat_complete.choices[0].message.content)