uptrain

0

UpTrain is an open-source unified platform to evaluate and improve Generative AI applications. We provide grades for 20+ preconfigured checks (covering language, code, embedding use-cases), perform…

Analytics

llm-test
llm-prompting
experimentation
machine-learning

Logo of UpTrain - an open-source platform to evaluate and improve LLM applications

Try out Evaluation Read Docs Slack Community Request New Feature Daily Monitoring

GitHub Release GitHub commit activity GitHub License PyPI - Downloads

UpTrain is an open-source unified platform to evaluate and improve Generative AI applications. We provide grades for 20+ preconfigured evaluations (covering language, code, embedding use cases), perform root cause analysis on failure cases and give insights on how to resolve them.


Key Features πŸ”‘

Interactive Dashboards

UpTrain Dashboard is a web-based interface that runs on your local machine. You can use the dashboard to evaluate your LLM applications, view the results, and perform a root cause analysis.

20+ Pre-configured Evaluations

Support for 20+ pre-configured evaluations such as Response Completeness, Factual Accuracy, Context Conciseness etc.

Data Security

All the evaluations and analysis run locally on your system, ensuring that the data never leaves your secure environment (except for LLM calls while using model grading checks)

Experimentation

Experiment with different embedding models like text-embedding-3-large/small, text-embedding-3-ada, baai/bge-large, etc. UpTrain supports HuggingFace models, Replicate endpoints, or custom models hosted on your endpoint.

Root Cause Analysis

You can perform root cause analysis on cases with either negative user feedback or low evaluation scores to understand which part of your LLM pipeline is giving suboptimal results. Check out the supported RCA templates.

Select from a Variety of Evaluation LLMs

We allow you to use any of OpenAI, Anthropic, Mistral, Azure's Openai endpoints or open-source LLMs hosted on Anyscale to be used as evaluators.

Customize Evaluations

UpTrain provides tons of ways to customize evaluations. You can customize the evaluation method (chain of thought vs classify), few-shot examples, and scenario description. You can also create custom evaluators.

Coming Soon:

  1. Collaborate with your team
  2. Embedding visualization via UMAP and Clustering
  3. Pattern recognition among failure cases
  4. Prompt improvement suggestions

Getting Started πŸ™Œ

Method 1: Using the Locally Hosted Dashboard

The UpTrain dashboard is a web-based interface that allows you to evaluate your LLM applications. It is a self-hosted dashboard that runs on your local machine. You don't need to write any code to use the dashboard. You can use the dashboard to evaluate your LLM applications, view the results, and perform a root cause analysis.

Before you start, ensure you have docker installed on your machine. If not, you can install it from here.

The following commands will download the UpTrain dashboard and start it on your local machine.

# Clone the repository
git clone https://github.com/uptrain-ai/uptrain
cd uptrain

# Run UpTrain
bash run_uptrain.sh

NOTE: UpTrain Dashboard is currently in Beta version. We would love your feedback to improve it.

Method 2: Using the UpTrain package

If you are a developer and want to integrate UpTrain evaluations into your application, you can use the UpTrain package. This allows for a more programmatic way to evaluate your LLM applications.

Install the package through pip:

pip install uptrain

How to use UpTrain:

You can evaluate your responses via the open-source version by providing your OpenAI API key to run evaluations.

from uptrain import EvalLLM, Evals
import json

OPENAI_API_KEY = "sk-***************"

data = [{
    'question': 'Which is the most popular global sport?',
    'context': "The popularity of sports can be measured in various ways, including TV viewership, social media presence, number of participants, and economic impact. Football is undoubtedly the world's most popular sport with major events like the FIFA World Cup and sports personalities like Ronaldo and Messi, drawing a followership of more than 4 billion people. Cricket is particularly popular in countries like India, Pakistan, Australia, and England. The ICC Cricket World Cup and Indian Premier League (IPL) have substantial viewership. The NBA has made basketball popular worldwide, especially in countries like the USA, Canada, China, and the Philippines. Major tennis tournaments like Wimbledon, the US Open, French Open, and Australian Open have large global audiences. Players like Roger Federer, Serena Williams, and Rafael Nadal have boosted the sport's popularity. Field Hockey is very popular in countries like India, Netherlands, and Australia. It has a considerable following in many parts of the world.",
    'response': 'Football is the most popular sport with around 4 billion followers worldwide'
}]

eval_llm = EvalLLM(openai_api_key=OPENAI_API_KEY)

results = eval_llm.evaluate(
    data=data,
    checks=[Evals.CONTEXT_RELEVANCE, Evals.FACTUAL_ACCURACY, Evals.RESPONSE_COMPLETENESS]
)

print(json.dumps(results, indent=3))

If you have any questions, please join our Slack community

Speak directly with the maintainers of UpTrain by booking a call here.


Pre-built Evaluations We Offer πŸ“

quality of your responses
EvalDescription
Response CompletenessGrades whether the response has answered all the aspects of the question specified.
Response ConcisenessGrades how concise the generated response is or if it has any additional irrelevant information for the question asked.
Response RelevanceGrades how relevant the generated context was to the question specified.
Response ValidityGrades if the response generated is valid or not. A response is considered to be valid if it contains any information.
Response ConsistencyGrades how consistent the response is with the question asked as well as with the context provided.
quality of retrieved context and response groundedness
EvalDescription
Context RelevanceGrades how relevant the context was to the question specified.
Context UtilizationGrades how complete the generated response was for the question specified, given the information provided in the context.
Factual AccuracyGrades whether the response generated is factually correct and grounded by the provided context.
Context ConcisenessEvaluates the concise context cited from an original context for irrelevant information.
Context RerankingEvaluates how efficient the reranked context is compared to the original context.
language quality of the response
EvalDescription
Language FeaturesGrades the quality and effectiveness of language in a response, focusing on factors such as clarity, coherence, conciseness, and overall communication.
TonalityGrades whether the generated response matches the required persona's tone
language quality of the response
EvalDescription
Code HallucinationGrades whether the code present in the generated response is grounded by the context.
conversation as a whole
EvalDescription
User SatisfactionGrades how well the user's concerns are addressed and assesses their satisfaction based on provided conversation.
custom evaluations and others
EvalDescription
Custom GuidelineAllows you to specify a guideline and grades how well the LLM adheres to the provided guideline when giving a response.
Custom PromptsAllows you to create your own set of evaluations.
compare responses with ground truth
EvalDescription
Response MatchingCompares and grades how well the response generated by the LLM aligns with the provided ground truth.
safeguard system prompts and avoid LLM mis-use
EvalDescription
Prompt InjectionGrades whether the user's prompt is an attempt to make the LLM reveal its system prompts.
Jailbreak DetectionGrades whether the user's prompt is an attempt to jailbreak (i.e. generate illegal or harmful responses).
evaluate the clarity of user queries
EvalDescription
Sub-Query CompletenessEvaluate whether all of the sub-questions generated from a user's query, taken together, cover all aspects of the user's query or not
Multi-Query AccuracyEvaluate whether the variants generated accurately represent the original query

Integrations 🀝

Eval FrameworksLLM ProvidersLLM PackagesServing frameworksLLM ObservabilityVector DBs
OpenAI EvalsOpenAILlamaIndexOllamaLangfuseQdrant
AzureTogether AIHeliconeFAISS
ClaudeAnyscaleZenoChroma
MistralReplicate
HuggingFace

More integrations are coming soon. If you have a specific integration in mind, please let us know by creating an issue.


Monitoring Prompt Drift in LLMs: Benchmark by UpTrain

Most popular LLMs like GPT-4, GPT-3.5-turbo, Claude-2.1 etc., are closed-source, i.e. exposed via an API with very little visibility on what happens under the hood. There are many reported instances of prompt drift (or GPT-4 becoming lazy) and research work exploring the degradation in model quality. This benchmark is an attempt to track the change in model behaviour by evaluating its response on a fixed dataset.

image

You can find the benchmark here.

Resources πŸ’‘

  1. How to evaluate your LLM application
  2. How to detect jailbreaks
  3. Dealing with hallucinations

Why we are building UpTrain πŸ€”

Having worked with ML and NLP models for the last 8 years, we were continuosly frustated with numerous hidden failures in our models which led to us building UpTrain. UpTrain was initially started as an ML observability tool with checks to identify regression in accuracy.

However we soon released that LLM developers face an even bigger problem -- there is no good way to measure accuracy of their LLM applications, let alone identify regression.

We also saw release of OpenAI evals, where they proposed the use of LLMs to grade the model responses. Furthermore, we gained confidence to approach this after reading how Anthropic leverages RLAIF and dived right into the LLM evaluations research (We are soon releasing a repository of awesome evaluations research).

So, come today, UpTrain is our attempt to bring order to LLM chaos and contribute back to the community. While a majority of developers still rely on intuition and productionise prompt changes by reviewing a couple of cases, we have heard enough regression stories to believe "evaluations and improvement" will be a key part of LLM ecosystem as the space matures.

  1. Robust evaluations allows you to systematically experiment with different configurations and prevent any regressions by helping objectively select the best choice.

  2. It helps you understand where your systems are going wrong, find the root cause(s) and fix them - long before your end users complain and potentially churn out.

  3. Evaluations like prompt injection and jailbreak detection are essential to maintain safety and security of your LLM applications.

  4. Evaluations help you provide transparency and build trust with your end-users - especially relevant if you are selling to enterprises.


Why open-source?

  1. We understand that there is no one-size-fits-all solution when it come to evaluations. We are increasingly seeing the desire from developers to modify the evaluation prompt or set of choices or the few shot examples, etc. We believe the best developer experience lies in open-source, instead of exposing 20 different parameters.

  2. Foster innovation: The field of LLM evaluations and using LLM-as-a-judge is still pretty nascent. We see a lot of exciting research happening, almost on a daily basis and being open-source provides the right platform to us and our community to implement those techniques and innovate faster.


How You Can Help πŸ™

We are continuously striving to enhance UpTrain, and there are several ways you can contribute:

  1. Notice any issues or areas for improvement: If you spot anything wrong or have ideas for enhancements, please create an issue on our GitHub repository.

  2. Contribute directly: If you see an issue you can fix or have code improvements to suggest, feel free to contribute directly to the repository.

  3. Request custom evaluations: If your application requires a tailored evaluation, let us know, and we'll add it to the repository.

  4. Integrate with your tools: Need integration with your existing tools? Reach out, and we'll work on it.

  5. Assistance with evaluations: If you need assistance with evaluations, post your query on our Slack channel, and we'll resolve it promptly.

  6. Show your support: Show your support by starring us ⭐ on GitHub to track our progress.

  7. Spread the word: If you like what we've built, give us a shoutout on Twitter!

Your contributions and support are greatly appreciated! Thank you for being a part of UpTrain's journey.


License πŸ’»

This repo is published under Apache 2.0 license and we are committed to adding more functionalities to the UpTrain open-source repo. We also have a managed version if you just want a more hands-off experience. Please book a demo call here.


Provide feedback (Harsher the better πŸ˜‰)

We are building UpTrain in public. Help us improve by giving your feedback here.


Contributors πŸ–₯️

We welcome contributions to UpTrain. Please see our contribution guide for details.