Data Engineering Agent – Übersicht

Mit dem Data-Engineering-Agenten können Sie Datenpipelines in BigQuery mithilfe von Prompts in natürlicher Sprache erstellen, ändern und Fehler beheben. Der Data-Engineering-Agent bietet die folgenden Funktionen, um Ihre Data-Engineering-Workflows für die Aufnahme von Daten in BigQuery zu optimieren:

  • Dataform-Integration: Der Agent generiert und organisiert Datenpipeline code direkt in Dataform-Repositories und -Arbeitsbereichen.
  • Planerstellung: Der Agent kann seine Überlegungen zusammenfassen und einen Plan erstellen, mit dem Sie den Plan des Agenten überprüfen können, bevor Sie fortfahren.
  • Codevalidierung: Der Agent validiert und korrigiert automatisch Kompilierungs fehler in jedem generierten Code, um sicherzustellen, dass die Datenpipeline funktioniert.
  • Automatische Datenaufbereitung: Der Agent führt die Datenaufbereitung durch und wandelt Rohdaten ohne manuelle Eingriffe in strukturierte Tabellen um.
  • Benutzerdefinierte Anweisungen: Der Agent unterstützt benutzerdefinierte Anweisungen, mit denen Sie bestimmte Regeln und wiederverwendbare Richtlinien in natürlicher Sprache definieren können.
  • Externer Kontext: Der Agent ist für zusätzlichen Kontext in Knowledge Catalog integriert.
  • Pipeline-Steuerung: Sie können generierte Agentenpläne überprüfen und anpassen bevor Aktionen ausgeführt werden.
  • Optimierung: Der Agent kann die Leistung Ihrer Datenpipeline optimieren.
  • Fehlerbehebung und Reparatur: Der Agent kann Fehler in Pipelines beheben und den Code korrigieren.

Wo kann der Data-Engineering-Agent verwendet werden?

Sie können den Data-Engineering-Agenten mit den folgenden Methoden verwenden:

So verwendet der Data-Engineering-Agent Ihre Daten

Um qualitativ hochwertigere Antworten zu generieren, kann der Data-Engineering-Agent zusätzliche Daten und Metadaten aus BigQuery und Knowledge Catalog abrufen, darunter Beispielzeilen aus BigQuery-Tabellen und Datenprofilscans, die in Knowledge Catalog generiert wurden. Der Agent verwendet diese Daten nicht für das Training, sondern nur als zusätzlichen Kontext während der Agentenkonversationen, um seine Antworten zu verbessern.

Wo verarbeitet der Data-Engineering-Agent Ihre Daten?

Weitere Informationen zu den Standorten, an denen der Data-Engineering-Agent Ihre Daten verarbeitet, finden Sie unter Wo verarbeitet Gemini in BigQuery Ihre Daten.

Beschränkungen

Für den Data-Engineering-Agenten gelten die folgenden Einschränkungen:

  • Der Data-Engineering-Agent unterstützt keine Befehle in natürlicher Sprache für die folgenden Dateitypen:
    • Notebooks
    • Datenvorbereitung
  • Der Data-Engineering-Agent kann keine Pipelines ausführen. Sie müssen Pipelines überprüfen und ausführen oder planen.
  • Der Data-Engineering-Agent kann keine Weblinks oder URLs durchsuchen, die in Anweisungen oder direkten Prompts angegeben sind.
  • Beim Importieren von Dateien in einer Agentenanweisungsdatei, unterstützt die @ Importsyntax nur Pfade, die mit ./, / oder einem Buchstaben beginnen.
  • Die Datenvorschau-Funktion wird nur für Tabellen, Deklarationen oder Abfragen unterstützt, bei denen das Flag hasOutput auf true gesetzt ist.
  • Der Data-Engineering-Agent unterliegt den allgemeinen Einschränkungen der KI Technologie.

Agentenfunktionen und -anpassungen

In den folgenden Abschnitten werden zusätzliche Agentenfunktionen und andere Methoden zum Anpassen des Data-Engineering-Agenten beschrieben.

Anweisungen für KI-Agenten

Anweisungen für KI-Agenten sind Anweisungen in natürlicher Sprache für den Data-Engineering-Agenten, mit denen Sie dauerhafte Anweisungen speichern können, damit der Agent eine Reihe benutzerdefinierter, vordefinierter Regeln befolgt. Verwenden Sie Anweisungen für KI-Agenten, wenn die Ergebnisse des Agenten in Ihrer gesamten Organisation einheitlich sein sollen, z. B. bei Namenskonventionen oder um einen Styleguide zu erzwingen.

Erstellen Sie eine GEMINI.MD Kontext Datei als Anweisungsdatei für den Data-Engineering-Agenten.

Best Practices für Anweisungsdateien für KI-Agenten

Wenn Sie Anweisungen für KI-Agenten verwenden, empfehlen wir Folgendes:

  • Alle Dateipfade in Dataform sind relativ zum Stammverzeichnis des Repository. Verwenden Sie relative Pfade für jede @file.md-Syntax, um Anweisungen ordnungsgemäß in GEMINI.md zu importieren.
  • In GEMINI.md importierte Dateien können selbst Importe enthalten, wodurch eine verschachtelte Struktur entsteht. Um eine unendliche Rekursion zu vermeiden, hat GEMINI.md eine maximale Importtiefe von fünf Ebenen.
  • Wenn Sie Anweisungen für mehrere Datenpipelines freigeben möchten, speichern Sie sie in einem zentralen Dataform-Repository und verknüpfen Sie sie mit dem Dataform-Arbeitsrepository. Sie können lokale Anweisungen verwenden, um zentrale Regeln für pipelinespezifisches Verhalten zu überschreiben.
  • Um die Konsistenz in Ihrem Projekt zu gewährleisten, können Sie auf Dateien mit Namenskonventionen oder Styleguides verlinken und den Agenten anweisen, diese Richtlinien bei der Arbeit mit Ihren Datenpipelines zu befolgen.
  • Sie können in der Anweisungsdatei Datenschichten vorschlagen, um verschiedene Datentypen zu gruppieren.
  • Durch die Verwendung von Überschriften und Listen in der Anweisungsdatei für den Agenten können Anweisungen für den Data-Engineering-Agenten organisiert und verdeutlicht werden.
  • Geben Sie aussagekräftige Dateinamen an und gruppieren Sie ähnliche Anweisungen in einer Datei. Organisieren Sie Regeln logisch nach Kategorie, Funktion oder Funktionalität mit Markdown-Überschriften.
  • Um widersprüchliche Anweisungen zu vermeiden, definieren Sie klar die spezifischen Bedingungen, unter denen jede Anweisung gilt.
  • Iterieren und optimieren Sie Ihre Prompts und Ihren Workflow. Das Verhalten des Agenten ändert sich im Laufe der Zeit mit der Einführung neuer Agentenversionen und Modellupgrades. Wir empfehlen daher, Ihre Regeln mit verschiedenen Prompts zu testen, um Bereiche zu identifizieren, die möglicherweise verbessert werden müssen. Halten Sie Ihre Regeldatei mit allen Änderungen an Ihrer Datenpipeline auf dem neuesten Stand.

Das folgende Beispiel zeigt eine Anweisungsdatei für den Agenten mit dem Namen GEMINI.md, in der unsere Best Practices für die effektive Verwendung des Data-Engineering-Agenten angewendet werden:

  ### Naming Conventions

  * Datasets: [business_domain]_[use_case] (e.g., ecommerce_sales)

  * Tables:
      - Raw/External: raw_[source_name]
      - Staging: stg_[business_entity]
      - Dimension: dim_[dimension_name]
      - Fact: fct_[fact_name]

  * Dataform Folders:
      - sources
      - staging
      - marts
      - dataProducts

  * Views: vw_[view_name]

  * Columns: snake_case (e.g., order_id, customer_name)

  ## Cloud Storage data load
  * When ingesting data from Cloud Storage, create external tables.

  ## Null handling
  * Filter out null id values

  ## String normalization
  * Standardize string columns by converting to lower case

  ## Data Cleaning Guidelines
  @./generic_cleaning.md

Zusätzliche lokale Dateien als Anweisungen für den Agenten importieren

Sie können auch andere Anweisungsdateien für den Data-Engineering-Agenten mit der @file.md-Syntax in die Datei GEMINI.md importieren. Weitere Informationen finden Sie unter Memory Import Processor.

Automatische Datenaufbereitung

Mit dem Data-Engineering-Agenten können Sie rohe, unverarbeitete Daten in strukturierte Tabellen umwandeln, die für die Datenanalyse geeignet sind. Auf Anfrage nimmt der Agent zuerst bis zu 1.000.000 Datensätze aus jeder Standard- oder externen Tabelle auf. Anschließend führt der Agent eine detaillierte Datenanalyse durch, indem er Profilierungsabfragen für diese Stichprobe ausführt. Nachdem Datentransformationen generiert wurden, wiederholt der Agent diesen Stichproben- und Profilierungsprozess, um die Qualität der Transformationen zu bewerten. Diese Transformationen zur Datenaufbereitung können das Beheben von Dateninkonsistenzen, Ausreißern oder Typkonflikten umfassen. Der Data-Engineering-Agent erstellt dann einen Plan mit den vorgeschlagenen Schritten zur Datenaufbereitung, den Sie überprüfen und optimieren können, bevor Aktionen ausgeführt werden.

Der Data-Engineering-Agent initiiert auch die Datenaufbereitungsanalyse, wenn Sie eine Rohdatentabelle hinzufügen, z. B. eine CSV-basierte externe Tabelle. Sie können den Plan zur Datenaufbereitung überprüfen und mit Konversationsbefehlen anpassen.

Für die Datenstichprobenerhebung und -profilierung werden BigQuery-Ressourcen verwendet. Es gelten die BigQuery Preise.

Der Data-Engineering-Agent unterstützt die folgenden Transformationen zur Datenaufbereitung:

  • Datenbereinigung. Der Agent kann Rohdaten analysieren und Bereinigungsmöglichkeiten vorschlagen, z. B. Ausreißer entfernen, fehlende oder inkonsistente Werte ergänzen (Datenimputation), doppelte Daten korrigieren oder Datenformate standardisieren, z. B. Telefonnummern oder Adressen.
  • Strukturtransformationen. Wenn ein Zielschema angegeben ist, kann der Agent Werte aus JSON-, ARRAY- oder STRUCT-Typen aufheben oder extrahieren, mehrere Spalten zusammenführen oder eine Spalte in mehrere Spalten aufteilen.
  • Datentyperkennung und ‑konvertierung. Der Agent kann die Daten analysieren, um die entsprechenden Feldtypen zu ermitteln. Anschließend kann der Agent eine sichere Typumwandlung durchführen, um Formatierungskonsistenzen in den Feldern für Datum, Uhrzeit, Datum/Uhrzeit oder Zeitstempel zu beheben.
  • Umrechnung von Maßeinheiten. Der Agent kann verschiedene Einheiten in einem Feld automatisch in eine einheitliche Einheit umwandeln, um Ihre Daten zu standardisieren.

Um die Genauigkeit zu gewährleisten, verwendet der Agent repräsentative Stichproben Ihrer Daten, um Probleme zu erkennen und die Transformationslogik zu validieren.

Agentenpläne erstellen und überprüfen

Der Data-Engineering-Agent kann Agentenpläne erstellen, die eine Zusammenfassung und einen Überblick über die Ziele und Schritte enthalten, die zur Erfüllung einer Anfrage erforderlich sind. Wenn Sie dem Agenten komplexe Anfragen stellen, die viele Änderungen erfordern, empfehlen wir, ihn um einen Agentenplan zu bitten, damit Sie die Absichten des Agenten überprüfen können, bevor er Aktionen ausführt. Ein Data-Engineering-Agent-Plan besteht in der Regel aus den folgenden Elementen:

  • Das Ziel des Agenten für eine bestimmte Anfrage
  • Ein allgemeiner Überblick über die Schritte, die der Agent ausführen möchte
  • Alle Annahmen, die der Agent trifft
  • Dateien, die der Agent ändern möchte
  • Alle Optimierungs- oder Bereinigungsschritte, die er ausführen möchte
  • Ein stufenweiser Ausführungsplan

In Ihrem Prompt können Sie angeben, dass der Plan überprüft und genehmigt werden muss, damit der Agent keine Maßnahmen ohne Ihre ausdrückliche Genehmigung ergreift. Beispiel:

Create a plan for a pipeline that finds the
top N pick up and drop off locations in NYC. I want to review the plan and
approve it before you create the pipeline.

Der Agent kann auch automatisch einen Agentenplan erstellen und Ihre Genehmigung anfordern. Dies kann vorkommen, wenn ein Prompt zu mehrdeutig ist oder der Agent mehr Klarheit benötigt, um Ihre Anfrage zu erfüllen.

Best Practices für die Verwendung von Agentenplänen finden Sie unter Best Practices.

Kontext aus Knowledge Catalog hinzufügen

Der Data-Engineering-Agent verwendet Knowledge Catalog, indem er Glossarbegriffe an BigQuery-Tabellen und ‑Spalten anhängt und Datenprofilscans generiert. Mit Glossarbegriffen können Spalten getaggt werden, die zusätzlichen Kontext erfordern, z. B. Spalten mit personenbezogenen Daten (personenidentifizierbare Informationen), für die spezielle Anweisungen zur Verarbeitung erforderlich sind, oder um übereinstimmende Spalten mit unterschiedlichen Namen in verschiedenen Tabellen zu identifizieren.

Knowledge Catalog verwendet auch die Daten profilierung, die dem Agenten ein besseres Verständnis der Datenverteilung in Tabellenspalten vermittelt und ihm hilft, genauere Zusicherungen zur Datenqualität zu erstellen.

Einer vorhandenen Tabelle Prüfungen der Datenqualität hinzufügen

Wenn Sie den Agenten auffordern, Qualitätsprüfungen hinzuzufügen, leitet er auf der Grundlage des Schemas und der Stichproben angemessene Prüfungen für die Tabelle ab. Sie können auch Meinungszusicherungen als Teil des Prompts hinzufügen. Beispiel:

  Add data quality checks for bigquery-public-data.thelook_ecommerce.users.

Datenpipelines optimieren

Sie können den Agenten auffordern, Ihre Datenpipelines zu optimieren. Beim Generieren von DDL für neue Tabellen empfiehlt der Data-Engineering-Agent die Partitionierung und das Clustering basierend auf den analysierten Datennutzungsmustern. Außerdem kann der Agent automatisch andere Pipeline-Optimierungen anwenden. Beispiele für mögliche Optimierungen:

  • Spaltenbereinigung, um die Menge der aus dem Speicher gelesenen Daten zu reduzieren, was sich hauptsächlich auf Kosten und Leistung auswirkt.
  • Prädikat-Pushdowns, um Daten frühzeitig im Ausführungsplan zu filtern und so die Menge der von nachfolgenden Vorgängen verarbeiteten Daten erheblich zu reduzieren.
  • Eliminierung gemeinsamer Unterausdrücke zur Verbesserung der Effizienz, indem die gemeinsame Transformationslogik nur einmal identifiziert und berechnet wird. So werden ineffiziente Praktiken wie das mehrmalige Scannen und Verknüpfen großer Tabellen vermieden.
  • Inkrementelle Modelle, um nur neue oder geänderte Daten seit der letzten Ausführung zu verarbeiten, anstatt bei jeder Ausführung ganze Tabellen neu zu erstellen.

Best Practices

Um die Ergebnisse bei der Arbeit mit dem Data-Engineering-Agenten und Dataform zu verbessern, empfehlen wir Folgendes:

Anweisungen für KI-Agenten für häufige Anfragen verwenden Wenn Sie häufig bestimmte Techniken anwenden oder dem Agenten immer wieder dieselben Korrekturen geben, verwenden Sie Anweisungen für KI-Agenten als zentralen Ort, um häufige Anweisungen und Anfragen zu speichern.

Agentenpläne verwenden : Agentenpläne können hilfreich sein, um komplexe Pipelineaufgaben aufzuschlüsseln. Agentenpläne zeigen Ihnen auch die Annahmen und Absichten des Agenten. Wir empfehlen daher, diese Pläne zu überprüfen, um sicherzustellen, dass der Agent den richtigen Kontext erhält.

Nachdem Sie einen Plan überprüft haben, können Sie ihn bearbeiten, indem Sie dem Data-Engineering-Agenten Feedback und Änderungen geben. Beispiel:

In the plan, ensure that all of the intermediate tables are views.

In einigen Fällen kann es hilfreich sein, den Agenten zu bitten, einen Plan zu erstellen, der keine ausdrückliche Genehmigung erfordert. Durch die Erstellung des Agentenplans wird der Data-Engineering-Agent gezwungen, seine Aktionen aufzuschlüsseln, was oft zu besseren Ergebnissen führt. Sie können den Agenten zwingen, einen Plan zu erstellen und ihn automatisch auszuführen. Beispiel:

Create a plan for a pipeline that finds the
top N pick up and drop off locations in NYC. You have my explicit pre-approval
to go ahead and execute this plan.

Deutlich schreiben Formulieren Sie Ihre Anfrage klar und vermeiden Sie Unklarheiten. Geben Sie nach Möglichkeit Quell- und Zieldatenquellen an, wenn Sie einen Prompt erstellen, wie im folgenden Beispiel gezeigt:

  Extract data from the sales.customers table in the us_west_1 region, and load
  it into the reporting.dim_customers table in BigQuery. Match the schema of the
  destination table.

Direkte und gezielte Anfragen stellen Stellen Sie jeweils nur eine Frage und halten Sie die Prompts kurz. Bei Prompts mit mehr als einer Frage sollten Sie jeden Teil der Frage einzeln auflisten, um die Klarheit zu verbessern, wie im folgenden Beispiel gezeigt:

  1. Create a new table named staging.events_cleaned. Use raw.events as the
     source. This new table should filter out any records where the user_agent
     matches the pattern '%bot%'. All original columns should be included.

  2. Next, create a table named analytics.user_sessions. Use
     staging.events_cleaned as the source. This table should calculate the
     duration for each session by grouping by session_id and finding the
     difference between the MAX(event_timestamp) and MIN(event_timestamp).

Explizite Anweisungen geben und wichtige Begriffe hervorheben Sie können wichtige Begriffe oder Konzepte in Ihren Prompts hervorheben und bestimmte Anforderungen als wichtig kennzeichnen, wie im folgenden Beispiel gezeigt:

  When creating the staging.customers table, it is *VERY IMPORTANT* that you
  transform the email column from the source table bronze.raw_customers.
  Coalesce any NULL values in the email column to an empty string ''.

Reihenfolge der Vorgänge angeben Bei geordneten Aufgaben strukturieren Sie Ihren Prompt in Listen, wobei die aufgeführten Elemente in kleine, fokussierte Schritte unterteilt sind, wie im folgenden Beispiel gezeigt:

  Create a pipeline with the following steps:
  1. Extract data from the ecomm.orders table.
  2. Join the extracted data with the marts.customers table on customer_id.
  3. Load the final result into the reporting.customer_orders table.

Optimieren und iterieren Probieren Sie verschiedene Formulierungen und Ansätze aus, um herauszufinden, was die besten Ergebnisse liefert. Wenn der Agent ungültiges SQL oder andere Fehler generiert, können Sie ihn mit Beispielen oder öffentlicher Dokumentation anleiten.

  The previous query was incorrect because it removed the timestamp. Please
  correct the SQL. Use the TIMESTAMP_TRUNC function to truncate the
  event_timestamp to the nearest hour, instead of casting it as a DATE. For
  example: TIMESTAMP_TRUNC(event_timestamp, HOUR).