25d74480aa
This PR restructures the commands of the Edit Prediction CLI (now called `ep`), to support some flows that are important for the training process: * generating zeta2 prompt and expected output, without running predictions * scoring outputs that are generated by a system other than the production code (to evaluate the model during training) To achieve this, we've restructured the CLI commands so that they all take as input, and produce as output, a consistent, uniform data format: a set of one or more `Example` structs, expressible either as the original markdown format, or as a JSON lines. The `Example` struct starts with the basic fields that are in human-readable eval format, but contain a number of optional fields that are filled in by different steps in the processing pipeline (`context`, `predict`, `format-prompt`, and `score`). ### To do * [x] Adjust the teacher model output parsing to use the full buffer contents * [x] Move udiff to cli * [x] Align `format-prompt` with Zeta2's production code * [x] Change score output to assume same provider * [x] Move pretty reporting to `eval` command * [x] Store cursor point in addition to cursor offset * [x] Rename `edit_prediction_cli2` -> `edit_prediction_cli` (nuke the old one) Release Notes: - N/A --------- Co-authored-by: Oleksiy Syvokon <oleksiy@zed.dev> Co-authored-by: Agus Zubiaga <agus@zed.dev> Co-authored-by: Ben Kunkle <ben@zed.dev>
73 lines
1.9 KiB
TOML
73 lines
1.9 KiB
TOML
[package]
|
|
name = "edit_prediction_cli"
|
|
version = "0.1.0"
|
|
edition.workspace = true
|
|
publish.workspace = true
|
|
license = "GPL-3.0-or-later"
|
|
|
|
[lints]
|
|
workspace = true
|
|
|
|
[[bin]]
|
|
name = "ep"
|
|
path = "src/main.rs"
|
|
|
|
[dependencies]
|
|
anyhow.workspace = true
|
|
anthropic.workspace = true
|
|
http_client.workspace = true
|
|
chrono.workspace = true
|
|
clap.workspace = true
|
|
client.workspace = true
|
|
cloud_llm_client.workspace= true
|
|
collections.workspace = true
|
|
debug_adapter_extension.workspace = true
|
|
dirs.workspace = true
|
|
extension.workspace = true
|
|
fs.workspace = true
|
|
futures.workspace = true
|
|
gpui.workspace = true
|
|
gpui_tokio.workspace = true
|
|
indoc.workspace = true
|
|
language.workspace = true
|
|
language_extension.workspace = true
|
|
language_model.workspace = true
|
|
language_models.workspace = true
|
|
languages = { workspace = true, features = ["load-grammars"] }
|
|
log.workspace = true
|
|
node_runtime.workspace = true
|
|
paths.workspace = true
|
|
project.workspace = true
|
|
prompt_store.workspace = true
|
|
pulldown-cmark.workspace = true
|
|
release_channel.workspace = true
|
|
reqwest_client.workspace = true
|
|
serde.workspace = true
|
|
serde_json.workspace = true
|
|
settings.workspace = true
|
|
shellexpand.workspace = true
|
|
smol.workspace = true
|
|
sqlez.workspace = true
|
|
sqlez_macros.workspace = true
|
|
terminal_view.workspace = true
|
|
util.workspace = true
|
|
watch.workspace = true
|
|
edit_prediction = { workspace = true, features = ["eval-support"] }
|
|
wasmtime.workspace = true
|
|
zeta_prompt.workspace = true
|
|
zlog.workspace = true
|
|
|
|
# Wasmtime is included as a dependency in order to enable the same
|
|
# features that are enabled in Zed.
|
|
#
|
|
# If we don't enable these features we get crashes when creating
|
|
# a Tree-sitter WasmStore.
|
|
[package.metadata.cargo-machete]
|
|
ignored = ["wasmtime"]
|
|
|
|
[dev-dependencies]
|
|
indoc.workspace = true
|
|
gpui = { workspace = true, features = ["test-support"] }
|
|
project = { workspace = true, features = ["test-support"] }
|
|
pretty_assertions.workspace = true
|