Google BigQuery agent#

Note

To use the BigQuery agent, you must pip install flytekitplugins-bigquery.

class flytekitplugins.bigquery.BigQueryConfig(ProjectID, Location=None, QueryJobConfig=None)#

BigQueryConfig should be used to configure a BigQuery Task.

Parameters:

  • ProjectID (str)

  • Location (str | None)

  • QueryJobConfig (QueryJobConfig | None)

class flytekitplugins.bigquery.BigQueryTask(*args, **kwargs)#

This is the simplest form of a BigQuery Task, that can be used even for tasks that do not produce any output.

get_custom(settings)#

Return additional plugin-specific custom data (if any) as a serializable dictionary.

Parameters:

settings (SerializationSettings)

Return type:

Dict[str, Any]

get_sql(settings)#

Returns the Sql definition (if any) that is used to run the task on hosted Flyte.

Parameters:

settings (SerializationSettings)

Return type:

Sql | None