strata-forge

Typed, async-first toolkit for AI/LLM work — inference, evaluation, fine-tuning, retrieval, and remote compute orchestration across the major foundation-model providers.

Built on LiteLLM for provider breadth and Pydantic v2 for strict typing. The public API is async, with sync wrappers in strata_forge.sync for CLI and notebook ergonomics. Heavy optional dependencies (training, serving, remote compute) live behind extras, so the base install stays light.

Install

pip install strata-forge

Requires Python 3.14+.

Optional extras pull in the heavy dependencies only when you need them:

pip install "strata-forge[rag]"        # embeddings, vector store, reranking
pip install "strata-forge[compute]"    # SkyPilot + SSH remote execution
pip install "strata-forge[serving]"    # vLLM
pip install "strata-forge[finetuning]" # torch, transformers, trl, peft
pip install "strata-forge[all]"        # everything

Full set: redis, multimodal, inspect, langfuse, hf, evals, rag, compute, serving, bedrock, storage, finetuning, all.

Quickstart

import asyncio

from strata_forge.llm import LLMClient, Message


async def main() -> None:
    client = LLMClient("gpt-5.5", provider="openai")
    response = await client.complete([Message.user("Say hello in one short sentence.")])
    print(response.text)


asyncio.run(main())

Provider credentials are read from the environment (for example OPENAI_API_KEY). A CLI is installed as strata-forge.

What's in it

Module Purpose
strata_forge.llm Provider abstraction, structured output, tool calling, two-axis fallback
strata_forge.core Cross-cutting errors, retry, logging, budget ceilings, reproducibility
strata_forge.config Pydantic Settings root with YAML overlays and .env support
strata_forge.prompts Jinja2 and Langfuse-backed prompt registry
strata_forge.tracing Langfuse observability
strata_forge.datasets Dataset CRUD with a Hugging Face bridge
strata_forge.evals Experiment runner, graders, metrics
strata_forge.agents Agent runtime, memory, multi-agent composition
strata_forge.rag Embed, chunk, store, retrieve, rerank
strata_forge.storage fsspec gateway and Hugging Face Hub
strata_forge.compute SkyPilot and SSH backends, serving, batch inference
strata_forge.training SFT / DPO / ORPO / KTO / GRPO with PEFT
strata_forge.pipelines Runnable entrypoints executed on a compute target

Documentation

Reference docs live in the repository:

Development

uv sync        # materialize the venv + lockfile
make check     # lint (ruff) + type-check (pyright strict)
make test      # unit tests

License

Apache License 2.0.

Project links

Data verified by PyPI on Aug 27, 2026
Data provided by the project maintainers, verified at the time the release was uploaded to PyPI.

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Latest release

1 maintainer

PyPI data
Data sourced directly from PyPI's database.
Avatar for Gemovic from gravatar.com Gemovic

Credits

Author: Nemanja Gemovic

GitHub Statistics

Data verified by PyPI on Aug 27, 2026
The GitHub source repository was provided by the project maintainers and verified by PyPI at the time of upload. Stars, forks, and open issues/PRs are derived from that repository and have not been independently verified.

License expression

Apache-2.0
View SPDX License List

Requires

Python >=3.14

Provides Extra

all bedrock compute evals finetuning hf inspect langfuse multimodal rag redis serving storage

Tags

agents ai async evaluation fine-tuning llm rag

Classifiers

Report project as malware

Download files

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

Source Distribution

strata_forge-0.2.0.tar.gz (253.0 kB view details)

Uploaded Source

Built Distribution

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

strata_forge-0.2.0-py3-none-any.whl (345.7 kB view details)

Uploaded Python 3

File details

Details for the file strata_forge-0.2.0.tar.gz.

File metadata

  • Download URL: strata_forge-0.2.0.tar.gz
  • Upload date:
  • Size: 253.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for strata_forge-0.2.0.tar.gz
Algorithm Hash digest
SHA256 c57c70ce4fcfa6851bd0cc424cd41237cedbd155bce0a7051f40b565c5872aed
MD5 31521140cb47903e6e98bdb1628ac4a8
BLAKE2b-256 c6222f250ac98e657dd36464c07ee770fc22fb0e38c6615e0ec41e3196504bf3

See more details on using hashes here.

Provenance

The following attestation bundles were made for strata_forge-0.2.0.tar.gz:

Publisher: release.yml on GemovicNemanja/strata-forge

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file strata_forge-0.2.0-py3-none-any.whl.

File metadata

  • Download URL: strata_forge-0.2.0-py3-none-any.whl
  • Upload date:
  • Size: 345.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for strata_forge-0.2.0-py3-none-any.whl
Algorithm Hash digest
SHA256 c8f758ed8a83e2be9e529c1598e885e0ac773926620684aa3c18eebb70cb3513
MD5 490f27e6dbe3711ea31fbab9ed85ffca
BLAKE2b-256 3b08c2d3352830a28ed808ae2e2b5f9447eafcbdf888302fbcbb9838835d131e

See more details on using hashes here.

Provenance

The following attestation bundles were made for strata_forge-0.2.0-py3-none-any.whl:

Publisher: release.yml on GemovicNemanja/strata-forge

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

This release

0.2.0 This release

2 files

0.1.1

2 files

0.1.0

2 files

0.0.1

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page