Skip to main content

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.

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.0.1.tar.gz (205.8 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.0.1-py3-none-any.whl (292.2 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: strata_forge-0.0.1.tar.gz
  • Upload date:
  • Size: 205.8 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.0.1.tar.gz
Algorithm Hash digest
SHA256 89e3c56600bd0c71a7ea5cfa7b8f92c93dd2e219d7b1b15b44e833598733e325
MD5 e4b0210d1e4a47bfd90c5d1643adb0e7
BLAKE2b-256 f4460761216ea4526daca1a448e35cd2a78fe3506a0d9cf4b37cc8d53c5d79b7

See more details on using hashes here.

Provenance

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

File metadata

  • Download URL: strata_forge-0.0.1-py3-none-any.whl
  • Upload date:
  • Size: 292.2 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.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 6c639fe5286e68ed1e0d2008e6ac33f651635fec811129b864dedc93f9ac8f7f
MD5 fb8d96b1ba5a6fc14b13a7232c32b57d
BLAKE2b-256 cb07df35bbf2833f8a0a516dadc35aa2e81df7f1164699a6508b83f3569537c4

See more details on using hashes here.

Provenance

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

Supported by

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