- Read our (pre-print) paper: read the paper
- Join our Discord: discord
- Read our short blog post: read the blog post
- Check out our Colab Demo: colab demo
- Check out the video: MIT 2024
You can install the Palimpzest package and CLI on your machine by cloning this repository and running:
$ git clone [email protected]:mitdbg/palimpzest.git
$ cd palimpzest
$ pip install .
Installing Palimpzest also installs its CLI tool pz
which provides users with basic utilities for creating and managing their own Palimpzest system. Running pz --help
diplays an overview of the CLI's commands:
$ pz --help
Usage: pz [OPTIONS] COMMAND [ARGS]...
The CLI tool for Palimpzest.
Options:
--help Show this message and exit.
Commands:
help (h) Print the help message for PZ.
init (i) Initialize data directory for PZ.
ls-data (ls,lsdata) Print a table listing the datasets
registered with PZ.
register-data (r,reg,register) Register a data file or data directory with
PZ.
rm-data (rm,rmdata) Remove a dataset that was registered with
PZ.
Users can initialize their own system by running pz init
. This will create Palimpzest's working directory in ~/.palimpzest
:
$ pz init
Palimpzest system initialized in: /Users/matthewrusso/.palimpzest
If we list the set of datasets registered with Palimpzest, we'll see there currently are none:
$ pz ls
+------+------+------+
| Name | Type | Path |
+------+------+------+
+------+------+------+
Total datasets: 0
To add (or "register") a dataset with Palimpzest, we can use the pz register-data
command (also aliased as pz reg
) to specify that a file or directory at a given --path
should be registered as a dataset with the specified --name
:
$ pz reg --path README.md --name rdme
Registered rdme
If we list Palimpzest's datasets again we will see that README.md
has been registered under the dataset named rdme
:
$ pz ls
+------+------+------------------------------------------+
| Name | Type | Path |
+------+------+------------------------------------------+
| rdme | file | /Users/matthewrusso/palimpzest/README.md |
+------+------+------------------------------------------+
Total datasets: 1
To remove a dataset from Palimpzest, simply use the pz rm-data
command (also aliased as pz rm
) and specify the --name
of the dataset you would like to remove:
$ pz rm --name rdme
Deleted rdme
Finally, listing our datasets once more will show that the dataset has been deleted:
$ pz ls
+------+------+------+
| Name | Type | Path |
+------+------+------+
+------+------+------+
Total datasets: 0
Palimpzest will cache intermediate results by default. It can be useful to remove them from the cache when trying to evaluate the performance improvement(s) of code changes. We provide a utility command pz clear-cache
(also aliased as pz clr
) to clear the cache:
$ pz clr
Cache cleared
You may wish to work with multiple configurations of Palimpzest in order to, e.g., evaluate the difference in performance between various LLM services for your data extraction task. To see the config Palimpzest is currently using, you can run the pz print-config
command (also aliased as pz config
):
$ pz config
--- default ---
filecachedir: /some/local/filepath
llmservice: openai
name: default
parallel: false
By default, Palimpzest uses the configuration named default
. As shown above, if you run a script using Palimpzest out-of-the-box, it will use OpenAI endpoints for all of its API calls.
Now, let's say you wanted to try using together.ai's for your API calls, you could do this by creating a new config with the pz create-config
command (also aliased as pz cc
):
$ pz cc --name together-conf --llmservice together --parallel True --set
Created and set config: together-conf
The --name
parameter is required and specifies the unique name for your config. The --llmservice
and --parallel
options specify the service to use and whether or not to process files in parallel. Finally, if the --set
flag is present, Palimpzest will update its current config to point to the newly created config.
We can confirm that Palimpzest checked out our new config by running pz config
:
$ pz config
--- together-conf ---
filecachedir: /some/local/filepath
llmservice: together
name: together-conf
parallel: true
You can switch which config you are using at any time by using the pz set-config
command (also aliased as pz set
):
$ pz set --name default
Set config: default
$ pz config
--- default ---
filecachedir: /some/local/filepath
llmservice: openai
name: default
parallel: false
$ pz set --name together-conf
Set config: together-conf
$ pz config
--- together-conf ---
filecachedir: /some/local/filepath
llmservice: together
name: together-conf
parallel: true
Finally, you can delete a config with the pz rm-config
command (also aliased as pz rmc
):
$ pz rmc --name together-conf
Deleted config: together-conf
Note that you cannot delete the default
config, and if you delete the config that you currently have set, Palimpzest will set the current config to be default
.
There are a few things you need to do in order to use remote parallel services.
If you want to use parallel LLM execution on together.ai, you have to modify the config.yaml (by default, Palimpzest uses ~/.palimpzest/config_default.yaml
) so that llmservice: together
and parallel: True
are set.
If you want to use parallel PDF processing at modal.com, you have to:
- Set
pdfprocessing: modal
in the config.yaml file. - Run
modal deploy src/palimpzest/tools/allenpdf.py
. This will remotely install the modal function so you can run it. (Actually, it's probably already installed there, but do this just in case. Also do it if there's been a change to the server-side function inside that file.)
Below are simple instructions to run pz on a test data set of enron emails that is included with the system:
-
Initialize the configuration by running
pz init
. -
Add the enron data set with:
pz reg --path testdata/enron-tiny --name enron-tiny
then run it through the test program with:python demos/simpleDemo.py --task enron --datasetid enron-tiny
-
Add the test paper set with:
pz reg --path testdata/pdfs-tiny --name pdfs-tiny
then run it through the test program with:python demos/simpleDemo.py --task paper --datasetid pdfs-tiny
-
Palimpzest defaults to using OpenAI. You’ll need to export an environment variable
OPENAI_API_KEY