Using your AI-VERDE API key to integrate with LlamaIndex¶
Prerequisites¶
- Obtain your AI-VERDE API key. Instructions can be found here.
- Note the model name you want to use. Instructions for listing available models can be found here.
- Install Python 3 with
pip.
1. Install LlamaIndex libraries¶
pip install llama-index-core llama-index-llms-litellm
2. Write a Python script¶
Replace [MODEL NAME] with a model name from your model list, keeping the litellm_proxy/ prefix (e.g. litellm_proxy/gemma-4-26b-it). LiteLLM's litellm_proxy provider takes the base URL without /v1.
from llama_index.llms.litellm import LiteLLM
from llama_index.core.llms import ChatMessage
llm = LiteLLM(
model="litellm_proxy/[MODEL NAME]",
api_base="https://llm-api.cyverse.ai",
api_key="[AI-VERDE API KEY]",
)
message = ChatMessage(role="user", content="Hey! how's it going?")
response = llm.chat([message])
print(response)
Alternatively, you can include the API key as an environment variable or secret to avoid storing it in plain text:
import getpass
import os
if not os.environ.get("VERDE_API_KEY"):
os.environ["VERDE_API_KEY"] = getpass.getpass("Enter AI-VERDE API key: ")
api_key = os.environ["VERDE_API_KEY"]
from llama_index.llms.litellm import LiteLLM
from llama_index.core.llms import ChatMessage
llm = LiteLLM(
model="litellm_proxy/[MODEL NAME]",
api_base="https://llm-api.cyverse.ai",
api_key=api_key,
)
message = ChatMessage(role="user", content="Hey! how's it going?")
response = llm.chat([message])
print(response)
Note
LlamaIndex embedding support is outlined here, but functionality depends on access to an embedding model through AI-VERDE.