Skip to content

Commit

Permalink
Deployment documentation for Bigquery agent (#4034)
Browse files Browse the repository at this point in the history
Signed-off-by: Kevin Su <[email protected]>
  • Loading branch information
pingsutw authored Sep 25, 2023
1 parent ef000f3 commit 86e3a2c
Show file tree
Hide file tree
Showing 4 changed files with 126 additions and 0 deletions.
90 changes: 90 additions & 0 deletions rsts/deployment/agents/bigquery.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,90 @@
.. _deployment-agent-setup-bigquery:

Google BigQuery Agent
======================

This guide provides an overview of setting up BigQuery agent in your Flyte deployment.
Please note that the BigQuery agent requires Flyte deployment in the GCP cloud;
it is not compatible with demo/AWS/Azure.

Set up the GCP Flyte cluster
----------------------------

* Ensure you have a functional Flyte cluster running in `GCP <https://docs.flyte.org/en/latest/deployment/gcp/index.html#deployment-gcp>`__.
* Create a service account for BigQuery. For more details, refer to: https://cloud.google.com/bigquery/docs/quickstarts/quickstart-client-libraries.
* Verify that you have the correct kubeconfig and have selected the appropriate Kubernetes context.
* Confirm that you have the correct Flytectl configuration at ``~/.flyte/config.yaml``.

Specify agent configuration
----------------------------

.. tabs::

.. group-tab:: Flyte binary

Edit the relevant YAML file to specify the agent.

.. code-block:: yaml
:emphasize-lines: 7,11
tasks:
task-plugins:
enabled-plugins:
- container
- sidecar
- k8s-array
- bigquery
default-for-task-types:
- container: container
- container_array: k8s-array
- bigquery_query_job_task: agent-service
.. group-tab:: Flyte core

Create a file named ``values-override.yaml`` and add the following configuration to it.

.. code-block:: yaml
configmap:
enabled_plugins:
# -- Tasks specific configuration [structure](https://pkg.go.dev/github.com/flyteorg/flytepropeller/pkg/controller/nodes/task/config#GetConfig)
tasks:
# -- Plugins configuration, [structure](https://pkg.go.dev/github.com/flyteorg/flytepropeller/pkg/controller/nodes/task/config#TaskPluginConfig)
task-plugins:
# -- [Enabled Plugins](https://pkg.go.dev/github.com/flyteorg/flyteplugins/go/tasks/config#Config). Enable sagemaker*, athena if you install the backend
enabled-plugins:
- container
- sidecar
- k8s-array
- bigquery
default-for-task-types:
container: container
sidecar: sidecar
container_array: k8s-array
bigquery_query_job_task: agent-service
Ensure that the propeller has the correct service account for BigQuery.

Upgrade the Flyte Helm release
------------------------------

.. tabs::

.. group-tab:: Flyte binary

.. code-block:: bash
helm upgrade <RELEASE_NAME> flyteorg/flyte-binary -n <YOUR_NAMESPACE> --values <YOUR_YAML_FILE>
Replace ``<RELEASE_NAME>`` with the name of your release (e.g., ``flyte-backend``),
``<YOUR_NAMESPACE>`` with the name of your namespace (e.g., ``flyte``),
and ``<YOUR_YAML_FILE>`` with the name of your YAML file.

.. group-tab:: Flyte core

.. code-block:: bash
helm upgrade <RELEASE_NAME> flyte/flyte-core -n <YOUR_NAMESPACE> --values values-override.yaml
Replace ``<RELEASE_NAME>`` with the name of your release (e.g., ``flyte``)
and ``<YOUR_NAMESPACE>`` with the name of your namespace (e.g., ``flyte``).
26 changes: 26 additions & 0 deletions rsts/deployment/agents/index.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
.. _deployment-agent-setup:

Agent Setup
===========

.. tags:: Agent, Integration, Data, Advanced

Discover the process of setting up Agents for Flyte.

.. panels::
:header: text-center
:column: col-lg-12 p-2

.. link-button:: deployment-agent-setup-bigquery
:type: ref
:text: Bigquery Agent
:classes: btn-block stretched-link
^^^^^^^^^^^^
Guide to setting up the Bigquery agent.

.. toctree::
:maxdepth: 1
:name: Agent setup
:hidden:

bigquery
9 changes: 9 additions & 0 deletions rsts/deployment/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,15 @@ plugins, authentication, performance tuning, and maintaining Flyte as a producti

---

.. link-button:: deployment-agent-setup
:type: ref
:text: 🤖 Agent Setup
:classes: btn-block stretched-link
^^^^^^^^^^^^
Enable Flyte agents to extend Flyte's capabilities, including features like File sesnor, Databricks job, and Snowflake query services.

---

.. link-button:: deployment-configuration
:type: ref
:text: 🎛 Cluster Configuration
Expand Down
1 change: 1 addition & 0 deletions rsts/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@
deployment/index
deployment/deployment/index
deployment/plugins/index
deployment/agents/index
deployment/configuration/index
deployment/configuration/generated/index
deployment/security/index
Expand Down

0 comments on commit 86e3a2c

Please sign in to comment.