Orchestrate and observe tools built with langchain.
Project description
langchain-prefect
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 aContextDecorator
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:
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()
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:
- Fork the repository
- Clone the forked repository
- Install the repository and its dependencies:
pip install -e ".[dev]"
- Make desired changes
- Add tests
- Insert an entry to CHANGELOG.md
- Install
pre-commit
to perform quality checks prior to commit:
pre-commit install
git commit
,git push
, and create a pull request
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for langchain_prefect-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | dcf0f4e21c31d55a9fe41dd6e3c3fd55bd51ea4308f80c34cc4a82d76c4b73b7 |
|
MD5 | 75b35e5e265bdbef38f06c0b6533cde7 |
|
BLAKE2b-256 | 8583330cef6b1271c084246505f69caa6dcd9f0c72e70c5d20bc7282926e425b |