Patches VCR.py to include non-network tooling for use with LangChain. Refactor with confidence as you record and replay all your LLM logic in a contained environment, free from any and all side effects.
pip install vcr-langchain
Use it with pytest:
import vcr_langchain as vcr
from langchain.llms import OpenAI
@vcr.use_cassette()
def test_use_as_test_decorator():
llm = OpenAI(model_name="text-ada-001")
assert llm("Tell me a surreal joke") == "<put the output here>"
The next time you run it:
- the output is now deterministic
- it executes a lot faster by replaying from cache
- no command executions or other side effects actually happen
- you no longer need to have real API keys defined for test execution in CI environments
For more examples, see the usages test file.
Note that tools, if initialized outside of the vcr_langchain
decorator, will not have recording capabilities patched in. This is true even if an agent using those tools is initialized within the decorator.
For more information on how VCR works and what other options there are, please see the VCR docs.
For more information on how to use langchain, please see the langchain docs.
Please note that there is a lot of langchain functionality that I haven't gotten around to hijacking for recording. If there's anything you need to record in a cassette, please open a PR or issue.