xAI

import os
from langtrace_python_sdk import langtrace # Must precede any llm module imports
langtrace.init(api_key = os.environ['LANGTRACE_API_KEY'])
from openai import OpenAI
client = OpenAI(
  # This is the default and can be omitted
  api_key=os.environ["XAI_API_KEY"],
  base_url="https://api.x.ai/v1",
)

# Generate a simple output with the llama3 MODEL

chat_completion = client.chat.completions.create(
  messages=[
      {
          "role": "user",
          "content": "What is LangChain?",
      }
  ],
  model="grok-beta",
)
print(chat_completion.choices[0].message.content)