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)