Check out the accompanying blog post here.
$ pip install compress-gptSimply change your existing imports of langchain.PromptTemplate to compress_gpt.langchain.CompressTemplate (to compress prompts before populating variables) or compress_gpt.langchain.CompressPrompt (to compress prompts after populating variables).
-from langchain import PromptTemplate
+from compress_gpt.langchain import CompressPrompt as PromptTemplateFor very simple prompts, use CompressSimplePrompt and CompressSimpleTemplate instead.
If compression ever fails or results in extra tokens, the original prompt will be used. Each compression result is aggressively cached, but the first run can take a hot sec.
import compress_gpt
compress_gpt.clear_cache()My blog post helps explain the below image.