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 13, 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:
Newer release available (0.2.0)

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 13, 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.1.1.tar.gz (224.1 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.1.1-py3-none-any.whl (311.3 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: strata_forge-0.1.1.tar.gz
  • Upload date:
  • Size: 224.1 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.1.1.tar.gz
Algorithm Hash digest
SHA256 571cd254addb0c37831527cc645bb4cceff5f65eba3fdccfb9e6768a0bdf368e
MD5 00be214972fac302502f09636ba9263a
BLAKE2b-256 9b1739007a7673841c965c1a2f57363338b217cabd561fc76663817980af33f8

See more details on using hashes here.

Provenance

The following attestation bundles were made for strata_forge-0.1.1.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.1.1-py3-none-any.whl.

File metadata

  • Download URL: strata_forge-0.1.1-py3-none-any.whl
  • Upload date:
  • Size: 311.3 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.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 d2f6e4c4574c04a046a369f22f13eb426c4a01ab74dfecc1de77a44558a50c7c
MD5 1af3a248ad27f61df1ef36b8e5ad0bdc
BLAKE2b-256 3dd40cd2b5e49be1edb53c2f36832b36e8f17883ad7adf9e64ede7526a47d094

See more details on using hashes here.

Provenance

The following attestation bundles were made for strata_forge-0.1.1-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

0.2.0

2 files

This release

0.1.1 This release

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