from piapi_ai.Entity import Entity


class Flux(Entity):

    def create_task(self, data):
        result = self.raw_request("post", data=data)

        return result

    def get_task(self, task_id):
        self.endpoint = self.endpoint + "/" + task_id
        result = self.raw_request("get")

        return result