Skip to main content

AutoRAG is a flexible and scalable solution for building Retrieval-Augmented Generation (RAG) systems.

Project description

AutoRAG

Powering seamless retrieval and generation workflows for our internal AI systems

Python Version Code Formatter Code Linter Code Checker Code Coverage

Overview

AutoRAG is a flexible and scalable solution for building Retrieval-Augmented Generation (RAG) systems.

This SDK provides out-of-the-box functionality for creating and managing retrieval-augmented generation workflows, offering a modular, highly-configurable interface. It supports multiple vector stores and leverages http clients like httpx for handling requests, ensuring seamless integration.

Features

  • Modular architecture: The SDK allows you to swap, extend, or customize components like retrieval models, vector stores, and response generation strategies.
  • High scalability: Built to handle large-scale data retrieval and generation, enabling robust, production-ready applications.
  • Celery for dependency injection: Efficient background tasks with support for distributed task execution.
  • Multi-flow support: Easily integrate various vector databases (ex: Qdrant, Azure AI Search) with various language models providers (ex: OpenAI, vLLM, Ollama) using standardized public methods for seamless development.

Installation

  1. Create a virtual environment, we recommend Miniconda for environment management:
    conda create -n autorag python=3.12
    conda activate autorag
    
  2. Install the package:
    pip install autonomize-autorag
    

Usage

The full set of API can be found in api.md

import os
from autorag.language_models import OpenAI

client = OpenAI(
    api_key=os.environ.get("OPENAI_API_KEY"),
)

generation = client.generate(
    message="What is GPT?"
    model="gpt-4o"
)

Contribution

To contribute in our AutoRAG SDK, please refer to our Contribution Guidelines.

License

Copyright (C) Autonomize AI - All Rights Reserved

This file is part of this project.

This project can not be copied and/or distributed without the express permission of Autonomize AI.

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

autonomize_autorag-0.1.6.tar.gz (6.2 kB view details)

Uploaded Source

Built Distribution

autonomize_autorag-0.1.6-py3-none-any.whl (9.2 kB view details)

Uploaded Python 3

File details

Details for the file autonomize_autorag-0.1.6.tar.gz.

File metadata

  • Download URL: autonomize_autorag-0.1.6.tar.gz
  • Upload date:
  • Size: 6.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.8.3 CPython/3.12.5 Linux/6.5.0-1025-azure

File hashes

Hashes for autonomize_autorag-0.1.6.tar.gz
Algorithm Hash digest
SHA256 2311e6b5672e8dcfcf940ef2c5e5ac5d2cc9dc8fdc9eb1127aa15020a48c6b18
MD5 61ef322ace35d736a6cf99fc46e1b9a4
BLAKE2b-256 031cd60c7348ce5615d7b3e4dc81c91527e45b3f76f8d36eef4a879ad03c657f

See more details on using hashes here.

File details

Details for the file autonomize_autorag-0.1.6-py3-none-any.whl.

File metadata

File hashes

Hashes for autonomize_autorag-0.1.6-py3-none-any.whl
Algorithm Hash digest
SHA256 efea1cec7bf0e4acc08aa35cec53869506aeee93a215782834ae3c618ab4fbc7
MD5 1aef95589973e7836eba80a95e8869ae
BLAKE2b-256 12f98f5ca5b03b62bad4f8a3886ffe3f859f4d2ecbdea3b5b2299ea177f7bf46

See more details on using hashes here.

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