Skip to main content

A unified toolkit for working with multiple LLM providers

Project description

llm-kit

llm-kit is a unified, async-first Python toolkit for interacting with multiple Large Language Model (LLM) providers through a consistent, provider-agnostic API.

It supports:

  • Multiple LLM providers (OpenAI, Gemini, Anthropic, etc.)
  • Text and structured JSON generation
  • Multimodal inputs (PDFs, images, text files)
  • Clean abstractions and strong typing

✨ Core Ideas

1. Outputs are explicit

LLMs generate outputs, not OCR results.

llm-kit exposes two primary operations:

  • generate_text → free-form text
  • generate_json → structured, schema-driven output

2. Inputs can be text and/or files

Files (PDFs, images, etc.) are first-class inputs and can be passed directly to generation methods.

OCR and file parsing are treated as provider implementation details.


📦 Installation

pip install llm-kit

Optional provider support (example):

pip install llm-kit[openai]
pip install llm-kit[gemini]

🚀 Quick Start

Text-only generation

text = await llm.generate_text(
    prompt="Explain what power factor is in simple terms"
)

Text generation with a file (PDF, image, etc.)

from llm_kit.core import LLMFile

pdf = LLMFile(
    content=pdf_bytes,
    mime_type="application/pdf",
    filename="bill.pdf",
)

summary = await llm.generate_text(
    prompt="Summarize this electricity bill",
    files=[pdf],
)

Structured JSON extraction

schema = {
    "type": "object",
    "properties": {
        "consumer_name": {"type": "string"},
        "bill_amount": {"type": "number"},
        "due_date": {"type": "string"},
    },
    "required": ["consumer_name", "bill_amount"],
}

data = await llm.generate_json(
    prompt="Extract billing details from this document",
    schema=schema,
    files=[pdf],
)

🧠 Design Philosophy

  • Provider-agnostic: No OpenAI/Gemini specifics in the public API
  • Async-first: Built for modern Python backends
  • Composable: Easy to plug into pipelines (FastAPI, background jobs, ETL)
  • Explicit contracts: Clear separation of inputs, outputs, and providers

🧩 Core Abstractions

BaseLLMClient

All providers implement the same interface:

class BaseLLMClient:
    async def generate_text(...)
    async def generate_json(...)

LLMFile

A provider-agnostic representation of file inputs:

from llm_kit.core import LLMFile

🔌 Providers

Each provider is implemented as an adapter that conforms to BaseLLMClient.

Supported / planned:

  • OpenAI
  • Gemini
  • Anthropic
  • Local / OSS models (future)

🛠 Development

Install dependencies

poetry install

Format & lint

poetry run ruff check . --fix
poetry run black .

Run tests

poetry run pytest

📍 Status

🚧 Under active development

The public API is stabilizing. Expect rapid iteration until v1.0.


📄 License

MIT License

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

llm_kit_pro-0.1.2.tar.gz (7.2 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

llm_kit_pro-0.1.2-py3-none-any.whl (9.4 kB view details)

Uploaded Python 3

File details

Details for the file llm_kit_pro-0.1.2.tar.gz.

File metadata

  • Download URL: llm_kit_pro-0.1.2.tar.gz
  • Upload date:
  • Size: 7.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.8.4 CPython/3.11.12 Darwin/25.1.0

File hashes

Hashes for llm_kit_pro-0.1.2.tar.gz
Algorithm Hash digest
SHA256 be7bf115e325391161df299d077efeeca2fde0693be642ef8f2c48be5845b815
MD5 107a2790cdfe15332138235beed8134a
BLAKE2b-256 f2944ca6c8cf39b5f2f987d2cf78611f2a4e7142c7f86578d8e3718093a18275

See more details on using hashes here.

File details

Details for the file llm_kit_pro-0.1.2-py3-none-any.whl.

File metadata

  • Download URL: llm_kit_pro-0.1.2-py3-none-any.whl
  • Upload date:
  • Size: 9.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.8.4 CPython/3.11.12 Darwin/25.1.0

File hashes

Hashes for llm_kit_pro-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 f80ac354ced8d466b18ccdf9b0ca0abdfd2e9a526f63a30a64e0dfc834dddbfa
MD5 f07c12bd2abfe587669877aee1eaa3ff
BLAKE2b-256 dea7989604d8eae7e770ba92883edba67621f1eb6576391b264853472097b9c2

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page