Onyx GenAI SDK

The goal of this project is to simplify the developer experience when interacting with Onyx GenAI Services. This project provides wrappers around the the underlying APIs provided by the service.

Table of Contents

1. Installation
2. Using the SDK in Onyx
3. Running Unit Tests
4. Running Code Quality Checks

Installation

To install with pip, run:

pip install onyxgenai

To install with conda, run:

conda install -c conda-forge onyxgenai

Using the SDK in Onyx

1. Create a Conda Store Environment with all dependencies listed in the requirements.txt

2. Start your JupyterLab Server

3. Create a new Jupyter Notebook

4. Install the SDK as indicated above

5. Add the onyxgenai client imports to your project

For more in depth examples, see notebooks section of this repo.

Embedding Client

The Embedding Client provides access to the Onyx GenAI Embedding Service. The client provides access to functionality such as:

• Generating Text and Image Embeddings and Vector Storage
• Retrieving Vector Store Collections
• Vector Database Search

Model Client

The Model Client provides access to the Onyx GenAI Model Store Service. The client provides access to functionality such as:

• Retrieving Model Info
• Retrieving Active Model Deployment Info
• Deploying and Deleting Model Deployments
• Performing Text and Image Prediction and Embedding
• Generating Text Completions from an LLM

Running Unit Tests

1. To run unit tests, run the following:

pytest

Running Code Quality Checks

1. To run code quality checks, run the following:

ruff check .