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)