Skip to content

langchain-prefect

PyPI

Read the accompanying article for background

Orchestrate and observe langchain using Prefect

Large Language Models (LLMs) are interesting and useful  -  building apps that use them responsibly feels like a no-brainer. Tools like Langchain make it easier to build apps using LLMs. We need to know details about how our apps work, even when we want to use tools with convenient abstractions that may obfuscate those details.

Prefect is built to help data people build, run, and observe event-driven workflows wherever they want. It provides a framework for creating deployments on a whole slew of runtime environments (from Lambda to Kubernetes), and is cloud agnostic (best supports AWS, GCP, Azure). For this reason, it could be a great fit for observing apps that use LLMs.

Features

  • RecordLLMCalls is a ContextDecorator that can be used to track LLM calls made by Langchain LLMs as Prefect flows.

Call an LLM and track the invocation with Prefect:

from langchain.llms import OpenAI
from langchain_prefect.plugins import RecordLLMCalls

with RecordLLMCalls():
    llm = OpenAI(temperature=0.9)
    text = (
        "What would be a good company name for a company that makes colorful socks?"
    )
    llm(text)
and a flow run will be created to track the invocation of the LLM:

LLM invocation UI

Run several LLM calls via langchain agent as Prefect subflows:

from langchain.agents import initialize_agent, load_tools
from langchain.llms import OpenAI

from prefect import flow

llm = OpenAI(temperature=0)
tools = load_tools(["llm-math"], llm=llm)
agent = initialize_agent(tools, llm)

@flow
def my_flow():
    agent.run(
        "How old is the current Dalai Lama? "
        "What is his age divided by 2 (rounded to the nearest integer)?"
    )

with RecordLLMCalls(tags={"agent"}):
    my_flow()

LLM agent UI

Find more examples here.

How do I get a Prefect UI?

  • The easiest way is to use the Prefect Cloud UI for free. You can find details on getting setup here.

  • If you don't want to sign up for cloud, you can use the dashboard locally by running prefect server start in your terminal - more details here.

Resources

Installation

pip install langchain-prefect

Requires an installation of Python 3.10+.

Feedback

If you encounter any bugs while using langchain-prefect, feel free to open an issue in the langchain-prefect repository.

Feel free to star or watch langchain-prefect for updates too!

Contributing

If you'd like to help contribute to fix an issue or add a feature to langchain-prefect, please propose changes through a pull request from a fork of the repository.

Here are the steps:

  1. Fork the repository
  2. Clone the forked repository
  3. Install the repository and its dependencies:
    pip install -e ".[dev]"
    
  4. Make desired changes
  5. Add tests
  6. Insert an entry to CHANGELOG.md
  7. Install pre-commit to perform quality checks prior to commit:
    pre-commit install
    
  8. git commit, git push, and create a pull request