llamacloud_managed_index_rag_query_quickstart.py

1import os
2from llama_index.indices.managed.llama_cloud import LlamaCloudIndex
3from llama_index.core import Settings
4
5# Ensure you have your API keys set in your environment variables
6# os.environ["LLAMA_CLOUD_API_KEY"] = "your-llama-cloud-api-key"
7# os.environ["OPENAI_API_KEY"] = "your-openai-api-key"
8
9# Define the index parameters
10index_name = "quickstart-index"
11project_name = "default"
12organization_id = None  # Optional: specific org ID if applicable
13
14# 1. Create or connect to a Managed Index
15# This will create the index on LlamaCloud if it doesn't exist
16index = LlamaCloudIndex(
17    name=index_name,
18    project_name=project_name,
19    organization_id=organization_id,
20    api_key=os.environ["LLAMA_CLOUD_API_KEY"]
21)
22
23# 2. Setup query engine
24# LlamaCloud handles the retrieval; by default it uses OpenAI for the synthesis
25query_engine = index.as_query_engine()
26
27# 3. Execute a query
28response = query_engine.query("What are the main features of LlamaCloud?")
29
30print(str(response))