Then load the environment variables in your Python code:
Copy
from dotenv import load_dotenvimport os# Load environment variables from .env fileload_dotenv()# Set up environment variables with fallback valuesos.environ["XAI_API_KEY"] = os.getenv("XAI_API_KEY")os.environ["AGENTOPS_API_KEY"] = os.getenv("AGENTOPS_API_KEY")
Initialize AgentOps at the beginning of your application. Then, use the OpenAI SDK with xAI’s base URL to interact with Grok. AgentOps will automatically track all API calls.
Copy
import osimport agentopsfrom openai import OpenAI# Initialize AgentOpsagentops.init()# Create OpenAI client configured for xAIclient = OpenAI( api_key=os.getenv("XAI_API_KEY"), base_url="https://api.x.ai/v1",)# Basic chat completioncompletion = client.chat.completions.create( model="grok-3-latest", messages=[ {"role": "system", "content": "You are a helpful AI assistant."}, {"role": "user", "content": "Explain the concept of AI observability in simple terms."}, ],)print(completion.choices[0].message.content)