LangKit is an open-source text metrics toolkit for monitoring language models. It offers an array of methods for extracting relevant signals from the input and/or output text, which are compatible with the open-source data logging library whylogs.
The generated profiles can be visualized and monitored in the WhyLabs platform or they can be further analyzed by the user on their own accord.
Productionizing language models, including LLMs, comes with a range of risks due to the infinite amount of input combinations, which can elicit an infinite amount of outputs. The unstructured nature of text poses a challenge in the ML observability space - a challenge worth solving, since the lack of visibility on the model's behavior can have serious consequences.
The currently supported metrics include:
- readability score
- complexity and grade scores
- sentiment analysis
- patterns - count of strings matching a user-defined regex pattern group
- jailbreaks - similarity scores with respect to known jailbreak attempts and prompt injection attacks
- refusals - similarity scores with respect to known LLM refusal of service responses
To install LangKit, use the Python Package Index (PyPI) as follows:
pip install langkitLangKit modules contain UDFs that automatically wire into the collection of UDFs on String features provided by whylogs by default. All we have to do is import the LangKit modules and then instantiate a custom schema as shown in the example below.
from whylogs.experimental.core.metrics.udf_metric import generate_udf_schema
from whylogs.core.schema import DeclarativeSchema
import whylogs as why
from langkit.sentiment import *
from langkit.textstat import *
text_schema = DeclarativeSchema(generate_udf_schema())
results = why.log({"prompt": "hello!", "response": "world!"}, schema=text_schema)The code above will produce a set of metrics comprised of the default whylogs metrics for text features and all the metrics defined in the imported modules.
More examples are available here.