Mayo commited on
Commit
ccd8887
·
unverified ·
1 Parent(s): f46098f

chore: fix tests

Browse files
Files changed (2) hide show
  1. koharu-llm/bin/llm.rs +1 -1
  2. 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"