Add paint function for agent David

src/aios_kernel/compute_task.py

@@ -64,6 +64,17 @@ class ComputeTask:
         else:
             self.params["model_name"] = "text-embedding-ada-002"
         self.params["input"] = input
+    
+    def set_text_2_image_params(self, prompt: str, model_name, callchain_id=None):
+        self.task_type = ComputeTaskType.TEXT_2_IMAGE
+        self.create_time = time.time()
+        self.task_id = uuid.uuid4().hex
+        self.callchain_id = callchain_id
+        self.params["prompt"] = prompt
+        if model_name is not None:
+            self.params["model_name"] = model_name
+        else:
+            self.params["model_name"] = "v1-5-pruned-emaonly"
 
     def display(self) -> str:
         return f"ComputeTask: {self.task_id} {self.task_type} {self.state}"