aymara-ai 0.5.2

A Python SDK for using Aymara to test your AI alignment.

Aymara Python SDK

Hi! 👋 We're Aymara. We built this library to help you measure & improve the alignment of any text-to-text genAI model (e.g., a fine-tuned Llama model) or application (e.g., a chatbot powered by GPT).

Use the Aymara REST API to create custom red team tests.

1. 🦺 Safety. Input text describing the content your genAI is(n't) allowed to generate & get a test to assess your genAI's compliance. (Text-to-image coming soon.)

2. 🧨 Jailbreaks. Input your genAI's system prompt & get a test to assess your genAI's ability to follow your instructions when tested across hundreds of jailbreaks.

3. 🎯 Accuracy (coming soon). Input text from the knowledge base your genAI should know & get a test to assess the accuracy (and hallucinations) of your genAI's answers.

And use the API to score your genAI's test answers, get detailed explanations of failing test answers, & receive specific advice to improve the safety & accuracy of your genAI.

API Access

The API is accessible in a free trial with limited functionality or as a paid service with full functionality.

Our Python SDK provides convenient access to the API from Python 3.9+. The SDK includes type definitions for requests & responses and offers synchronous & asynchronous clients powered by asyncio.

Documentation

docs.aymara.ai has our full library API & guides to walk you through safety tests (including the free trial version) & jailbreak tests.

Installation

Install the SDK with pip. We suggest using a virtual environment to manage dependencies.

pip install aymara-ai

Configuration

Get an Aymara API key and store it as an env variable:

export AYMARA_API_KEY=[AYMARA_API_KEY]

Or supply your key directly to the client:

client = AymaraAI(api_key="AYMARA_API_KEY")

Support and Requests

If you found a bug, have a question, or want to request a feature, reach out at [e-mail] or open an issue on our GitHub repo.

Versioning

This package generally follows SemVer conventions. Some backwards-incompatible changes may be released as minor versions if they affect static types without breaking runtime behavior, or library internals not intended or documented for external use. (Please open an issue if you are relying on internals).

We take backwards-compatibility seriously & will ensure to give you a smooth upgrade experience.

Requirements

Python 3.9 or higher.