Skip to content

A framework to analyse the Stochastic Consistency of repeated requests to an LLM or LLM-based Agent

License

Notifications You must be signed in to change notification settings

thompsonson/det

Repository files navigation

Setup Poetry, check formatting and style, and run the tests

LLM Response Analysis Framework

Welcome to a LLM Response Analysis Framework! This tool is designed to dive deep into the heart of Language Models (LLMs) and their intriguing responses. Designed for researchers, developers, and LLM enthusiasts, the framework offers a way to examine the consistency of Large Language Models and Agents build on them.

Features | Screenshots | Getting Started | Development

Features

  • Dynamic LLM Integration Seamlessly connect with various LLM providers and models to fetch responses using a flexible architecture. Following integrations are available.

    • Openai
    • Groq
    • Ollama
  • LangChain Structured Output Chain Analysis Seamlessly connect with a LangChain Structured Output and check for the consistency of responses. See this documentation for further information.

  • Semantic Similarity Calculation Understand the nuanced differences between responses by calculating their semantic distances.

  • Diverse Response Analysis Group, count, and analyze responses to highlight both their uniqueness and redundancy.

  • Rich Presentation Utilize beautiful tables and text differences to present analysis results in an understandable and visually appealing manner.

Screenshots

Below are some screenshots showcasing the framework in action:

GPT-3.5 Example

GPT-3.5 Analysis

GPT-4 Example

GPT-4 Analysis

LangChain Structure Output example (using gpt-4o)

LangChain Structure Output example

These visuals provide a glimpse into how the framework processes and presents data from different LLM versions, highlighting the flexibility and depth of analysis possible with this tool.

Getting Started

Prerequisites

  • Ensure you have Python 3.10 or higher installed on your system.

Installation

Option 1. pipx is recommended

Use pipx for det rather than install it on the base python. pipx is awesome for installing and running Python applications in isolated environments.

Get pipx here: https://pipx.pypa.io/stable/

Install det using pipx:

pipx install det

Option 2. If you live danagerously, chaos is your middle name, or you don't want to use pipx

I understand that this approach heavily depends on the state of my system, it may not work and may result in pythonic module hell induced headaches .

Are you sure you want to do this?

I really know what I'm doing or plan on throwing away my computer

OK, no more checking - fill your boots.

Install det using pip:

pip install det

Configuration

Before using det, configure your LLM and embeddings provider API keys

export OPENAI_API_KEY=sk-makeSureThisIsaRealKey

for groq to configure groq client API key.

export GROQ_API_KEY=gsk-DUMMYKEYISTHIS

Basic Usage

To get a list of all the arguments and their descriptions, use:

det --help

a basic analysis of OpenAI's gpt-4o-mini model

det check-responses \
  --iterations 2 \
  --llm-provider OpenAI \
  --llm-model gpt-4o-mini \
  --embeddings-provider OpenAI \
  --embeddings-model text-embedding-ada-002

for Groq use --llm-provider as Groq for Ollama use --llm-provider as Ollama

LangChain Structured Output Chains

a LangChain Structured Output example

note, this requires the prompt details /resources/prompt.json and a pydantic output class /resources/risk_definition.py

det check-chain \
  --iterations 20 \
  --embeddings-provider OpenAI \
  --embeddings-model text-embedding-ada-002 \
  --prompt-config ./resources/prompts.json \
  --prompt-group RiskDefinition \
  --input-variables-str "risk_statement=There is a risk that failure to enforce multi-factor authentication can cause unauthorized access to user accounts to occur, leading to account takeover that could lead to financial fraud and identity theft issues for customers."

Development

Prerequisites

Setup Instructions

  1. Clone the repository:

    git clone https://github.com/thompsonson/det.git
    cd det
    
  2. Set up the Poetry environment:

    poetry install
    
  3. Activate the Poetry shell:

    poetry shell
    

You're now ready to start development on the det project!

Documentation

The documentation is in the module headings. I'll probably move it out at some point but that's good for now :)

Support and Contribution

For support, please open an issue on the GitHub repository. Contributions are welcome.

License

This project is licensed under the MIT License - see the LICENSE file for details.

About

A framework to analyse the Stochastic Consistency of repeated requests to an LLM or LLM-based Agent

Topics

Resources

License

Stars

Watchers

Forks