import openai
client = openai.OpenAI(
# The custom base URL points to Serverless Inference
base_url='https://api.inference.wandb.ai/v1',
# Create an API key at https://forge.coreweave.com/settings
api_key="[YOUR-API-KEY]",
# Optional: Team and project for usage tracking
project="[YOUR-TEAM]/[YOUR-PROJECT]",
)
response = client.chat.completions.create(
model="meta-llama/Llama-3.1-8B-Instruct",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Tell me a joke."}
],
)
print(response.choices[0].message.content)