Add embeding compute task

This commit is contained in:
tsukasa
2023-08-31 15:45:02 +08:00
parent 39eb96f9e3
commit febe28eecc
3 changed files with 40 additions and 24 deletions
+8
View File
@@ -38,6 +38,14 @@ class ComputeTask:
self.params["model_name"] = "gpt-4-0613"
self.params["max_token_size"] = max_token_size
def set_embeding_params(self, model_name, input, callchain_id = None):
self.task_type = "embeding"
self.create_time = time.time()
self.task_id = uuid.uuid4().hex
self.callchain_id = callchain_id
self.params["model_name"] = model_name
self.params["input"] = input
def display(self) -> str:
return f"ComputeTask: {self.task_id} {self.task_type} {self.state}"