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

Documentation: ecologits.ai

⚙️ Installation

pip install ecologits

For integration with a specific provider, use pip install ecologits[openai]. We are currently supporting the following providers: anthropic, cohere, google-generativeai, huggingface-hub, mistralai and openai. See the full list of providers.

🚀 Usage

from ecologits import EcoLogits
from openai import OpenAI

# Initialize EcoLogits
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 of the inference
print(f"Energy consumption: {response.impacts.energy.value} kWh")
print(f"GHG emissions: {response.impacts.gwp.value} kgCO2eq")

See package documentation on EcoLogits

💪 Contributing

To get started with setting up a development environment and making a contribution to EcoLogits, see Contributing to EcoLogits.

⚖️ License

This project is licensed under the terms of the Mozilla Public License Version 2.0 (MPL-2.0).