LlamaIndex

import os
from llama_index.llms.openai import OpenAI
from llama_index.core import Settings

# Set your Adaptive API key
os.environ["OPENAI_API_KEY"] = os.environ["ADAPTIVE_API_KEY"]

# Initialize OpenAI LLM with Adaptive endpoint
llm = OpenAI(
    model="",  # Empty string enables intelligent routing
    api_base="https://api.llmadaptive.uk/v1",
    api_key=os.environ["ADAPTIVE_API_KEY"],
)

# Set as global LLM
Settings.llm = llm

# Use with simple queries
response = llm.complete("What is retrieval-augmented generation?")
print(response)

from llama_index.core import VectorStoreIndex, SimpleDirectoryReader

# Load documents
documents = SimpleDirectoryReader("data").load_data()

# Create index
index = VectorStoreIndex.from_documents(documents)

# Query with intelligent routing
query_engine = index.as_query_engine()
response = query_engine.query("What are the benefits of RAG?")
print(response)