Mayo commited on
chore: fix tests
Browse files- koharu-llm/bin/llm.rs +1 -1
- koharu-llm/tests/llm.rs +1 -1
koharu-llm/bin/llm.rs
CHANGED
|
@@ -110,7 +110,7 @@ async fn main() -> anyhow::Result<()> {
|
|
| 110 |
..args.model.default_generate_options()
|
| 111 |
};
|
| 112 |
|
| 113 |
-
let out = llm.generate(&args.prompt, &opts, target_language)?;
|
| 114 |
|
| 115 |
println!("{}", out);
|
| 116 |
println!(
|
|
|
|
| 110 |
..args.model.default_generate_options()
|
| 111 |
};
|
| 112 |
|
| 113 |
+
let out = llm.generate(&args.prompt, &opts, target_language, None)?;
|
| 114 |
|
| 115 |
println!("{}", out);
|
| 116 |
println!(
|
koharu-llm/tests/llm.rs
CHANGED
|
@@ -27,7 +27,7 @@ async fn llm_generates_text_for_all_models() -> anyhow::Result<()> {
|
|
| 27 |
..model.default_generate_options()
|
| 28 |
};
|
| 29 |
|
| 30 |
-
let generated = llm.generate(prompt, &opts, Language::English)?;
|
| 31 |
assert!(
|
| 32 |
!generated.trim().is_empty(),
|
| 33 |
"model {model:?} should return some text"
|
|
|
|
| 27 |
..model.default_generate_options()
|
| 28 |
};
|
| 29 |
|
| 30 |
+
let generated = llm.generate(prompt, &opts, Language::English, None)?;
|
| 31 |
assert!(
|
| 32 |
!generated.trim().is_empty(),
|
| 33 |
"model {model:?} should return some text"
|