From 5fd57e468a66961368bd17243de897976f792dad Mon Sep 17 00:00:00 2001 From: Michal Moskal Date: Wed, 31 Jan 2024 00:08:22 +0000 Subject: [PATCH] merge fixes --- promptlib/promptlib/aici.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/promptlib/promptlib/aici.py b/promptlib/promptlib/aici.py index 4e1f32c2..49a5e850 100644 --- a/promptlib/promptlib/aici.py +++ b/promptlib/promptlib/aici.py @@ -56,5 +56,5 @@ def _upload_wasm(base_url, wasm_runner_path): ) -def _submit_program(base_url, aici_module, aici_arg, temperature=0, max_tokens=200, n=1, log=False): - return aici_rest.completion("", aici_module, aici_arg, temperature, max_tokens, n, ignore_eos=False, base_url=base_url) +def _submit_program(base_url, aici_module, aici_arg, temperature=0, max_tokens=200, log=False): + return aici_rest.run_controller(controller=aici_module, controller_arg=aici_arg, temperature=temperature, max_tokens=max_tokens, base_url=base_url)