arize-ai/arize-skills

arize-ai-provider-integration

Creates, reads, updates, and deletes Arize AI integrations that store LLM provider credentials used by evaluators and other Arize features.

Ver código fuente
Documento original del Skill

Contenido del repositorio de origen con títulos, ejemplos, código, tablas, enlaces e imágenes preservados.

Arize AI Integration Skill

`SPACE` — --space flags accept a space name (e.g., my-workspace) or a base64 space ID (e.g., U3BhY2U6...). Find yours with ax spaces list. Note: ai-integrations create does not accept --space — AI integrations are account-scoped. Use --space only with list, get, update, and delete.

Concepts

  • AI Integration = stored LLM provider credentials registered in Arize; used by evaluators to call a judge model and by other Arize features that need to invoke an LLM on your behalf
  • Provider = the LLM service backing the integration (e.g., OPEN_AI, ANTHROPIC, AWS_BEDROCK)
  • Integration ID = a base64-encoded global identifier for an integration (e.g., TGxtSW50ZWdyYXRpb246MTI6YUJjRA==); required for evaluator creation and other downstream operations
  • Scoping = visibility rules controlling which spaces or users can use an integration
  • Auth type = how Arize authenticates with the provider: DEFAULT (provider API key), PROXY_WITH_HEADERS (proxy via custom headers), or BEARER_TOKEN (bearer token auth)

Prerequisites

Proceed directly with the task — run the ax command you need. Do NOT check versions, env vars, or profiles upfront.

If an ax command fails, troubleshoot based on the error:

  • command not found or version error → see references/ax-setup.md
  • 401 Unauthorized / missing API key → run ax profiles show to inspect the current profile. If the profile is missing or the API key is wrong, follow references/ax-profiles.md to create/update it. If the user doesn't have their key, direct them to https://app.arize.com/admin > API Keys
  • Space unknown → see Space
  • LLM provider call fails (missing provider credentials) → run ax ai-integrations list --space SPACE to check for platform-managed credentials. If none exist:
  • Preferred: Give the user the exact ax ai-integrations create command from Supported Providers below (reference env var names like $OPENAI_API_KEY, never raw values). Ask them to export the provider key in their own terminal and run the command there — never paste the key into chat. Do not run the create command yourself unless the var is already exported in this terminal session.
  • Bedrock/Vertex: use --provider-metadata instead of --api-key (see Supported Providers below).
  • Security: Never read .env files or search the filesystem for credentials. Exception: the non-secret ARIZE_SPACE_ID line (see Space). Use ax profiles for Arize credentials and ax ai-integrations for LLM provider keys. Never ask the user to paste secrets into chat. For missing credentials, see references/ax-profiles.md.

List AI Integrations

List all integrations accessible in a space:

bash
ax ai-integrations list --space SPACE

Filter by name (case-insensitive substring match):

bash
ax ai-integrations list --space SPACE --name "openai"

Paginate large result sets:

bash
# Get first page
ax ai-integrations list --space SPACE --limit 20 -o json

# Get next page using cursor from previous response
ax ai-integrations list --space SPACE --limit 20 --cursor CURSOR_TOKEN -o json

Key flags:

FlagDescription
--spaceSpace name or ID to filter integrations
--nameCase-insensitive substring filter on integration name
--limitMax results (1–100, default 15)
--cursorPagination token from a previous response
-o, --outputOutput format: table (default) or json

Response fields:

FieldDescription
idBase64 integration ID — copy this for downstream commands
nameHuman-readable name
providerLLM provider enum (see Supported Providers below)
has_api_keytrue if credentials are stored
model_namesAllowed model list, or null if all models are enabled
enable_default_modelsWhether default models for this provider are allowed
function_calling_enabledWhether tool/function calling is enabled
auth_typeAuthentication method: DEFAULT, PROXY_WITH_HEADERS, or BEARER_TOKEN

Get a Specific Integration

bash
ax ai-integrations get NAME_OR_ID
ax ai-integrations get NAME_OR_ID -o json
ax ai-integrations get NAME_OR_ID --space SPACE   # required when using name instead of ID

Use this to inspect an integration's full configuration or to confirm its ID after creation.


Create an AI Integration

Before creating, always list integrations first — the user may already have a suitable one:

bash
ax ai-integrations list --space SPACE

If no suitable integration exists, create one. The required flags depend on the provider.

Every create command follows this shape; the provider-specific flags come from the table below:

bash
ax ai-integrations create \
  --name "My OpenAI Integration" \
  --provider OPEN_AI \
  --api-key $OPENAI_API_KEY \
  --enable-default-models

Every integration except LiteLLM, Fireworks AI and Together AI needs a model source: pass --enable-default-models (Arize's default model list for the provider) and/or one or more --model-name flags. Without either, the server rejects the create with 400: An integration must have at least one model available.

For a provider's full example and notes (Bedrock/Vertex metadata, base URLs, flags to avoid), open only that provider's file from the Details column below.

Supported Providers

ProviderRequired extra flags ("model source" = --enable-default-models or --model-name)Key env varDetails
OPEN_AI--api-key <key>, model source$OPENAI_API_KEYprovider-openai.md
ANTHROPIC--api-key <key>, model source$ANTHROPIC_API_KEYprovider-anthropic.md
AZURE_OPEN_AI--api-key <key>, --base-url <azure-endpoint>, --model-name <deployment>$AZURE_OPENAI_API_KEYprovider-azure-openai.md
AWS_BEDROCK--provider-metadata '{"role_arn": "<arn>"}' (no API key), model source—provider-aws-bedrock.md
VERTEX_AI--provider-metadata '{"project_id": "<gcp-project>", "location": "<region>", "project_access_label": "<label>"}' (no API key), model source—provider-vertex-ai.md
GEMINI--api-key <key>, model source$GEMINI_API_KEYprovider-gemini.md
NVIDIA_NIMmodel source; --api-key and --base-url optional (default: NVIDIA hosted)$NVIDIA_API_KEYprovider-nvidia-nim.md
CUSTOM--base-url <endpoint>, model source$CUSTOM_LLM_API_KEYprovider-custom.md
LITELLM--base-url <endpoint>, --api-key <key>$LITELLM_API_KEYprovider-litellm.md
FIREWORKS--api-key <key> only — no --base-url or --headers$FIREWORKS_API_KEYprovider-fireworks.md
TOGETHER_AI--api-key <key> only — no --base-url or --headers$TOGETHER_API_KEYprovider-together-ai.md

Optional flags for any provider

FlagDescription
--model-nameAllowed model name (repeat for multiple, e.g. --model-name gpt-4o --model-name gpt-4o-mini); satisfies the model-source requirement
--enable-default-modelsEnable Arize's default model list for the provider; satisfies the model-source requirement
--function-calling-enabledEnable tool/function calling support
--auth-typeAuthentication type: DEFAULT, PROXY_WITH_HEADERS, BEARER_TOKEN, or OAUTH2_CLIENT_CREDENTIALS
--headersCustom headers as JSON object or file path (for proxy auth)
--provider-metadataProvider-specific metadata as JSON object or file path

After creation

Capture the returned integration ID (e.g., TGxtSW50ZWdyYXRpb246MTI6YUJjRA==) — it is needed for evaluator creation and other downstream commands. If you missed it, retrieve it:

bash
ax ai-integrations list --space SPACE -o json
# or by name/ID directly:
ax ai-integrations get NAME_OR_ID

Update an AI Integration

update is a partial update — only the flags you provide are changed. Omitted fields stay as-is.

bash
# Rename
ax ai-integrations update NAME_OR_ID --name "New Name"

# Rotate the API key
ax ai-integrations update NAME_OR_ID --api-key $OPENAI_API_KEY

# Change the model list (replaces all existing model names)
ax ai-integrations update NAME_OR_ID --model-name gpt-4o --model-name gpt-4o-mini

# Update base URL (for Azure, custom, or NIM)
ax ai-integrations update NAME_OR_ID --base-url "https://new-endpoint.example.com/v1"

# Restrict visibility to specific spaces (full replace — lists all spaces that should have access)
ax ai-integrations update NAME_OR_ID \
  --scopings '[{"space_id": "SPACE_GLOBAL_ID", "scoping_type": "include"}]'

Add --space SPACE when using a name instead of ID. Any flag accepted by create can be passed to update.

`--scopings` flag: Controls which spaces can use this integration. Accepts a JSON array of scoping rules. Replaces all existing scopings on update. Use ax spaces list -o json to find space global IDs.


Delete an AI Integration

Warning: Deletion is permanent. Evaluators that reference this integration will no longer be able to run.

bash
ax ai-integrations delete NAME_OR_ID --force
ax ai-integrations delete NAME_OR_ID --space SPACE --force   # required when using name instead of ID

Omit --force to get a confirmation prompt instead of deleting immediately.


Troubleshooting

ProblemSolution
ax: command not foundSee references/ax-setup.md
401 UnauthorizedAPI key may not have access to this space. Verify key and space ID at https://app.arize.com/admin > API Keys
No profile foundRun ax profiles show --expand; set ARIZE_API_KEY env var or write ~/.arize/config.toml
Integration not foundVerify with ax ai-integrations list --space SPACE
has_api_key: false after createCredentials were not saved — re-run update with the correct --api-key or --provider-metadata
Evaluator runs fail with LLM errorsCheck integration credentials with ax ai-integrations get INT_ID; rotate the API key if needed
Invalid value for '--provider' with FIREWORKS or TOGETHER_AIThe installed ax CLI predates these providers — upgrade the ax CLI (see references/ax-setup.md)
An integration must have at least one model availableAdd --enable-default-models or at least one --model-name (required for every provider except LiteLLM, Fireworks AI and Together AI)
Integration has the wrong providerax ai-integrations update NAME_OR_ID --provider NEW_PROVIDER changes it, but other fields are kept as-is — pass the new provider's required flags (API key, base URL, model source) in the same call. When the two providers need very different settings, deleting and recreating is cleaner

Related Skills

  • arize-evaluator: Create LLM-as-judge evaluators that use an AI integration → use arize-evaluator
  • arize-experiment: Run experiments that use evaluators backed by an AI integration → use arize-experiment

Save Credentials for Future Use

See references/ax-profiles.md § Save Credentials for Future Use.

del mismo repositorio

Más Skills

Todos los Skills
arize-ai
Comunidad

arize-prompt-optimization

Optimizes, improves, and debugs LLM prompts using production trace data, evaluations, and annotations. Extracts prompts from spans, gathers performance signal, and runs a data-driven optimization loop using the ax CLI. Use when the user mentions optimize prompt, improve prompt, make AI respond better, improve output quality, prompt engineering, prompt tuning, or system prompt improvement.

instalaciones
2
GitHub Stars
63
Actualizado
14 oct
arize-ai
Comunidad

arize-admin

Manages Arize users, organizations, spaces, projects, roles, role bindings, resource restrictions, and API keys via the ax CLI. Use for enterprise admin workflows: inviting and offboarding users, onboarding new teams, creating custom roles for SAML/SSO mappings, assigning roles to users, restricting project-level access, and managing service keys for multi-tenant architectures. Covers ax users, ax organizations, ax spaces, ax projects, ax roles, ax role-bindings, and ax api-keys.

instalaciones
3
GitHub Stars
58
Actualizado
7 oct
arize-ai
Comunidad

arize-annotation

Creates and manages annotation configs (categorical, continuous, freeform label schemas) and annotation queues (human review workflows) on Arize. Applies human annotations to project spans via the Python SDK. Use when the user mentions annotation config, annotation queue, label schema, human feedback, bulk annotate spans, updateannotations, labeling queue, annotate record, or human review.

instalaciones
3
GitHub Stars
58
Actualizado
7 oct
arize-ai
Comunidad

arize-compliance-audit

INVOKE THIS SKILL when auditing an AI agent or LLM app for regulatory compliance. Covers EU AI Act, GPAI Code of Practice, GDPR, NIST AI RMF, Colorado AI Act, HIPAA, and ISO 42001. Scans the codebase for compliance gaps, cross-references Arize instrumentation for audit trail coverage, and produces an actionable remediation checklist tailored to the selected frameworks.

instalaciones
3
GitHub Stars
58
Actualizado
7 oct