Skip to content

Record and replay LLM interactions for langchain

License

Notifications You must be signed in to change notification settings

develop-health/vcr-langchain

 
 

Repository files navigation

VCR LangChain

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.

Quickstart

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.

Pitfalls

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.

Documentation

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.

Projects that use this

About

Record and replay LLM interactions for langchain

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 91.5%
  • Makefile 8.5%