flytekitplugins.bigquery.BigQueryAgent#

class flytekitplugins.bigquery.BigQueryAgent[source]#

Methods

create(output_prefix, task_template, inputs=None, **kwargs)[source]#

Return a Unique ID for the task that was created. It should return error code if the task creation failed.

Parameters

  • output_prefix (str) –

  • task_template (flytekit.models.task.TaskTemplate) –

  • inputs (Optional[flytekit.models.literals.LiteralMap]) –

Return type

flyteidl.admin.agent_pb2.CreateTaskResponse

delete(resource_meta, **kwargs)[source]#

Delete the task. This call should be idempotent.

Parameters

resource_meta (bytes) –

Return type

flyteidl.admin.agent_pb2.DeleteTaskResponse

get(resource_meta, **kwargs)[source]#

Return the status of the task, and return the outputs in some cases. For example, bigquery job can’t write the structured dataset to the output location, so it returns the output literals to the propeller, and the propeller will write the structured dataset to the blob store.

Parameters

resource_meta (bytes) –

Return type

flyteidl.admin.agent_pb2.GetTaskResponse

Attributes

name = 'Bigquery Agent'
task_type

task_type is the name of the task type that this agent supports.