import openai
client = openai.OpenAI(
# 커스텀 base URL은 W&B Inference를 가리킵니다.
base_url='https://api.inference.wandb.ai/v1',
# https://wandb.ai/settings 에서 API 키를 생성하세요.
api_key="<your-api-key>",
# 선택 사항: 사용량 추적을 위한 Team 및 프로젝트
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)