Skip to content

devsgnr/krisis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Krisis

Clinical evaluation framework for testing LLM safety behavior in medical reasoning.

Krisis evaluates not only whether an LLM is correct, but whether it knows when to abstain, defer, or express uncertainty in high-stakes clinical tasks.

Why Krisis

Krisis grew out of Cady AI, an earlier CKD detection chatbot presented at a national AI hackathon. Cady AI used a model trained on the UCI Chronic Kidney Disease dataset to predict CKD/not-CKD, return class probabilities, and attribute which lab results pushed risk upward.

That project exposed the next safety question: as LLMs become more fluent in clinical reasoning, can they recognize cases where they should not confidently answer? Krisis turns that question into a reusable evaluation framework: a human-in-the-loop type system for checking whether LLMs can defer, abstain, and express uncertainty before their outputs are trusted.

What Krisis Does

Krisis provides:

  • clinical task suites that produce structured patient records
  • provider backends for OpenAI, Anthropic, Grok, and Google Gemini
  • batched and concurrent benchmark execution
  • retry/backoff handling for transient provider failures
  • structured parsing of model predictions, confidence, and abstentions
  • abstention-aware metrics beyond plain accuracy
  • text, full JSON, and metrics-only JSON reports
  • execution metadata such as runtime, throughput, batch size, concurrency, and token usage

Research Status And Limitations

Krisis v0.1 currently includes one implemented suite: Chronic Kidney Disease (CKD), based on the UCI CKD dataset.

Supported CKD tasks:

  • detection: CKD vs not CKD
  • staging: CKD stage classification
  • progression: synthetic progression stress test

Important limitations:

  • CKD is the only available suite in v0.1.
  • The UCI CKD dataset is small and cross-sectional.
  • Progression is synthetic because the source dataset is not longitudinal.
  • Krisis is for research and evaluation only. It is not a medical device and must not be used to diagnose or treat patients.
  • Results depend on model version, prompts, provider behavior, dataset quality, and benchmark settings.

Installation

Install Krisis:

pip install krisis

Install provider-specific dependencies:

pip install "krisis[openai]"
pip install "krisis[anthropic]"
pip install "krisis[grok]"
pip install "krisis[gemini]"

Quickstart

Warning Krisis v0.1 only includes the CKD suite. The UCI CKD CSV is not bundled with the package; download it locally and pass its path to CKDSuite.

from krisis.backends.openai import OpenAIBackend
from krisis.benchmark import Benchmark
from krisis.data.base import FeatureSet, SuiteConfig, Task
from krisis.data.ckd.suite import CKDSuite
from krisis.results.report import format_report

suite = CKDSuite(
    config=SuiteConfig(
        features=FeatureSet.FULL,
        task=Task.DETECTION,
        seed=42,
        n_synthetic=80,
        test_size=0.2,
    ),
    data_path="datasets/ckd/ckd_full.csv",
)

backend = OpenAIBackend(
    model="gpt-5.5",
    api_key="YOUR_API_KEY",
)

result = Benchmark(
    suite,
    backend,
    batch_size=8,
    max_concurrency=2,
).run()

print(format_report(result))

Outputs

Krisis supports three report styles.

Text report:

from krisis.results.report import format_report

print(format_report(result))

Full JSON report:

from krisis.results.report import format_json_report

print(format_json_report(result, include_results=True))

Metrics-only JSON report for plotting/model comparison:

from krisis.results.report import format_metrics_json_report

print(format_metrics_json_report(result))

The execution block includes benchmark runtime and operational metadata:

{
  "batch_size": 8,
  "max_concurrency": 2,
  "n_input_records": 160,
  "n_api_batches": 20,
  "elapsed_seconds": 42.18,
  "records_per_second": 3.79,
  "input_tokens": 12000,
  "output_tokens": 2400,
  "token_total": 14400
}

Core Concepts

  • Suite: prepares a clinical dataset/task and returns patient records.
  • Backend: adapts a model provider to Krisis' standard response shape.
  • Benchmark: runs records through a backend with batching, concurrency, and retries.
  • Metric: scores model behavior across correctness, uncertainty, and deferral.
  • Report: serializes results as text or JSON for review, plotting, or papers.

Metrics

Krisis includes:

  • Accuracy
  • Balanced Accuracy
  • Selective Accuracy (answered only)
  • Abstention Rate
  • Answer Rate / Coverage
  • Deferral Alignment
  • Expected Calibration Error
  • Brier Score where applicable

Selective accuracy separates how often the model was right when it answered from how often it chose not to answer.

Model Backends

Provider Backend Default model
OpenAI OpenAIBackend gpt-5.5
Anthropic AnthropicBackend claude-opus-4-7
Grok GrokBackend grok-4.3
Google Gemini GeminiBackend gemini-3-pro-preview

All backends return the same structured fields:

prediction
abstained
confidence
raw_response
input_tokens
output_tokens
total_tokens

Citation

If you use Krisis in research, please cite it as software:

@software{watila_krisis_2026,
  author = {Watila, Emmanuel},
  title = {Krisis: A Clinical Evaluation Framework for Large Language Models},
  year = {2026},
  version = {0.1.0},
  url = {https://github.com/devsgnr/krisis}
}

License

Apache-2.0

About

A clinical evaluation framework for LLMs that measures accuracy, abstention, calibration, deferral behavior, runtime, token use, and structured-output reliability across high-stakes medical reasoning tasks.

Topics

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors

Languages