├── LICENSE ├── README.md ├── collie ├── __init__.py ├── ccnews_extractor.py ├── constraint_renderer.py ├── constraints.py ├── english_extractor.py ├── examples │ ├── extract.py │ ├── para_constraints.py │ ├── passage_constraints.py │ ├── sent_constraints.py │ ├── sources.py │ ├── validate.py │ └── word_constraints.py ├── extract_constraints.py ├── extractor_utils.py ├── gutenberg_extractor.py ├── models.py └── wiki_extractor.py ├── data └── all_data.dill ├── docs ├── constraint_spec.md └── extraction.md ├── download.sh ├── logs ├── alpaca-7b-1trial-no0-prompt.json ├── alpaca-7b-1trial-no1-prompt.json ├── alpaca-7b-1trial-no2-prompt.json ├── alpaca-7b-1trial-no3-prompt.json ├── alpaca-7b-1trial-no4-prompt.json ├── gpt-3.5-turbo-20trial-prompt.json ├── gpt-3.5-turbo-2nd-round-generation.json ├── gpt-3.5-turbo-multi-round-generation-1trial.json ├── gpt-4-20trial-prompt.json ├── gpt-4-2nd-round-generation.json ├── gpt-4-multi-round-generation-1trial.json ├── palm-text-bison-001-20trial-prompt.json ├── pass_coherence_report.json ├── vicuna-7b-1trial-no0-prompt.json ├── vicuna-7b-1trial-no1-prompt.json ├── vicuna-7b-1trial-no2-prompt.json ├── vicuna-7b-1trial-no3-prompt.json └── vicuna-7b-1trial-no4-prompt.json ├── pyproject.toml ├── requirements.txt ├── scripts ├── analysis.ipynb ├── run_api_models.py └── run_gpu_models.py ├── setup.py ├── teaser.png └── tests └── test_constraints.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/README.md -------------------------------------------------------------------------------- /collie/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /collie/ccnews_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/ccnews_extractor.py -------------------------------------------------------------------------------- /collie/constraint_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/constraint_renderer.py -------------------------------------------------------------------------------- /collie/constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/constraints.py -------------------------------------------------------------------------------- /collie/english_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/english_extractor.py -------------------------------------------------------------------------------- /collie/examples/extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/examples/extract.py -------------------------------------------------------------------------------- /collie/examples/para_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/examples/para_constraints.py -------------------------------------------------------------------------------- /collie/examples/passage_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/examples/passage_constraints.py -------------------------------------------------------------------------------- /collie/examples/sent_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/examples/sent_constraints.py -------------------------------------------------------------------------------- /collie/examples/sources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/examples/sources.py -------------------------------------------------------------------------------- /collie/examples/validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/examples/validate.py -------------------------------------------------------------------------------- /collie/examples/word_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/examples/word_constraints.py -------------------------------------------------------------------------------- /collie/extract_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/extract_constraints.py -------------------------------------------------------------------------------- /collie/extractor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/extractor_utils.py -------------------------------------------------------------------------------- /collie/gutenberg_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/gutenberg_extractor.py -------------------------------------------------------------------------------- /collie/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/models.py -------------------------------------------------------------------------------- /collie/wiki_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/collie/wiki_extractor.py -------------------------------------------------------------------------------- /data/all_data.dill: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/data/all_data.dill -------------------------------------------------------------------------------- /docs/constraint_spec.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/docs/constraint_spec.md -------------------------------------------------------------------------------- /docs/extraction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/docs/extraction.md -------------------------------------------------------------------------------- /download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/download.sh -------------------------------------------------------------------------------- /logs/alpaca-7b-1trial-no0-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/alpaca-7b-1trial-no0-prompt.json -------------------------------------------------------------------------------- /logs/alpaca-7b-1trial-no1-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/alpaca-7b-1trial-no1-prompt.json -------------------------------------------------------------------------------- /logs/alpaca-7b-1trial-no2-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/alpaca-7b-1trial-no2-prompt.json -------------------------------------------------------------------------------- /logs/alpaca-7b-1trial-no3-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/alpaca-7b-1trial-no3-prompt.json -------------------------------------------------------------------------------- /logs/alpaca-7b-1trial-no4-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/alpaca-7b-1trial-no4-prompt.json -------------------------------------------------------------------------------- /logs/gpt-3.5-turbo-20trial-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/gpt-3.5-turbo-20trial-prompt.json -------------------------------------------------------------------------------- /logs/gpt-3.5-turbo-2nd-round-generation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/gpt-3.5-turbo-2nd-round-generation.json -------------------------------------------------------------------------------- /logs/gpt-3.5-turbo-multi-round-generation-1trial.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/gpt-3.5-turbo-multi-round-generation-1trial.json -------------------------------------------------------------------------------- /logs/gpt-4-20trial-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/gpt-4-20trial-prompt.json -------------------------------------------------------------------------------- /logs/gpt-4-2nd-round-generation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/gpt-4-2nd-round-generation.json -------------------------------------------------------------------------------- /logs/gpt-4-multi-round-generation-1trial.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/gpt-4-multi-round-generation-1trial.json -------------------------------------------------------------------------------- /logs/palm-text-bison-001-20trial-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/palm-text-bison-001-20trial-prompt.json -------------------------------------------------------------------------------- /logs/pass_coherence_report.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/pass_coherence_report.json -------------------------------------------------------------------------------- /logs/vicuna-7b-1trial-no0-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/vicuna-7b-1trial-no0-prompt.json -------------------------------------------------------------------------------- /logs/vicuna-7b-1trial-no1-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/vicuna-7b-1trial-no1-prompt.json -------------------------------------------------------------------------------- /logs/vicuna-7b-1trial-no2-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/vicuna-7b-1trial-no2-prompt.json -------------------------------------------------------------------------------- /logs/vicuna-7b-1trial-no3-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/vicuna-7b-1trial-no3-prompt.json -------------------------------------------------------------------------------- /logs/vicuna-7b-1trial-no4-prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/logs/vicuna-7b-1trial-no4-prompt.json -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | openai 2 | aiolimiter 3 | rich 4 | fschat 5 | -------------------------------------------------------------------------------- /scripts/analysis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/scripts/analysis.ipynb -------------------------------------------------------------------------------- /scripts/run_api_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/scripts/run_api_models.py -------------------------------------------------------------------------------- /scripts/run_gpu_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/scripts/run_gpu_models.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/setup.py -------------------------------------------------------------------------------- /teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/teaser.png -------------------------------------------------------------------------------- /tests/test_constraints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/princeton-nlp/Collie/HEAD/tests/test_constraints.py --------------------------------------------------------------------------------