Skip to main content

Self-hosted SRE investigation copilot with YAML tools, SSH execution, SSE streaming, and secret redaction.

Project description

ops-copilot

CI PyPI License: MIT Python

Self-hosted SRE investigation copilot for production systems.

ops-copilot lets an LLM call tools defined in YAML, execute safe remote commands over SSH, redact secrets from outputs, and stream investigation events through LangGraph or an optional FastAPI SSE server.

Who this is for

  • SREs and platform engineers running self-hosted infrastructure.
  • Open source maintainers operating docs, bots, CI runners, demos, or package services.
  • Teams that want reviewed operational tools instead of free-form shell access.
  • Developers building incident-investigation UIs around LangGraph or LangChain.

Maintenance workflows

This repository is maintained with CI, build checks, smoke tests, release workflows, Dependabot, issue templates, PR checklists, a security model, and PyPI releases.

Typical maintainer tasks include reviewing YAML tools, triaging operational edge cases, adding tests for sanitizer and command-rendering behavior, and preparing safe releases.

Architecture

User question -> InvestigationGraph -> LLM -> YAML tools -> SSH host
                                      <- redacted tool output <- command result

The package is intentionally generic. You can start with shell tools from YAML, then inject custom Python RemoteTool classes for richer workflows.

Install

uv add ops-copilot

Optional extras:

uv add 'ops-copilot[server]'
uv add 'ops-copilot[openai]'
uv add 'ops-copilot[ollama]'

YAML tools

tools:
  - name: disk_usage
    type: shell
    description: Show filesystem usage.
    command: df -h

  - name: journalctl_service
    type: shell
    description: Show recent logs for a systemd service.
    command: journalctl -u {service} --since '{since}' --no-pager
    parameters:
      service:
        type: string
      since:
        type: string
        required: false
        default: "30 minutes ago"

Minimal usage

from ops_copilot import InvestigationGraph, SSHClient, ToolRegistry

ssh = SSHClient(host="server.example.com", user="deploy", key_path="~/.ssh/id_ed25519")
tools = ToolRegistry(ssh, config_path="tools.yaml").load()

graph = InvestigationGraph(
    llm=your_langchain_chat_model,
    tools=tools,
    system_prompt="You are an SRE copilot. Investigate safely and report evidence.",
)

async for event in graph.stream("The API is slow. What should I check?"):
    print(event)

Streaming events

InvestigationGraph.stream() yields dictionaries with these event names:

Event Meaning
token streamed model text
tool_start tool call started with input and step id
tool_end tool call finished with redacted output
error graph or stream error
done investigation complete

Optional FastAPI server

The ops_copilot.server.create_app() helper exposes:

  • POST /investigate
  • POST /investigate/stream

If OPS_COPILOT_API_KEY is set, clients must send X-API-Key.

Security notes

This project executes commands on servers you control. Treat tools.yaml as privileged code.

Recommendations:

  • Use SSH key auth with least-privilege users.
  • Review every command template before exposing it to an LLM.
  • Avoid destructive commands in YAML.
  • Keep parameterized commands narrow.
  • Store no secrets in YAML or prompts.
  • Rely on built-in redaction as a safety net, not as your only control.

Built-in redaction covers env-style secret lines, Bearer tokens, OpenAI-style keys, JWTs, long hex runs, and inline image data URLs.

Shell tools also apply a conservative command policy. Obvious destructive commands such as rm, dd, mkfs, shutdown, docker rm, docker prune, and systemctl restart are blocked unless the YAML tool explicitly opts in with policy.allow_destructive: true. Use dry_run: true to review rendered commands without executing them.

Audit logs

Use JsonlAuditLog to append redacted investigation events for incident review:

from ops_copilot import InvestigationGraph, JsonlAuditLog

graph = InvestigationGraph(
    llm=your_langchain_chat_model,
    tools=tools,
    system_prompt="Investigate safely and cite evidence.",
    audit_log=JsonlAuditLog("audit/investigation.jsonl"),
)

Documentation and examples

  • docs/security-model.md documents threat boundaries and deployment controls.
  • docs/why-ops-copilot.md explains the project scope and ecosystem need.
  • docs/demo.md shows a local demo that runs without real SSH credentials.
  • docs/codex-maintenance.md documents safe Codex-style maintenance workflows.
  • docs/writing-tools.md explains YAML and custom Python tools.
  • docs/server.md covers the optional FastAPI/SSE integration.
  • docs/maintenance-workflows.md describes maintainer workflows and review checklists.
  • docs/toolpacks.md documents reviewed example toolpacks.
  • docs/incident-fixtures.md documents fake incidents for demos and regression tests.
  • examples/local_demo.py runs without a real SSH host using fake outputs.
  • examples/replay_incident.py replays fake incident fixtures for demos.
  • examples/custom_tool.py shows how to inject a custom RemoteTool class.

Roadmap

  • Persistent investigation sessions.
  • More incident fixture coverage for regression tests.

Development

uv sync --dev
uv run ruff check .
uv run pytest
uv run python scripts/smoke.py
uv build

License

MIT

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

ops_copilot-0.1.5.tar.gz (212.9 kB view details)

Uploaded Source

Built Distribution

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

ops_copilot-0.1.5-py3-none-any.whl (20.6 kB view details)

Uploaded Python 3

File details

Details for the file ops_copilot-0.1.5.tar.gz.

File metadata

  • Download URL: ops_copilot-0.1.5.tar.gz
  • Upload date:
  • Size: 212.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for ops_copilot-0.1.5.tar.gz
Algorithm Hash digest
SHA256 cba4a03a11667f36775e553b1c8645c973bd9e70eda42eeed1fa0da35588411c
MD5 ae803584f2b50128a860ee6d9803ea2c
BLAKE2b-256 cea8edfe71c60a991c3233980212bef0bad4d62e75308e251bc1b9c8c7a57920

See more details on using hashes here.

Provenance

The following attestation bundles were made for ops_copilot-0.1.5.tar.gz:

Publisher: publish.yml on BenjaminJornet/ops-copilot

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

File details

Details for the file ops_copilot-0.1.5-py3-none-any.whl.

File metadata

  • Download URL: ops_copilot-0.1.5-py3-none-any.whl
  • Upload date:
  • Size: 20.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for ops_copilot-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 c280e276f52a9b286a81416ecc8e6dcf03ea26e2ab202a336520a94f61d7de15
MD5 35a997e8d211e8081d413d5626f81b8e
BLAKE2b-256 dae7a0d8f5358722a881cfcce57ce8f33489546b509e4e5f21d5f0502b51ff38

See more details on using hashes here.

Provenance

The following attestation bundles were made for ops_copilot-0.1.5-py3-none-any.whl:

Publisher: publish.yml on BenjaminJornet/ops-copilot

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 Pingdom Monitoring Sentry Error logging StatusPage Status page