mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-11-24 06:19:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

🎨

Browse source
This commit is contained in:
Antonio Scandurra 2024-11-22 16:33:31 +01:00
parent 1bc342ff5a
commit 79a70b72b3
1 changed files with 23 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
crates/zeta/src/zeta.rs
View file

@ -912,7 +912,7 @@ mod tests {
#[gpui::test]
async fn test_new_cli_arg(cx: &mut TestAppContext) {
cx.executor().allow_parking();
let zeta = zeta(cx);
let zeta = Zeta::test(cx);
let buffer = open_buffer(
"crates/cli/src/main.rs",
@ -971,7 +971,7 @@ mod tests {
cx: &mut TestAppContext,
) {
cx.executor().allow_parking();
let zeta = zeta(cx);
let zeta = Zeta::test(cx);
let buffer = open_buffer(filename, initial, &zeta, cx);
let cursor_start = edited
@ -992,7 +992,7 @@ mod tests {
cx: &mut TestAppContext,
) {
cx.executor().allow_parking();
let zeta = zeta(cx);
let zeta = Zeta::test(cx);
let buffer = open_buffer(filename, initial, &zeta, cx);
let cursor_start = edited
@ -1081,24 +1081,26 @@ mod tests {
);
}
fn zeta(cx: &mut TestAppContext) -> Model<Zeta> {
cx.new_model(|_| {
let (api_url, api_key, model) = match std::env::var("FIREWORKS_API_KEY") {
Ok(api_key) => (
Arc::from("https://api.fireworks.ai/inference/v1"),
Arc::from(api_key),
Arc::from(std::env::var("FIREWORKS_MODEL").unwrap_or_else(|_| {
"accounts/fireworks/models/qwen2p5-coder-32b-instruct".to_string()
})),
),
Err(_) => (
Arc::from("http://localhost:11434/v1"),
Arc::from(""),
Arc::from("qwen2.5-coder:32b"),
),
};
Zeta::new(api_url, api_key, model, Arc::new(ReqwestClient::new()))
})
impl Zeta {
fn test(cx: &mut TestAppContext) -> Model<Zeta> {
cx.new_model(|_| {
let (api_url, api_key, model) = match std::env::var("FIREWORKS_API_KEY") {
Ok(api_key) => (
Arc::from("https://api.fireworks.ai/inference/v1"),
Arc::from(api_key),
Arc::from(std::env::var("FIREWORKS_MODEL").unwrap_or_else(|_| {
"accounts/fireworks/models/qwen2p5-coder-32b-instruct".to_string()
})),
),
Err(_) => (
Arc::from("http://localhost:11434/v1"),
Arc::from(""),
Arc::from("qwen2.5-coder:32b"),
),
};
Zeta::new(api_url, api_key, model, Arc::new(ReqwestClient::new()))
})
}
}
fn edit(buffer: &Model<Buffer>, text: &str, cx: &mut TestAppContext) {