ecologits 0.1.2

EcoLogits tracks and estimates the energy consumption and environmental impacts of using generative AI models through APIs.

EcoLogits

EcoLogits tracks and estimates the energy consumption and environmental impacts of using generative AI models through APIs.

⚙️ Installation

pip install ecologits

🚀 Usage

from ecologits import EcoLogits
from openai import OpenAI

EcoLogits.init()

client = OpenAI(
    api_key="<OPENAI_API_KEY>",
)

response = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "user", "content": "Tell me a funny joke!"}
    ]
)

# Get estimated environmental impacts for that inference.
print(response.impacts)  # Impacts(energy=0.025, energy_unit='Wh', ...)

See package documentation on EcoLogits

💪 Contributing

Requirements

Have poetry installed on your system.

Easy install using a virtualenv:

Create a venv:

python3 -m venv .venv
source .venv/bin/activate

Install poetry:

pip install poetry

Install project

poetry install --all-extras --with dev,docs

Run tests

poetry run pytest

Run pre-commit hooks locally

Install pre-commit

pre-commit run --all-files