├── .gitignore ├── LICENSE.md ├── README.md ├── assets ├── banner.ai ├── banner.pdf ├── banner.png ├── example-focus.png ├── examples-overview.png ├── tradeoff-plots.png └── viz.png ├── cartridges ├── __init__.py ├── cache.py ├── clients │ ├── __init__.py │ ├── base.py │ ├── openai.py │ ├── together.py │ ├── tokasaurus.py │ └── usage.py ├── data │ ├── __init__.py │ ├── chunkers.py │ ├── gmail │ │ ├── resources.py │ │ ├── server.py │ │ ├── tools.py │ │ └── utils.py │ ├── longhealth │ │ ├── __init__.py │ │ ├── evals.py │ │ ├── resources.py │ │ └── utils.py │ ├── mcp │ │ ├── mixin.py │ │ └── tools.py │ ├── mtob │ │ ├── __init__.py │ │ ├── baseline.py │ │ ├── evals.py │ │ ├── load.py │ │ ├── requirements.txt │ │ └── resources.py │ ├── qasper │ │ ├── __init__.py │ │ ├── evals.py │ │ ├── resources.py │ │ └── rewrite.py │ ├── resources.py │ ├── retrieval │ │ ├── __init__.py │ │ ├── load.py │ │ ├── retrievers.py │ │ └── tools.py │ ├── ruler │ │ ├── README.md │ │ ├── __init__.py │ │ ├── constants.py │ │ ├── download_paulgraham_essay.py │ │ ├── evals.py │ │ ├── niah.py │ │ ├── requirements.txt │ │ ├── resources.py │ │ └── variable_tracking.py │ ├── slack │ │ ├── resources.py │ │ └── tools.py │ ├── tex │ │ ├── __init__.py │ │ ├── evals.py │ │ └── resources.py │ └── tools.py ├── datasets.py ├── evaluate.py ├── generation.py ├── initialization │ ├── __init__.py │ ├── data │ │ └── gradient.txt │ ├── pretrained.py │ ├── random.py │ ├── text.py │ └── tokenization_utils.py ├── models │ ├── __init__.py │ ├── attention.py │ ├── config.py │ ├── llama │ │ ├── __init__.py │ │ ├── configuration_llama.py │ │ └── modeling_llama.py │ └── qwen │ │ ├── __init__.py │ │ ├── configuration_qwen3.py │ │ └── modeling_qwen3.py ├── structs.py ├── synthesize.py ├── synthesizers │ ├── __init__.py │ ├── base.py │ ├── self_study.py │ └── utils.py ├── tests │ ├── README.md │ ├── __init__.py │ ├── clients │ │ ├── test_base.py │ │ ├── test_openai.py │ │ └── test_tokasaurus.py │ ├── mocks.py │ ├── models │ │ ├── common.py │ │ ├── test_llama.py │ │ └── test_qwen.py │ ├── resources │ │ ├── __init__.py │ │ ├── gmail │ │ │ ├── __init__.py │ │ │ └── test_gmail.py │ │ ├── mcp │ │ │ ├── __init__.py │ │ │ └── test_tools.py │ │ ├── slack │ │ │ ├── __init__.py │ │ │ └── test_slack.py │ │ └── test_resources.py │ └── synthesizers │ │ ├── __init__.py │ │ └── test_self_study.py ├── train.py └── utils │ ├── __init__.py │ ├── cache_size.py │ ├── chat.py │ ├── hf.py │ ├── llm_messages.py │ ├── thinking.py │ └── wandb.py ├── docs └── training.md ├── examples ├── __init__.py ├── arxiv │ ├── README.md │ ├── arxiv_baseline.py │ ├── arxiv_synthesize.py │ ├── arxiv_synthesize_eval.py │ ├── arxiv_train.py │ └── cartridges.tex ├── benchmarks │ ├── __init__.py │ ├── longhealth │ │ ├── baseline_longhealth.py │ │ ├── longhealth_synthesize.py │ │ └── longhealth_train.py │ ├── mtob │ │ ├── mtob_baseline.py │ │ ├── mtob_synthesize.py │ │ └── mtob_train.py │ └── niah │ │ ├── niah_baseline.py │ │ ├── niah_synthesize.py │ │ └── niah_train.py └── cartridge_chat.py ├── infra ├── modal_deploy_sglang.py ├── modal_deploy_tokasaurus.py ├── modal_deploy_tokasaurus_flash.py ├── modal_deploy_vllm.py ├── modal_ssh.py └── tuning │ ├── examples │ └── longhealth_tune.py │ └── tune_toka.py ├── pyproject.toml └── viz ├── .gitignore ├── README.md ├── eslint.config.js ├── index.html ├── package-lock.json ├── package.json ├── public └── vite.svg ├── requirements.txt ├── src ├── App.css ├── App.jsx ├── assets │ └── react.svg ├── dashboards │ ├── __init__.py │ ├── base.py │ └── codehop.py ├── index.css ├── main.jsx ├── pages │ ├── DatasetsPage.jsx │ └── TrainingPage.jsx └── server.py ├── start.sh └── vite.config.js /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/README.md -------------------------------------------------------------------------------- /assets/banner.ai: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/assets/banner.ai -------------------------------------------------------------------------------- /assets/banner.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/assets/banner.pdf -------------------------------------------------------------------------------- /assets/banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/assets/banner.png -------------------------------------------------------------------------------- /assets/example-focus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/assets/example-focus.png -------------------------------------------------------------------------------- /assets/examples-overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/assets/examples-overview.png -------------------------------------------------------------------------------- /assets/tradeoff-plots.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/assets/tradeoff-plots.png -------------------------------------------------------------------------------- /assets/viz.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/assets/viz.png -------------------------------------------------------------------------------- /cartridges/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/__init__.py -------------------------------------------------------------------------------- /cartridges/cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/cache.py -------------------------------------------------------------------------------- /cartridges/clients/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/clients/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/clients/base.py -------------------------------------------------------------------------------- /cartridges/clients/openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/clients/openai.py -------------------------------------------------------------------------------- /cartridges/clients/together.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/clients/together.py -------------------------------------------------------------------------------- /cartridges/clients/tokasaurus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/clients/tokasaurus.py -------------------------------------------------------------------------------- /cartridges/clients/usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/clients/usage.py -------------------------------------------------------------------------------- /cartridges/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/__init__.py -------------------------------------------------------------------------------- /cartridges/data/chunkers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/chunkers.py -------------------------------------------------------------------------------- /cartridges/data/gmail/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/gmail/resources.py -------------------------------------------------------------------------------- /cartridges/data/gmail/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/gmail/server.py -------------------------------------------------------------------------------- /cartridges/data/gmail/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/gmail/tools.py -------------------------------------------------------------------------------- /cartridges/data/gmail/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/gmail/utils.py -------------------------------------------------------------------------------- /cartridges/data/longhealth/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/data/longhealth/evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/longhealth/evals.py -------------------------------------------------------------------------------- /cartridges/data/longhealth/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/longhealth/resources.py -------------------------------------------------------------------------------- /cartridges/data/longhealth/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/longhealth/utils.py -------------------------------------------------------------------------------- /cartridges/data/mcp/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/mcp/mixin.py -------------------------------------------------------------------------------- /cartridges/data/mcp/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/mcp/tools.py -------------------------------------------------------------------------------- /cartridges/data/mtob/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/data/mtob/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/mtob/baseline.py -------------------------------------------------------------------------------- /cartridges/data/mtob/evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/mtob/evals.py -------------------------------------------------------------------------------- /cartridges/data/mtob/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/mtob/load.py -------------------------------------------------------------------------------- /cartridges/data/mtob/requirements.txt: -------------------------------------------------------------------------------- 1 | sacrebleu -------------------------------------------------------------------------------- /cartridges/data/mtob/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/mtob/resources.py -------------------------------------------------------------------------------- /cartridges/data/qasper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/data/qasper/evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/qasper/evals.py -------------------------------------------------------------------------------- /cartridges/data/qasper/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/qasper/resources.py -------------------------------------------------------------------------------- /cartridges/data/qasper/rewrite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/qasper/rewrite.py -------------------------------------------------------------------------------- /cartridges/data/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/resources.py -------------------------------------------------------------------------------- /cartridges/data/retrieval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/retrieval/__init__.py -------------------------------------------------------------------------------- /cartridges/data/retrieval/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/retrieval/load.py -------------------------------------------------------------------------------- /cartridges/data/retrieval/retrievers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/retrieval/retrievers.py -------------------------------------------------------------------------------- /cartridges/data/retrieval/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/retrieval/tools.py -------------------------------------------------------------------------------- /cartridges/data/ruler/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/ruler/README.md -------------------------------------------------------------------------------- /cartridges/data/ruler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/data/ruler/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/ruler/constants.py -------------------------------------------------------------------------------- /cartridges/data/ruler/download_paulgraham_essay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/ruler/download_paulgraham_essay.py -------------------------------------------------------------------------------- /cartridges/data/ruler/evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/ruler/evals.py -------------------------------------------------------------------------------- /cartridges/data/ruler/niah.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/ruler/niah.py -------------------------------------------------------------------------------- /cartridges/data/ruler/requirements.txt: -------------------------------------------------------------------------------- 1 | html2 2 | bs4 3 | nltk 4 | wonderwords -------------------------------------------------------------------------------- /cartridges/data/ruler/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/ruler/resources.py -------------------------------------------------------------------------------- /cartridges/data/ruler/variable_tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/ruler/variable_tracking.py -------------------------------------------------------------------------------- /cartridges/data/slack/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/slack/resources.py -------------------------------------------------------------------------------- /cartridges/data/slack/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/slack/tools.py -------------------------------------------------------------------------------- /cartridges/data/tex/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/data/tex/evals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/tex/evals.py -------------------------------------------------------------------------------- /cartridges/data/tex/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/tex/resources.py -------------------------------------------------------------------------------- /cartridges/data/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/data/tools.py -------------------------------------------------------------------------------- /cartridges/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/datasets.py -------------------------------------------------------------------------------- /cartridges/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/evaluate.py -------------------------------------------------------------------------------- /cartridges/generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/generation.py -------------------------------------------------------------------------------- /cartridges/initialization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/initialization/__init__.py -------------------------------------------------------------------------------- /cartridges/initialization/data/gradient.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/initialization/data/gradient.txt -------------------------------------------------------------------------------- /cartridges/initialization/pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/initialization/pretrained.py -------------------------------------------------------------------------------- /cartridges/initialization/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/initialization/random.py -------------------------------------------------------------------------------- /cartridges/initialization/text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/initialization/text.py -------------------------------------------------------------------------------- /cartridges/initialization/tokenization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/initialization/tokenization_utils.py -------------------------------------------------------------------------------- /cartridges/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/models/__init__.py -------------------------------------------------------------------------------- /cartridges/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/models/attention.py -------------------------------------------------------------------------------- /cartridges/models/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/models/config.py -------------------------------------------------------------------------------- /cartridges/models/llama/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/models/llama/configuration_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/models/llama/configuration_llama.py -------------------------------------------------------------------------------- /cartridges/models/llama/modeling_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/models/llama/modeling_llama.py -------------------------------------------------------------------------------- /cartridges/models/qwen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/models/qwen/configuration_qwen3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/models/qwen/configuration_qwen3.py -------------------------------------------------------------------------------- /cartridges/models/qwen/modeling_qwen3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/models/qwen/modeling_qwen3.py -------------------------------------------------------------------------------- /cartridges/structs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/structs.py -------------------------------------------------------------------------------- /cartridges/synthesize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/synthesize.py -------------------------------------------------------------------------------- /cartridges/synthesizers/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /cartridges/synthesizers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/synthesizers/base.py -------------------------------------------------------------------------------- /cartridges/synthesizers/self_study.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/synthesizers/self_study.py -------------------------------------------------------------------------------- /cartridges/synthesizers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/synthesizers/utils.py -------------------------------------------------------------------------------- /cartridges/tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/README.md -------------------------------------------------------------------------------- /cartridges/tests/__init__.py: -------------------------------------------------------------------------------- 1 | """Test package for cartridges.""" -------------------------------------------------------------------------------- /cartridges/tests/clients/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/clients/test_base.py -------------------------------------------------------------------------------- /cartridges/tests/clients/test_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/clients/test_openai.py -------------------------------------------------------------------------------- /cartridges/tests/clients/test_tokasaurus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/clients/test_tokasaurus.py -------------------------------------------------------------------------------- /cartridges/tests/mocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/mocks.py -------------------------------------------------------------------------------- /cartridges/tests/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/models/common.py -------------------------------------------------------------------------------- /cartridges/tests/models/test_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/models/test_llama.py -------------------------------------------------------------------------------- /cartridges/tests/models/test_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/models/test_qwen.py -------------------------------------------------------------------------------- /cartridges/tests/resources/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/tests/resources/gmail/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/tests/resources/gmail/test_gmail.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/resources/gmail/test_gmail.py -------------------------------------------------------------------------------- /cartridges/tests/resources/mcp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/tests/resources/mcp/test_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/resources/mcp/test_tools.py -------------------------------------------------------------------------------- /cartridges/tests/resources/slack/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cartridges/tests/resources/slack/test_slack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/resources/slack/test_slack.py -------------------------------------------------------------------------------- /cartridges/tests/resources/test_resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/resources/test_resources.py -------------------------------------------------------------------------------- /cartridges/tests/synthesizers/__init__.py: -------------------------------------------------------------------------------- 1 | """Tests for synthesizers.""" -------------------------------------------------------------------------------- /cartridges/tests/synthesizers/test_self_study.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/tests/synthesizers/test_self_study.py -------------------------------------------------------------------------------- /cartridges/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/train.py -------------------------------------------------------------------------------- /cartridges/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/utils/__init__.py -------------------------------------------------------------------------------- /cartridges/utils/cache_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/utils/cache_size.py -------------------------------------------------------------------------------- /cartridges/utils/chat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/utils/chat.py -------------------------------------------------------------------------------- /cartridges/utils/hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/utils/hf.py -------------------------------------------------------------------------------- /cartridges/utils/llm_messages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/utils/llm_messages.py -------------------------------------------------------------------------------- /cartridges/utils/thinking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/utils/thinking.py -------------------------------------------------------------------------------- /cartridges/utils/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/cartridges/utils/wandb.py -------------------------------------------------------------------------------- /docs/training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/docs/training.md -------------------------------------------------------------------------------- /examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/arxiv/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/arxiv/README.md -------------------------------------------------------------------------------- /examples/arxiv/arxiv_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/arxiv/arxiv_baseline.py -------------------------------------------------------------------------------- /examples/arxiv/arxiv_synthesize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/arxiv/arxiv_synthesize.py -------------------------------------------------------------------------------- /examples/arxiv/arxiv_synthesize_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/arxiv/arxiv_synthesize_eval.py -------------------------------------------------------------------------------- /examples/arxiv/arxiv_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/arxiv/arxiv_train.py -------------------------------------------------------------------------------- /examples/arxiv/cartridges.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/arxiv/cartridges.tex -------------------------------------------------------------------------------- /examples/benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /examples/benchmarks/longhealth/baseline_longhealth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/longhealth/baseline_longhealth.py -------------------------------------------------------------------------------- /examples/benchmarks/longhealth/longhealth_synthesize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/longhealth/longhealth_synthesize.py -------------------------------------------------------------------------------- /examples/benchmarks/longhealth/longhealth_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/longhealth/longhealth_train.py -------------------------------------------------------------------------------- /examples/benchmarks/mtob/mtob_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/mtob/mtob_baseline.py -------------------------------------------------------------------------------- /examples/benchmarks/mtob/mtob_synthesize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/mtob/mtob_synthesize.py -------------------------------------------------------------------------------- /examples/benchmarks/mtob/mtob_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/mtob/mtob_train.py -------------------------------------------------------------------------------- /examples/benchmarks/niah/niah_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/niah/niah_baseline.py -------------------------------------------------------------------------------- /examples/benchmarks/niah/niah_synthesize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/niah/niah_synthesize.py -------------------------------------------------------------------------------- /examples/benchmarks/niah/niah_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/benchmarks/niah/niah_train.py -------------------------------------------------------------------------------- /examples/cartridge_chat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/examples/cartridge_chat.py -------------------------------------------------------------------------------- /infra/modal_deploy_sglang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/infra/modal_deploy_sglang.py -------------------------------------------------------------------------------- /infra/modal_deploy_tokasaurus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/infra/modal_deploy_tokasaurus.py -------------------------------------------------------------------------------- /infra/modal_deploy_tokasaurus_flash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/infra/modal_deploy_tokasaurus_flash.py -------------------------------------------------------------------------------- /infra/modal_deploy_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/infra/modal_deploy_vllm.py -------------------------------------------------------------------------------- /infra/modal_ssh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/infra/modal_ssh.py -------------------------------------------------------------------------------- /infra/tuning/examples/longhealth_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/infra/tuning/examples/longhealth_tune.py -------------------------------------------------------------------------------- /infra/tuning/tune_toka.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/infra/tuning/tune_toka.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/pyproject.toml -------------------------------------------------------------------------------- /viz/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/.gitignore -------------------------------------------------------------------------------- /viz/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/README.md -------------------------------------------------------------------------------- /viz/eslint.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/eslint.config.js -------------------------------------------------------------------------------- /viz/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/index.html -------------------------------------------------------------------------------- /viz/package-lock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/package-lock.json -------------------------------------------------------------------------------- /viz/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/package.json -------------------------------------------------------------------------------- /viz/public/vite.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/public/vite.svg -------------------------------------------------------------------------------- /viz/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/requirements.txt -------------------------------------------------------------------------------- /viz/src/App.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/App.css -------------------------------------------------------------------------------- /viz/src/App.jsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/App.jsx -------------------------------------------------------------------------------- /viz/src/assets/react.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/assets/react.svg -------------------------------------------------------------------------------- /viz/src/dashboards/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/dashboards/__init__.py -------------------------------------------------------------------------------- /viz/src/dashboards/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/dashboards/base.py -------------------------------------------------------------------------------- /viz/src/dashboards/codehop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/dashboards/codehop.py -------------------------------------------------------------------------------- /viz/src/index.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/index.css -------------------------------------------------------------------------------- /viz/src/main.jsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/main.jsx -------------------------------------------------------------------------------- /viz/src/pages/DatasetsPage.jsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/pages/DatasetsPage.jsx -------------------------------------------------------------------------------- /viz/src/pages/TrainingPage.jsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/pages/TrainingPage.jsx -------------------------------------------------------------------------------- /viz/src/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/src/server.py -------------------------------------------------------------------------------- /viz/start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/start.sh -------------------------------------------------------------------------------- /viz/vite.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/cartridges/HEAD/viz/vite.config.js --------------------------------------------------------------------------------