diff --git a/src/aios_kernel/compute_task.py b/src/aios_kernel/compute_task.py index c6053db..dd875a2 100644 --- a/src/aios_kernel/compute_task.py +++ b/src/aios_kernel/compute_task.py @@ -63,6 +63,7 @@ class ComputeTask: self.params["model_name"] = model_name else: self.params["model_name"] = "text-embedding-ada-002" + self.params["input"] = input def display(self) -> str: