🚀 Rax AI Python SDK

Install

pip install rax-ai

Quick Start

from rax_ai import RaxAI
rax = RaxAI(api_key="your-api-key")
response = rax.chat(
 model="rax-4.0",
 messages=[
 {"role": "user", "content": "Hello!"}
 ]
)
print(response["choices"][0]["message"]["content"])

Features

Configuration

from rax_ai import RaxAI
rax = RaxAI(
 api_key="your-key",
 base_url="https://ai.raxcore.dev/api", # default
 timeout=30 # seconds, default
)

Core Methods

Chat Completions

response = rax.chat(
 model="rax-4.0",
 messages=[
 {"role": "system", "content": "You are a helpful assistant."},
 {"role": "user", "content": "Explain quantum computing"}
 ],
 max_tokens=150,
 temperature=0.7,
 top_p=0.9
)
print(response["choices"][0]["message"]["content"])

Streaming Chat

for chunk in rax.chat_stream(
 model="rax-4.0",
 messages=[{"role": "user", "content": "Tell me a story..."}]
):
 print(chunk, end="", flush=True)

Models Management

# Get available models
models = rax.get_models()
print("Available models:", [m["id"] for m in models["data"]])
# Validate API key
is_valid = rax.validate_key()
print("Key valid:", is_valid)

Usage Analytics

# Get current usage
usage = rax.get_usage()
print("Total requests:", usage["total_requests"])
print("Total tokens:", usage["total_tokens"])
# Get usage for date range
monthly_usage = rax.get_usage(
 start_date="2024-01-01", 
 end_date="2024-01-31"
)
print("Monthly breakdown:", monthly_usage["daily_breakdown"])

Error Handling

from rax_ai import RaxAI, RaxAIError
try:
 response = rax.chat(
 model="rax-4.0",
 messages=[{"role": "user", "content": "Hello!"}]
 )
except RaxAIError as e:
 print(f"API Error: {e.message}")
 print(f"Status: {e.status_code}")
 print(f"Type: {e.error_type}")
 print(f"Details: {e.to_dict()}")

Environment Variables

# .env
RAX_AI_API_KEY=your-key-here

import os
from rax_ai import RaxAI
rax = RaxAI(api_key=os.getenv("RAX_AI_API_KEY"))

Framework Integration

FastAPI

from fastapi import FastAPI
from rax_ai import RaxAI
app = FastAPI()
rax = RaxAI(api_key=os.getenv("RAX_AI_API_KEY"))
@app.post("/chat")
async def chat_endpoint(message: str):
 response = rax.chat(
 model="rax-4.0",
 messages=[{"role": "user", "content": message}]
 )
 return response

Flask

from flask import Flask, request, jsonify
from rax_ai import RaxAI
app = Flask(__name__)
rax = RaxAI(api_key=os.getenv("RAX_AI_API_KEY"))
@app.route("/chat", methods=["POST"])
def chat():
 message = request.json["message"]
 response = rax.chat(
 model="rax-4.0",
 messages=[{"role": "user", "content": message}]
 )
 return jsonify(response)

Django

# views.py
from django.http import JsonResponse
from rax_ai import RaxAI
import json
rax = RaxAI(api_key=settings.RAX_AI_API_KEY)
def chat_view(request):
 data = json.loads(request.body)
 response = rax.chat(
 model="rax-4.0",
 messages=[{"role": "user", "content": data["message"]}]
 )
 return JsonResponse(response)

Advanced Usage

Configuration Management

# Get current config
config = rax.get_config()
print("Base URL:", config["base_url"])
print("Timeout:", config["timeout"])

Retry Logic

Python Version Support

Get Your API Key

Type Hints

from rax_ai.types import ChatMessage, ChatRequest, ChatResponse
messages: List[ChatMessage] = [
 {"role": "user", "content": "Hello!"}
]
response: ChatResponse = rax.chat(
 model="rax-4.0",
 messages=messages
)

Support