oak-team/oak-gpui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

zeta cli: Print log paths when running predict (#42396)

Browse Source 
Release Notes:

- N/A

Co-authored-by: Michael Sloan <mgsloan@gmail.com>
Co-authored-by: Ben Kunkle <ben@zed.dev>
This commit is contained in:
Agus Zubiaga
2025-11-11 09:56:20 -03:00
committed by GitHub
co-authored by Michael Sloan Ben Kunkle
parent 777b46533f
commit f2ad0d716f
2 changed files with 25 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/zeta_cli/src/paths.rs
+8
View File
@@ -6,3 +6,11 @@ pub static CACHE_DIR: LazyLock<PathBuf> =
pub static REPOS_DIR: LazyLock<PathBuf> = LazyLock::new(|| TARGET_DIR.join("zeta-repos"));
pub static WORKTREES_DIR: LazyLock<PathBuf> = LazyLock::new(|| TARGET_DIR.join("zeta-worktrees"));
pub static LOGS_DIR: LazyLock<PathBuf> = LazyLock::new(|| TARGET_DIR.join("zeta-logs"));
pub static LOGS_SEARCH_PROMPT: LazyLock<PathBuf> =
LazyLock::new(|| LOGS_DIR.join("search_prompt.md"));
pub static LOGS_SEARCH_QUERIES: LazyLock<PathBuf> =
LazyLock::new(|| LOGS_DIR.join("search_queries.json"));
pub static LOGS_PREDICTION_PROMPT: LazyLock<PathBuf> =
LazyLock::new(|| LOGS_DIR.join("prediction_prompt.md"));
pub static LOGS_PREDICTION_RESPONSE: LazyLock<PathBuf> =
LazyLock::new(|| LOGS_DIR.join("prediction_response.md"));
crates/zeta_cli/src/predict.rs
+17 -5
View File
@@ -1,7 +1,10 @@
use crate::PromptFormat;
use crate::example::{ActualExcerpt, ExpectedExcerpt, NamedExample};
use crate::headless::ZetaCliAppState;
use crate::paths::{CACHE_DIR, LOGS_DIR};
use crate::paths::{
CACHE_DIR, LOGS_DIR, LOGS_PREDICTION_PROMPT, LOGS_PREDICTION_RESPONSE, LOGS_SEARCH_PROMPT,
LOGS_SEARCH_QUERIES,
};
use ::serde::Serialize;
use anyhow::{Result, anyhow};
use clap::Args;
@@ -61,6 +64,15 @@ pub async fn run_zeta2_predict(
.await
.unwrap();
result.write(args.format, std::io::stdout()).unwrap();
println!("## Logs\n");
println!("Search prompt: {}", LOGS_SEARCH_PROMPT.display());
println!("Search queries: {}", LOGS_SEARCH_QUERIES.display());
println!("Prediction prompt: {}", LOGS_PREDICTION_PROMPT.display());
println!(
"Prediction response: {}",
LOGS_PREDICTION_RESPONSE.display()
);
}
thread_local! {
@@ -147,12 +159,12 @@ pub async fn zeta2_predict(
match event {
zeta2::ZetaDebugInfo::ContextRetrievalStarted(info) => {
start_time = Some(info.timestamp);
fs::write(LOGS_DIR.join("search_prompt.md"), &info.search_prompt)?;
fs::write(&*LOGS_SEARCH_PROMPT, &info.search_prompt)?;
}
zeta2::ZetaDebugInfo::SearchQueriesGenerated(info) => {
search_queries_generated_at = Some(info.timestamp);
fs::write(
LOGS_DIR.join("search_queries.json"),
&*LOGS_SEARCH_QUERIES,
serde_json::to_string_pretty(&info.search_queries).unwrap(),
)?;
}
@@ -164,7 +176,7 @@ pub async fn zeta2_predict(
let prediction_started_at = Instant::now();
start_time.get_or_insert(prediction_started_at);
fs::write(
LOGS_DIR.join("prediction_prompt.md"),
&*LOGS_PREDICTION_PROMPT,
&request.local_prompt.unwrap_or_default(),
)?;
@@ -198,7 +210,7 @@ pub async fn zeta2_predict(
let response = request.response_rx.await?.0.map_err(|err| anyhow!(err))?;
let response = zeta2::text_from_response(response).unwrap_or_default();
let prediction_finished_at = Instant::now();
fs::write(LOGS_DIR.join("prediction_response.md"), &response)?;
fs::write(&*LOGS_PREDICTION_RESPONSE, &response)?;
let mut result = result.lock().unwrap();