Skip to main content

A framework for AI engineering

Project description

Aixen: AI x Engineering

PyPI version

An open-source toolkit for the rapid development of AI-powered applications.

Features:

  • Clean Pipelines: Write AI pipelines as simple Python functions for readability and easy team collaboration.
  • Declarative: Use type annotations and Pydantic for function signatures, enabling auto-checks and code generation.
  • Tracking: Track and debug AI runs with statistics, performance evaluation, and API cost tracking. Gather fine-tuning samples.
  • Learnable Functions: (Coming soon) Optimize functions with hyperparameters automatically.

Quick Start

Install using pip:

pip install aixen

Add API keys to the environment or .env file in your working directory:

OPENAI_API_KEY=sk-xxx

A simple pipeline:

import aixen as ai
from pydantic import BaseModel

@ai.fn
def add(a: int, b: int) -> int:
    """
    Adds two numbers
    """
    return a + b


@ai.chat_fn
def factorial(n: int) -> int:
    """
    Calculates the factorial of a number
    """


class GreetingCard(BaseModel):
    text: str
    image_description: str


@ai.chat_fn
def greet(name: str) -> GreetingCard:
    """
    Creates a greeting media message for a person specified by the user.
    """


with ai.Context() as context:
    result = add(a=1, b=2)
    fact5 = factorial(5)
    card = greet(name="Lev")

    print(f"Greeting card: {card}")
    print(f"Used: ${context.usage_cost_usd}")

See the examples folder.

Community and Support

Join our community to stay updated and get support:

If you encounter any issues, please open an issue on GitHub.

Contributing

We welcome contributions from the community!

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

aixen-0.1.1.tar.gz (28.1 kB view hashes)

Uploaded Source

Built Distribution

aixen-0.1.1-py3-none-any.whl (23.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page