Airflow ai sdk with gemini

Dockerfile

@@ -18,6 +18,10 @@ RUN pip uninstall -y -r requirements-uninstall.txt && \
   apache-airflow-providers-microsoft-mssql==3.9.0 \
   apache-airflow-providers-common-sql==1.16.0
 
+# Copy and install MAIN requirements.txt
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
 # Copy and install requirements.txt
 COPY tests-requirements.txt .
 RUN pip install --no-cache-dir -r tests-requirements.txt

dag_confs/examples_and_tests/termos_interesse_arquivo_nacional_inlabs_copra.yaml

@@ -0,0 +1,29 @@
+dag:
+  id: busca_termos_arquivo_nacional_inlabs_copra
+  description: Realiza busca no DOU das publicações no âmbito do Copra (Arquivo Nacional)
+  tags:
+    - arquivo_nacional
+    - inlabs
+  owner:
+    - cdata
+  dataset: inlabs
+  search:
+    sources:
+      - INLABS
+    terms:
+      - Arquivo Nacional
+    is_exact_search: On
+    ignore_signature_match: On
+  report:
+    emails:
+      - copra.gabin@gestao.gov.br
+      - daiana.ribeiro@gestao.gov.br
+      - carlos.burity@gestao.an.gov.br
+      - ro-dou-suporte@gestao.gov.br
+    subject: "Ro-DOU: Publicações do dia no DOU"
+    skip_null: False
+  doc_md: >-
+    ## Clipping Arquivo Nacional
+
+    Realiza a busca de termos no âmbito do COPRA/Arquivo Nacional.
+    Contato: Daiana Ribeiro daiana.ribeiro@gestao.gov.br

requirements.txt

@@ -2,4 +2,6 @@ pandas==2.1.4
 unidecode==1.2.0
 html2text==2024.2.26
 markdown==3.6.0
-Jinja2==3.1.4
+Jinja2==3.1.4
+google-generativeai==0.8.6
+airflow-ai-sdk==0.1.7