switchai 0.1.4

A unified library for interacting with various AI APIs through a standardized interface.

SwitchAI

SwitchAI is a lightweight and flexible library that provides a standardized interface for interacting with various AI APIs like OpenAI, Anthropic, Mistral, and more. With SwitchAI, you can easily switch between AI providers or use multiple APIs simultaneously, all with a simple and consistent interface.

Installation

Install with pip:

pip install switchai  

Getting Started

To use SwitchAI, you will need API keys for the AI providers you intend to interact with. You can set these keys either as environment variables or pass them as configuration to the SwitchAI client.

If you choose to use environment variables, ensure you follow the naming conventions for each provider as outlined in the documentation.

Example Usage

Chat Example

from switchai import SwitchAI

# Initialize the client with the desired AI model
client = SwitchAI("gpt-4o")

# Send a message and receive a response
response = client.chat(
    messages=[
        {"role": "user", "content": "Hello, how are you?"}
    ]
)

# Print the response
print(response)

Text Embedding Example

from switchai import SwitchAI

# Initialize the client with the chosen embedding model
client = SwitchAI("models/text-embedding-004")

# Generate embeddings for a list of text inputs
response = client.embed(
    input=[
        "I am feeling great today!",
        "I am feeling sad today."
    ]
)

# Print the response
print(response)

Documentation

For full documentation, visit SwitchAI Documentation.

Contributing

Contributions are always welcome! If you'd like to help enhance SwitchAI, feel free to make a contribution.