csai-helper 0.1.2

Lightweight helpers for AI prompts, summarization, and formatting.

csai-helper

Lightweight helpers for AI prompting, summarization, and response formatting. Ships with a pluggable provider interface and optional OpenAI provider.

Installation

From source (editable):

pip install -e .[dev]

For runtime only (from PyPI):

pip install csai-helper

Optional OpenAI support:

pip install "csai-helper[openai]"

Google Gemini support (requires GOOGLE_API_KEY and a model such as gemini-1.5-flash):

export GOOGLE_API_KEY=your-key
ai-helper --provider gemini --model gemini-1.5-flash ask "Hi"

Quickstart

from ai_helper import get_response, summarize_text, format_response
from ai_helper.providers.local import LocalProvider
from ai_helper.providers.gemini import GeminiProvider

provider = LocalProvider()
print(get_response("Hello, AI!", provider=provider))

gemini = GeminiProvider()
print(get_response("Say hi in 3 words", provider=gemini))

text = """Python is a popular programming language created by Guido van Rossum.
It is used for web development, data science, automation, and more."""
print(summarize_text(text, provider=provider, max_words=20))

print(format_response("  Hello\n\nWorld  "))

CLI

ai-helper ask "Tell me a joke"
ai-helper summarize "Long text here..." --max-words 30
ai-helper --provider gemini --model gemini-pro ask "Send me a haiku"

Set OPENAI_API_KEY in your environment to use the OpenAI provider:

export OPENAI_API_KEY=sk-...
ai-helper --provider openai --model gpt-4o-mini ask "Explain transformers in 2 sentences"

API

• get_response(prompt, provider=None, **kwargs) → str
• summarize_text(text, provider=None, max_words=120, **kwargs) → str
• format_response(text) → str

Providers implement AIProvider.get_response(prompt, system_prompt=None, **kwargs).

Development

Run formatting and tests:

ruff check .
pytest

Publishing

1. Build distributions:

   python -m build
   

2. Upload to PyPI (requires twine):

   python -m pip install --upgrade twine
   twine upload dist/*
   

License

MIT