nit

Signed-off-by: David Korczynski <david@adalogics.com>
google · Jul 15, 2024 · b8b8081 · b8b8081
1 parent 19fc62e
commit b8b8081
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_toolkit/corpus_generator.py b/llm_toolkit/corpus_generator.py
@@ -72,7 +72,7 @@ def get_corpus_generator_script(
   prompt_path = os.path.join(response_dir, 'prompt.txt')
   prompt.save(prompt_path)
 
-  corpus_model.generate_code(prompt, response_dir)
+  corpus_model.query_llm(prompt, response_dir)
   for file in os.listdir(response_dir):
     if not parser.is_raw_output(file):
       continue